Skip to content

Commit f5b6ec9

Browse files
Add WOQ int8 test with Inductor Freeze (#362)
1 parent cb3bd8c commit f5b6ec9

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test/integration/test_integration.py

+8
Original file line numberDiff line numberDiff line change
@@ -798,6 +798,14 @@ def test_int8_weight_only_quant_subclass_api(self, device, dtype):
798798
_int8wo_api, device, 40, test_dtype=dtype
799799
)
800800

801+
@parameterized.expand(COMMON_DEVICE_DTYPE)
802+
@torch._inductor.config.patch({"freezing": True})
803+
@unittest.skipIf(not TORCH_VERSION_AFTER_2_4, "freeze requires torch 2.4 and after.")
804+
def test_int8_weight_only_quant_with_freeze(self, device, dtype):
805+
self._test_lin_weight_subclass_api_impl(
806+
_int8wo_api, device, 40, test_dtype=dtype
807+
)
808+
801809
@parameterized.expand(COMMON_DEVICE_DTYPE)
802810
@unittest.skipIf(not TORCH_VERSION_AFTER_2_3, "int4 requires torch nightly.")
803811
def test_int4_weight_only_quant_subclass_api(self, device, dtype):

0 commit comments

Comments
 (0)