vLLM/tests/kernels/attention at 25efcfa7887c4a9541b6328af69dbd5fee4e8173 - vLLM

conftest.py

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

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

test_amx_mla.py

[CPU] Add AMX-only high-performance MLA backend for DeepSeek V2/V3/R1 (#52616 )

2026-08-19 10:28:52 +08:00

test_attention_selector.py

[SM100] Hdim 256 optimized (#52980 )

2026-08-25 20:28:40 -07:00

test_attention.py

[ROCm][CI] Add float16 dtype and unsupported head size tests for paged attention (#53177 )

2026-08-21 15:16:45 -05:00

test_cache.py

[Attention][DSA] Enable W4A16 DSA (#51724 )

2026-09-01 12:37:47 +08:00

test_cascade_flash_attn.py

[Platform] Deprecate seed_everything (#31659 )

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

test_cpu_attn.py

[CPU][BugFix] Remove redundant kv cache write (#50437 )

2026-07-31 15:28:44 +08:00

test_cutlass_mla_decode.py

[ROCM][CI] Attention test speedup (#52763 )

2026-08-18 16:40:30 -05:00

test_deepgemm_attention.py

[Attention][DSA] Take the native decode path for MTP=3 on SM90 (#52164 )

2026-08-14 10:09:51 +00:00

test_flash_attn.py

[SM100] Hdim 256 optimized (#52980 )

2026-08-25 20:28:40 -07:00

test_flashinfer_mla_decode.py

[Bugfix] Use int8 workspace for FlashInfer MLA decode (#48046 )

2026-07-08 23:39:40 +00:00

test_flashinfer_trtllm_attention.py

[Attention][Spec Decode] Support varlen trtllm-gen decode for adaptive verification (#52157 )

2026-08-24 14:43:57 -04: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

[Bugfix][SM120] DSv4: pass contiguous C128A decode topk indices on SM120 (#53574 )

2026-08-31 09:44:42 +00:00

test_flashmla.py

[ROCM][CI] Attention test speedup (#52763 )

2026-08-18 16:40:30 -05:00

test_kimi_k3_mla_fused_epilogue.py

[Attention][MLA] Fuse Kimi-K3 chunked-context K/V packing (#51772 )

2026-08-12 22:42:57 -07:00

test_kimi_k3_mla_key_concat_kv_cache.py

[Kimi-K3] DCP support (#50484 )

2026-08-10 12:54:26 -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

[XPU][Tests] Make tests device-agnostic (#51968 )

2026-08-20 02:25:41 -07:00

test_mha_attn.py

[ROCm][CI] Restore Torch defaults and type DSV4 scratch buffers (#52566 )

2026-08-17 18:32:04 +00:00

test_minimax_m3_msa_cutlass_sparse_decode.py

[Bugfix][MiniMax-M3] Keep FP8 query allocation stable across CUDA graph replay (#51203 )

2026-08-20 10:45:28 -07:00

test_minimax_m3.py

[ROCm][CI] Restore attention coverage after KV-cache layout refactor (#53351 )

2026-08-22 20:00:25 -05: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

Add FlashMLA H100 tests to CI, fix them after #32810 (#50322 )

2026-07-29 22:06:18 -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

[Kernel] Make prefix-prefill tiling independent of the KV page size (#54194 )

2026-09-01 09:19:55 +08:00

test_rocm_aiter_fa.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_rocm_aiter_mla_causal_verify_mask.py

[ROCm][MLA][DCP] Support causal multi-token verification (#51705 )

2026-08-31 10:42:28 -07:00

test_rocm_aiter_mla_decode_metadata.py

[Bugfix][ROCm][CI] Give the AITER MLA decode metadata stub its MLA dims (#52139 )

2026-08-13 13:35:00 -05:00

test_rocm_aiter_mla_decode.py

[ROCm][CI] Add MLA decode accuracy and determinism tests (#50480 )

2026-08-05 20:35:56 -05:00

test_rocm_aiter_mla_fp8_prefill.py

[ROCm][K3] Extend FP8 asm MLA prefill to non-divisor small head counts (#51040 )

2026-08-26 17:47:16 +00:00

test_rocm_aiter_mla_fp8_support.py

[ROCm][CI] Move ROCm AITER quantization tests (#40938 )

2026-08-18 14:50:35 +08:00

test_rocm_aiter_mla_head_padding.py

[ROCm][MLA][DCP] Support causal multi-token verification (#51705 )

2026-08-31 10:42:28 -07:00

test_rocm_aiter_mla_op_registration.py

[Test] Add ROCm AITER MLA op registration and env gating tests (#50930 )

2026-08-07 14:07:31 -05:00

test_rocm_aiter_mla_sparse_metadata_sync.py

[ROCm] [BugFix] Fix Quark GLM-5.2 Checkpoint inference: indexer wk per-channel FP8 dequant + missing sparse-MLA metadata fields (#48886 )

2026-07-27 22:38:22 +00:00

test_rocm_aiter_unified_attn.py

Enable gfx1250 ROCm architecture (#46516 )

2026-07-31 01:41:12 +00: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

[ROCm][CI] Restore Torch defaults and type DSV4 scratch buffers (#52566 )

2026-08-17 18:32:04 +00:00

test_triton_decode_attention.py

[ROCM][CI] Attention test speedup (#52763 )

2026-08-18 16:40:30 -05: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

[ROCM][CI] Attention test speedup (#52763 )

2026-08-18 16:40:30 -05:00

test_triton_unified_attention.py

[Kernel][Gemma4] Prune Triton sliding-window tiles for multimodal prefixes (#53147 )

2026-08-31 22:21:22 +00:00

test_trtllm_kvfp8_dequant.py

[6/N][KV-Cache Layout Refactor] Standardize KV cache layout (#51718 )

2026-08-21 17:47:49 -07:00

test_use_trtllm_attention.py

[Attention] Add FlashInfer XQA decode support on SM12x (#49718 )

2026-08-11 16:32:49 -07:00

test_xpu_mla_sparse.py

[Bugfix] Prevent NaN poisoning in xpu_mla_sparse for fully-masked index chunks (#48366 )

2026-07-27 09:06:07 +08:00