-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathhamr_cuda_malloc_host_allocator.h
143 lines (111 loc) · 4.45 KB
/
hamr_cuda_malloc_host_allocator.h
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
#ifndef hamr_cuda_malloc_host_allocator_h
#define hamr_cuda_malloc_host_allocator_h
#include "hamr_config.h"
#include <type_traits>
#include <memory>
namespace hamr
{
/// a deleter for arrays allocated with cudaMallocHost
template <typename T, typename E = void>
class cuda_malloc_host_deleter {};
/// a deleter for arrays allocated with cudaMallocHost, specialized for objects
template <typename T>
class HAMR_EXPORT cuda_malloc_host_deleter<T, typename std::enable_if<!std::is_arithmetic<T>::value>::type>
{
public:
/** constructs the deleter
* @param[in] ptr the pointer to the array to delete
* @param[in] n the number of elements in the array
*/
cuda_malloc_host_deleter(T *ptr, size_t n);
/** deletes the array
* @param[in] ptr the pointer to the array to delete. must be the same as
* that passed during construction.
*/
void operator()(T *ptr);
private:
T *m_ptr;
size_t m_elem;
};
/// a deleter for arrays allocated with cudaMallocHost, specialized for numbers
template <typename T>
class HAMR_EXPORT cuda_malloc_host_deleter<T, typename std::enable_if<std::is_arithmetic<T>::value>::type>
{
public:
/** constructs the deleter
* @param[in] ptr the pointer to the array to delete
* @param[in] n the number of elements in the array
*/
cuda_malloc_host_deleter(T *ptr, size_t n);
/** deletes the array
* @param[in] ptr the pointer to the array to delete. must be the same as
* that passed during construction.
*/
void operator()(T *ptr);
private:
T *m_ptr;
size_t m_elem;
};
/** A class for allocating arrays with cudaMallocHost. Use this allocator for
* host accessible memory when you want to overlap data movement and computation
* with CUDA.
*/
template <typename T, typename E = void>
struct cuda_malloc_host_allocator {};
/** a class for allocating arrays with cudaMallocHost, specialized for objects
* Use this allocator for host accessible memory when you want to overlap data movement and computation
* with CUDA
*/
template <typename T>
struct HAMR_EXPORT cuda_malloc_host_allocator<T, typename std::enable_if<!std::is_arithmetic<T>::value>::type>
{
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @returns a shared pointer to the array that holds a deleter for the memory
*/
static std::shared_ptr<T> allocate(size_t n) HAMR_EXPORT;
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @param[in] val a value to initialize the elements to
* @returns a shared pointer to the array that holds a deleter for the memory
*/
static std::shared_ptr<T> allocate(size_t n, const T &val) HAMR_EXPORT;
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @param[in] vals an array of n elements to initialize the elements with
* @returns a shared pointer to the array that holds a deleter for the memory
*/
template <typename U>
static std::shared_ptr<T> allocate(size_t n, const U *vals) HAMR_EXPORT;
};
/** a class for allocating arrays with cudaMallocHost, specialized for numbers.
* Use this allocator for host accessible memory when you want to overlap data
* movement and computation with CUDA
*/
template <typename T>
struct HAMR_EXPORT cuda_malloc_host_allocator<T, typename std::enable_if<std::is_arithmetic<T>::value>::type>
{
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @returns a shared pointer to the array that holds a deleter for the memory
*/
static std::shared_ptr<T> allocate(size_t n) HAMR_EXPORT;
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @param[in] val a value to initialize the elements to
* @returns a shared pointer to the array that holds a deleter for the memory
*/
static std::shared_ptr<T> allocate(size_t n, const T &val) HAMR_EXPORT;
/** allocate an array of n elements.
* @param[in] n the number of elements to allocate
* @param[in] vals an array of n elements to initialize the elements with
* @returns a shared pointer to the array that holds a deleter for the memory
*/
template <typename U>
static std::shared_ptr<T> allocate(size_t n, const U *vals) HAMR_EXPORT;
};
}
#if !defined(HAMR_SEPARATE_IMPL)
#include "hamr_cuda_malloc_host_allocator_impl.h"
#endif
#endif