Browse Source

!8468 Fix ReduceEltwiseFusionPass not set output_used_num

From: @irmo
Reviewed-by: @jjfeing,@kisnwang
Signed-off-by: @jjfeing
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c5056e41a1
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/reduce_eltwise_fusion_pass.cc

+ 2
- 0
mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/reduce_eltwise_fusion_pass.cc View File

@@ -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);
}
}


Loading…
Cancel
Save