Polish remove_global_guarded_variable (#254)

PaddlePaddle · Jul 7, 2023 · a28875e · a28875e
1 parent 7953427
commit a28875e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sot/opcode_translator/executor/opcode_executor.py b/sot/opcode_translator/executor/opcode_executor.py
@@ -1766,14 +1766,14 @@ def FOR_ITER(self, instr):
             if not isinstance(
                 iterator, (SequenceIterVariable, DictIterVariable)
             ):
-                self._graph.remove_global_guarded_variable(iterator)
                 raise BreakGraphError()
             backup_iter_idx = iterator.idx
             self._inline_call_for_loop(iterator, instr)
             self._lasti = self.indexof(instr.jump_to)
         except BreakGraphError as e:
             if backup_iter_idx:
                 iterator.idx = backup_iter_idx
+            self._graph.remove_global_guarded_variable(iterator)
             self._break_graph_in_for_loop(iterator, instr)
             return Stop()