【pir_save_load】modify GRUcell test for predcitor (#68723)

xiaoguoguo626807 · web-flow · commit a8158b63833c · 2024-10-16T18:50:57.000+08:00
* modify topk

* modify GRUCell for same fetchvar filter
diff --git a/python/paddle/jit/api.py b/python/paddle/jit/api.py
@@ -660,7 +660,9 @@ def _get_output_vars(outputs, output_spec, with_hook=False):
         from paddle.autograd.backward_utils import ValueSet
 
         for var in paddle.utils.flatten(outputs):
-            if isinstance(var, paddle.pir.Value):
+            if isinstance(var, paddle.pir.Value) and var not in ValueSet(
+                result_list
+            ):
                 result_list.append(var)
 
         if output_spec is not None:
diff --git a/python/paddle/static/pir_io.py b/python/paddle/static/pir_io.py
@@ -329,6 +329,8 @@ def normalize_pir_program(program, feed_vars, fetch_vars, **kwargs):
         scale_op = var.get_defining_op()
         if scale_op.name() == "pd_op.scale":
             orig_var = scale_op.operand_source(0)
+        else:
+            orig_var = var
         if orig_var.has_name:
             fetch_vars_tuple.append((orig_var, orig_var.name))
         else:
