Commit aeb92efd authored by dmyershov's avatar dmyershov Committed by Robert Kimball

IntelGPU backend: ArgMin/ArgMax functionality was fixed; Tests on unsupported…

IntelGPU backend: ArgMin/ArgMax functionality was fixed; Tests on unsupported functionality were disabled (#1834)

* IntelGPU backend: ArgMin/ArgMax functionality was fixed; Tests on unsupported functionality were disabled

* Additional improvement of ArgMin/ArgMax algorithm
parent 61be3814
......@@ -1565,14 +1565,13 @@ void runtime::intelgpu::do_arg_max_min_operation(cldnn::topology& topology,
{
gws = generate_loops(writer, output_shape, true);
writer << get_opencl_type_name(output_type) << " " << var_name << " = " << infinity
<< ";\n";
writer << "uint index = -1;\n";
writer << get_opencl_type_name(input_type) << " " << var_name << " = " << infinity << ";\n";
writer << get_opencl_type_name(output_type) << " index = 0;\n";
writer << "for (uint i = 0; i < " << input_shape.at(reduction_axis) << "; ++i)\n";
writer.block_begin();
{
writer << "if(i == 0 || input0" << dims_buffer << operation_sign << var_name << ")\n";
writer << "if (input0" << dims_buffer << operation_sign << var_name << ")\n";
writer.block_begin();
{
writer << var_name << " = input0" << dims_buffer << ";\n";
......
......@@ -7,6 +7,7 @@ backwards_dot_scalar_tensor
backwards_dot_tensor3_tensor3
backwards_dot_tensor_scalar
backwards_dot_tensor_vector
backwards_exp
backwards_maxpool_n2_c1_hw5_3x3_str2_max
backwards_maxpool_n4_c1_hw4_2x2_max
backwards_replace_slice
......@@ -20,6 +21,9 @@ dequantize
dequantize_axes
dequantize_int8
divide_by_zero_int32
dot_3d_multi_axis
dot_4d_5d_multi_axis
dot_4d_5d_multi_axis_more
function_call
max_pool_3d
numeric_double_inf
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment