Use cuda.bindings layout.

rapidsai · Jan 17, 2025 · b8591b2 · b8591b2
1 parent 097ac45
commit b8591b2
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/python/pylibraft/pylibraft/common/cuda.pxd b/python/pylibraft/pylibraft/common/cuda.pxd
@@ -14,7 +14,7 @@
 # limitations under the License.
 #
 
-from cuda.ccudart cimport cudaStream_t
+from cuda.bindings.cyruntime cimport cudaStream_t
 
 
 cdef class Stream:

diff --git a/python/pylibraft/pylibraft/common/cuda.pyx b/python/pylibraft/pylibraft/common/cuda.pyx
@@ -19,7 +19,7 @@
 # cython: embedsignature = True
 # cython: language_level = 3
 
-from cuda.ccudart cimport (
+from cuda.bindings.cyruntime cimport (
     cudaError_t,
     cudaGetErrorName,
     cudaGetErrorString,

diff --git a/python/pylibraft/pylibraft/common/handle.pyx b/python/pylibraft/pylibraft/common/handle.pyx
@@ -21,7 +21,7 @@
 
 import functools
 
-from cuda.ccudart cimport cudaStream_t
+from cuda.bindings.cyruntime cimport cudaStream_t
 from libc.stdint cimport uintptr_t
 
 from rmm.librmm.cuda_stream_view cimport (

diff --git a/python/pylibraft/pylibraft/common/interruptible.pyx b/python/pylibraft/pylibraft/common/interruptible.pyx
@@ -22,7 +22,7 @@
 import contextlib
 import signal
 
-from cuda.ccudart cimport cudaStream_t
+from cuda.bindings.cyruntime cimport cudaStream_t
 from cython.operator cimport dereference
 
 from rmm.librmm.cuda_stream_view cimport cuda_stream_view