From 19109739c6b575c6dc92c4b3ffada1fc5ef6d4f7 Mon Sep 17 00:00:00 2001
From: gaoyong10 <gaoyong10@huawei.com>
Date: Wed, 10 Aug 2022 17:43:16 +0800
Subject: [PATCH] Unset ref node copy in graph run mode.

---
 .../ccsrc/runtime/graph_scheduler/control_node_scheduler.cc     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.cc b/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.cc
index 85b720e8261..8c6296961db 100644
--- a/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.cc
+++ b/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.cc
@@ -58,7 +58,7 @@ bool is_need_copy_device_tensor(const AnfNodePtr &backend_node, size_t index) {
 
   auto kernel_graph = AnfAlgo::FetchKernelGraph(backend_node);
   MS_EXCEPTION_IF_NULL(kernel_graph);
-  if (kernel_graph->IsInRefOutputMap({backend_node, index})) {
+  if ((!kernel_graph->is_graph_run_mode()) && kernel_graph->IsInRefOutputMap({backend_node, index})) {
     return false;
   }