Browse Source

Fix HandleMemoryRWConflict remove identity on while subgraph.

tags/v1.1.0
unknown 4 years ago
parent
commit
d9679121e4
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      ge/graph/optimize/mem_rw_conflict_optimize.cc

+ 5
- 0
ge/graph/optimize/mem_rw_conflict_optimize.cc View File

@@ -680,6 +680,11 @@ Status GraphOptimize::HandleMemoryRWConflict(ComputeGraphPtr &compute_graph) {
} }
// 2.loop all node, including node in subgraph and handle memory rw conflict // 2.loop all node, including node in subgraph and handle memory rw conflict
for (auto &node : compute_graph->GetAllNodes()) { for (auto &node : compute_graph->GetAllNodes()) {
// ignore while subgraph node
const auto parent_node = node->GetOwnerComputeGraph()->GetParentNode();
if ((parent_node != nullptr) && (kWhileOpTypes.count(parent_node->GetType()) > 0)) {
continue;
}
// ignore data / netoutput of subgraph // ignore data / netoutput of subgraph
if (node->GetType() == DATA && AttrUtils::HasAttr(node->GetOpDesc(), ATTR_NAME_PARENT_NODE_INDEX)) { if (node->GetType() == DATA && AttrUtils::HasAttr(node->GetOpDesc(), ATTR_NAME_PARENT_NODE_INDEX)) {
continue; continue;


Loading…
Cancel
Save