From 27b9b2870e142b0384a8e3af2baafef40951fa6c Mon Sep 17 00:00:00 2001 From: huanghui Date: Wed, 11 Nov 2020 17:30:23 +0800 Subject: [PATCH] fix ReduceEltwiseFusionPass not set output_used_num --- .../ascend/buffer_fusion/reduce_eltwise_fusion_pass.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git 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 index 8c9fe78df9..da09950669 100644 --- 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 output_used_num{SizeToLong(kernel_graph.manager()->node_users()[eltwise_input].size())}; + AnfAlgo::SetNodeAttr(kAttrOutputUsedNum, MakeValue(output_used_num), eltwise_input); SetRecordFusionId(record); } }