We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3760194 commit 805b254Copy full SHA for 805b254
1 file changed
cuda_bindings/cuda/bindings/driver.pyx.in
@@ -6908,8 +6908,11 @@ cdef class CUstream:
6908
return <void_ptr>self._pvt_ptr[0]
6909
def getPtr(self):
6910
return <void_ptr>self._pvt_ptr
6911
+ def __cuda_stream__(self):
6912
+ return (0, <uintptr_t><void_ptr>(self._pvt_ptr[0]))
6913
{{endif}}
6914
6915
+
6916
{{if 'CUgraphicsResource' in found_types}}
6917
6918
cdef class CUgraphicsResource:
0 commit comments