From 1a492ec8927417d7ed6d9232a816583fa1ca3859 Mon Sep 17 00:00:00 2001
From: dingshihao2 <dingshihao2@hisilicon.com>
Date: Fri, 16 Apr 2021 19:23:20 +0800
Subject: [PATCH] fix gen_mask control-edges bug

---
 ge/graph/passes/link_gen_mask_nodes_pass.cc | 9 ---------
 1 file changed, 9 deletions(-)

diff --git a/ge/graph/passes/link_gen_mask_nodes_pass.cc b/ge/graph/passes/link_gen_mask_nodes_pass.cc
index 27b12ffc..2788bc43 100755
--- a/ge/graph/passes/link_gen_mask_nodes_pass.cc
+++ b/ge/graph/passes/link_gen_mask_nodes_pass.cc
@@ -28,10 +28,6 @@ using std::vector;
 
 namespace ge {
 namespace {
-<<<<<<< Updated upstream
-const size_t kGenMaskInputIndex = 1;
-=======
->>>>>>> Stashed changes
 const size_t kDefaultMaxParallelNum = 1;
 }  // namespace
 
@@ -108,10 +104,6 @@ void LinkGenMaskNodesPass::GetAllGenMaskNodes(ComputeGraphPtr graph, vector<Node
     }
 
     auto in_data_nodes = node->GetInDataNodes();
-<<<<<<< Updated upstream
-    if (in_data_nodes.size() > kGenMaskInputIndex) {
-      NodePtr &gen_mask = in_data_nodes.at(kGenMaskInputIndex);
-=======
     for (auto &in_data_node : in_data_nodes) {
       // node gen_mask is located at different place in the fused node
       if (in_data_node->GetName().find(DROPOUTGENMASK) == in_data_node->GetName().npos) {
@@ -119,7 +111,6 @@ void LinkGenMaskNodesPass::GetAllGenMaskNodes(ComputeGraphPtr graph, vector<Node
       }
       NodePtr &gen_mask = in_data_node;
 
->>>>>>> Stashed changes
       if ((gen_mask->GetOpDesc() == nullptr) || (gen_mask->GetOpDesc()->HasAttr(ATTR_NAME_STREAM_LABEL))) {
         continue;
       }