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

IntelGPU backend: Broadcast bug fix: (output_shape.at(0) == 1) doesn't mean that…

IntelGPU backend: Broadcast bug fix: (output_shape.at(0) == 1) doesn't mean that it is scalar (#1754)
parent adb38ab4
......@@ -138,7 +138,7 @@ void runtime::intelgpu::do_bcast_sum_operation(cldnn::topology& topology,
else
{
// corner case with scalar
if (output_shape.empty() || (!output_shape.empty() && (output_shape.at(0) == 1)))
if (output_shape.empty() || (!output_shape.empty() && (shape_size(output_shape) == 1)))
{
return do_sum_to_scalar_operation(topology,
input_name,
......
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