fix ReduceEltwiseFusionPass not set output_used_num

4 years ago · 27b9b2870e
parent 466fdcbd42
commit 27b9b2870e
1 changed files with 2 additions and 0 deletions
--- a/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/reduce_eltwise_fusion_pass.cc
+++ b/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/reduce_eltwise_fusion_pass.cc
@ -64,6 +64,8 @@ void ReduceEltwiseFusionPass::MatchReduceEltwise(const CNodePtr &cnode, const se
      }
    }
    candidate_fusion->push_back(record);
+    std::vector<int64_t> output_used_num{SizeToLong(kernel_graph.manager()->node_users()[eltwise_input].size())};
+    AnfAlgo::SetNodeAttr(kAttrOutputUsedNum, MakeValue(output_used_num), eltwise_input);
    SetRecordFusionId(record);
  }
 }