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

[GPU] Small fix for dynamic_shape_gather_opts pass (#21807)

Browse Source
This commit is contained in:
Roman Lyamin 2023-12-21 11:42:03 +04:00 committed by GitHub
parent 12a9d55c3e
commit 0709a35cb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_gpu/src/graph/graph_optimizer/dynamic_shape_gather_opts.cpp
View File

@ -25,11 +25,11 @@ void dynamic_shape_gather_opts::run(program& p) {
continue; continue;
auto idx_rank = impl_params->get_input_layout(1).get_partial_shape().size(); auto idx_rank = impl_params->get_input_layout(1).get_partial_shape().size();
if (idx_rank > 1) { if (idx_rank != 1) {
continue; continue;
} }
auto axis = impl_params->typed_desc<gather>()->axis; auto axis = impl_params->typed_desc<gather>()->axis;
if (impl_params->get_input_layout(0).get_partial_shape()[axis] == -1 if (impl_params->get_input_layout(0).get_partial_shape()[axis] == -1
|| impl_params->get_input_layout(1).get_partial_shape()[0] == -1 || impl_params->get_input_layout(1).get_partial_shape()[0] == -1
|| impl_params->get_input_layout(0).get_partial_shape()[axis] == impl_params->get_input_layout(1).get_partial_shape()[0]) { || impl_params->get_input_layout(0).get_partial_shape()[axis] == impl_params->get_input_layout(1).get_partial_shape()[0]) {
// May be skipepd // May be skipepd