[LIR] CleanupLoopOffsets fix

openvinotoolkit · Sep 12, 2023 · 3f554be · 3f554be
1 parent 2c11c2f
commit 3f554be
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/common/snippets/src/lowered/pass/cleanup_loop_offsets.cpp b/src/common/snippets/src/lowered/pass/cleanup_loop_offsets.cpp
@@ -48,16 +48,23 @@ bool CleanupLoopOffsets::run(LinearIR& linear_ir) {
                         const auto& found = per_port_connector_offset.find(managed_connector);
                         if (found != per_port_connector_offset.end()) {
                             // Since data ptr is incremented on [ptr_increment x increment],
-                            // we should guarantee proportionality of ptr shifts
+                            // we should guarantee proportionality of ptr shifts.
+                            // If the data ptr can't be proportionally shifted, the optimization is not applied
                             // For example,
                             // Inner Loop: WA = 32, Inc = 1, ptr_increment[0] = 20, final_offset[0] = -640
                             // Outer Loop: WA = 70, Inc = 32, ptr_increment[0] = 20, final_offset[0] = -1400
                             // To save data ptr shift proportionality, we have to calculate so:
                             //    outer_ptr_increment[0] = (inner_final_offset[0] + outer_ptr_increment[0] * outer_Inc) / outer_Inc
                             //    outer_ptr_increment[0] = (-640 + 20 x 32) / 32 = 0
-                            outer_ptr_increments[i] = (fin_offsets[found->second] + outer_ptr_increments[i] * outer_increment) / outer_increment;
-                            fin_offsets[found->second] = 0;
-                            is_modified = true;
+
+                            const auto full_outer_increment = outer_ptr_increments[i] * outer_increment;
+                            const auto new_final_outer_increment = full_outer_increment + fin_offsets[found->second];
+
+                            if (new_final_outer_increment % outer_increment == 0) {
+                                outer_ptr_increments[i] = new_final_outer_increment / outer_increment;
+                                fin_offsets[found->second] = 0;
+                                is_modified = true;
+                            }
                         }
                     }
                     outer_loop_end->set_ptr_increments(outer_ptr_increments);