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

Fixup

parent 19f36b03
No related merge requests found
Pipeline #20590 passed with stage
in 5 minutes and 4 seconds
...@@ -150,7 +150,9 @@ def test_torch_native_compilation_gpu(): ...@@ -150,7 +150,9 @@ def test_torch_native_compilation_gpu():
assert 'call_backward' in dir(torch_extension) 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): def test_execute_torch(target):
import pycuda.autoinit import pycuda.autoinit
module_name = "Ololol" + target module_name = "Ololol" + target
......
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