Browse Source

!15511 FuncGrap::ReplaceInOrder heap_use-after-free fix

From: @zhangzhaoju
Reviewed-by: @zh_qh,@ginfung
Signed-off-by: @zh_qh
pull/15511/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
c2c16ea927
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc

+ 2
- 1
mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/fused_batch_norm_fusion.cc View File

@@ -177,7 +177,8 @@ void FusedBatchNormFusion::EliminateMonadNodes(const FuncGraphPtr &func_graph, c
MS_EXCEPTION_IF_NULL(manager); MS_EXCEPTION_IF_NULL(manager);
auto assign_sub1 = GetAnfNodeByVar(equiv, assign_sub1_var_); auto assign_sub1 = GetAnfNodeByVar(equiv, assign_sub1_var_);
MS_EXCEPTION_IF_NULL(assign_sub1); 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; const AnfNodePtr &output = node_index.first;
MS_EXCEPTION_IF_NULL(output); MS_EXCEPTION_IF_NULL(output);
if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimUpdateState)) { if (AnfAlgo::CheckPrimitiveType(output, prim::kPrimUpdateState)) {


Loading…
Cancel
Save