Skip to content
Snippets Groups Projects
Commit 476eb2e5 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Skip nvcc tests on TRAVIS

parent 232ad850
No related branches found
No related tags found
No related merge requests found
Pipeline #17410 failed
......@@ -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
......
......@@ -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')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment