From 0cec9f8ddefd469380951828229f68d83d7ed697 Mon Sep 17 00:00:00 2001
From: wjm <wangjiming5@huawei.com>
Date: Sat, 5 Jun 2021 07:25:44 +0800
Subject: [PATCH] fix

---
 ge/graph/optimize/mem_rw_conflict_optimize.cc | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/ge/graph/optimize/mem_rw_conflict_optimize.cc b/ge/graph/optimize/mem_rw_conflict_optimize.cc
index 30fcabef..7e7ab908 100644
--- a/ge/graph/optimize/mem_rw_conflict_optimize.cc
+++ b/ge/graph/optimize/mem_rw_conflict_optimize.cc
@@ -743,10 +743,8 @@ Status GraphOptimize::HandleMemoryRWConflict(ComputeGraphPtr &compute_graph) {
       continue;
     }
     // ignore data / netoutput of subgraph
-    if (AttrUtils::HasAttr(node->GetOpDesc(), ATTR_NAME_PARENT_NODE_INDEX)) {
-      if (node->GetType() == DATA || node->GetType() == NETOUTPUT) {
-        continue;
-      }
+    if (IsSubgraphInputNode(node) || IsSubgraphOutputNode(node)) {
+      continue;
     }
 
     bool identity_reserved = false;