Skip to content

Commit

Permalink
Replace raw device_memory_resource pointer in pylibcudf Cython (#16674)
Browse files Browse the repository at this point in the history
Replaces a single `device_memory_resource*` in pylibcudf Cython inline C++ function with `rmm::device_async_resource_ref` to help smooth RMM refactoring effort.

Authors:
  - Mark Harris (https://github.com/harrism)

Approvers:
  - Vyas Ramasubramani (https://github.com/vyasr)
  - Lawrence Mitchell (https://github.com/wence-)

URL: #16674
  • Loading branch information
harrism authored Aug 28, 2024
1 parent fbd6114 commit 9e9efcc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/pylibcudf/pylibcudf/libcudf/interop.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ cdef extern from *:
ArrowArray* to_arrow_host_raw(
cudf::table_view const& tbl,
rmm::cuda_stream_view stream = cudf::get_default_stream(),
rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource()) {
rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource()) {
// Assumes the sync event is null and the data is already on the host.
ArrowArray *arr = new ArrowArray();
auto device_arr = cudf::to_arrow_host(tbl, stream, mr);
Expand Down

0 comments on commit 9e9efcc

Please sign in to comment.