vllm/tests/kernels/moe at e729cc823d313aa7623ecadefe4305ea241c3dce - vllm

modular_kernel_tools

[MoE Refactor] Remove MoE DP chunking (#39107 )

2026-04-14 09:48:05 -04:00

__init__.py

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

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

conftest.py

[MoE Refactor][Test] FusedMoE layer test (#24675 )

2026-04-06 17:17:23 +00:00

parallel_utils.py

[MoE] Move PF Methods to Folder (#35927 )

2026-03-22 02:42:59 -06:00

test_batched_deepgemm.py

[MoE] Move DEEP_GEMM into experts/ subdirectory (#39005 )

2026-04-08 19:23:08 +00: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] Move DEEP_GEMM into experts/ subdirectory (#39005 )

2026-04-08 19:23:08 +00: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

[CPU] Support gelu act in cpu_fused_moe (#38770 )

2026-04-02 14:14:32 +08:00

test_cutedsl_moe.py

[Bugfix] [Tests] Enforce out tensor device in kernel/moe/test_cutedsl_moe.py (#39644 )

2026-04-12 17:08:08 -07:00

test_cutlass_moe.py

[Feature]: Remove Chunking From FusedMoE (#34086 )

2026-03-12 14:24:38 -04:00

test_cutlass_mxfp8_grouped_mm.py

[Kernel] Integrate SM100 MXFP8 blockscaled grouped MM and quant kernels (#34448 )

2026-03-01 23:31:19 -08:00

test_deepep_deepgemm_moe.py

[MoE Refactor] Remove MoE DP chunking (#39107 )

2026-04-14 09:48:05 -04:00

test_deepep_moe.py

[MoE] Move PF Methods to Folder (#35927 )

2026-03-22 02:42:59 -06:00

test_deepgemm.py

[MoE] Move DEEP_GEMM into experts/ subdirectory (#39005 )

2026-04-08 19:23:08 +00:00

test_flashinfer_moe.py

[MoE Refactor] Remove MoE DP chunking (#39107 )

2026-04-14 09:48:05 -04:00

test_flashinfer.py

[MoE Refactor] Remove MoE DP chunking (#39107 )

2026-04-14 09:48:05 -04:00

test_fused_topk.py

[Kernel] Add topk_sigmoid kernel (#31246 )

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

test_gemma4router.py

[FEAT] [Perf] [Gemma4] Fused Gemma4 Routing Function Triton (#39083 )

2026-04-19 09:57:39 +00:00

test_gpt_oss_triton_kernels.py

[MoE] Move GPT OSS Triton kernel experts into fused_moe/experts/ (#39007 )

2026-04-14 19:27:39 +00:00

test_grouped_topk.py

[Bugfix] Register VLLM_BATCH_INVARIANT in envs.py to fix spurious unknown env var warning (#35007 )

2026-03-23 18:31:14 -04:00

test_marlin_vs_trtllm_mxint4.py

Add platform manual_seed_all API (#38468 )

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

test_modular_kernel_combinations.py

replace cuda_device_count_stateless() to current_platform.device_count() (#37841 )

2026-03-31 22:32:54 +08:00

test_modular_oai_triton_moe.py

[MoE] Move GPT OSS Triton kernel experts into fused_moe/experts/ (#39007 )

2026-04-14 19:27:39 +00:00

test_moe_align_block_size.py

[Bugfix] Fix expert_ids padding values in moe_align_block_size kernel (#35161 )

2026-02-24 17:14:24 -08:00

test_moe_layer.py

[Bugfix] Temporarily disable B200 fp4 MoE layer tests (#40057 )

2026-04-16 19:26:55 -04:00

test_moe_permute_unpermute.py

[Refactor] Remove align block size logic in moe_permute (#33449 )

2026-02-06 10:57:06 -08:00

test_moe_weight_loading_padded.py

[fix][MOE] Fix MOE experts intermediate_size dimension not being narrowed before weight loading (#39688 )

2026-04-14 09:35:28 +00:00

test_moe.py

[CI] Speed up test_fused_marlin_moe (#40178 )

2026-04-17 19:26:21 -07:00

test_mxfp4_moe.py

[Kernel] Add MXFP4 W4A4 CUTLASS MoE kernel for SM100 (#37463 )

2026-04-17 16:42:32 -07:00

test_nvfp4_moe.py

[MoE Refactor] Create MK for TRTLLM Kernels (#32564 )

2026-03-03 10:39:50 -08:00

test_ocp_mx_moe.py

[MoE Refactor] Mxfp4 oracle rebased (#37128 )

2026-03-21 03:37:04 +00:00

test_rocm_aiter_topk.py

[CI][BugFix][AMD] Don't set VLLM_ROCM_USE_AITER anymore in test_rocm_aiter_topk since its not necessary (#36996 )

2026-03-20 07:12:45 +08:00

test_routing_simulator.py

[ROCm][CI] Fix AITER state leak in shared_fused_moe_routed_transform test (#38137 )

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

test_routing.py

[EPLB] Optmize eplb mapping and record in router for prefill (#36261 )

2026-03-30 19:48:33 +00:00

test_shared_fused_moe_routed_transform.py

Add platform manual_seed_all API (#38468 )

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

test_silu_mul_fp8_quant_deep_gemm.py

[UX] Integrate DeepGEMM into vLLM wheel via CMake (#37980 )

2026-04-08 18:56:32 -07: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

[Refactor] Replace activation: str with MoEActivation enum (#33843 )

2026-02-11 17:29:32 -08: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

test_unquantized_backend_selection.py

Fix MoE backend selection for LoRA (unquantized MoE) (#40273 )

2026-04-19 17:18:40 +00:00

test_zero_expert_moe.py

[MoE Refactor] Refactor ZeroExpertFusedMoE into new framework (#35549 )

2026-04-14 16:11:20 -04:00

utils.py

[MoE Refactor] Remove MoE DP chunking (#39107 )

2026-04-14 09:48:05 -04:00