Skip to content

Commit f4986e4

Browse files
committed
skip failed distributed on xpu
1 parent 6697253 commit f4986e4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/dtypes/test_nf4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -757,7 +757,7 @@ def world_size(self) -> int:
757757
return 2
758758

759759
@skip_if_lt_x_gpu(2)
760-
@unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
760+
@unittest.skipIf(not torch.cuda.is_available(), "Need GPU available")
761761
def test_comm(self):
762762
self.run_subtests(
763763
{"input_size": [512, 2048]},

0 commit comments

Comments
 (0)