fix debug 2

allnes · Nov 2, 2023 · 6f40540 · 6f40540
1 parent 625f6c2
commit 6f40540
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_ie_scheduler.cpp b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_ie_scheduler.cpp
@@ -24,10 +24,11 @@ void ACLScheduler::set_num_threads(unsigned int num_threads) {}
 
 void ACLScheduler::schedule_custom(ICPPKernel *kernel, const Hints &hints, const Window &window, ITensorPack &tensors) {
     const Window & max_window = window;
-    const unsigned int num_iterations = max_window.num_iterations_total();
+    const unsigned int num_iterations =
+            max_window.num_iterations(hints.split_dimension()) == 1 ? 1 : max_window.num_iterations_total();
     const auto _num_threads = std::min(num_iterations, static_cast<unsigned int>(parallel_get_num_threads()));
 
-    if (num_iterations == 0) {
+    if (num_iterations < 1) {
         return;
     }