Dan BlanaruandMatthew Bonanni d29125c085 Xqa decode kernels (#43232) ...

Signed-off-by: Dan Blanaru <48605845+DanBlanaru@users.noreply.github.com>
Co-authored-by: Matthew Bonanni <mbonanni@redhat.com>

2026-07-02 12:32:05 -07:00

..

conftest.py

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

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

test_aiter_flash_attn.py

[ROCm][AITER] Fix AITER import regression for explicit backend selection (#33749)

2026-02-06 15:08:16 +00:00

test_attention_selector.py

[SpecDecode] Support DFlash with FlashInfer (#43081)

2026-06-22 04:55:30 +00:00

test_attention.py

Delete PagedAttention (#47361)

2026-07-02 12:31:26 -07:00

test_cache.py

[Feature] Support DCP with FP8 KV cache in MLA decode path (#44044)

2026-06-24 19:28:17 +00:00

test_cascade_flash_attn.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_cpu_attn.py

[CPU] Support Gemma Diffusion (#45690)

2026-06-16 14:39:56 +08:00

test_cutlass_mla_decode.py

[Attention] Re-enable cross-layer KV cache layout for MLA via stride-aware kernels (#45111)

2026-06-22 06:57:02 -07:00

test_deepgemm_attention.py

[Feat] DeepSeek V4 Rebased (#40860)

2026-04-26 18:31:08 -07:00

test_flash_attn.py

[Platform] Deprecate seed_everything (#31659)

2026-01-04 18:34:04 -08:00

test_flashinfer_mla_decode.py

[CI/Build][AMD] Fix import errors in tests/kernels/attention (#29032)

2025-11-20 17:48:09 +08:00

test_flashinfer_trtllm_attention.py

Add nvfp4 kv cache support (#40177)

2026-05-01 04:55:16 +00:00

test_flashinfer.py

[Update] Use FlashInfer fast_decode_plan directly instead of replication (#34687)

2026-02-26 16:31:43 -08:00

test_flashmla_sparse.py

[DSv4 Perf] DSv4 flashinfer sparse index cache for metadata, 2%~4% TTFT improvement (#45863)

2026-06-17 10:55:48 -04:00

test_flashmla.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_lightning_attn.py

[Kernel][Test] Extend lightning_attn and awq_triton kernel tests to XPU (#43307)

2026-06-04 14:25:59 -04:00

test_merge_attn_states.py

[Kernel] Fuse FP8 output quantization into merge_attn_states (#36518)

2026-04-03 01:47:04 +00:00

test_mha_attn.py

[MM][OOT] Support CPU seq_lens for OOT MMEncoderAttention kernels (#36605)

2026-03-12 03:28:23 -07:00

test_minimax_m3.py

[Minimax-M3] BF16/FP8 Indexer using MSA (#45892)

2026-06-23 10:28:49 -07:00

test_mixed_causal_attn.py

[Model] Add DiffusionGemma Support (#45163)

2026-06-11 22:17:35 -07:00

test_mla_cross_layer_kernel_equivalence.py

[Attention] Re-enable cross-layer KV cache layout for MLA via stride-aware kernels (#45111)

2026-06-22 06:57:02 -07:00

test_mla_decode_cpu.py

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

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

test_pack_unpack_triton.py

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

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

test_prefix_prefill.py

[Hardware][AMD][CI] Fix Kernels Attention test groups (#46080)

2026-06-21 17:10:51 -05:00

test_rocm_aiter_mla_decode_metadata.py

[Bugfix][ROCm][MLA] Pass q/kv dtypes to get_mla_metadata_v1 in FP8 decode (#46997)

2026-06-30 05:31:16 -07:00

test_rocm_aiter_unified_attn.py

[ROCm] Fix AITER_UNIFIED_ATTN Dispatching After AITER Bump (#46780)

2026-06-26 02:09:56 -07:00

test_rocm_attention_selector.py

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

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

test_rocm_triton_attn_dsv4.py

[Hardware][AMD][CI] Fix Kernels Attention test groups (#46080)

2026-06-21 17:10:51 -05:00

test_triton_decode_attention.py

[Attention] Re-enable cross-layer KV cache layout for MLA via stride-aware kernels (#45111)

2026-06-22 06:57:02 -07:00

test_triton_prefill_attention.py

[XPU] enable triton attention test on XPU by removing cuda device binding (#39627)

2026-04-20 20:57:11 +08:00

test_triton_unified_attention_diffkv.py

[Attention] add triton diff-kv backend for mimo (#41797)

2026-06-11 11:36:31 -04:00

test_triton_unified_attention.py

[Hardware][AMD][CI] Fix Kernels Attention test groups (#46080)

2026-06-21 17:10:51 -05:00

test_trtllm_kvfp8_dequant.py

[ROCm][CI] Skip trtllm kvfp8 dequant tests on ROCm (#37330)

2026-03-18 11:12:26 +08:00

test_use_trtllm_attention.py

Xqa decode kernels (#43232)

2026-07-02 12:32:05 -07:00

test_xpu_mla_sparse.py

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

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