From 288b5326cba2132179c162869aec80bb3c236981 Mon Sep 17 00:00:00 2001 From: Wilson Seok Date: Tue, 13 Dec 2022 09:10:00 +0900 Subject: [PATCH] fix PR14422 regression for using arg_max_min opt kernel (#14487) (#14533) --- src/plugins/intel_gpu/src/graph/layout_optimizer.cpp | 5 +++++ src/plugins/intel_gpu/src/graph/program.cpp | 1 + 2 files changed, 6 insertions(+) diff --git a/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp b/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp index 319dd82e983..73d8ead21c5 100644 --- a/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp +++ b/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp @@ -11,6 +11,7 @@ #include "reorder_inst.h" #include "resample_inst.h" #include "reshape_inst.h" +#include "arg_max_min_inst.h" #include "generic_layer.hpp" #include @@ -1790,6 +1791,10 @@ format layout_optimizer::get_preferred_format(program_node& node) { else if (input_layout.format.dimension() == 4) expected = format::bfyx; } + } else if (node.is_type()) { + // Set default format for issue 92967/98750 + // TODO: will remove when arg_max_min_ref supports blocked format + expected = format::get_default_format(node.get_input_layouts()[0].get_rank(), false, false); } return expected; diff --git a/src/plugins/intel_gpu/src/graph/program.cpp b/src/plugins/intel_gpu/src/graph/program.cpp index a81f34d63e3..5d13b23bbef 100644 --- a/src/plugins/intel_gpu/src/graph/program.cpp +++ b/src/plugins/intel_gpu/src/graph/program.cpp @@ -1487,6 +1487,7 @@ void program::set_layout_optimizer_attributes(layout_optimizer& lo) { || (prim.as().input().get_output_layout().data_type != data_types::u8 && prim.as().input().get_output_layout().data_type != data_types::i8) || prim.as().get_primitive()->across_channels) && + prim.type() != cldnn::arg_max_min::type_id() && prim.type() != cldnn::dft::type_id() && prim.type() != cldnn::grid_sample::type_id() && prim.type() != cldnn::mutable_data::type_id() &&