From bce317e3420b214b0b26aa644eb410b3998153f8 Mon Sep 17 00:00:00 2001 From: wilson-seok Date: Mon, 21 Oct 2024 09:43:14 +0000 Subject: [PATCH] update condition --- .../intel_gpu/src/kernel_selector/jitter.cpp | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/src/plugins/intel_gpu/src/kernel_selector/jitter.cpp b/src/plugins/intel_gpu/src/kernel_selector/jitter.cpp index 9225451c00eb69..33d13429fdcf3f 100644 --- a/src/plugins/intel_gpu/src/kernel_selector/jitter.cpp +++ b/src/plugins/intel_gpu/src/kernel_selector/jitter.cpp @@ -2199,16 +2199,19 @@ std::string FusedOpsCodeGenerator::GetJitLoad(const FusedOpsConfiguration& conf, return block_read; } - bool multiple_element = false; - for (auto dim : input_tensor.GetDims()) { - auto v = dim.v; - if (v > 1) { - multiple_element = true; - break; + bool multiple_elements = false; + // For dynamic shape input tensor, check any one of static dimension has more than one element. + if (input_tensor.is_dynamic()) { + for (auto dim : input_tensor.GetDims()) { + auto v = dim.v; + if (v > 1) { + multiple_elements = true; + break; + } } } - if (input_tensor.LogicalSize() > 1 || multiple_element) { + if (input_tensor.LogicalSize() > 1 || multiple_elements) { // Currently we assume that in such scenario we can safely load sub_group_size elements from the pointer return Broadcast(block_read, input_dt, vec_size); } else {