Skip to content

Commit 950b4fb

Browse files
committed
Enable GPTQ on Intel GPU
1 parent 9fed009 commit 950b4fb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/dtypes/test_nf4.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -771,6 +771,7 @@ def world_size(self) -> int:
771771

772772
@skip_if_lt_x_gpu(2)
773773
@unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
774+
@unittest.skipIf(torch.xpu.is_available(), "XPU enablement in progress")
774775
def test_comm(self):
775776
self.run_subtests(
776777
{"input_size": [512, 2048]},

0 commit comments

Comments
 (0)