Skip to content

Commit e13e268

Browse files
committed
fix ut
Signed-off-by: Enwei Zhu <[email protected]>
1 parent 2b17db2 commit e13e268

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/modules/test_fused_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1370,7 +1370,7 @@ def test_fused_moe_nvfp4(dtype, moe_backend):
13701370
SEQ_LEN = 4
13711371
HIDDEN_SIZE = 512
13721372
INTERMEDIATE_SIZE = 512
1373-
NUM_EXPERTS = 4
1373+
NUM_EXPERTS = 8
13741374
TOP_K = 2
13751375
routing_method = RenormalizeMoeRoutingMethod(top_k=TOP_K)
13761376
torch.manual_seed(0)

0 commit comments

Comments
 (0)