20 #if defined(_HAS_CUDA_) 22 #include <cuda_runtime_api.h> 36 MemoryManager(
unsigned int size, memory_t mem_type, device_t device_id);
79 magmadnn_error_t
copy_from_host(T *src,
unsigned int begin_idx,
unsigned int size);
82 #if defined(_HAS_CUDA_) 88 magmadnn_error_t copy_from_device(T *src,
unsigned int begin_idx,
unsigned int size);
95 magmadnn_error_t copy_from_managed(T *host_src, T *device_src,
unsigned int begin_idx,
unsigned int size);
102 magmadnn_error_t copy_from_cudamanaged(T *src,
unsigned int begin_idx,
unsigned int size);
112 magmadnn_error_t
sync(
bool gpu_was_modified=
true);
118 magmadnn_error_t
set_device(device_t device_id);
124 T
get(
unsigned int idx)
const;
130 void set(
unsigned int idx, T val);
137 #if defined(_HAS_CUDA_) 146 T* get_cuda_managed_ptr();
170 #if defined(_HAS_CUDA_) 178 void init_cuda_managed();
187 #if defined(_HAS_CUDA_) ~MemoryManager()
Definition: memorymanager.cpp:77
MemoryManager & operator=(const MemoryManager &that)
Definition: memorymanager.cpp:66
unsigned int get_size() const
Definition: memorymanager.h:158
magmadnn_error_t copy_from(const MemoryManager< T > &src, unsigned int begin_idx, unsigned int size)
Definition: memorymanager.cpp:94
T * get_ptr()
Definition: memorymanager.cpp:337
T * get_host_ptr()
Definition: memorymanager.cpp:320
memory_t get_memory_type() const
Definition: memorymanager.h:163
magmadnn_error_t sync(bool gpu_was_modified=true)
Definition: memorymanager.cpp:238
magmadnn_error_t copy_from_host(T *src, unsigned int begin_idx, unsigned int size)
Definition: memorymanager.cpp:127
magmadnn_error_t set_device(device_t device_id)
Definition: memorymanager.cpp:304
MemoryManager(unsigned int size, memory_t mem_type, device_t device_id)
Definition: memorymanager.cpp:14
Definition: memorymanager.h:29