Skip to content

Commit 6929f8b

Browse files
authored
[Misc] fixed nvfp4_moe test failures due to invalid kwargs (#21246)
Signed-off-by: Yang Chen <[email protected]>
1 parent 32ec9e2 commit 6929f8b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/kernels/moe/test_nvfp4_moe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,11 +93,11 @@ def test_cutlass_fp4_moe_no_graph(m: int, n: int, k: int, e: int, topk: int,
9393
a1_gscale=a1_gs,
9494
w1_fp4=w1_q,
9595
w1_blockscale=w1_blockscale,
96-
w1_alphas=(1 / w1_gs),
96+
g1_alphas=(1 / w1_gs),
9797
a2_gscale=a2_gs,
9898
w2_fp4=w2_q,
9999
w2_blockscale=w2_blockscale,
100-
w2_alphas=(1 / w2_gs),
100+
g2_alphas=(1 / w2_gs),
101101
topk_weights=topk_weights,
102102
topk_ids=topk_ids,
103103
m=m,

0 commit comments

Comments
 (0)