ops.h File Reference

#include <optional>
#include "mlx/distributed/distributed.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)
array	mlx::core::distributed::all_gather (const array &x, std::optional< Group > group=std::nullopt)