From d1a1a868666cf90ca78f440f814863dc94e66481 Mon Sep 17 00:00:00 2001 From: chenyemeng Date: Wed, 25 Nov 2020 09:56:27 +0800 Subject: [PATCH] modify control passes --- ge/graph/passes/attach_stream_label_pass.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/passes/attach_stream_label_pass.cc b/ge/graph/passes/attach_stream_label_pass.cc index fe8d0e7d..9a5dc507 100644 --- a/ge/graph/passes/attach_stream_label_pass.cc +++ b/ge/graph/passes/attach_stream_label_pass.cc @@ -58,7 +58,7 @@ void AttachStreamLabelPass::FindNodes(const ComputeGraphPtr &graph) { const std::string &type = op_desc->GetType(); if ((type == STREAMSWITCH) && op_desc->HasAttr(ATTR_NAME_SWITCH_TRUE_BRANCH_FLAG)) { stream_switch_nodes_.emplace_back(node); - } else if ((type == STREAMMERGE)&& !op_desc->HasAttr(ATTR_NAME_NEXT_ITERATION)) { + } else if ((type == STREAMMERGE) && !op_desc->HasAttr(ATTR_NAME_NEXT_ITERATION)) { need_label_nodes_.emplace_back(node); } else if ((type == ENTER) || (type == REFENTER)) { enter_nodes_.emplace_back(node);