[incubator-mxnet] branch master updated: fix bug, issue 12613 (#12614)

[ha...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

haibin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new d4e202a  fix bug, issue 12613 (#12614)
d4e202a is described below

commit d4e202a5d4152b540be970525f2bb91b62ebf148
Author: lovickie <zh...@126.com>
AuthorDate: Thu Sep 27 01:59:39 2018 +0800

    fix bug, issue 12613 (#12614)
---
 src/executor/graph_executor.cc | 1 +
 src/executor/graph_executor.h  | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
index 54a8d22..922917f 100644
--- 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
index b94bb43..f5f032e 100644
--- 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