Fix CI failure - Flashinfer Kernel tests (#34316)

Signed-off-by: wzhao18 <wzhao18.sz@gmail.com>
2026-02-11 17:17:16 -05:00
parent 527ca32197
commit 5aff2699bd
3 changed files with 3 additions and 0 deletions
--- 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(),