Optimize exceptional iterations

slothy-optimizer · Jul 12, 2024 · 8867e41 · 8867e41
1 parent 6b061ca
commit 8867e41
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/slothy/core/slothy.py b/slothy/core/slothy.py
@@ -490,8 +490,12 @@ def loop_lbl_iter(i):
         if self.config.sw_pipelining.unknown_iteration_count:
             optimized_code += indented(self.arch.Branch.unconditional(loop_lbl_end))
             for i in range(1, num_exceptional):
+                exceptional = i * body
+                c2 = c.copy()
+                c2.sw_pipelining.enabled = False
+                res = Heuristics.linear(exceptional, logger.getChild(f"exceptional_{i}"), c2)
                 optimized_code += [SourceLine(f"{loop_lbl_iter(i)}:")]
-                optimized_code += i * indented(body)
+                optimized_code += indented(res.code)
                 optimized_code += [SourceLine(f"{loop_lbl_iter(i)}_end:")]
                 if i != num_exceptional - 1:
                     optimized_code += indented(self.arch.Branch.unconditional(loop_lbl_end))

diff --git a/slothy/helper.py b/slothy/helper.py
@@ -37,7 +37,7 @@ def _extract_comments_from_text(self):
             return
         s = list(self._raw.split("//"))
         self._raw = s[0]
-        self._comments += map(str.strip, s[1:])
+        self._comments += map(str.lstrip, s[1:])
         self._trim_comments()
 
     def _extract_indentation_from_text(self):