From 8858e2280704ea632da5df8df079f56b9825bc8f Mon Sep 17 00:00:00 2001
From: dingshihao2 <dingshihao2@hisilicon.com>
Date: Sat, 17 Apr 2021 14:30:42 +0800
Subject: [PATCH] fix gen_mask control-edges bug

---
 ge/graph/passes/link_gen_mask_nodes_pass.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ge/graph/passes/link_gen_mask_nodes_pass.cc b/ge/graph/passes/link_gen_mask_nodes_pass.cc
index 8dfd447d..b9adc51b 100755
--- a/ge/graph/passes/link_gen_mask_nodes_pass.cc
+++ b/ge/graph/passes/link_gen_mask_nodes_pass.cc
@@ -111,6 +111,8 @@ void LinkGenMaskNodesPass::GetAllGenMaskNodes(ComputeGraphPtr graph, vector<Node
         // node gen_mask is located at different place in the fused node
         if (in_data_node->GetName().find(DROPOUTGENMASK) != in_data_node->GetName().npos) {
           gen_mask = in_data_node;
+          GELOGD("The fused node type [%s], paired with input node name [%s].", node->GetType(), gen_mask->GetName());
+          break;
         }
       }