distributed_impl.h File Reference

#include "mlx/distributed/distributed.h"

Go to the source code of this file.

Classes
class	mlx::core::distributed::detail::GroupImpl
	Abstract base class of a distributed group implementation. More...

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

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