Skip to content

Commit 1777304

Browse files
白永斌845473182
authored andcommitted
fix unquantized ut
Signed-off-by: 白永斌 <[email protected]> Signed-off-by: 欧派果奶我还要 <[email protected]>
1 parent 2627e98 commit 1777304

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/ut/ops/test_fused_moe.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -409,9 +409,9 @@ def test_unified_apply_mlp_without_quantization(self,
409409
topk_scales = torch.randn(10, 1, dtype=torch.float16)
410410

411411
result = unified_apply_mlp(hidden_states=hidden_states,
412-
w1=[w1],
412+
w1=w1,
413413
w1_scale=None,
414-
w2=[w2],
414+
w2=w2,
415415
w2_scale=None,
416416
group_list=group_list,
417417
dynamic_scale=None,
@@ -512,9 +512,9 @@ def test_unified_apply_mlp_without_quantization_310p(
512512
topk_scales = torch.randn(10, 1, dtype=torch.float16)
513513

514514
result = unified_apply_mlp(hidden_states=hidden_states,
515-
w1=[w1],
515+
w1=w1,
516516
w1_scale=None,
517-
w2=[w2],
517+
w2=w2,
518518
w2_scale=None,
519519
group_list=group_list,
520520
dynamic_scale=None,

0 commit comments

Comments
 (0)