a946079208
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!30896 add passing fracz_group through optimizer's output 

Merge pull request !30896 from yuchaojie/op_select
This commit is contained in:
i-robot 2022-03-07 12:35:14 +00:00 committed by Gitee
parent 77cd10ce04 4490ef82de
commit 513663ea0d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/set_fracz_group_attr.cc
View File

@ -125,6 +125,7 @@ std::vector<KernelWithIndex> 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<KernelWithIndex> 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};
});
}
}
}