From 330253533409f86c8738aa082726c87e168724b7 Mon Sep 17 00:00:00 2001
From: Bogdan Pereanu <bogdan.pereanu@intel.com>
Date: Wed, 15 May 2024 15:08:26 +0300
Subject: [PATCH] [SAMPLES] Get the correct batch size when running with
 pre-compiled models (#24507)

### Details:
 - *get the correct batch size when running with pre-compiled models*

### Tickets:
 - *CVS-140703*
---
 samples/cpp/benchmark_app/main.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/samples/cpp/benchmark_app/main.cpp b/samples/cpp/benchmark_app/main.cpp
index fdc5a5288ff2cf..41b72211f28555 100644
--- a/samples/cpp/benchmark_app/main.cpp
+++ b/samples/cpp/benchmark_app/main.cpp
@@ -792,6 +792,11 @@ int main(int argc, char* argv[]) {
                                               FLAGS_scale_values,
                                               FLAGS_mean_values,
                                               compiledModel.inputs());
+
+            batchSize = get_batch_size(app_inputs_info.at(0));
+            warn_if_no_batch(app_inputs_info.at(0));
+            slog::info << "Model batch size: " << batchSize << slog::endl;
+
             if (batchSize == 0) {
                 batchSize = 1;
             }