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

Fixup

parent 19f36b03
No related branches found
No related tags found
No related merge requests found
Pipeline #20590 passed
...@@ -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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment