diff --git a/paddle/fluid/pybind/distributed_py.cc b/paddle/fluid/pybind/distributed_py.cc index cc06861d8d052d04c4b744d618fb12c5359a11f2..0b4a14d51b016d4f2377d79d4ff39318d298a63a 100644 --- a/paddle/fluid/pybind/distributed_py.cc +++ b/paddle/fluid/pybind/distributed_py.cc @@ -1251,7 +1251,9 @@ void BindDistributed(py::module *m) { py::arg("rank"), py::arg("world_size"), py::arg("group_id") = 0, - py::call_guard()); + py::call_guard()) + .def_static("group_start", distributed::ProcessGroupBKCL::GroupStart) + .def_static("group_end", distributed::ProcessGroupBKCL::GroupEnd); #endif py::class_