build/html/distributed_2ops_8h_source.html

// Copyright © 2024 Apple Inc.


#pragma once


#include <optional>


#include "mlx/distributed/distributed.h"


namespace mlx::core::distributed {


array all_sum(const array& x, std::optional<Group> group = std::nullopt);

array all_gather(const array& x, std::optional<Group> group = std::nullopt);


} // namespace mlx::core::distributed

mlx::core::array
Definition array.h:20

distributed.h

mlx::core::distributed
Definition distributed.h:9

mlx::core::distributed::all_gather
array all_gather(const array &x, std::optional< Group > group=std::nullopt)

mlx::core::distributed::all_sum
array all_sum(const array &x, std::optional< Group > group=std::nullopt)