Skip to content

Commit 331c1f1

Browse files
committed
Use the new cuda-python modules.
1 parent 59d3293 commit 331c1f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernel_tuner/backends/nvcuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# embedded in try block to be able to generate documentation
1111
# and run tests without cuda-python installed
1212
try:
13-
from cuda import cuda, cudart, nvrtc
13+
from cuda.bindings import driver, runtime, nvrtc
1414
except ImportError:
1515
cuda = None
1616

0 commit comments

Comments
 (0)