Woosuk Kwon 006aea17d7 [BugFix] Remove incorrect assert in split_decodes_and_prefills (#36553) ...

Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>

2026-03-09 20:02:02 -07:00

..

mla

[ROCm][CI] Fix ROCm attention backend validation for head sizes, block sizes, and compute capability checks (#36292)

2026-03-09 12:02:41 -05:00

__init__.py

[V1] Implement vLLM V1 [1/N] (#9289)

2024-10-22 01:24:07 -07:00

cpu_attn.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

fa_utils.py

[BugFix] Fallback from FA4->FA2 for Batch Invariance (#36059)

2026-03-05 14:05:56 -05:00

flash_attn_diffkv.py

[1/N][Attention] Restructure attention: move files (#31916)

2026-01-09 13:10:24 -08:00

flash_attn.py

[Core] Add All-to-All communication backend for DCP (#34883)

2026-03-04 10:01:57 -05:00

flashinfer.py

[CI] Bump mypy version (#34950)

2026-03-04 20:55:11 +00:00

flex_attention.py

feat(attention): extract KV-cache update from FlexAttention backend (#36263)

2026-03-08 20:40:12 -07:00

gdn_attn.py

[CI] Bump mypy version (#34950)

2026-03-04 20:55:11 +00:00

linear_attn.py

[V1][Hybrid] Mamba Prefix Caching with align mode (#30877)

2026-01-23 09:56:48 -08:00

mamba1_attn.py

[Mamba1] - Kernel Level Chunk Alignment for Prefix Caching (#34798)

2026-03-01 20:40:23 +08:00

mamba2_attn.py

[Mamba1] - Kernel Level Chunk Alignment for Prefix Caching (#34798)

2026-03-01 20:40:23 +08:00

mamba_attn.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

registry.py

[Attention] Add FlashInfer Sparse MLA backend (#33451)

2026-02-12 17:21:54 +00:00

rocm_aiter_fa.py

Fix CUDA graph decode capture crash in AITER FlashAttention (#36042)

2026-03-06 18:12:07 -08:00

rocm_aiter_unified_attn.py

[ROCm][CI] Fix ROCm attention backend validation for head sizes, block sizes, and compute capability checks (#36292)

2026-03-09 12:02:41 -05:00

rocm_attn.py

[ROCm][CI] Fix ROCm attention backend validation for head sizes, block sizes, and compute capability checks (#36292)

2026-03-09 12:02:41 -05:00

short_conv_attn.py

[Misc] Add get_name to missing AttentionBackends (#32698)

2026-01-23 10:35:44 +00:00

tree_attn.py

[Performance] Extract KV-cache update from TreeAttention backend (#35384)

2026-03-06 00:22:43 +00:00

triton_attn.py

[ROCm][CI] Fix ROCm attention backend validation for head sizes, block sizes, and compute capability checks (#36292)

2026-03-09 12:02:41 -05:00

utils.py

[BugFix] Remove incorrect assert in split_decodes_and_prefills (#36553)

2026-03-09 20:02:02 -07:00