[CustomDevice] fix release ProcessGroupCustom error when process exit

PaddlePaddle · Dec 7, 2023 · 67f4b25 · 67f4b25
1 parent ff9a5f6
commit 67f4b25
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/fluid/pybind/distributed_py.cc b/paddle/fluid/pybind/distributed_py.cc
@@ -1271,6 +1271,7 @@ void BindDistributed(py::module *m) {
                   py::arg("rank"),
                   py::arg("world_size"),
                   py::arg("group_id") = 0,
+                  py::return_value_policy::renference_internal,
                   py::call_guard<py::gil_scoped_release>());
 
 #endif