Skip to content

Commit

Permalink
Fix to use rank size instead of 4.
Browse files Browse the repository at this point in the history
Signed-off-by: hyunback <[email protected]>
  • Loading branch information
hyunback committed Oct 10, 2024
1 parent 2ec0ae1 commit e5467b1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/plugins/intel_gpu/src/graph/program_node.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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<size_t>(4)) - in_pshape.size();
size_t ones_to_add = std::max(out_pshape.size(), static_cast<size_t>(rank)) - in_pshape.size();
if (ones_to_add > 0) {
layout new_layout = in;
ov::PartialShape new_input_pshape;
Expand Down

0 comments on commit e5467b1

Please sign in to comment.