We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2cf7971 commit b1a6bafCopy full SHA for b1a6baf
1 file changed
cuda_bindings/cuda/bindings/driver.pyx.in
@@ -7101,8 +7101,11 @@ cdef class CUstream:
7101
return <void_ptr>self._pvt_ptr[0]
7102
def getPtr(self):
7103
return <void_ptr>self._pvt_ptr
7104
+ def __cuda_stream__(self):
7105
+ return (0, <uintptr_t><void_ptr>(self._pvt_ptr[0]))
7106
{{endif}}
7107
7108
+
7109
{{if 'CUgraphicsResource' in found_types}}
7110
7111
cdef class CUgraphicsResource:
0 commit comments