build/html/mpi_8h_source.html

// Copyright © 2024 Apple Inc.


#include "mlx/distributed/distributed.h"


namespace mlx::core::distributed::mpi {


using GroupImpl = mlx::core::distributed::detail::GroupImpl;


bool is_available();

std::shared_ptr<GroupImpl> init(bool strict = false);


} // namespace mlx::core::distributed::mpi


mlx::core::distributed::detail::GroupImpl
Abstract base class of a distributed group implementation.
Definition distributed_impl.h:12

distributed.h

mlx::core::distributed::mpi
Definition mpi.h:5

mlx::core::distributed::mpi::is_available
bool is_available()

mlx::core::distributed::mpi::init
std::shared_ptr< GroupImpl > init(bool strict=false)

mlx::core::distributed::mpi::GroupImpl
mlx::core::distributed::detail::GroupImpl GroupImpl
Definition mpi.h:7