1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
|
//------------------------------------------------------------------------------
// rmm_wrap/rmm_wrap.hpp
//------------------------------------------------------------------------------
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#pragma once
#include "stddef.h"
#include <cuda.h>
//#include <rmm/cuda_stream.hpp>
//#include <rmm/device_buffer.hpp>
//#include <rmm/detail/aligned.hpp>
#include <rmm/mr/host/host_memory_resource.hpp>
#include <rmm/mr/host/new_delete_resource.hpp>
#include <rmm/mr/host/pinned_memory_resource.hpp>
#include <rmm/mr/device/owning_wrapper.hpp>
#include <rmm/mr/device/device_memory_resource.hpp>
#include <rmm/mr/device/cuda_memory_resource.hpp>
#include <rmm/mr/device/polymorphic_allocator.hpp>
#include <rmm/mr/device/thread_safe_resource_adaptor.hpp>
#include <rmm/mr/device/managed_memory_resource.hpp>
#include <rmm/mr/device/pool_memory_resource.hpp>
#include <rmm/mr/device/limiting_resource_adaptor.hpp>
#include <rmm/cuda_stream_view.hpp>
#include <rmm/cuda_stream_pool.hpp>
#include <cstdint>
#include <memory>
#include <memory_resource>
#include <unordered_map>
#include "rmm_wrap.h"
typedef rmm::mr::new_delete_resource host_mr;
typedef rmm::mr::pinned_memory_resource pinned_mr;
typedef rmm::mr::cuda_memory_resource device_mr;
typedef rmm::mr::managed_memory_resource managed_mr;
typedef rmm::mr::pool_memory_resource<rmm::mr::device_memory_resource> pool_mr;
typedef rmm::mr::pool_memory_resource<host_mr> host_pool_mr;
typedef rmm::mr::pool_memory_resource<pinned_mr> host_pinned_pool_mr;
typedef rmm::mr::pool_memory_resource<device_mr> device_pool_mr;
typedef rmm::mr::pool_memory_resource<managed_mr> managed_pool_mr;
typedef std::unordered_map< std::size_t, std::size_t> alloc_map;
typedef rmm::cuda_stream_pool cuda_stream_pool;
typedef rmm::cuda_stream_view cuda_stream_view;
|