From 72f361336ddcce3d4745de6701c6b1298f2d230c Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Sun, 17 May 2026 22:35:14 +0000 Subject: [PATCH] Use async APIs in random MR ref tests --- cpp/tests/mr/mr_ref_test.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cpp/tests/mr/mr_ref_test.hpp b/cpp/tests/mr/mr_ref_test.hpp index 4b9ad7360..841e8a6d9 100644 --- a/cpp/tests/mr/mr_ref_test.hpp +++ b/cpp/tests/mr/mr_ref_test.hpp @@ -257,14 +257,14 @@ inline void test_random_async_allocations(rmm::device_async_resource_ref ref, allocations.end(), [&generator, &distribution, &ref, stream](allocation& alloc) { alloc.size = distribution(generator); - EXPECT_NO_THROW(alloc.ptr = ref.allocate_sync(alloc.size, rmm::CUDA_ALLOCATION_ALIGNMENT)); + EXPECT_NO_THROW(alloc.ptr = ref.allocate(stream, alloc.size, rmm::CUDA_ALLOCATION_ALIGNMENT)); if (not stream.is_default()) { stream.synchronize(); } EXPECT_NE(nullptr, alloc.ptr); EXPECT_TRUE(is_properly_aligned(alloc.ptr)); }); std::for_each(allocations.begin(), allocations.end(), [stream, &ref](allocation& alloc) { - EXPECT_NO_THROW(ref.deallocate_sync(alloc.ptr, alloc.size, rmm::CUDA_ALLOCATION_ALIGNMENT)); + EXPECT_NO_THROW(ref.deallocate(stream, alloc.ptr, alloc.size, rmm::CUDA_ALLOCATION_ALIGNMENT)); if (not stream.is_default()) { stream.synchronize(); } }); }