youkaichao 558f0907dc

[attention][DCP] use AttentionImpl.need_to_return_lse_for_decode (#24372) ...

Signed-off-by: youkaichao <youkaichao@gmail.com>

2025-09-07 01:18:59 +00:00

..

mla

[Doc]: fix typos in Python comments (#24173)

2025-09-04 08:52:17 -07:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

[attention][DCP] use AttentionImpl.need_to_return_lse_for_decode (#24372)

2025-09-07 01:18:59 +00:00

differential_flash_attn.py

[Docs] Fix warnings in mkdocs build (#23649)

2025-08-26 18:19:23 +00:00

dual_chunk_flash_attn.py

[NVIDIA][torch.compile] Support Flashinfer TRTLLM FP8-q/kv NVFP4-out Attention Kernel (#22703)

2025-08-22 22:09:05 +00:00

flash_attn.py

[Docs] Fix warnings in mkdocs build (#23649)

2025-08-26 18:19:23 +00:00

flashmla.py

[V0 Deprecation] Remove advance_step (#22969)

2025-08-15 08:22:31 -07:00

placeholder_attn.py

[V0 Deprecation] Remove advance_step (#22969)

2025-08-15 08:22:31 -07:00

rocm_aiter_mla.py

[V0 Deprecation] Remove advance_step (#22969)

2025-08-15 08:22:31 -07:00

rocm_flash_attn.py

[Docs] Fix warnings in mkdocs build (#23649)

2025-08-26 18:19:23 +00:00

triton_mla.py

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

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

utils.py

[Docs] Fix warnings in mkdocs build (#23649)

2025-08-26 18:19:23 +00:00

xformers.py

[Docs] Fix warnings in mkdocs build (#23649)

2025-08-26 18:19:23 +00:00