fix func args

PaddlePaddle · Feb 7, 2025 · d1f8d0d · d1f8d0d
1 parent 6384ec4
commit d1f8d0d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/paddle/jit/dy2static/pir_partial_program.py b/python/paddle/jit/dy2static/pir_partial_program.py
@@ -774,7 +774,12 @@ def _get_scope(self, program_id=None, use_scope_cache=False):
     def _create_program(self, is_infer_mode=False) -> RunnableProgram:
         if is_infer_mode:
 
-            def pass_fn(forward_program, backward_program, program_name_attr):
+            def pass_fn(
+                forward_program,
+                backward_program,
+                program_name_attr,
+                whole_program,
+            ):
                 # common pass
                 pm = paddle.base.libpaddle.pir.PassManager()
                 paddle.base.libpaddle.pir.infer_symbolic_shape_pass(