#18 Handle multiple stack vars instruction in condensed graph

costa-group · Dec 23, 2024 · 4c29c4f · 4c29c4f
1 parent 6c6da3e
commit 4c29c4f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/greedy/greedy_new_version.py b/src/greedy/greedy_new_version.py
@@ -596,8 +596,9 @@ def _is_condensed(self, node):
         """
         Whether to consider the instruction associated to the node
         """
-        return "STORE" in node or any(self._stack_var_copies_needed[out_stack] > 1 or out_stack in self._final_stack
-                                      for out_stack in self._id2instr[node]["outpt_sk"])
+        return "STORE" in node or self._id2instr[node]["outpt_sk"] > 1 or \
+            any(self._stack_var_copies_needed[out_stack] > 1 or out_stack in self._final_stack
+                for out_stack in self._id2instr[node]["outpt_sk"])
 
     def _compute_top_can_used(self, instr: instr_JSON_T, top_can_be_used: Dict[var_id_T, Set[var_id_T]]) -> Set[
         var_id_T]: