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

fix PR14422 regression for using arg_max_min opt kernel (#14487) (#14533)

Browse Source
This commit is contained in:
Wilson Seok
2022-12-13 09:10:00 +09:00
committed by GitHub
parent 6f5edfa942
commit 288b5326cb
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/plugins/intel_gpu/src/graph/layout_optimizer.cpp
View File

@@ -11,6 +11,7 @@
#include "reorder_inst.h" #include "reorder_inst.h"
#include "resample_inst.h" #include "resample_inst.h"
#include "reshape_inst.h" #include "reshape_inst.h"
#include "arg_max_min_inst.h"
#include "generic_layer.hpp" #include "generic_layer.hpp"
#include <sstream> #include <sstream>
@@ -1790,6 +1791,10 @@ format layout_optimizer::get_preferred_format(program_node& node) {
else if (input_layout.format.dimension() == 4) else if (input_layout.format.dimension() == 4)
expected = format::bfyx; expected = format::bfyx;
} }
} else if (node.is_type<arg_max_min>()) {
// 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; return expected;

1
src/plugins/intel_gpu/src/graph/program.cpp
View File

@@ -1487,6 +1487,7 @@ void program::set_layout_optimizer_attributes(layout_optimizer& lo) {
|| (prim.as<mvn>().input().get_output_layout().data_type != data_types::u8 && || (prim.as<mvn>().input().get_output_layout().data_type != data_types::u8 &&
prim.as<mvn>().input().get_output_layout().data_type != data_types::i8) prim.as<mvn>().input().get_output_layout().data_type != data_types::i8)
|| prim.as<mvn>().get_primitive()->across_channels) && || prim.as<mvn>().get_primitive()->across_channels) &&
prim.type() != cldnn::arg_max_min::type_id() &&
prim.type() != cldnn::dft::type_id() && prim.type() != cldnn::dft::type_id() &&
prim.type() != cldnn::grid_sample::type_id() && prim.type() != cldnn::grid_sample::type_id() &&
prim.type() != cldnn::mutable_data::type_id() && prim.type() != cldnn::mutable_data::type_id() &&