Go to the source code of this file.
|
template<typename T , typename U , typename Op , int N_READS = REDUCE_N_READS> |
void | all_reduce (const device T *in, device U *out, const constant size_t &in_size, const constant size_t &row_size, uint3 gid, uint3 lid, uint3 lsize, uint simd_per_group, uint simd_lane_id, uint simd_group_id) |
|
◆ all_reduce()
template<typename T , typename U , typename Op , int N_READS = REDUCE_N_READS>
void all_reduce |
( |
const device T * | in, |
|
|
device U * | out, |
|
|
const constant size_t & | in_size, |
|
|
const constant size_t & | row_size, |
|
|
uint3 | gid, |
|
|
uint3 | lid, |
|
|
uint3 | lsize, |
|
|
uint | simd_per_group, |
|
|
uint | simd_lane_id, |
|
|
uint | simd_group_id ) |