fix python gil problem

PaddlePaddle · Dec 6, 2021 · f933e14 · f933e14 · paddle-bot-old · Dec 6, 2021
1 parent 7b1bb87
commit f933e14
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/pybind/bind_fleet_executor.cc b/paddle/fluid/pybind/bind_fleet_executor.cc
@@ -33,7 +33,8 @@ void BindFleetExecutor(py::module* m) {
   py::class_<FleetExecutor>(*m, "FleetExecutor")
       .def(py::init<const std::string&>())
       .def("init", &FleetExecutor::Init)
-      .def("run", &FleetExecutor::Run);
+      .def("run", &FleetExecutor::Run,
+           py::call_guard<py::gil_scoped_release>());
 
   py::class_<TaskNode>(*m, "TaskNode")
       .def(py::init<const framework::ProgramDesc&, int64_t, int64_t, int64_t>())