From af0a11ff6999cfa653bf308e16c6d06431bb334c Mon Sep 17 00:00:00 2001 From: huanghui Date: Sat, 19 Sep 2020 11:00:43 +0800 Subject: [PATCH] Update the pattern of conv2dbackprop_eltwise_fusion pass. Only match Conv2dBackpropInput+ReluGradV2. --- .../buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc b/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc index 9918c530c8..7df785b27a 100644 --- a/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc +++ b/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc @@ -58,9 +58,7 @@ void Conv2DBackpropEltwiseFusionPass::MatchSingleFusionPattern(const session::Ke } auto cnode = node->cast(); MS_EXCEPTION_IF_NULL(cnode); - if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && - AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && - (cnode->inputs().size() == ELTWISE_INPUT_SIZE || cnode->inputs().size() == ELTWISE_DOUBLE_IN_INPUT_SIZE)) { + if (AnfAlgo::GetCNodeName(cnode) == kReluGradV2OpName) { MatchConv2DBackpropInputEltwise(cnode, kernel_graph, candidate_fusion); } }