diff --git a/pystencils_tests/test_opencl.py b/pystencils_tests/test_opencl.py index 2917d0dae2c00b6fc2c16071d94d8484cd82a8b6..adffeb4750a62f5bb08ded29fffe00ac5900f706 100644 --- a/pystencils_tests/test_opencl.py +++ b/pystencils_tests/test_opencl.py @@ -40,6 +40,8 @@ def test_print_opencl(): @pytest.mark.skipif(not HAS_OPENCL, reason="Test requires pyopencl") def test_opencl_jit_fixed_size(): + pytest.importorskip('pycuda') + z, y, x = pystencils.fields("z, y, x: [20,30]") assignments = pystencils.AssignmentCollection({ @@ -92,6 +94,8 @@ def test_opencl_jit_fixed_size(): @pytest.mark.skipif(not HAS_OPENCL, reason="Test requires pyopencl") def test_opencl_jit(): + pytest.importorskip('pycuda') + z, y, x = pystencils.fields("z, y, x: [2d]") assignments = pystencils.AssignmentCollection({ @@ -144,6 +148,8 @@ def test_opencl_jit(): @pytest.mark.skipif(not HAS_OPENCL, reason="Test requires pyopencl") def test_opencl_jit_with_parameter(): + pytest.importorskip('pycuda') + z, y, x = pystencils.fields("z, y, x: [2d]") a = sp.Symbol('a')