From 5aff2699bdcedd9ee91fe936fc21b26466203ae1 Mon Sep 17 00:00:00 2001 From: Wei Zhao <51183510+wzhao18@users.noreply.github.com> Date: Wed, 11 Feb 2026 17:17:16 -0500 Subject: [PATCH] Fix CI failure - Flashinfer Kernel tests (#34316) Signed-off-by: wzhao18 --- tests/kernels/moe/test_flashinfer.py | 1 + tests/kernels/moe/test_flashinfer_moe.py | 1 + tests/kernels/moe/test_pplx_cutlass_moe.py | 1 + 3 files changed, 3 insertions(+) diff --git a/tests/kernels/moe/test_flashinfer.py b/tests/kernels/moe/test_flashinfer.py index ddcd221ef..c5d34ef0b 100644 --- a/tests/kernels/moe/test_flashinfer.py +++ b/tests/kernels/moe/test_flashinfer.py @@ -287,6 +287,7 @@ def test_flashinfer_cutlass_moe_fp8_no_graph( hidden_dim=k, intermediate_size_per_partition=n, num_local_experts=e, + num_logical_experts=e, activation=activation, device="cuda", moe_parallel_config=FusedMoEParallelConfig.make_no_parallel(), diff --git a/tests/kernels/moe/test_flashinfer_moe.py b/tests/kernels/moe/test_flashinfer_moe.py index 113649afe..c61bca313 100644 --- a/tests/kernels/moe/test_flashinfer_moe.py +++ b/tests/kernels/moe/test_flashinfer_moe.py @@ -97,6 +97,7 @@ def test_flashinfer_fp4_moe_no_graph( hidden_dim=k, intermediate_size_per_partition=n, num_local_experts=e, + num_logical_experts=e, activation=activation, device="cuda", moe_parallel_config=FusedMoEParallelConfig.make_no_parallel(), diff --git a/tests/kernels/moe/test_pplx_cutlass_moe.py b/tests/kernels/moe/test_pplx_cutlass_moe.py index 213d28cda..894e57fe2 100644 --- a/tests/kernels/moe/test_pplx_cutlass_moe.py +++ b/tests/kernels/moe/test_pplx_cutlass_moe.py @@ -147,6 +147,7 @@ def pplx_cutlass_moe( hidden_dim=hidden_dim, intermediate_size_per_partition=intermediate_dim, num_local_experts=num_local_experts, + num_logical_experts=num_experts, moe_parallel_config=FusedMoEParallelConfig.make_no_parallel(), activation="silu", in_dtype=torch.bfloat16,