Skip to content

Commit 434609a

Browse files
committed
Fix non-gptoss tests failing
Signed-off-by: Dongfeng Yu <[email protected]>
1 parent d974c3c commit 434609a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/unittest/_torch/thop/parallel/test_moe.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1483,8 +1483,13 @@ def run_moe_fp4_test(self, num_tokens: int, hidden_size: int,
14831483
hidden_states_scale_linear_fp4,
14841484
gemm1_weights_fp4_shuffled,
14851485
gemm1_scales_fp4_shuffled,
1486+
None, # Bias
1487+
None, # Alpha
1488+
None, # Beta
1489+
None, # Limit
14861490
gemm2_weights_fp4_shuffled,
14871491
gemm2_scales_fp4_shuffled,
1492+
None, # Bias
14881493
scale_c_fc1,
14891494
scale_gate_fc1,
14901495
scale_c_fc2,

0 commit comments

Comments
 (0)