From 8137d01e84ca5a4b9a668535ac948a903052402a Mon Sep 17 00:00:00 2001
From: liangzelang <liangzelang1@huawei.com>
Date: Sat, 24 Apr 2021 11:13:41 +0800
Subject: [PATCH] [control flow] handle same label id in one labelswitch

---
 mindspore/ccsrc/backend/session/ascend_auto_monad.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/session/ascend_auto_monad.cc b/mindspore/ccsrc/backend/session/ascend_auto_monad.cc
index 985d376cdf0..c9f629e0530 100644
--- a/mindspore/ccsrc/backend/session/ascend_auto_monad.cc
+++ b/mindspore/ccsrc/backend/session/ascend_auto_monad.cc
@@ -1308,13 +1308,14 @@ class ExecuteOrderGenerator {
       // Use new label if find repeated label.
       if (iter == labels->end()) {
         new_labels.emplace_back(label_id);
+        labels->emplace_back(label_id);
         continue;
       }
       new_labels.emplace_back(++max_label_);
       labels_multimap->insert(std::pair<uint32_t, uint32_t>(*iter, max_label_));
+      labels->emplace_back(max_label_);
       is_new_labels = true;
     }
-    labels->insert(labels->end(), new_labels.begin(), new_labels.end());
     switch_labels->insert(switch_labels->end(), new_labels.begin(), new_labels.end());
     if (is_new_labels) {
       AnfAlgo::SetNodeAttr(kAttrLabelSwitchList, MakeValue(new_labels), node);