diff --git a/python/rmm/_cuda/stream.pyx b/python/rmm/_cuda/stream.pyx index f9d924e71..4c9890d51 100644 --- a/python/rmm/_cuda/stream.pyx +++ b/python/rmm/_cuda/stream.pyx @@ -94,10 +94,7 @@ cdef class Stream: return self.c_is_default() def _init_from_numba_stream(self, obj): - if obj.handle.value is None: - self._cuda_stream = cuda_stream_default.value() - else: - self._cuda_stream = (obj.handle.value) + self._cuda_stream = (int(obj)) self._owner = obj def _init_from_cupy_stream(self, obj):