Skip to content

Commit

Permalink
Re-enable test_8da4w_quantizer on 2.3 (#107)
Browse files Browse the repository at this point in the history
* Re-enable test_8da4w_quantizer on 2.3

* Update test_quant_api.py

* Update test_quant_api.py
  • Loading branch information
cpuhrsch authored Apr 30, 2024
1 parent e3ed90f commit 894312d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/quantization/test_quant_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def test_int8_wo_quant_save_load(self):

torch.testing.assert_close(ref, res.cpu())

@unittest.skipIf(not TORCH_VERSION_AFTER_2_4, "skipping when torch verion is 2.4 or lower")
@unittest.skipIf(not TORCH_VERSION_AFTER_2_3, "skipping when torch verion is 2.3 or lower")
def test_8da4w_quantizer(self):
from torchao.quantization.quant_api import Int8DynActInt4WeightQuantizer
from torchao.quantization.GPTQ import Int8DynActInt4WeightLinear
Expand Down

0 comments on commit 894312d

Please sign in to comment.