From 1442290e63ed024402f6e98332eb83fbfdf6e1c0 Mon Sep 17 00:00:00 2001
From: zhangzhaoju <zhangzhaoju@huawei.com>
Date: Thu, 22 Apr 2021 11:13:31 +0800
Subject: [PATCH] FuncGrap::ReplaceInOrder heap_use-after-free fix

---
 .../optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc      | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc b/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc
index 78e2d3e5fa..82708121d6 100644
--- a/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc
+++ b/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc
@@ -177,7 +177,8 @@ void FusedBatchNormFusion::EliminateMonadNodes(const FuncGraphPtr &func_graph, c
   MS_EXCEPTION_IF_NULL(manager);
   auto assign_sub1 = GetAnfNodeByVar(equiv, assign_sub1_var_);
   MS_EXCEPTION_IF_NULL(assign_sub1);
-  for (const auto &node_index : manager->node_users()[assign_sub1]) {
+  auto users = manager->node_users()[assign_sub1];
+  for (const auto &node_index : users) {
     const AnfNodePtr &output = node_index.first;
     MS_EXCEPTION_IF_NULL(output);
     if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimUpdateState)) {