| commit | 9ca8b321f5d8914652f3fa4a43cfd8e4bc251ff3 | [log] [tgz] |
|---|---|---|
| author | Edward Z. Yang <ezyang@fb.com> | Thu Nov 02 02:21:10 2017 -0400 |
| committer | Edward Z. Yang <ezyang@fb.com> | Thu Nov 02 02:21:10 2017 -0400 |
| tree | 6e995f616a2add8e2280911a62059193fd8feb45 | |
| parent | 53b01527f446c1940e2dc1c2023575e51d93919c [diff] |
Skip cpp tests if CUDA not available. Signed-off-by: Edward Z. Yang <ezyang@fb.com>
diff --git a/test/test_jit.py b/test/test_jit.py index 2bfbc34..3de9903 100644 --- a/test/test_jit.py +++ b/test/test_jit.py
@@ -613,6 +613,7 @@ 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()