Pleaplusone 6cae1e5332 [ROCm][MLA] Support block-size > 1 for AITER MLA backend (#27224) ...

Signed-off-by: ganyi <ygan@amd.com>
Co-authored-by: wuhuikx <hattie.wu@amd.com>

2025-11-05 10:43:02 -05:00

..

mla

[ROCm][MLA] Support block-size > 1 for AITER MLA backend (#27224)

2025-11-05 10:43:02 -05: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

[XPU] Add gpt-oss model support for Intel GPU (#27786)

2025-11-05 02:17:23 +00: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

[V1] [Hybrid] Mamba1 Automatic Prefix Caching (#26377)

2025-11-02 04:16:23 -08:00

mamba2_attn.py

[V1] [Hybrid] Mamba1 Automatic Prefix Caching (#26377)

2025-11-02 04:16:23 -08:00

mamba_attn.py

[V1] [Hybrid] Mamba1 Automatic Prefix Caching (#26377)

2025-11-02 04:16:23 -08: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

[ROCm][Perf] New design on ROCm AITER MHA backend Implementation (#25763)

2025-11-04 18:05:33 +00: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] Fix DCP Assert (AssertionError: DCP not support reorder_batch_threshold > 1 now.) (#28100)

2025-11-05 14:54:43 +08:00

xformers.py

remove attn output view kernel (#26680)

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