vllm/tests/kernels/moe at 2b465570e6dd327e8422ef9c87e9b2b1454ceaed - vllm

modular_kernel_tools

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567 )

2026-01-27 01:28:02 +00:00

__init__.py

[Kernel] DeepEP dispatch-combine kernel integration (#18434 )

2025-06-03 12:30:02 -07:00

parallel_utils.py

[Chore] Separate out optional dependency checks from vllm.utils (#27207 )

2025-10-22 10:44:21 -04:00

test_batched_deepgemm.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_batched_moe.py

[MoE Refactor] Separate Router into OO Classes (#30623 )

2026-01-18 11:40:49 -05:00

test_block_fp8.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_block_int8.py

[Misc] Make SchedulerConfig.max_model_len init-only (#28733 )

2025-11-15 01:59:31 -08:00

test_count_expert_num_tokens.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633 )

2025-10-12 09:51:31 -07:00

test_cpu_fused_moe.py

[Bugfix] Fix _CPU_MOE_ACT AssertionError when vLLM config not set (#32777 )

2026-01-23 08:22:37 +00:00

test_cutedsl_moe.py

[MoE] Nvfp4 Masked Gemm: Add flashinfer grouped_gemm_nt_masked (#25990 )

2025-11-19 13:29:06 -08:00

test_cutlass_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_deepep_deepgemm_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_deepep_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_deepgemm.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_flashinfer_moe.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567 )

2026-01-27 01:28:02 +00:00

test_flashinfer.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567 )

2026-01-27 01:28:02 +00:00

test_fused_topk.py

[Kernel] Add topk_sigmoid kernel (#31246 )

2026-01-21 22:49:51 +00:00

test_gpt_oss_triton_kernels.py

[Attention] Use sparse prefill kernel for fp8 kv-cache in DeepSeek-v3.2 (#27532 )

2025-12-12 05:57:47 -08:00

test_grouped_topk.py

[MoE Refactor] Separate Router into OO Classes (#30623 )

2026-01-18 11:40:49 -05:00

test_marlin_vs_trtllm_mxint4.py

[Hardware][SM100] Add TRTLLM Kernel for INT4 W4A16 Kernel. (#32437 )

2026-01-30 10:30:46 -08:00

test_modular_kernel_combinations.py

[Platform] Deprecate seed_everything (#31659 )

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

test_modular_oai_triton_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_moe_align_block_size.py

AMD CI Test - unskip moe_sum test and moe_align_block_size tests (#32039 )

2026-01-13 23:25:10 -08:00

test_moe_permute_unpermute.py

[MoE Refactor] Separate Router into OO Classes (#30623 )

2026-01-18 11:40:49 -05:00

test_moe.py

Fix test_moe.py for Transformers v5 (#33413 )

2026-01-30 14:03:25 +00:00

test_nvfp4_moe.py

[MoE Refactor] Integrate Naive Prepare Finalize into MK (#32567 )

2026-01-27 01:28:02 +00:00

test_ocp_mx_moe.py

Bump Flashinfer to v0.6.1 (#30993 )

2026-01-21 08:49:50 -08:00

test_pplx_cutlass_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_pplx_moe.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_rocm_aiter_topk.py

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

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

test_routing_simulator.py

[MoE Refactor] Separate Router into OO Classes (#30623 )

2026-01-18 11:40:49 -05:00

test_routing.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_silu_mul_fp8_quant_deep_gemm.py

[Platform] Deprecate seed_everything (#31659 )

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

test_silu_mul_per_token_group_quant_fp8_colmajor.py

[Platform] Deprecate seed_everything (#31659 )

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

test_triton_moe_no_act_mul.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00

test_triton_moe_ptpc_fp8.py

[CI/Build] Only use supported types and features on ROCm in MoE kernel tests (#29149 )

2025-11-21 20:34:33 -07:00

utils.py

[MoE Refactor] Oracle Select FP8+NVFP4 Kernels In Priority (#32414 )

2026-01-21 08:22:33 -05:00