Skip to content

Commit 5576da6

Browse files
authored
Restore option to run testing without cupy installed. (#741)
1 parent f13f9eb commit 5576da6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

cuda_core/tests/test_launcher.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55
import os
66
import pathlib
77

8-
import cupy as cp
8+
try:
9+
import cupy as cp
10+
except ImportError:
11+
cp = None
912
import numpy as np
1013
import pytest
1114
from conftest import skipif_need_cuda_headers
@@ -209,6 +212,7 @@ def test_cooperative_launch():
209212
s.sync()
210213

211214

215+
@pytest.mark.skipif(cp is None, reason="cupy not installed")
212216
@pytest.mark.parametrize(
213217
"memory_resource_class",
214218
[

0 commit comments

Comments
 (0)