|
|
|
@ -50,6 +50,7 @@ limitations under the License. */
|
|
|
|
|
#include "paddle/fluid/platform/profiler.h"
|
|
|
|
|
#include "paddle/fluid/pybind/async_executor_py.h"
|
|
|
|
|
#include "paddle/fluid/pybind/const_value.h"
|
|
|
|
|
#include "paddle/fluid/pybind/data_set_py.h"
|
|
|
|
|
#include "paddle/fluid/pybind/exception.h"
|
|
|
|
|
#include "paddle/fluid/pybind/fleet_wrapper_py.h"
|
|
|
|
|
#include "paddle/fluid/pybind/imperative.h"
|
|
|
|
@ -61,7 +62,6 @@ limitations under the License. */
|
|
|
|
|
#include "paddle/fluid/pybind/recordio.h"
|
|
|
|
|
#include "paddle/fluid/pybind/tensor_py.h"
|
|
|
|
|
#include "paddle/fluid/string/to_string.h"
|
|
|
|
|
#include "paddle/fluid/pybind/data_set_py.h"
|
|
|
|
|
|
|
|
|
|
#ifdef PADDLE_WITH_CUDA
|
|
|
|
|
#ifndef _WIN32
|
|
|
|
@ -923,6 +923,7 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
|
py::class_<framework::Executor>(m, "Executor")
|
|
|
|
|
.def(py::init<const platform::Place &>())
|
|
|
|
|
.def("close", &Executor::Close)
|
|
|
|
|
.def("run_from_dataset", &Executor::RunFromDataset)
|
|
|
|
|
.def("run", [](Executor &self, const ProgramDesc &prog, Scope *scope,
|
|
|
|
|
int block_id, bool create_local_scope, bool create_vars,
|
|
|
|
|
const std::vector<std::string> &fetch_vars) {
|
|
|
|
|