Commit 08285051 authored by Sergey Shalnov's avatar Sergey Shalnov Committed by Robert Kimball

IntelGPU backend: Concat operation fix for single element (#2579)

parent 407c0bc0
......@@ -1195,15 +1195,16 @@ void runtime::intelgpu::do_concat_operation(cldnn::topology& topology,
}
else
{
if (idx == input_shapes.size() - 1)
{
// last kernel should produce the output name as overall node required
name_suffix = "";
}
kernel_input.push_back(aux_output_name);
kernel_arguments = get_kernel_args(2, 1);
}
// last kernel should produce the output name as overall node required
if (idx == input_shapes.size() - 1)
{
name_suffix = "";
}
const cldnn::custom_gpu_primitive op_concat(output_name + name_suffix,
kernel_input,
{writer.get_code()},
......
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