From 6c491b8d3e1595b69253fd9b9560b594e4c53370 Mon Sep 17 00:00:00 2001
From: caifubi <caifubi1@huawei.com>
Date: Thu, 28 May 2020 22:15:29 +0800
Subject: [PATCH] Only release runtime resource in GRAPH_MODE

---
 mindspore/ccsrc/session/kernel_graph.cc | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/session/kernel_graph.cc b/mindspore/ccsrc/session/kernel_graph.cc
index 6fd7cf892c..58a310ce5e 100644
--- a/mindspore/ccsrc/session/kernel_graph.cc
+++ b/mindspore/ccsrc/session/kernel_graph.cc
@@ -712,6 +712,14 @@ void KernelGraph::UpdateCallRealInput() {
 
 std::string KernelGraph::ToString() const { return std::string("kernel_graph_").append(std::to_string(graph_id_)); }
 
-KernelGraph::~KernelGraph() { device::KernelRuntimeManager::Instance().ClearGraphResource(graph_id_); }
+KernelGraph::~KernelGraph() {
+  auto context = MsContext::GetInstance();
+  if (!context) {
+    return;
+  }
+  if (context->execution_mode() == kGraphMode) {
+    device::KernelRuntimeManager::Instance().ClearGraphResource(graph_id_);
+  }
+}
 }  // namespace session
 }  // namespace mindspore