fix

PaddlePaddle · Sep 2, 2023 · 5824aa0 · 5824aa0
1 parent 2cdbd94
commit 5824aa0
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/sot/opcode_translator/executor/opcode_executor.py b/sot/opcode_translator/executor/opcode_executor.py
@@ -1862,11 +1862,13 @@ def _break_graph_in_call(
 
         for i, stack_arg in enumerate(self.stack):
             # Avoid passing NULL as a parameter to the resume function
-            if isinstance(stack_arg, NullVariable):
-                if i < len(self.stack) - pop_n:
-                    self._graph.pycode_gen.gen_load_object(
-                        NullVariable(), f'dummy_var{i}', push_null=False
-                    )
+            if (
+                isinstance(stack_arg, NullVariable)
+                and i < len(self.stack) - pop_n
+            ):
+                self._graph.pycode_gen.gen_load_object(
+                    NullVariable(), f'dummy_var{i}', push_null=False
+                )
             else:
                 var_loader.load(stack_arg)