Browse Source

rm redundant Memcpy before Merge

tags/v1.1.0
chenyemeng 3 years ago
parent
commit
8b2c538fb7
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      ge/graph/passes/merge_pass.cc

+ 10
- 0
ge/graph/passes/merge_pass.cc View File

@@ -79,6 +79,16 @@ Status MergePass::Run(NodePtr &node) {
return FAILED;
}
}
auto in_node = in_data_nodes.at(0);
bool memcpy_optimize_flag = (in_node != nullptr) &&
((in_node->GetType() == MEMCPYASYNC) || (in_node->GetType() == MEMCPYADDRASYNC)) &&
(in_node->GetInDataNodes().size() == 1);
if (memcpy_optimize_flag) {
if (IsolateAndDeleteNode(in_node, {0}) != SUCCESS) {
GELOGE(FAILED, "Isolate and delete node %s failed.", in_node->GetName().c_str());
return FAILED;
}
}
return IsolateAndDeleteNode(node, merge_io_map);
}
default: {


Loading…
Cancel
Save