distributed.h File Reference

#include <memory>
#include "mlx/array.h"

Go to the source code of this file.

Classes
struct	mlx::core::distributed::Group
	A distributed::Group represents a group of independent mlx processes that can communicate. More...

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

Functions
bool	mlx::core::distributed::is_available ()
Group	mlx::core::distributed::init (bool strict=false)
	Initialize the distributed backend and return the group containing all discoverable processes.
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)