Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
sandyhouse committed Mar 30, 2022
1 parent 01a3cf6 commit 3ef390b
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions paddle/fluid/pybind/distributed_py.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -215,7 +215,8 @@ void BindDistributed(py::module *m) {
*m, "ProcessGroupNCCL", ProcessGroup)
.def(py::init<const std::shared_ptr<distributed::Store> &, int, int,
int>(),
py::call_guard<py::gil_scoped_release>());
py::arg("store"), py::arg("rank"), py::arg("world_size"),
py::arg("group_id") = 0, py::call_guard<py::gil_scoped_release>());
#endif

#if defined(PADDLE_WITH_ASCEND_CL)
Expand All @@ -224,7 +225,8 @@ void BindDistributed(py::module *m) {
*m, "ProcessGroupHCCL", ProcessGroup)
.def(py::init<const std::shared_ptr<distributed::Store> &, int, int,
int>(),
py::call_guard<py::gil_scoped_release>());
py::arg("store"), py::arg("rank"), py::arg("world_size"),
py::arg("group_id") = 0, py::call_guard<py::gil_scoped_release>());
#endif

py::class_<distributed::ProcessGroup::Task,
Expand Down

1 comment on commit 3ef390b

@paddle-bot-old
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

✅Congratulation! Your pull request passed all required CI. You could ask reviewer(s) to approve and merge. 🎉

Please sign in to comment.