pinned_memory.hpp

/*
 * SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>
 
#include <optional>
 
namespace CUDF_EXPORT cudf {
 
rmm::host_device_async_resource_ref set_pinned_memory_resource(
  rmm::host_device_async_resource_ref mr);
 
rmm::host_device_async_resource_ref get_pinned_memory_resource();
 
struct pinned_mr_options {
  std::optional<size_t> pool_size;  
};
 
bool config_default_pinned_memory_resource(pinned_mr_options const& opts);
 
void set_kernel_pinned_copy_threshold(size_t threshold);
 
size_t get_kernel_pinned_copy_threshold();
 
void set_allocate_host_as_pinned_threshold(size_t threshold);
 
size_t get_allocate_host_as_pinned_threshold();
 
}  // namespace CUDF_EXPORT cudf