diff --git a/tests/test_post_train_quant.py b/tests/test_post_train_quant.py index d54ea330a14c8b898e83819d663074fcee0dcdd4..200ea26530fe1167e3d6879806b6a585895f871f 100644 --- a/tests/test_post_train_quant.py +++ b/tests/test_post_train_quant.py @@ -803,4 +803,4 @@ def test_linear_layer_wrapper_params_only(per_channel, bias): x = torch.rand(5, in_features) - _test_wts_only_quant(layer, x, per_channel, bias) + _test_wts_only_quant(layer, x, per_channel, bias, 8)