Skip to content

Commit 34f38ca

Browse files
Add WOQ int8 test with Inductor Freeze
1 parent bc2f8b7 commit 34f38ca

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test/integration/test_integration.py

Lines changed: 8 additions & 0 deletions
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)