Skip to content

Commit 9bb2fcb

Browse files
committed
Backward compatibility hack for older cuda-python versions.
1 parent 4e255fd commit 9bb2fcb

2 files changed

Lines changed: 10 additions & 2 deletions

File tree

kernel_tuner/backends/nvcuda.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,11 @@
1313
try:
1414
from cuda.bindings import driver, runtime, nvrtc
1515
except ImportError:
16-
driver = None
16+
try:
17+
# backward compatibility hack for older cuda-python versions
18+
from cuda import cuda as driver, cudart as runtime, nvrtc as nvrtc
19+
except ImportError:
20+
driver = None
1721

1822

1923
class CudaFunctions(GPUBackend):

kernel_tuner/observers/nvcuda.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,11 @@
33
try:
44
from cuda.bindings import runtime
55
except ImportError:
6-
cuda = None
6+
try:
7+
# backward compatibility hack for older cuda-python versions
8+
from cuda import cudart as runtime
9+
except ImportError:
10+
cuda = None
711

812
from kernel_tuner.observers.observer import BenchmarkObserver
913
from kernel_tuner.utils.nvcuda import cuda_error_check

0 commit comments

Comments
 (0)