Update on "Set up CI"

[ghstack-poisoned]

Author: zou3519

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")