Ning Xie 71baf85ae1

[Kernel] mark TorchSDPABackend swap_blocks NotImplementedError (#19749)

2025-06-20 18:18:11 +00:00

..

mla

[torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass (#16756)

2025-06-12 08:31:04 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

abstract.py

[torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass (#16756)

2025-06-12 08:31:04 -07:00

blocksparse_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

cpu_mla.py

[CPU] V1 support for the CPU backend (#16441)

2025-06-03 18:43:01 -07:00

dual_chunk_flash_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

flash_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

flashinfer.py

[V1][Kernel] Flashinfer HND KV cache layout (#19280)

2025-06-17 09:09:22 -04:00

flashmla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

hpu_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

ipex_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

pallas.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

placeholder_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

rocm_aiter_mla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

rocm_flash_attn.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00

torch_sdpa.py

[Kernel] mark TorchSDPABackend swap_blocks NotImplementedError (#19749)

2025-06-20 18:18:11 +00:00

triton_mla.py

[V1] Support cross-layer KV sharing (#18212)

2025-06-03 20:33:07 +00:00

utils.py

[CI] change spell checker from codespell to typos (#18711)

2025-06-11 19:57:10 -07:00

xformers.py

[Kernel] Raise verbose error and consolidate num_heads/num_kv_heads divisibility check (#19339)

2025-06-15 13:43:48 +08:00