R3hankhan 8e27663b6a [CPU] Add head sizes 80 and 112 with vec16 fallback (#31968) ...

Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com>

2026-01-09 22:14:46 +08:00

..

mla

[Chore] Migrate V0 attention utils (#31891)

2026-01-07 13:44:36 +00:00

__init__.py

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

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

cpu_attn.py

[CPU] Add head sizes 80 and 112 with vec16 fallback (#31968)

2026-01-09 22:14:46 +08:00

flash_attn_diffkv.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

flash_attn.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

flashinfer.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

flex_attention.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

gdn_attn.py

[Chore] Migrate V0 attention utils (#31891)

2026-01-07 13:44:36 +00:00

linear_attn.py

[Attention] add _cudagraph_support for linear attention (#28934)

2025-11-25 12:25:20 +08:00

mamba1_attn.py

[Mamba] - Consolidate Mambas Attention Logic (#28133)

2025-12-23 21:57:00 +01:00

mamba2_attn.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

mamba_attn.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

rocm_aiter_fa.py

[Attention][3/n] Remove usage of deprecated seq_lens_cpu and num_computed_tokens_cpu CommonAttentionMetadata properties (#31850)

2026-01-07 13:31:34 +08:00

rocm_aiter_unified_attn.py

[ROCm] Support for Whisper v1 with Aiter Unified Attention and Aiter Flash Attention (#28376)

2025-11-24 03:26:00 +00:00

rocm_attn.py

[Bugfix][ROCm]Fix Qwen3-Next-80B-A3B-Thinking inference and optimize non-standard block size (544) support under rocm_atten (#31380)

2026-01-09 19:28:02 +08:00

short_conv_attn.py

[Mamba] - Consolidate Mambas Attention Logic (#28133)

2025-12-23 21:57:00 +01:00

tree_attn.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00

triton_attn.py

[Attention][1/n] Remove usage of deprecated seq_lens_cpu and num_computed_tokens_cpu CommonAttentionMetadata properties (#31773)

2026-01-06 04:05:17 -08:00

utils.py

fixed mypy warnings for files vllm/v1/attention with TEMPORARY workaround (#31465)

2026-01-07 04:08:47 +00:00