From e5467b165b60e9d0fc6507f22a051c0a17056ea4 Mon Sep 17 00:00:00 2001 From: hyunback Date: Tue, 8 Oct 2024 17:17:47 +0900 Subject: [PATCH] Fix to use rank size instead of 4. Signed-off-by: hyunback --- src/plugins/intel_gpu/src/graph/program_node.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plugins/intel_gpu/src/graph/program_node.cpp b/src/plugins/intel_gpu/src/graph/program_node.cpp index 48e00a68e77e78..fc9648b90e444c 100644 --- a/src/plugins/intel_gpu/src/graph/program_node.cpp +++ b/src/plugins/intel_gpu/src/graph/program_node.cpp @@ -1550,7 +1550,7 @@ void program_node::create_onednn_primitive_attributes( size_t rank = cldnn::format::dimension(in.format); auto in_pshape = in.get_partial_shape(); auto out_pshape = get_output_layout().get_partial_shape(); - size_t ones_to_add = std::max(out_pshape.size(), static_cast(4)) - in_pshape.size(); + size_t ones_to_add = std::max(out_pshape.size(), static_cast(rank)) - in_pshape.size(); if (ones_to_add > 0) { layout new_layout = in; ov::PartialShape new_input_pshape;