ops.h File Reference

#include <optional>
#include "mlx/distributed/distributed.h"
#include "mlx/utils.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::distributed

Functions
array	mlx::core::distributed::all_sum (const array &x, std::optional< Group > group=std::nullopt, StreamOrDevice s={})
array	mlx::core::distributed::all_gather (const array &x, std::optional< Group > group=std::nullopt, StreamOrDevice S={})
array	mlx::core::distributed::send (const array &x, int dst, std::optional< Group > group=std::nullopt, StreamOrDevice s={})
array	mlx::core::distributed::recv (std::vector< int > shape, Dtype dtype, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})
array	mlx::core::distributed::recv_like (const array &x, int src, std::optional< Group > group=std::nullopt, StreamOrDevice s={})