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
void	mlx::core::copy_gpu_inplace (const array &in, array &out, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)
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 &in, array &out, CopyType ctype, const Stream &s)
void	mlx::core::copy_gpu_inplace (const array &in, array &out, const Strides &i_strides, int64_t i_offset, CopyType ctype, const Stream &s)
void	mlx::core::fill_gpu (const array &val, array &out, const Stream &s)