From 079d24d729f26aaa3445dce92f8b12fcc9aaa288 Mon Sep 17 00:00:00 2001
From: Peter Andreas Entschev <peter@entschev.com>
Date: Fri, 11 Dec 2020 13:52:22 -0800
Subject: [PATCH] Simplify handling of Numba streams

---
 python/rmm/_cuda/stream.pyx | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

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 = <cudaStream_t><uintptr_t>(obj.handle.value)
+        self._cuda_stream = <cudaStream_t><uintptr_t>(int(obj))
         self._owner = obj
 
     def _init_from_cupy_stream(self, obj):