update

PaddlePaddle · Nov 16, 2023 · 88f7b16 · 88f7b16
1 parent fbc5064
commit 88f7b16
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/paddle/jit/dy2static/partial_program.py b/python/paddle/jit/dy2static/partial_program.py
@@ -277,6 +277,11 @@ def __call__(self, inputs):
         self._cast_fp16_if_pure_fp16(in_vars)
         attrs = self._prepare_attributes()
 
+        for idx, val in enumerate(attrs):
+            if val == "x_names":
+                attrs[idx + 1] = in_var_names
+                break
+
         self._sync_lr_value_with_scheduler()
 
         _legacy_C_ops.run_program(