fix ReduceEltwiseFusionPass not set output_used_num

pull/8468/head
huanghui 4 years ago
parent 466fdcbd42
commit 27b9b2870e

@ -64,6 +64,8 @@ void ReduceEltwiseFusionPass::MatchReduceEltwise(const CNodePtr &cnode, const se
} }
} }
candidate_fusion->push_back(record); 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); SetRecordFusionId(record);
} }
} }

Loading…
Cancel
Save