【pir】modify name of Fetch input (#68530)

xiaoguoguo626807 · phlrain · web-flow · commit 8fdba0b1a4af · 2024-10-09T14:18:34.000+08:00
* modify old ir put_along_axis api

* upate

* modify paddletest

* modify fetch input name

---------

Co-authored-by: phlrain &lt;phliuhongyu@126.com&gt;
diff --git a/python/paddle/static/pir_io.py b/python/paddle/static/pir_io.py
@@ -326,10 +326,11 @@ def normalize_pir_program(program, feed_vars, fetch_vars, **kwargs):
 
     fetch_vars_tuple = []
     for i, var in enumerate(clone_fetch_vars):
-        if "name" in var.get_defining_op().attrs():
-            fetch_vars_tuple.append(
-                (var, var.get_defining_op().attrs()['name'])
-            )
+        scale_op = var.get_defining_op()
+        if scale_op.name() == "pd_op.scale":
+            orig_var = scale_op.operand_source(0)
+        if orig_var.has_name:
+            fetch_vars_tuple.append((orig_var, orig_var.name))
         else:
             fetch_vars_tuple.append((var, "fetch_name_" + str(i)))
     with paddle.static.program_guard(copy_program):
