diff --git a/tests/backends/test_torch_native_compilation.py b/tests/backends/test_torch_native_compilation.py index c4db892234c9a4bd2f238612df49a9adb78b6747..1fb4eb76e93365b0d8f779a449ee05455b71fb6e 100644 --- a/tests/backends/test_torch_native_compilation.py +++ b/tests/backends/test_torch_native_compilation.py @@ -150,7 +150,9 @@ def test_torch_native_compilation_gpu(): assert 'call_backward' in dir(torch_extension) -@pytest.mark.skipif('CI' in os.environ, reason="GPU too old on GITLAB CI") +@pytest.mark.parametrize('target', ( + pytest.param('gpu', marks=pytest.mark.skipif('CI' in os.environ, reason="GPU too old on GITLAB CI")), + 'cpu')) def test_execute_torch(target): import pycuda.autoinit module_name = "Ololol" + target