diff --git a/cuda_bindings/cuda/bindings/driver.pyx.in b/cuda_bindings/cuda/bindings/driver.pyx.in index 2921fe181..33c1c770f 100644 --- a/cuda_bindings/cuda/bindings/driver.pyx.in +++ b/cuda_bindings/cuda/bindings/driver.pyx.in @@ -7101,8 +7101,11 @@ cdef class CUstream: return self._pvt_ptr[0] def getPtr(self): return self._pvt_ptr + def __cuda_stream__(self): + return (0, (self._pvt_ptr[0])) {{endif}} + {{if 'CUgraphicsResource' in found_types}} cdef class CUgraphicsResource: