Matthew Bonanni f29aeb5a25

Add FLASHINFER_MLA to test_mla_backends and add B200 CI run (#27663) ...

Signed-off-by: Matthew Bonanni <mbonanni@redhat.com>

2025-10-31 11:12:19 -07:00

..

mla

Add FLASHINFER_MLA to test_mla_backends and add B200 CI run (#27663)

2025-10-31 11:12:19 -07:00

__init__.py

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

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

cpu_attn.py

[fix][cpu] fix prefill attention in CPU attention backend (#27035)

2025-10-18 13:30:21 +00:00

flash_attn.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

flashinfer.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

flex_attention.py

[Bugfix] Avoid too small block m/n for FlexAttention kernel option (#27853)

2025-10-31 19:33:12 +08:00

gdn_attn.py

[MISC] cudagraph_capture_sizes related improvements (#26016)

2025-10-24 05:11:05 -07:00

linear_attn.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

mamba1_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

mamba2_attn.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

mamba_attn.py

[MISC] cudagraph_capture_sizes related improvements (#26016)

2025-10-24 05:11:05 -07:00

pallas.py

[Chore]:Extract math and argparse utilities to separate modules (#27188)

2025-10-26 04:03:32 -07:00

rocm_aiter_fa.py

[Chore] Clean up pytorch helper functions in vllm.utils (#26908)

2025-10-18 09:48:22 -07:00

rocm_aiter_unified_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

rocm_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

short_conv_attn.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

tree_attn.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00

triton_attn.py

Move query quantization to attention layer for Flashinfer & Triton. (#26534)

2025-10-15 19:01:38 -04:00

utils.py

[BugFix] Reordering extend logic fix (#27739)

2025-10-29 21:39:34 -07:00

xformers.py

remove attn output view kernel (#26680)

2025-10-14 22:53:10 +00:00