/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/host/device_memory.hpp Source File#
device_memory.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
__global__ void set_buffer_value(T *p, T x, uint64_t buffer_element_size)
Definition: device_memory.hpp:14
Manages device memory allocation and host-device data transfers.
Definition: device_memory.hpp:52
void ToDevice(const void *p, const std::size_t cpySize) const
Definition: device_memory.hpp:108
std::size_t mMemSize
size of device buffer in bytes
Definition: device_memory.hpp:192
void FromDevice(void *p, const std::size_t cpySize) const
Definition: device_memory.hpp:127
HostTensor< T > ToHost(std::size_t cpySize)
Definition: device_memory.hpp:137
void * mpDeviceBuf
pointer to device buffer
Definition: device_memory.hpp:191
Definition: host_tensor.hpp:336