diff --git a/kernel_tuner/backends/cupy.py b/kernel_tuner/backends/cupy.py index 51613be7..2eaa94b9 100644 --- a/kernel_tuner/backends/cupy.py +++ b/kernel_tuner/backends/cupy.py @@ -10,6 +10,7 @@ # and run tests without cupy installed try: import cupy as cp + from cupyx import get_runtime_info except ImportError: cp = None @@ -68,7 +69,7 @@ def __init__(self, device=0, iterations=7, compiler_options=None, observers=None # collect environment information env = dict() - cupy_info = str(cp._cupyx.get_runtime_info()).split("\n")[:-1] + cupy_info = str(get_runtime_info()).split("\n")[:-1] info_dict = { s.split(":")[0].strip(): s.split(":")[1].strip() for s in cupy_info }