From d42117e7422facdbffbd77d3f5b2841fe6ad5ed9 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 20 Mar 2018 18:42:40 +0800
Subject: [PATCH] Set NumThreads
---
paddle/fluid/pybind/pybind.cc | 21 +++++++++++----------
1 file changed, 11 insertions(+), 10 deletions(-)
diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc
index 929c343f7a..60662244cc 100644
--- a/paddle/fluid/pybind/pybind.cc
+++ b/paddle/fluid/pybind/pybind.cc
@@ -498,16 +498,17 @@ All parameter, weight, gradient are variables in Paddle.
m.def("reset_profiler", platform::ResetProfiler);
py::class_<ParallelExecutor>(m, "ParallelExecutor")
- .def(
- "__init__",
- [](ParallelExecutor &self, const std::vector<platform::Place> &places,
- const std::unordered_set<std::string> ¶ms,
- const ProgramDesc &startup_program,
- const ProgramDesc &main_program, const std::string &loss_var_name,
- Scope *scope) {
- new (&self) ParallelExecutor(places, params, startup_program,
- main_program, loss_var_name, scope);
- })
+ .def("__init__",
+ [](ParallelExecutor &self, size_t num_threads,
+ const std::vector<platform::Place> &places,
+ const std::unordered_set<std::string> ¶ms,
+ const ProgramDesc &startup_program,
+ const ProgramDesc &main_program, const std::string &loss_var_name,
+ Scope *scope) {
+ new (&self)
+ ParallelExecutor(num_threads, places, params, startup_program,
+ main_program, loss_var_name, scope);
+ })
.def("run", &ParallelExecutor::Run);
BindRecordIOWriter(m);