Adapt Python stream access to cuda::stream_ref

Author: bdice

python/rapidsmpf/rapidsmpf/memory/buffer_resource.pyx

@@ -595,7 +595,7 @@ cdef class BufferResource:
             If ``size`` exceeds the reservation size.
         """
         cdef unique_ptr[cpp_Buffer] handle
-        cdef stream_ref cpp_stream = stream_ref(stream.view().get())
+        cdef stream_ref cpp_stream = stream.view()
         with nogil:
             handle = move(
                 deref(self._handle).make_buffer(

python/rapidsmpf/rapidsmpf/memory/packed_data.pyx

@@ -201,7 +201,7 @@ cdef class PackedData:
         cdef const uint8_t* meta_ptr = NULL
         if meta_size > 0:
             meta_ptr = <const uint8_t*>&metadata[0]
-        cdef stream_ref sv = stream_ref(stream.view().get())
+        cdef stream_ref sv = stream.view()
         cdef unique_ptr[device_buffer] gpu = move(gpu_data.c_obj)
         cdef PackedData ret = cls.__new__(cls)
         with nogil:

python/rapidsmpf/rapidsmpf/memory/pinned_memory_resource.pyx

@@ -145,7 +145,7 @@ cdef class PinnedMemoryResource:
         Integer address of the allocated memory.
         """
         cdef void* ptr
-        cdef stream_ref cpp_stream = stream_ref(stream.view().get())
+        cdef stream_ref cpp_stream = stream.view()
         with nogil:
             ptr = self._handle.value().allocate(cpp_stream, nbytes)
         return <size_t>ptr
@@ -163,7 +163,7 @@ cdef class PinnedMemoryResource:
         stream
             CUDA stream associated with the allocation.
         """
-        cdef stream_ref cpp_stream = stream_ref(stream.view().get())
+        cdef stream_ref cpp_stream = stream.view()
         with nogil:
             self._handle.value().deallocate(cpp_stream, <void*>ptr, nbytes)