diff --git a/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/set_fracz_group_attr.cc b/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/set_fracz_group_attr.cc index 726271bac0..432ef33d1a 100644 --- a/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/set_fracz_group_attr.cc +++ b/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/set_fracz_group_attr.cc @@ -125,6 +125,7 @@ std::vector GetCNodeNeighborFraczNodes(const FuncGraphManagerPt input_num = 1; output_num = 1; } + // traverse input for (size_t i = 0; i < input_num; ++i) { if (AnfAlgo::GetInputFormat(cnode, i) == kOpFormat_FRAC_Z) { auto input = cnode->input(i + 1); @@ -140,16 +141,15 @@ std::vector GetCNodeNeighborFraczNodes(const FuncGraphManagerPt } } } - if (kOptOperatorSet.find(node_name) == kOptOperatorSet.end()) { - for (size_t i = 0; i < output_num; ++i) { - if (AnfAlgo::GetOutputFormat(cnode, i) == kOpFormat_FRAC_Z) { - auto output = GetOutputItem(manager, cnode, groups, i); - if (output != nullptr) { - std::transform(node_user[output].begin(), node_user[output].end(), std::back_inserter(ret), - [](KernelWithIndex node_index) { - return KernelWithIndex{node_index.first, node_index.second - 1}; - }); - } + // traverse output + for (size_t i = 0; i < output_num; ++i) { + if (AnfAlgo::GetOutputFormat(cnode, i) == kOpFormat_FRAC_Z) { + auto output = GetOutputItem(manager, cnode, groups, i); + if (output != nullptr) { + std::transform(node_user[output].begin(), node_user[output].end(), std::back_inserter(ret), + [](KernelWithIndex node_index) { + return KernelWithIndex{node_index.first, node_index.second - 1}; + }); } } }