We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4e255fd commit 9bb2fcbCopy full SHA for 9bb2fcb
2 files changed
kernel_tuner/backends/nvcuda.py
@@ -13,7 +13,11 @@
13
try:
14
from cuda.bindings import driver, runtime, nvrtc
15
except ImportError:
16
- driver = None
+ 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
21
22
23
class CudaFunctions(GPUBackend):
kernel_tuner/observers/nvcuda.py
@@ -3,7 +3,11 @@
3
4
from cuda.bindings import runtime
5
6
- cuda = None
7
8
+ from cuda import cudart as runtime
9
10
+ cuda = None
11
12
from kernel_tuner.observers.observer import BenchmarkObserver
from kernel_tuner.utils.nvcuda import cuda_error_check
0 commit comments