Michael Goin 4082338a25 Remove unneeded ROCm platform import when using CUDA (#22765) ...

Signed-off-by: mgoin <mgoin64@gmail.com>

2025-08-12 21:26:38 -07:00

..

mla

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

[Attention] Support multiple attention metadata builders per kv_cache_spec + proper local attention no hybrid kv cache fix (#21588)

2025-08-06 18:40:52 -07:00

differential_flash_attn.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

dual_chunk_flash_attn.py

[Bugfix][Qwen][DCA] fixes bug in dual-chunk-flash-attn backend for qwen 1m models. (#21364)

2025-07-23 06:34:37 -07:00

flash_attn.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

flashinfer.py

[NVIDIA] Support Flashinfer TRT-LLM Prefill Attention Kernel (#22095)

2025-08-05 02:45:34 -07:00

flashmla.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

placeholder_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

rocm_aiter_mla.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

rocm_flash_attn.py

Remove unneeded ROCm platform import when using CUDA (#22765)

2025-08-12 21:26:38 -07:00

triton_mla.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

utils.py

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

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

xformers.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00