pinned_memory.hpp

/*
 * 版权所有 (c) 2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可以在以下位置获得许可证的副本
 *
 * https://apache.ac.cn/licenses/LICENSE-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