vLLM/tests/kernels at ebbb2d55ace74b7066bca0ff8f333012bb8c4299 - vLLM

attention

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

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

core

[Kernel] Add weightless RMSNorm CUDA kernels for has_weight=False (#41430 ) (#44109 )

2026-06-16 23:45:55 -07:00

helion

[Kernel][Helion][1/N] Add Helion kernel for rms_norm_dynamic_per_token_quant (#34432 )

2026-06-17 23:03:54 +08:00

ir

[ROCm][CI] Fix ROCm LoRA Transformers fallback with full CUDA graphs (#41577 )

2026-05-23 04:31:32 +00:00

mamba

[Kernel][Test] Make kernel tests for mamba dual-HW (CUDA + XPU) (#42736 )

2026-06-08 08:22:47 +08:00

moe

[Log] Update deepgemm log (#45857 )

2026-06-17 15:34:22 +00:00

quantization

[Refactor] Remove dead quantization code and tests (#45454 )

2026-06-17 16:12:01 -04: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

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_cache_kernels.py

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

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

test_compressor_kv_cache.py

[DSv4] Adding TRTLLM gen attention kernel (#43827 )

2026-06-04 07:35:09 -07: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

[Attention][Perf] Optimize cp_gather_and_upconvert_fp8_kv_cache - DeepSeek-v3.2 (#35290 )

2026-03-09 09:46:57 -07:00

test_fla_layernorm_guard.py

[Bugfix] Fix dtype mismatch in RMSNormGated.forward_native() during torch.compile (#35256 )

2026-03-01 15:14:46 -05:00

test_flex_attention.py

[FlexAttention] make custom mask mods fully cudagraphable (#45232 )

2026-06-17 11:53:12 +08:00

test_fp32_router_gemm.py

[Model] Add MiniMax M3 support (#45381 )

2026-06-16 01:01:25 +08:00

test_fused_deepseek_v4_qnorm_rope_kv_insert.py

[DSv4] Adding TRTLLM gen attention kernel (#43827 )

2026-06-04 07:35:09 -07:00

test_fused_gdn_post_conv.py

[Perf] fuse kernels in gdn (#37813 )

2026-04-02 11:52:18 +00: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

[Feat][DSV4] Fuse q pad into deepseek v4 fused kernel (#43162 )

2026-05-26 05:12:54 -10:00

test_fused_minimax_m3_qknorm_rope_kv_insert.py

[M3] Enable FP8 sparse GQA (#45744 )

2026-06-16 21:38:03 -07: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

[Bugfix] fix main branch pre-commit error (1 line change) (#36897 )

2026-03-12 09:08:37 -07:00

test_fused_sigmoid_gating_delta_rule.py

[Kernel] Add fused_sigmoid_gating_delta_rule_update kernel for Qwen3 Next (#35777 )

2026-03-08 23:41:01 -07:00

test_kda.py

[Perf][KDA] Fuse gate softplus, chunk-local cumsum, and RCP_LN2 scaling (#43667 )

2026-05-28 13:47:08 +00:00

test_mhc_kernels.py

[DSv4] Move mHC tilelang kernels & Don't use CustomOP in dsv4/nvidia (#43905 )

2026-05-29 10:25:02 +08:00

test_minimax_m3_amd_ops.py

[Model] Add MiniMax M3 support (#45381 )

2026-06-16 01:01:25 +08: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

[Feat] DeepSeek V4 Rebased (#40860 )

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

utils.py

[ROCm] Fix MoE kernel test failures on gfx950 (#37833 )

2026-03-25 13:46:40 -05:00