From 4678c91d94ac00a8a60f7279189e8378109b4712 Mon Sep 17 00:00:00 2001 From: Vukasin Milovanovic Date: Fri, 28 Feb 2025 16:43:25 -0800 Subject: [PATCH] more --- cpp/src/utilities/host_memory.cpp | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/cpp/src/utilities/host_memory.cpp b/cpp/src/utilities/host_memory.cpp index d261048ca80..e41d772a479 100644 --- a/cpp/src/utilities/host_memory.cpp +++ b/cpp/src/utilities/host_memory.cpp @@ -25,6 +25,7 @@ #include #include #include + namespace cudf { namespace { @@ -48,17 +49,14 @@ class fixed_pinned_pool_memory_resource { pool_size_{rmm::align_up(size, rmm::CUDA_ALLOCATION_ALIGNMENT)}, pool_{new host_pooled_mr(upstream_mr_, pool_size_, pool_size_)} { + CUDF_LOG_INFO("Pinned pool size = %zu", pool_size_); + // Allocate full size from the pinned pool to figure out the beginning and end address pool_begin_ = pool_->allocate_async(pool_size_, stream_); pool_end_ = static_cast(static_cast(pool_begin_) + pool_size_); pool_->deallocate_async(pool_begin_, pool_size_, stream_); } - fixed_pinned_pool_memory_resource(fixed_pinned_pool_memory_resource const&) = delete; - fixed_pinned_pool_memory_resource(fixed_pinned_pool_memory_resource&&) = delete; - fixed_pinned_pool_memory_resource& operator=(fixed_pinned_pool_memory_resource const&) = delete; - fixed_pinned_pool_memory_resource& operator=(fixed_pinned_pool_memory_resource&&) = delete; - void* allocate_async(std::size_t bytes, std::size_t alignment, cuda::stream_ref stream) { if (bytes <= pool_size_) {