Files
mlx/mlx/distributed/ibv/ibv.h
Angelos Katharopoulos 873df2e0f7 Initial working all reduce
2025-11-20 12:36:16 -08:00

13 lines
297 B
C++

// Copyright © 2025 Apple Inc.
#include "mlx/distributed/distributed.h"
namespace mlx::core::distributed::ibv {
using GroupImpl = mlx::core::distributed::detail::GroupImpl;
bool is_available();
std::shared_ptr<GroupImpl> init(bool strict = false);
} // namespace mlx::core::distributed::ibv