Skip to content

Commit e02305c

Browse files
committed
fix(ut): fix ut
Signed-off-by: zhoux77899 <zhouxiang100@huawei.com>
1 parent dde4126 commit e02305c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/ut/ops/test_fused_ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -841,6 +841,7 @@ def test_unified_apply_mlp_with_quantization_and_fusion_mlp(
841841
w1_scale_bias=w1_scale_bias,
842842
w2_scale_bias=w2_scale_bias,
843843
topk_scales=None,
844+
with_quant=True,
844845
fusion=True)
845846

846847
mock_get_forward_context.assert_called()

0 commit comments

Comments
 (0)