distributed_impl.h File Reference

#include "mlx/distributed/distributed.h"

Go to the source code of this file.

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

Functions
Stream	mlx::core::distributed::detail::communication_stream ()
void	mlx::core::distributed::detail::all_sum (Group group, const array &input, array &output)
void	mlx::core::distributed::detail::all_gather (Group group, const array &input, array &output)
void	mlx::core::distributed::detail::send (Group group, const array &input, int dst)
	Send an array to the dst rank.
void	mlx::core::distributed::detail::recv (Group group, array &out, int src)
	Recv an array from the src rank.