Browse Source

!444 add memcpy

From: @chen_yemeng
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @wqtshg
tags/v1.1.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
0fcf6fd664
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/switch_data_edges_bypass.cc

+ 1
- 1
ge/graph/passes/switch_data_edges_bypass.cc View File

@@ -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);


Loading…
Cancel
Save