update for cpu reservation

openvinotoolkit · Dec 24, 2024 · bd4a132 · bd4a132
1 parent e684864
commit bd4a132
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/plugins/intel_cpu/src/cpu_streams_calculation.cpp b/src/plugins/intel_cpu/src/cpu_streams_calculation.cpp
@@ -210,8 +210,12 @@ std::vector<std::vector<int>> get_streams_info_table(
         ((input_streams_changed == true) && (input_streams == 1))) {
         n_streams = 1;
         stream_info[NUMBER_OF_STREAMS] = n_streams;
-        current_socket_id =
-            proc_type_table.size() == 1 ? proc_type_table[0][PROC_SOCKET_ID] : proc_type_table[1][PROC_SOCKET_ID];
+        for (size_t n = 0; n < proc_socket_table.size(); n++) {
+            if (proc_socket_table[n][ALL_PROC] > 0) {
+                current_socket_id = proc_socket_table[n][PROC_SOCKET_ID];
+                break;
+            }
+        }
         if (input_threads > 0) {
             if (hint_model_distribution_policy.size() == 0) {
                 n_threads_per_stream = std::min(input_threads, proc_type_table[0][ALL_PROC]);