|
4 | 4 |
|
5 | 5 | #include "zero_compiler_in_driver.hpp"
|
6 | 6 |
|
7 |
| -#include <fstream> |
8 | 7 | #include <regex>
|
9 | 8 | #include <string_view>
|
10 | 9 |
|
@@ -272,13 +271,19 @@ std::string LevelZeroCompilerInDriver<TableExtension>::serializeIOInfo(const std
|
272 | 271 |
|
273 | 272 | inputsPrecisionSS << INPUTS_PRECISIONS_KEY << KEY_VALUE_SEPARATOR << VALUE_DELIMITER;
|
274 | 273 | inputsLayoutSS << INPUTS_LAYOUTS_KEY << KEY_VALUE_SEPARATOR << VALUE_DELIMITER;
|
| 274 | + const auto getRankOrThrow = [](const ov::PartialShape& shape) -> size_t { |
| 275 | + if (shape.rank().is_dynamic()) { |
| 276 | + OPENVINO_THROW("Dynamic rank is not supported for NPU plugin"); |
| 277 | + } |
| 278 | + return shape.rank().get_length(); |
| 279 | + }; |
275 | 280 |
|
276 | 281 | if (!parameters.empty()) {
|
277 | 282 | size_t parameterIndex = 0;
|
278 | 283 |
|
279 | 284 | for (const std::shared_ptr<ov::op::v0::Parameter>& parameter : parameters) {
|
280 |
| - const ov::element::Type& precision = parameter->get_element_type(); |
281 |
| - const size_t rank = parameter->get_shape().size(); |
| 285 | + const auto precision = parameter->get_element_type(); |
| 286 | + const auto rank = getRankOrThrow(parameter->get_partial_shape()); |
282 | 287 |
|
283 | 288 | if (parameterIndex != 0) {
|
284 | 289 | inputsPrecisionSS << VALUES_SEPARATOR;
|
@@ -310,10 +315,9 @@ std::string LevelZeroCompilerInDriver<TableExtension>::serializeIOInfo(const std
|
310 | 315 | outputsLayoutSS << OUTPUTS_LAYOUTS_KEY << KEY_VALUE_SEPARATOR << VALUE_DELIMITER;
|
311 | 316 |
|
312 | 317 | size_t resultIndex = 0;
|
313 |
| - |
314 | 318 | for (const std::shared_ptr<ov::op::v0::Result>& result : results) {
|
315 |
| - const ov::element::Type_t precision = result->get_element_type(); |
316 |
| - const size_t rank = result->get_shape().size(); |
| 319 | + const auto precision = result->get_element_type(); |
| 320 | + const auto rank = getRankOrThrow(result->get_output_partial_shape(0)); |
317 | 321 |
|
318 | 322 | if (resultIndex != 0) {
|
319 | 323 | outputsPrecisionSS << VALUES_SEPARATOR;
|
|
0 commit comments