DeviceMem Struct Reference

DeviceMem Struct Reference#

Composable Kernel: ck_tile::DeviceMem Struct Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::DeviceMem Struct Reference

Container for storing data in GPU device memory. More...

#include <device_memory.hpp>

Public Member Functions

 DeviceMem ()
 
 DeviceMem (std::size_t mem_size)
 
template<typename T >
 DeviceMem (const HostTensor< T > &t)
 
void Realloc (std::size_t mem_size)
 
void * GetDeviceBuffer () const
 
std::size_t GetBufferSize () const
 
void ToDevice (const void *p) const
 
void ToDevice (const void *p, const std::size_t cpySize) const
 
void FromDevice (void *p) const
 
void FromDevice (void *p, const std::size_t cpySize) const
 
template<typename T >
HostTensor< T > ToHost (std::size_t cpySize)
 
template<typename T >
HostTensor< T > ToHost ()
 
void SetZero () const
 
template<typename T >
void SetValue (T x) const
 
 ~DeviceMem ()
 

Public Attributes

void * mpDeviceBuf
 
std::size_t mMemSize
 

Detailed Description

Container for storing data in GPU device memory.

Constructor & Destructor Documentation

◆ DeviceMem() [1/3]

ck_tile::DeviceMem::DeviceMem ( )
inline

◆ DeviceMem() [2/3]

ck_tile::DeviceMem::DeviceMem ( std::size_t  mem_size)
inline

◆ DeviceMem() [3/3]

template<typename T >
ck_tile::DeviceMem::DeviceMem ( const HostTensor< T > &  t)
inline

◆ ~DeviceMem()

ck_tile::DeviceMem::~DeviceMem ( )
inline

Member Function Documentation

◆ FromDevice() [1/2]

void ck_tile::DeviceMem::FromDevice ( void *  p) const
inline

◆ FromDevice() [2/2]

void ck_tile::DeviceMem::FromDevice ( void *  p,
const std::size_t  cpySize 
) const
inline

◆ GetBufferSize()

std::size_t ck_tile::DeviceMem::GetBufferSize ( ) const
inline

◆ GetDeviceBuffer()

void* ck_tile::DeviceMem::GetDeviceBuffer ( ) const
inline

◆ Realloc()

void ck_tile::DeviceMem::Realloc ( std::size_t  mem_size)
inline

◆ SetValue()

template<typename T >
void ck_tile::DeviceMem::SetValue ( T  x) const
inline

◆ SetZero()

void ck_tile::DeviceMem::SetZero ( ) const
inline

◆ ToDevice() [1/2]

void ck_tile::DeviceMem::ToDevice ( const void *  p) const
inline

◆ ToDevice() [2/2]

void ck_tile::DeviceMem::ToDevice ( const void *  p,
const std::size_t  cpySize 
) const
inline

◆ ToHost() [1/2]

template<typename T >
HostTensor<T> ck_tile::DeviceMem::ToHost ( )
inline

◆ ToHost() [2/2]

template<typename T >
HostTensor<T> ck_tile::DeviceMem::ToHost ( std::size_t  cpySize)
inline

Member Data Documentation

◆ mMemSize

std::size_t ck_tile::DeviceMem::mMemSize

◆ mpDeviceBuf

void* ck_tile::DeviceMem::mpDeviceBuf
The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/host/device_memory.hpp