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;
|
