Shamrock 2025.10.0
Astrophysical Code
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
shamalgs::collective::DDSCommCache Struct Reference

Public Member Functions

template<sham::USMKindTarget target>
std::vector< std::unique_ptr< sham::DeviceBuffer< u8, target > > > & get_cache1 ()
 
template<sham::USMKindTarget target>
std::vector< std::unique_ptr< sham::DeviceBuffer< u8, target > > > & get_cache2 ()
 
template<sham::USMKindTarget target>
void set_sizes (sham::DeviceScheduler_ptr dev_sched, const std::vector< size_t > &sizes_cache1, const std::vector< size_t > &sizes_cache2)
 
void send_cache_write_buf_at (size_t buf_id, size_t offset, const sham::DeviceBuffer< u8 > &buf)
 
void send_cache_read_buf_at (size_t buf_id, size_t offset, size_t size, sham::DeviceBuffer< u8 > &buf)
 
void recv_cache_write_buf_at (size_t buf_id, size_t offset, const sham::DeviceBuffer< u8 > &buf)
 
void recv_cache_read_buf_at (size_t buf_id, size_t offset, size_t size, sham::DeviceBuffer< u8 > &buf)
 

Public Attributes

std::variant< DDSCommCacheTarget< sham::device >, DDSCommCacheTarget< sham::host > > cache
 

Detailed Description

Definition at line 97 of file distributedDataComm.hpp.

Member Function Documentation

◆ get_cache1()

template<sham::USMKindTarget target>
std::vector< std::unique_ptr< sham::DeviceBuffer< u8, target > > > & shamalgs::collective::DDSCommCache::get_cache1 ( )
inline

Definition at line 101 of file distributedDataComm.hpp.

◆ get_cache2()

template<sham::USMKindTarget target>
std::vector< std::unique_ptr< sham::DeviceBuffer< u8, target > > > & shamalgs::collective::DDSCommCache::get_cache2 ( )
inline

Definition at line 106 of file distributedDataComm.hpp.

◆ recv_cache_read_buf_at()

void shamalgs::collective::DDSCommCache::recv_cache_read_buf_at ( size_t  buf_id,
size_t  offset,
size_t  size,
sham::DeviceBuffer< u8 > &  buf 
)
inline

Definition at line 154 of file distributedDataComm.hpp.

◆ recv_cache_write_buf_at()

void shamalgs::collective::DDSCommCache::recv_cache_write_buf_at ( size_t  buf_id,
size_t  offset,
const sham::DeviceBuffer< u8 > &  buf 
)
inline

Definition at line 145 of file distributedDataComm.hpp.

◆ send_cache_read_buf_at()

void shamalgs::collective::DDSCommCache::send_cache_read_buf_at ( size_t  buf_id,
size_t  offset,
size_t  size,
sham::DeviceBuffer< u8 > &  buf 
)
inline

Definition at line 136 of file distributedDataComm.hpp.

◆ send_cache_write_buf_at()

void shamalgs::collective::DDSCommCache::send_cache_write_buf_at ( size_t  buf_id,
size_t  offset,
const sham::DeviceBuffer< u8 > &  buf 
)
inline

Definition at line 127 of file distributedDataComm.hpp.

◆ set_sizes()

template<sham::USMKindTarget target>
void shamalgs::collective::DDSCommCache::set_sizes ( sham::DeviceScheduler_ptr  dev_sched,
const std::vector< size_t > &  sizes_cache1,
const std::vector< size_t > &  sizes_cache2 
)
inline

Definition at line 111 of file distributedDataComm.hpp.

Member Data Documentation

◆ cache

std::variant<DDSCommCacheTarget<sham::device>, DDSCommCacheTarget<sham::host> > shamalgs::collective::DDSCommCache::cache

Definition at line 98 of file distributedDataComm.hpp.


The documentation for this struct was generated from the following file: