copy.h File Reference

#include "mlx/backend/common/copy.h"
#include "mlx/stream.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
template<typename stride_t >
void	mlx::core::copy_gpu_inplace (const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)
void	mlx::core::copy_gpu (const array &src, array &out, CopyType ctype, const Stream &s)
void	mlx::core::copy_gpu (const array &src, array &out, CopyType ctype)
void	mlx::core::copy_gpu_inplace (const array &src, array &out, CopyType ctype, const Stream &s)
void	mlx::core::copy_gpu_inplace (const array &in, array &out, const std::vector< int64_t > &istride, int64_t ioffset, CopyType ctype, const Stream &s)
void	mlx::core::fill_gpu (const array &val, array &out, const Stream &s)