From aefe8e319f9b141a9b0571fe2e2973afe33809d0 Mon Sep 17 00:00:00 2001
From: chenyemeng <chenyemeng@huawei.com>
Date: Fri, 27 Nov 2020 17:47:42 +0800
Subject: [PATCH] add memcpy

---
 ge/graph/passes/switch_data_edges_bypass.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/graph/passes/switch_data_edges_bypass.cc b/ge/graph/passes/switch_data_edges_bypass.cc
index ce2b715b..f7453dd7 100644
--- a/ge/graph/passes/switch_data_edges_bypass.cc
+++ b/ge/graph/passes/switch_data_edges_bypass.cc
@@ -181,7 +181,7 @@ Status BypassSwitchOut(const NodePtr &switch_node, int out_index) {
 
     auto head_node = node_and_anchor.first;
     auto head_node_type = NodeUtils::GetNodeType(*head_node);
-    if (head_node_type == MERGE || head_node_type == REFMERGE) {
+    if (head_node_type == MEMCPYASYNC) {
       // if the switch connect to the merge directly, insert memcpy before merge
       auto memcpy_node = AddMemcpyBeforeNode(head_node, head_anchor->GetIdx());
       GE_CHECK_NOTNULL(memcpy_node);