test/hqq/test_hqq_affine.py

zxd1997066 · zxd1997066 · commit 7d32cdd12144 · 2025-12-05T10:17:39.000+08:00
diff --git a/test/hqq/test_hqq_affine.py b/test/hqq/test_hqq_affine.py
@@ -14,7 +14,7 @@
     ZeroPointDomain,
     quantize_,
 )
-from torchao.testing.utils import skip_if_rocm
+from torchao.testing.utils import skip_if_rocm, skip_if_xpu
 from torchao.utils import get_current_accelerator_device
 
 cuda_available = torch.accelerator.is_available()
@@ -116,6 +116,7 @@ def test_hqq_plain_5bit(self):
         )
 
     @skip_if_rocm("ROCm enablement in progress")
+    @skip_if_xpu("XPU enablement in progress")
     def test_hqq_plain_4bit(self):
         self._test_hqq(
             dtype=torch.uint4,

Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@`
`14`	`14`	`ZeroPointDomain,`
`15`	`15`	`quantize_,`
`16`	`16`	`)`
`17`		`-from torchao.testing.utils import skip_if_rocm`
	`17`	`+from torchao.testing.utils import skip_if_rocm, skip_if_xpu`
`18`	`18`	`from torchao.utils import get_current_accelerator_device`
`19`	`19`
`20`	`20`	`cuda_available = torch.accelerator.is_available()`
`@@ -116,6 +116,7 @@ def test_hqq_plain_5bit(self):`
`116`	`116`	`)`
`117`	`117`
`118`	`118`	`@skip_if_rocm("ROCm enablement in progress")`
	`119`	`+ @skip_if_xpu("XPU enablement in progress")`
`119`	`120`	`def test_hqq_plain_4bit(self):`
`120`	`121`	`self._test_hqq(`
`121`	`122`	`dtype=torch.uint4,`