fix bug, issue 12613 (apache#12614)

vandanavk · Sep 26, 2018 · d4e202a · d4e202a
1 parent 3042825
commit d4e202a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
@@ -44,6 +44,7 @@ GraphExecutor::GraphExecutor() {
   log_verbose_ = dmlc::GetEnv("MXNET_EXEC_VERBOSE_LOGGING", false);
   need_grad_ = false;
   subgraph_property_ = dmlc::GetEnv("MXNET_SUBGRAPH_BACKEND", std::string());
+  engine_ref_ = Engine::_GetSharedRef();
 }
 
 GraphExecutor::~GraphExecutor() {

diff --git a/src/executor/graph_executor.h b/src/executor/graph_executor.h
@@ -260,6 +260,8 @@ class GraphExecutor : public Executor {
   bool log_verbose_ = false;
   // subgraph property name
   std::string subgraph_property_;
+  // ref of engine
+  std::shared_ptr<Engine> engine_ref_;
 };
 
 }  // namespace exec