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

Go to the source code of this file.

Functions

template<typename T, typename U, typename Op>
void binary_ss (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T, typename U, typename Op>
void binary_sv (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T, typename U, typename Op>
void binary_vs (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T, typename U, typename Op>
void binary_vv (device const T *a, device const T *b, device U *c, device U *d, uint index)
 
template<typename T, typename U, typename Op>
void binary_sv2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
 
template<typename T, typename U, typename Op>
void binary_vs2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
 
template<typename T, typename U, typename Op>
void binary_vv2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
 
template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd1 (device const T *a, device const T *b, device U *c, device U *d, constant const int64_t &a_stride, constant const int64_t &b_stride, uint index)
 
template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd2 (device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], uint2 index, uint2 grid_dim)
 
template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd3 (device const T *a, device const T *b, device U *c, device U *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], uint3 index, uint3 grid_dim)
 
template<typename T, typename U, typename Op, int N = 1, typename IdxT = int64_t>
void binary_g (device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
 

Function Documentation

◆ binary_g()

template<typename T, typename U, typename Op, int N = 1, typename IdxT = int64_t>
void binary_g ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int * shape,
constant const int64_t * a_strides,
constant const int64_t * b_strides,
constant const int & ndim,
uint3 index,
uint3 grid_dim )

◆ binary_g_nd1()

template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd1 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int64_t & a_stride,
constant const int64_t & b_stride,
uint index )

◆ binary_g_nd2()

template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd2 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int64_t a_strides[2],
constant const int64_t b_strides[2],
uint2 index,
uint2 grid_dim )

◆ binary_g_nd3()

template<typename T, typename U, typename Op, typename IdxT = int64_t>
void binary_g_nd3 ( device const T * a,
device const T * b,
device U * c,
device U * d,
constant const int64_t a_strides[3],
constant const int64_t b_strides[3],
uint3 index,
uint3 grid_dim )

◆ binary_ss()

template<typename T, typename U, typename Op>
void binary_ss ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_sv()

template<typename T, typename U, typename Op>
void binary_sv ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_sv2()

template<typename T, typename U, typename Op>
void binary_sv2 ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint2 index,
uint2 grid_dim )

◆ binary_vs()

template<typename T, typename U, typename Op>
void binary_vs ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_vs2()

template<typename T, typename U, typename Op>
void binary_vs2 ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint2 index,
uint2 grid_dim )

◆ binary_vv()

template<typename T, typename U, typename Op>
void binary_vv ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint index )

◆ binary_vv2()

template<typename T, typename U, typename Op>
void binary_vv2 ( device const T * a,
device const T * b,
device U * c,
device U * d,
uint2 index,
uint2 grid_dim )