diff --git a/tests/backends/test_torch_native_compilation.py b/tests/backends/test_torch_native_compilation.py index b0c0dc6aee4368f4110498536a82de78b1f97586..bf72109a294879c22bd83caab3c22a0dc5233157 100644 --- a/tests/backends/test_torch_native_compilation.py +++ b/tests/backends/test_torch_native_compilation.py @@ -164,7 +164,8 @@ def test_execute_torch(): assert op_cpp is not None -@pytest.skip('NO_GPU_EXECUTION' in os.environ, reason='Skip GPU execution tests') +@pytest.mark.skipif('NO_GPU_EXECUTION' in os.environ, reason='Skip GPU execution tests') + def test_execute_torch_gpu(): x, y = pystencils.fields('x, y: float64[32,32]')