p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update the pattern of conv2dbackprop_eltwise_fusion pass. Only match Conv2dBackpropInput+ReluGradV2.

This commit is contained in:
huanghui 2020-09-19 11:00:43 +08:00
parent ba2fe29691
commit af0a11ff69
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc
View File

@ -58,9 +58,7 @@ void Conv2DBackpropEltwiseFusionPass::MatchSingleFusionPattern(const session::Ke
} }
auto cnode = node->cast<CNodePtr>(); auto cnode = node->cast<CNodePtr>();
MS_EXCEPTION_IF_NULL(cnode); MS_EXCEPTION_IF_NULL(cnode);
if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && if (AnfAlgo::GetCNodeName(cnode) == kReluGradV2OpName) {
AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE &&
(cnode->inputs().size() == ELTWISE_INPUT_SIZE || cnode->inputs().size() == ELTWISE_DOUBLE_IN_INPUT_SIZE)) {
MatchConv2DBackpropInputEltwise(cnode, kernel_graph, candidate_fusion); MatchConv2DBackpropInputEltwise(cnode, kernel_graph, candidate_fusion);
} }
} }