vllm/tests/kernels/moe at 73f48ce559e230fd0d738c52cb2e99bd0dd08754 - vllm

modular_kernel_tools

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

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

__init__.py

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

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

parallel_utils.py

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

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

test_batched_deepgemm.py

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

2026-03-03 10:39:50 -08: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

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

2026-03-12 14:24:38 -04: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

[Torch 2.11] Guard torch._C._cpu attribute checks for forward compatibility (#35673 )

2026-03-17 18:47:59 +00:00

test_cutedsl_moe.py

Revert "[MoE Kernel] Flashinfer nvfp4 cutedsl moe kernel integration" (#38050 ) (#38169 )

2026-03-26 07:59:09 -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] Move PF Methods to Folder (#35927 )

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

test_deepep_moe.py

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

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

test_deepgemm.py

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

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

test_flashinfer_moe.py

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

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

test_flashinfer.py

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

2026-03-12 14:24:38 -04: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

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

2026-03-25 13:46:40 -05: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

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

2026-03-03 10:39: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 Refactor] Create MK for TRTLLM Kernels (#32564 )

2026-03-03 10:39:50 -08: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_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

[Bugfix] Fix FusedMoE weight loading with padded hidden dimensions (#37010 )

2026-03-31 12:22:26 -04:00

test_moe.py

[Kernel] Add MXFP8 to Marlin GEMM/MoE and refactor Mxfp8LinearOp (#34664 )

2026-04-01 09:41:42 -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_router_gemm.py

[Kernel] Add gpt-oss Router GEMM kernel (#37205 )

2026-03-18 08:15:56 -07: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

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

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

test_silu_mul_fp8_quant_deep_gemm.py

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

2026-03-25 13:46:40 -05: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

[MoE Refactor] Migrate Unquantized to Full Oracle Flow (#36286 )

2026-03-31 15:43:33 -04:00

utils.py

[MoE Refactor] Make SharedExperts class for use with DefaultMoERunner (#35153 )

2026-04-01 09:44:08 -04:00