Browse Source

!8036 BnAddReluGrad fusion check

Merge pull request !8036 from chenweifeng/BnAddReluGrad-check
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
1014774ab6
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/optimizer/gpu/batch_norm_add_relu_grad_fusion.cc

+ 3
- 0
mindspore/ccsrc/backend/optimizer/gpu/batch_norm_add_relu_grad_fusion.cc View File

@@ -140,6 +140,9 @@ const AnfNodePtr BatchNormAddReluGradFusion::Process(const FuncGraphPtr &graph,
// process pattern as Relu(TensorAdd(BN#0, BN#1))
auto tuple_getitem = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 5);
MS_EXCEPTION_IF_NULL(tuple_getitem);
if (!utils::isa<CNodePtr>(tuple_getitem) || AnfAlgo::GetCNodeName(tuple_getitem) != prim::kPrimTupleGetItem->name()) {
return nullptr;
}
auto forward_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_getitem), 0);
if (AnfAlgo::GetCNodeName(forward_node) != kFusedBatchNormExWithAddAndActivation) {
return nullptr;


Loading…
Cancel
Save