MLX
Loading...
Searching...
No Matches
reduce_all.h File Reference

Go to the source code of this file.

Functions

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)
 

Function Documentation

◆ 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 )