vllm/tests/kernels at 54ab69b14eeafd5c8dcf818cfcc0eeb26d2ebddb - vllm

attention

[ROCm][Bugfix] Sanitize AITER paged-MQA logits before sparse top-k for DeepSeek-V4 (#49714 )

2026-07-28 11:08:50 -07:00

core

[ROCm] Fix and optimize GPT-J-style MRoPE (#49906 )

2026-07-28 10:50:10 -06:00

helion

[Kernel][Helion][1/N] Add Helion kernel for silu_and_mul_per_block_quant (#43994 )

2026-07-07 00:19:01 +08:00

ir

[ROCm] Make vllm_c RMSNorm output contiguous (#49913 )

2026-07-26 23:59:14 -05:00

mamba

[CPU][Spec Decode] Optimize GDN conv path for speculative decoding (#48577 )

2026-07-27 06:20:14 +00:00

moe

[MXFP8][ROCm] Fix MXFP8 MoE backend selection (#49747 )

2026-07-29 07:57:20 +08:00

quantization

[ROCm][Quark][6/N] Use MXFP4 linear kernel abstraction for aiter backend (#49348 )

2026-07-27 21:35:57 -05:00

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425 )

2024-05-13 23:50:09 +09:00

allclose_default.py

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

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

conftest.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423 )

2026-07-22 14:43:07 -06:00

quant_utils.py

[Perf] Create TMA-aligned input scale tensor for DeepGemm on Hopper (#32619 )

2026-01-22 15:47:04 -05:00

test_apply_repetition_penalties.py

[Platform] Deprecate seed_everything (#31659 )

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

test_awq_int4_to_int8.py

[CPU] Bump up to the latest CPU kernels (#41924 )

2026-05-07 05:45:59 -07:00

test_bf16x3_router_gemm_cutedsl.py

BF16x3 router GEMM (#47973 )

2026-07-16 17:04:20 +08:00

test_cache_kernels.py

[Bugfix] Restore gather_and_maybe_dequant_cache OOB guard (#49427 )

2026-07-22 13:47:37 -07:00

test_compressor_kv_cache.py

[DSv4 Perf] Skip empty c128 kernel launch, around 2x kernel performance improvement. (#48957 )

2026-07-22 10:55:20 -04:00

test_concat_mla_q.py

[Attention][Perf][Kernel] Replace torch.cat with vectorized CUDA kernel MLA query concat - DeepSeek-V3.2 (#34917 )

2026-03-09 09:50:36 -07:00

test_cp_gather_fp8.py

[Bugfix] Sparse MLA: enable fp8_ds_mla dense prefill (#48642 )

2026-07-16 22:44:04 +00:00

test_fla_layernorm_guard.py

[Refactor] Move fla to third party (#48500 )

2026-07-16 19:22:36 +01:00

test_flex_attention.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423 )

2026-07-22 14:43:07 -06:00

test_fp32_router_gemm.py

FP32 router GEMV optimization (#48335 )

2026-07-11 13:07:48 +00:00

test_fused_deepseek_v4_qnorm_rope_kv_insert.py

[ROCm][CI] Relax fused layernorm quant test tolerances for one-ULP outliers (#46658 )

2026-06-27 10:44:29 +08:00

test_fused_deepseek_v32_norm_rope.py

Revert "[Perf][GLM-5.2] Blackwell decode optimizations" (#49768 )

2026-07-24 16:04:59 -07:00

test_fused_gdn_post_conv.py

[Refactor] Move fla to third party (#48500 )

2026-07-16 19:22:36 +01:00

test_fused_indexer_q_rope_quant.py

[Model Refactoring] Move deepseek_v4_ops to models/deepseek_v4 [3/N] (#43073 )

2026-05-19 00:52:54 -07:00

test_fused_inv_rope_fp8_quant.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423 )

2026-07-22 14:43:07 -06:00

test_fused_minimax_m3_qknorm_rope_kv_insert.py

[CI] Wire untethered test files into CI jobs (#49340 )

2026-07-28 14:02:31 +00:00

test_fused_qk_norm_rope_gate.py

[Perf] fuse qk rmsnorm rope gate for qwen3.5 (#44176 )

2026-06-09 22:12:17 +08:00

test_fused_quant_activation.py

Add platform manual_seed_all API (#38468 )

2026-04-10 13:43:50 +08:00

test_fused_recurrent_packed_decode.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423 )

2026-07-22 14:43:07 -06:00

test_fused_sigmoid_gating_delta_rule.py

[CI] Fix stale/fragile untethered kernels-root tests (#49423 )

2026-07-22 14:43:07 -06:00

test_kda.py

[Refactor] Move fla to third party (#48500 )

2026-07-16 19:22:36 +01:00

test_ll_bf16_gemm.py

[Model] Add Inkling model support [1/N] (#48799 )

2026-07-15 23:40:07 -07:00

test_mhc_kernels.py

[ROCm][DSV4] Disable TileLang MHC dispatch on gfx942 (#45931 )

2026-06-22 09:26:54 +00:00

test_minimax_m3_amd_ops.py

[ROCm] fixed aiter master flag and expert parallelism compatibility on minimax-m3-mxfp8 (#47158 )

2026-07-08 15:26:17 +08:00

test_minimax_m3_sparse_attn_fp8_scale.py

[ROCm][MiniMax-M3] Add AITER sparse paged attention (#47287 )

2026-07-12 19:27:29 -07:00

test_onednn.py

[CPU] Refactor CPU attention backend (#27954 )

2025-11-12 09:43:06 +08:00

test_shuffle_rows.py

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

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

test_top_k_per_row.py

[Perf][DSv4/DSv3.2] Add cluster-cooperative topK kernel for low-latency scenarios (#43008 )

2026-06-23 16:11:00 -07:00

utils.py

[ROCm][CI] Relax fused layernorm quant test tolerances for one-ULP outliers (#46658 )

2026-06-27 10:44:29 +08:00