Skip to content

Commit 40577e0

Browse files
Merge pull request #376 from KernelTuner/363-attributeerror-module-cupy-has-no-attribute-_cupyx
Updated cupyx namespace usage
2 parents 57549ba + f55f9dd commit 40577e0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

kernel_tuner/backends/cupy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
# and run tests without cupy installed
1111
try:
1212
import cupy as cp
13+
from cupyx import get_runtime_info
1314
except ImportError:
1415
cp = None
1516

@@ -68,7 +69,7 @@ def __init__(self, device=0, iterations=7, compiler_options=None, observers=None
6869

6970
# collect environment information
7071
env = dict()
71-
cupy_info = str(cp._cupyx.get_runtime_info()).split("\n")[:-1]
72+
cupy_info = str(get_runtime_info()).split("\n")[:-1]
7273
info_dict = {
7374
s.split(":")[0].strip(): s.split(":")[1].strip() for s in cupy_info
7475
}

0 commit comments

Comments
 (0)