From 937b1a630e5eb068b2188fdcb8db240b548960af Mon Sep 17 00:00:00 2001 From: "Jiang, Yanbing" Date: Sun, 22 Sep 2024 19:04:16 -0700 Subject: [PATCH] Add test_int8_weight_only_quant_with_freeze back due to PyTorch fix --- test/integration/test_integration.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py index 6a5ea8ef9..3f1e0f474 100644 --- a/test/integration/test_integration.py +++ b/test/integration/test_integration.py @@ -829,9 +829,6 @@ def test_int8_weight_only_quant_subclass_api(self, device, dtype): @torch._inductor.config.patch({"freezing": True}) @unittest.skipIf(not TORCH_VERSION_AT_LEAST_2_4, "freeze requires torch 2.4 and after.") def test_int8_weight_only_quant_with_freeze(self, device, dtype): - if TORCH_VERSION_AT_LEAST_2_5 and device == "cpu": - self.skipTest("Regression introduced in PT nightlies") - self._test_lin_weight_subclass_api_impl( _int8wo_api, device, 40, test_dtype=dtype )