Update on "Set up CI"

[ghstack-poisoned]
pytorch · Apr 18, 2024 · bfce802 · bfce802
2 parents 5d90b73 + 67a4709
commit bfce802
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_extension.py b/test/test_extension.py
@@ -62,6 +62,7 @@ def _opcheck(self, device):
     def test_opcheck_cpu(self):
         self._opcheck("cpu")
 
+    @unittest.skipIf(not torch.cuda.is_available(), "requires cuda")
     def test_opcheck_cuda(self):
         self._opcheck("cuda")