diff --git a/tests/test_native_tensorflow_compilation.py b/tests/test_native_tensorflow_compilation.py index ad13bbb482f3a849053b8bcfadc59b79f6bfb4e5..f473afc66d23ad161d1ad5db8528f5142e24f4f1 100644 --- a/tests/test_native_tensorflow_compilation.py +++ b/tests/test_native_tensorflow_compilation.py @@ -71,6 +71,7 @@ def test_native_tensorflow_compilation_cpu(): @pytest.mark.skipif("TRAVIS" in os.environ, reason="nvcc compilation currently not working on TRAVIS") +@pytest.mark.skipif("GITLAB_CI" in os.environ, reason="linking against cudart does not work on GITLAB_CI") def test_native_tensorflow_compilation_gpu(): tf = pytest.importorskip('tensorflow') diff --git a/tests/test_tensorflow_jit.py b/tests/test_tensorflow_jit.py index 0f21d8d4f607ef57097e7d269c26368fb170e781..2b623745ad920142dffde03972e8ca8779705e20 100644 --- a/tests/test_tensorflow_jit.py +++ b/tests/test_tensorflow_jit.py @@ -11,14 +11,15 @@ from os.path import exists import pytest -import sympy import pystencils import pystencils_autodiff +import sympy from pystencils_autodiff import create_backward_assignments from pystencils_autodiff.backends.astnodes import TensorflowModule +@pytest.mark.xfail(reason="cannot link against cudart on GITLAB CI", strict=False) def test_tensorflow_jit_gpu(): pytest.importorskip('tensorflow')