IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Revert input's output layout in case of does_possible_implementation_exist failure (#8108)

Browse Source 
Signed-off-by: Kelvin Choi <kelvin.choi@intel.com>
This commit is contained in:
Kelvin Choi 2021-10-25 17:47:42 +09:00 committed by GitHub
parent e6f0b8fe88
commit de750b93fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
inference-engine/thirdparty/clDNN/src/graph_optimizer/remove_redundant_reorders.cpp vendored
View File

@ -308,12 +308,15 @@ void remove_redundant_reorders::run(program& p) {
if (!lo.can_fuse_reorder_to_prev(input, *node.get_users().front(), input.get_output_layout().format, output_layout.format))
continue;
auto old_output_layout_of_input = input.get_output_layout();
input.set_output_layout(output_layout, false);
if (input.type()->does_possible_implementation_exist(input)) {
p.replace_all_usages(node, input);
p.add_optimized_primitive_info(node.id());
p.remove_all_connections(node);
p.remove_if_dangling(node);
} else {
input.set_output_layout(old_output_layout_of_input, false);
}
}
}
@ -414,6 +417,7 @@ void remove_redundant_reorders::run(program& p) {
}
}
auto old_output_layout_of_input = input.get_output_layout();
auto output_layout = node->get_output_layout();
input.set_output_layout(output_layout, false);
if (input.type()->does_possible_implementation_exist(input)) {
@ -433,8 +437,11 @@ void remove_redundant_reorders::run(program& p) {
node->can_be_optimized(true);
p.add_optimized_primitive_info(node->id());
p.extract_and_remove(*node);
return true;
} else {
input.set_output_layout(old_output_layout_of_input, false);
return false;
}
return true;
};
if (enable_reorder_fusing) {