MLX
Loading...
Searching...
No Matches
Functions
copy.h File Reference

Go to the source code of this file.

Functions

template<typename T , typename U >
void copy_s (device const T *src, device U *dst, uint index)
 
template<typename T , typename U >
void copy_v (device const T *src, device U *dst, uint index)
 
template<typename T , typename U >
void copy_s2 (device const T *src, device U *dst, uint2 index, uint2 grid_dim)
 
template<typename T , typename U >
void copy_v2 (device const T *src, device U *dst, uint2 index, uint2 grid_dim)
 
template<typename T , typename U >
void copy_g_nd1 (device const T *src, device U *dst, constant const int64_t &src_stride, uint index)
 
template<typename T , typename U >
void copy_g_nd2 (device const T *src, device U *dst, constant const int64_t *src_strides, uint2 index, uint2 grid_dim)
 
template<typename T , typename U >
void copy_g_nd3 (device const T *src, device U *dst, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
 
template<typename T , typename U , int DIM>
void copy_g_nd (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
 
template<typename T , typename U >
void copy_g (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
 
template<typename T , typename U >
void copy_gg_nd1 (device const T *src, device U *dst, constant const int64_t &src_stride, constant const int64_t &dst_stride, uint index)
 
template<typename T , typename U >
void copy_gg_nd2 (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint2 index)
 
template<typename T , typename U >
void copy_gg_nd3 (device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
 
template<typename T , typename U , int DIM>
void copy_gg_nd (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
 
template<typename T , typename U >
void copy_gg (device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, constant const int &ndim, uint3 index)
 

Function Documentation

◆ copy_g()

template<typename T , typename U >
void copy_g ( device const T * src,
device U * dst,
constant const int * src_shape,
constant const int64_t * src_strides,
constant const int & ndim,
uint3 index,
uint3 grid_dim )

◆ copy_g_nd()

template<typename T , typename U , int DIM>
void copy_g_nd ( device const T * src,
device U * dst,
constant const int * src_shape,
constant const int64_t * src_strides,
uint3 index,
uint3 grid_dim )

◆ copy_g_nd1()

template<typename T , typename U >
void copy_g_nd1 ( device const T * src,
device U * dst,
constant const int64_t & src_stride,
uint index )

◆ copy_g_nd2()

template<typename T , typename U >
void copy_g_nd2 ( device const T * src,
device U * dst,
constant const int64_t * src_strides,
uint2 index,
uint2 grid_dim )

◆ copy_g_nd3()

template<typename T , typename U >
void copy_g_nd3 ( device const T * src,
device U * dst,
constant const int64_t * src_strides,
uint3 index,
uint3 grid_dim )

◆ copy_gg()

template<typename T , typename U >
void copy_gg ( device const T * src,
device U * dst,
constant const int * src_shape,
constant const int64_t * src_strides,
constant const int64_t * dst_strides,
constant const int & ndim,
uint3 index )

◆ copy_gg_nd()

template<typename T , typename U , int DIM>
void copy_gg_nd ( device const T * src,
device U * dst,
constant const int * src_shape,
constant const int64_t * src_strides,
constant const int64_t * dst_strides,
uint3 index )

◆ copy_gg_nd1()

template<typename T , typename U >
void copy_gg_nd1 ( device const T * src,
device U * dst,
constant const int64_t & src_stride,
constant const int64_t & dst_stride,
uint index )

◆ copy_gg_nd2()

template<typename T , typename U >
void copy_gg_nd2 ( device const T * src,
device U * dst,
constant const int64_t * src_strides,
constant const int64_t * dst_strides,
uint2 index )

◆ copy_gg_nd3()

template<typename T , typename U >
void copy_gg_nd3 ( device const T * src,
device U * dst,
constant const int64_t * src_strides,
constant const int64_t * dst_strides,
uint3 index )

◆ copy_s()

template<typename T , typename U >
void copy_s ( device const T * src,
device U * dst,
uint index )

◆ copy_s2()

template<typename T , typename U >
void copy_s2 ( device const T * src,
device U * dst,
uint2 index,
uint2 grid_dim )

◆ copy_v()

template<typename T , typename U >
void copy_v ( device const T * src,
device U * dst,
uint index )

◆ copy_v2()

template<typename T , typename U >
void copy_v2 ( device const T * src,
device U * dst,
uint2 index,
uint2 grid_dim )