From a1369c4b2166e84205e51980d4f9b7877af53026 Mon Sep 17 00:00:00 2001 From: "Skrebkov, Artemy" Date: Thu, 28 Nov 2024 17:56:16 +0000 Subject: [PATCH] Fix check for dynamic dim Signed-off-by: Skrebkov, Artemy --- src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp b/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp index a0e5d2d11c1fef..254aa21ed1c3da 100644 --- a/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp +++ b/src/plugins/intel_npu/src/backend/src/zero_infer_request.cpp @@ -54,9 +54,10 @@ void check_level_zero_attributes_match(const IODescriptor& ioDescriptor, const A "Maximum number of dimensions supported: " + std::to_string(ZE_MAX_GRAPH_ARGUMENT_DIMENSIONS_SIZE) + '\n' + "Given: " + std::to_string(ovDimensions.size())); + const uint32_t dynamicDim = std::numeric_limits::max(); for (size_t index = 0; index < ovDimensions.size(); ++index) { OPENVINO_ASSERT( - ioDescriptor.shapeFromCompiler.is_dynamic() || ovDimensions[index] == zeDescriptor.info.dims[index], + ovDimensions[index] == zeDescriptor.info.dims[index] || zeDescriptor.info.dims[index] == dynamicDim, "Shape mismatch for input/output named " + ioDescriptor.nameFromCompiler); } for (size_t index = ovDimensions.size(); index < ZE_MAX_GRAPH_ARGUMENT_DIMENSIONS_SIZE; ++index) {