Commit 8436249a authored by Sergey Shalnov's avatar Sergey Shalnov Committed by Robert Kimball

IntelGPU backend: Product operation fix (#2356)

parent 0241f1fd
......@@ -273,9 +273,12 @@ void runtime::intelgpu::do_product_operation(cldnn::topology& topology,
const size_t input_size = shape_size<Shape>(input_shape);
codegen::CodeWriter writer;
writer << "__kernel void " << function_name << "(const __global float input"
<< array_dims(input_shape) << ", __global float output" << array_dims(output_shape)
<< ")\n";
runtime::intelgpu::gen_func_def(writer,
function_name,
{get_opencl_type_name(output_type)},
{input_shape},
get_opencl_type_name(output_type),
output_shape);
writer.block_begin();
{
......@@ -309,7 +312,7 @@ void runtime::intelgpu::do_product_operation(cldnn::topology& topology,
++var_idx;
}
writer << "output" << access_dims(input_shape, "i", axis) << " *= input"
writer << "output" << access_dims(input_shape, "i", axis) << " *= input0"
<< access_dims(input_shape) << ";\n";
// Closing brackets for loop
......
......@@ -135,9 +135,6 @@ sum_stable_acc
sum_stable_acc_double
sum_stable_simple_double
sum_trivial_in_double
product_2d_to_scalar_int32
product_to_scalar_int32
product_to_scalar_int8
max_matrix_rows_zero_int32
max_to_scalar_int8
min_to_scalar_int8
......
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