Browse Source

!6539 Only match the pattern of Conv2DBackproInput+ReluGradV2

Merge pull request !6539 from huanghui/fix-ub-fusion-conv2dbpinput-eltwise
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
59f582faa3
1 changed files with 1 additions and 3 deletions
  1. +1
    -3
      mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/conv2dbackprop_eltwise_fusion_pass.cc

+ 1
- 3
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>();
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);
}
}


Loading…
Cancel
Save