Add MPI barrier

This commit is contained in:
Angelos Katharopoulos
2024-11-01 11:41:53 -07:00
parent 26be608470
commit c3ccd4919f
4 changed files with 17 additions and 1 deletions

View File

@@ -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",