Commit 8abd3aab authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fixup

parent 19f36b03
Pipeline #20590 passed with stage
in 5 minutes and 4 seconds
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment