From 9e9efcc9f5ed8411fb09f4d8384e14612a7f3b10 Mon Sep 17 00:00:00 2001 From: Mark Harris <783069+harrism@users.noreply.github.com> Date: Thu, 29 Aug 2024 09:19:48 +1000 Subject: [PATCH] Replace raw device_memory_resource pointer in pylibcudf Cython (#16674) 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: https://github.com/rapidsai/cudf/pull/16674 --- python/pylibcudf/pylibcudf/libcudf/interop.pxd | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pylibcudf/pylibcudf/libcudf/interop.pxd b/python/pylibcudf/pylibcudf/libcudf/interop.pxd index c7efff2340d..9228c017d93 100644 --- a/python/pylibcudf/pylibcudf/libcudf/interop.pxd +++ b/python/pylibcudf/pylibcudf/libcudf/interop.pxd @@ -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);