diff --git a/test/test_extension.py b/test/test_extension.py index 348ac9e..7bc29f4 100644 --- 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")