mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-07 20:58:17 +08:00
Add MPI barrier
This commit is contained in:
@@ -44,7 +44,8 @@ void init_distributed(nb::module_& parent_module) {
|
||||
color (int): A value to group processes into subgroups.
|
||||
key (int, optional): A key to optionally change the rank ordering
|
||||
of the processes.
|
||||
)pbdoc");
|
||||
)pbdoc")
|
||||
.def("barrier", &distributed::Group::barrier, "Make a synhronization point for all nodes in the group");
|
||||
|
||||
m.def(
|
||||
"is_available",
|
||||
|
||||
Reference in New Issue
Block a user