diff --git a/src/plugins/intel_gpu/src/graph/impls/onednn/concatenation_onednn.cpp b/src/plugins/intel_gpu/src/graph/impls/onednn/concatenation_onednn.cpp index 4a410289e90..1f8598c6bab 100644 --- a/src/plugins/intel_gpu/src/graph/impls/onednn/concatenation_onednn.cpp +++ b/src/plugins/intel_gpu/src/graph/impls/onednn/concatenation_onednn.cpp @@ -34,12 +34,14 @@ protected: int input_idx = DNNL_ARG_MULTIPLE_SRC; for (size_t i = 0; i < instance.inputs_memory_count(); i++) { auto& input = instance.input_memory(i); - args.insert({ input_idx++, input.get_onednn_memory(_pd.src_desc(static_cast(i))) }); + auto offset = onednn::get_f_offset(instance.get_input_layout(), _pd.dnnl::primitive_desc_base::src_desc(i)); + args.insert({input_idx++, input.get_onednn_memory(_pd.dnnl::primitive_desc_base::src_desc(i), offset)}); } { auto& output = instance.output_memory(); - args.insert({DNNL_ARG_DST, output.get_onednn_memory(_pd.dst_desc())}); + auto offset = onednn::get_f_offset(instance.get_output_layout(), _pd.dnnl::primitive_desc_base::dst_desc(0)); + args.insert({DNNL_ARG_DST, output.get_onednn_memory(_pd.dnnl::primitive_desc_base::dst_desc(0), offset)}); } configure_post_ops_arguments(instance, args);