Tuning code

7 years ago · e6bc7bb09a
parent 0ca28b8507
commit e6bc7bb09a
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/pybind/pybind.cc
+++ b/paddle/fluid/pybind/pybind.cc
@ -510,6 +510,11 @@ All parameter, weight, gradient are variables in Paddle.
             return &self.GetLocalScopes();
           },
           py::return_value_policy::reference)
+      .def("local_scopes_len",
+           [](ParallelExecutor &self) { return self.GetLocalScopes().size(); })
+      .def("local_scope", [](ParallelExecutor &self,
+                             size_t i) { return self.GetLocalScopes()[i]; },
+           py::return_value_policy::reference)
      .def("run", &ParallelExecutor::Run);

  BindRecordIOWriter(&m);