Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion cpp/src/memory/buffer_resource.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@ std::unique_ptr<rmm::device_buffer> BufferResource::move_to_device_buffer(
auto stream = buffer->stream();
auto ret = move(std::move(buffer), reservation)->release_device_buffer();
RAPIDSMPF_EXPECTS(
ret->stream().value() == stream.get(),
ret->stream().get() == stream.get(),
"something went wrong, the Buffer's stream and the device_buffer's stream "
"don't match"
);
Expand Down
4 changes: 2 additions & 2 deletions python/rapidsmpf/rapidsmpf/cuda_stream.pyx
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES.
# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

from rmm.pylibrmm.stream cimport Stream
Expand All @@ -19,4 +19,4 @@ def is_equal_streams(Stream s1 not None, Stream s2 not None):
-------
``True`` if both inputs reference the same stream; ``False`` otherwise.
"""
return s1.view().value() == s2.view().value()
return s1.view().get() == s2.view().get()
2 changes: 1 addition & 1 deletion python/rapidsmpf/rapidsmpf/memory/buffer_resource.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -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().value())
cdef stream_ref cpp_stream = stream_ref(stream.view().get())
with nogil:
handle = move(
deref(self._handle).make_buffer(
Expand Down
2 changes: 1 addition & 1 deletion python/rapidsmpf/rapidsmpf/memory/packed_data.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -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().value())
cdef stream_ref sv = stream_ref(stream.view().get())
cdef unique_ptr[device_buffer] gpu = move(gpu_data.c_obj)
cdef PackedData ret = cls.__new__(cls)
with nogil:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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().value())
cdef stream_ref cpp_stream = stream_ref(stream.view().get())
with nogil:
ptr = self._handle.value().allocate(cpp_stream, nbytes)
return <size_t>ptr
Expand All @@ -163,7 +163,7 @@ cdef class PinnedMemoryResource:
stream
CUDA stream associated with the allocation.
"""
cdef stream_ref cpp_stream = stream_ref(stream.view().value())
cdef stream_ref cpp_stream = stream_ref(stream.view().get())
with nogil:
self._handle.value().deallocate(cpp_stream, <void*>ptr, nbytes)

Expand Down