diff --git a/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp b/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp index 2669d3c918f2a3..72a75e289cb5e1 100644 --- a/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp +++ b/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp @@ -12,6 +12,7 @@ #include "resample_inst.h" #include "reshape_inst.h" #include "arg_max_min_inst.h" +#include "gather_inst.h" #include "shape_of_inst.h" #include @@ -1596,9 +1597,10 @@ format layout_optimizer::get_preferred_format(program_node& node) { bool allow_new_shape_infer = node.get_program().get_config().get_property(ov::intel_gpu::allow_new_shape_infer); if (allow_new_shape_infer) { - if (node.is_type()) + if (node.is_type() || + node.is_type()) { return format::get_default_format(node.get_dependency(0).get_output_layout(false).get_rank()); - + } // Let reorder_input pass to check input format instead of output_format in forward investigation, vice versa auto out_lay_rank = node.get_output_layout(false).get_rank(); auto dep_size = node.get_dependencies().size();