jennyyyyzhen a4cf9b22ba [ROCM][Bugfix] Use correct stride in cp_mha_gather_cache_kernel for hybrid model (#37228) (#37228) ...

Signed-off-by: jennyyyyzhen <yzhen@hmc.edu>
Co-authored-by: yZhen <yZhen@fb.com>

2026-03-26 10:33:39 -07:00

..

mla

[Bugfix] Restore CUDA graph persistent buffers for FP8 FlashMLA decode (#35175)

2026-03-26 16:13:39 +00:00

__init__.py

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

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

cpu_attn.py

[Torch 2.11] Guard torch._C._cpu attribute checks for forward compatibility (#35673)

2026-03-17 18:47:59 +00:00

fa_utils.py

[Bugfix] Register VLLM_BATCH_INVARIANT in envs.py to fix spurious unknown env var warning (#35007)

2026-03-23 18:31:14 -04:00

flash_attn_diffkv.py

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

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

flash_attn.py

[Bugfix] Register VLLM_BATCH_INVARIANT in envs.py to fix spurious unknown env var warning (#35007)

2026-03-23 18:31:14 -04:00

flashinfer.py

[Bugfix] Register VLLM_BATCH_INVARIANT in envs.py to fix spurious unknown env var warning (#35007)

2026-03-23 18:31:14 -04:00

flex_attention.py

[Bugfix][Minor] Fix potential NameError in mamba backend selector and misc typos (#35886)

2026-03-26 11:59:24 -04:00

gdn_attn.py

[Bugfix] Fix GDN attention crash with mixed decode/spec-decode batches (#34871)

2026-03-16 10:30:23 +01: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

[Attention] Support distinguishing between short extends and decodes (#37303)

2026-03-20 10:49:36 -07:00

registry.py

Add XPU MLA Sparse backend for DeepSeek v3.2 (#33230)

2026-03-11 19:19:15 +08:00

rocm_aiter_fa.py

[ROCM][Bugfix] Use correct stride in cp_mha_gather_cache_kernel for hybrid model (#37228) (#37228)

2026-03-26 10:33:39 -07:00

rocm_aiter_unified_attn.py

[ROCm][Bugfix] fix cache block size mismatch for aiter unified attention (#37606)

2026-03-20 00:00:08 +00:00

rocm_attn.py

[ROCm] Attention selector reordering (#36702)

2026-03-25 17:42:56 +08:00

short_conv_attn.py

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

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

tree_attn.py

[Misc] Add float16 to CacheDType (#37199)

2026-03-16 13:24:48 -07:00

triton_attn.py

[Misc] Add float16 to CacheDType (#37199)

2026-03-16 13:24:48 -07:00

utils.py

[Bugfix][Minor] Fix potential NameError in mamba backend selector and misc typos (#35886)

2026-03-26 11:59:24 -04:00