skip tests that needs cuda

Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
weifengpy · Apr 27, 2024 · 857b8db · 857b8db
1 parent 9e6b4ec
commit 857b8db
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/dtypes/test_nf4.py b/test/dtypes/test_nf4.py
@@ -381,6 +381,7 @@ def test_pin_memory(self):
         nf4_tensor = to_nf4(torch.randn(512 * 512, device='cuda'))
         self.assertFalse(nf4_tensor.is_pinned())
 
+    @unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
     def test_to_cuda(self):
         nf4_tensor = to_nf4(torch.randn(512 * 512))
         self.assertEqual(nf4_tensor.device.type, "cpu")