Sum< U > Struct Template Reference

#include <ops.h>

Public Member Functions
template<typename T >
T	simd_reduce (T val)
template<typename T >
void	atomic_update (device mlx_atomic< T > *out, T val, uint offset=0)
U	operator() (U a, U b)

Static Public Attributes
static constexpr constant U	init = U(0)

Member Function Documentation

atomic_update()

template<typename U >

template<typename T >

void Sum< U >::atomic_update ( device mlx_atomic< T > * out, T val, uint offset = 0 )

inline

operator()()

template<typename U >

U Sum< U >::operator() ( U a, U b )

inline

simd_reduce()

template<typename U >

template<typename T >

T Sum< U >::simd_reduce ( T val )

inline

Member Data Documentation

init

template<typename U >

constexpr constant U Sum< U >::init = U(0)

staticconstexpr

---

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

• mlx/backend/metal/kernels/reduction/ops.h