Skip cpp tests if CUDA not available.

Signed-off-by: Edward Z. Yang <[email protected]>
sungin-h · Nov 2, 2017 · 9ca8b32 · 9ca8b32
1 parent 53b0152
commit 9ca8b32
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_jit.py b/test/test_jit.py
@@ -613,6 +613,7 @@ def doit(x, y):
         assert(torch.equal(torch.ones([2, 2]), t_node.t("a")))
         self.assertExpected(str(g2))
 
+    @unittest.skipIf(not torch.cuda.is_available(), "cpp tests require CUDA")
     def test_cpp(self):
         torch._C._jit_run_cpp_tests()