Use rmm::device_async_resource_ref in multi_stream_allocation bench…

…mark (#1482)

We want to move away from using raw `device_memory_resource*` in code, so adapt the code to use `rmm::device_async_resource_ref`.

Fixes #1481

Authors:
  - Michael Schellenberger Costa (https://github.com/miscco)

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

URL: #1482

benchmarks/multi_stream_allocations/multi_stream_allocations_bench.cu

@@ -25,6 +25,7 @@
 #include <rmm/mr/device/owning_wrapper.hpp>
 #include <rmm/mr/device/per_device_resource.hpp>
 #include <rmm/mr/device/pool_memory_resource.hpp>
+#include <rmm/resource_ref.hpp>
 
 #include <cuda_runtime_api.h>
 
@@ -50,7 +51,7 @@ __global__ void compute_bound_kernel(int64_t* out)
 
 using MRFactoryFunc = std::function<std::shared_ptr<rmm::mr::device_memory_resource>()>;
 
-static void run_prewarm(rmm::cuda_stream_pool& stream_pool, rmm::mr::device_memory_resource* mr)
+static void run_prewarm(rmm::cuda_stream_pool& stream_pool, rmm::device_async_resource_ref mr)
 {
   auto buffers = std::vector<rmm::device_uvector<int64_t>>();
   for (int32_t i = 0; i < stream_pool.get_pool_size(); i++) {
@@ -61,7 +62,7 @@ static void run_prewarm(rmm::cuda_stream_pool& stream_pool, rmm::mr::device_memo
 
 static void run_test(std::size_t num_kernels,
                      rmm::cuda_stream_pool& stream_pool,
-                     rmm::mr::device_memory_resource* mr)
+                     rmm::device_async_resource_ref mr)
 {
   for (int32_t i = 0; i < num_kernels; i++) {
     auto stream = stream_pool.get_stream(i);