From 89f55e63d50dea96b3465fa311d9ae8f6d8dbe28 Mon Sep 17 00:00:00 2001 From: wangcong Date: Wed, 13 May 2020 15:11:04 +0800 Subject: [PATCH] reverse previous version --- .../ccsrc/pre_activate/ascend/buffer_fusion/buffer_fusion.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/pre_activate/ascend/buffer_fusion/buffer_fusion.cc b/mindspore/ccsrc/pre_activate/ascend/buffer_fusion/buffer_fusion.cc index 616bce1b66..cac3b425d3 100644 --- a/mindspore/ccsrc/pre_activate/ascend/buffer_fusion/buffer_fusion.cc +++ b/mindspore/ccsrc/pre_activate/ascend/buffer_fusion/buffer_fusion.cc @@ -566,8 +566,8 @@ void BufferFusion::MatchDepthwiseConvRelu(const CNodePtr &cnode, const session:: // Elemwise-->DepthwiseConvolution auto relu = cnode->input(1); MS_EXCEPTION_IF_NULL(relu); - if (cnode->isa() && AnfAlgo::GetCNodeName(relu) == prim::kPrimRelu->name() || - AnfAlgo::GetCNodeName() == kReluV2OpName) { + if (cnode->isa() && + (AnfAlgo::GetCNodeName(relu) == prim::kPrimRelu->name() || AnfAlgo::GetCNodeName(relu) == kReluV2OpName)) { std::vector output_used_num{SizeToInt(manager->node_users()[relu].size())}; AnfAlgo::SetNodeAttr(kAttrOutputUsedNum, MakeValue(output_used_num), relu); std::unordered_set record{cnode, relu};