diff --git a/tests/backends/test_torch_native_compilation.py b/tests/backends/test_torch_native_compilation.py index c4899693ea1a0329e139c50640c34dd09f40a59a..d9b350701cf013a3682af42359713310f6f1d0d4 100644 --- a/tests/backends/test_torch_native_compilation.py +++ b/tests/backends/test_torch_native_compilation.py @@ -16,6 +16,7 @@ from pystencils_autodiff._file_io import _write_file from pystencils_autodiff.backends.astnodes import TorchModule +@pytest.mark.skipif("TRAVIS" in os.environ, reason="nvcc compilation currently not working on TRAVIS") def test_torch_jit(): """ Test JIT compilation from example on git@github.com:pytorch/extension-cpp.git @@ -66,6 +67,7 @@ def test_torch_native_compilation_cpu(): assert 'call_backward' in dir(torch_extension) +@pytest.mark.skipif("TRAVIS" in os.environ, reason="nvcc compilation currently not working on TRAVIS") def test_torch_native_compilation_gpu(): from torch.utils.cpp_extension import load diff --git a/tests/test_native_tensorflow_compilation.py b/tests/test_native_tensorflow_compilation.py index 6c61237b302546a7b0312d6f02b72492fff76eb5..c7ec86c4ac8ccae03b3640192f86f47b5c2791a4 100644 --- a/tests/test_native_tensorflow_compilation.py +++ b/tests/test_native_tensorflow_compilation.py @@ -84,6 +84,7 @@ def test_native_tensorflow_compilation_cpu(): assert 'call_backward' in dir(lib) +@pytest.mark.skipif("TRAVIS" in os.environ, reason="nvcc compilation currently not working on TRAVIS") def test_native_tensorflow_compilation_gpu(): tf = pytest.importorskip('tensorflow')