From 88d173cc48724078289e3c18a93368c12f3fc0d5 Mon Sep 17 00:00:00 2001
From: simson <526422051@qq.com>
Date: Thu, 9 Apr 2020 09:20:08 +0800
Subject: [PATCH] rebuild graph if needed

---
 mindspore/ccsrc/transform/graph_runner.cc   | 7 +++++++
 mindspore/ccsrc/utils/context/ms_context.cc | 4 +++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/transform/graph_runner.cc b/mindspore/ccsrc/transform/graph_runner.cc
index e77b1bcd73..2bff1a740c 100644
--- a/mindspore/ccsrc/transform/graph_runner.cc
+++ b/mindspore/ccsrc/transform/graph_runner.cc
@@ -135,6 +135,13 @@ Status GraphRunner::RunGraph(const RunOptions& options, const std::vector<GeTens
     return Status::FAILED;
   }
 
+  // The information of some nodes could be changed after fusion in some cases
+  // Therefore a graph needs to be rebuilt in above situation
+  if (sess_->IsGraphNeedRebuild(wrap_ptr->id_)) {
+    sess_->RemoveGraph(wrap_ptr->id_);
+    sess_->AddGraph(wrap_ptr->id_, *(wrap_ptr->graph_ptr_), wrap_ptr->options_);
+  }
+
   ge::Status ret = sess_->RunGraph(wrap_ptr->id_, ge_inputs, ge_outputs);
   if (ret != ge::GRAPH_SUCCESS) {
     MS_LOG(ERROR) << "Call GE RunGraph Failed, ret is: " << ret;
diff --git a/mindspore/ccsrc/utils/context/ms_context.cc b/mindspore/ccsrc/utils/context/ms_context.cc
index bf05af9858..e9b4586b21 100644
--- a/mindspore/ccsrc/utils/context/ms_context.cc
+++ b/mindspore/ccsrc/utils/context/ms_context.cc
@@ -355,7 +355,9 @@ void MsContext::GetGeOptions(std::map<std::string, std::string>* ge_options) con
     MS_LOG(ERROR) << "Set proto lib path failed!";
   }
 
-  // Disbale the global variable acc, only enable it whlie adding training graph in pipeline
+  // Enable auto mixed precision according to the context options
+  (*ge_options)["ge.exec.auto_mix_precision"] = std::to_string(auto_mixed_precision_flag_);
+  // Disable the global variable acc, only enable it whlie adding training graph in pipeline
   (*ge_options)["ge.exec.variable_acc"] = "0";
 #endif
 }