Skip to content

Commit d764f5b

Browse files
committed
improve __cuda_stream__ performance
1 parent b713c5c commit d764f5b

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

cuda_core/cuda/core/experimental/_stream.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ cdef class Stream:
214214

215215
def __cuda_stream__(self) -> tuple[int, int]:
216216
"""Return an instance of a __cuda_stream__ protocol."""
217-
return (0, int(self.handle))
217+
return (0, <uintptr_t>(self._handle))
218218

219219
@property
220220
def handle(self) -> cuda.bindings.driver.CUstream:

0 commit comments

Comments
 (0)