Fix program cache bug (#65678)

zhangbo9674 · web-flow · commit a0db8a448344 · 2024-07-05T13:49:18.000+08:00
* fix

* fix

* fix
diff --git a/python/paddle/base/executor.py b/python/paddle/base/executor.py
@@ -1123,6 +1123,9 @@ def get_pir_program_and_executor(
         place,
         scope,
     ):
+        _add_pir_fetch_ops(
+            program, fetch_list=fetch_list, fetch_var_name=fetch_var_name
+        )
         return self._get_cached_program_and_executor_pir_mode(
             self._CachedData(
                 program,
@@ -1144,10 +1147,6 @@ def _get_pir_program_and_executor(self, cached_data):
         place = cached_data.place
         scope = cached_data.scope
 
-        _add_pir_fetch_ops(
-            program, fetch_list=fetch_list, fetch_var_name=fetch_var_name
-        )
-
         default_job = core.Job("default")
         type_to_program = {"default": program}
         plan = core.Plan([default_job], type_to_program)
diff --git a/test/legacy_test/test_lr_scheduler.py b/test/legacy_test/test_lr_scheduler.py
@@ -117,7 +117,7 @@ def _test_static(self, place, kwargs):
             adam = paddle.optimizer.Adam(learning_rate=scheduler)
             adam.minimize(loss)
             lr_var = adam._global_learning_rate()
-            test_prog = main_prog.clone()
+            test_prog = main_prog
 
         exe = paddle.static.Executor(place)
         exe.run(start_prog)
@@ -329,7 +329,7 @@ def _test_static(self, place, kwargs):
             loss = paddle.mean(x)
             adam.minimize(loss)
             lr_var = adam._global_learning_rate()
-            test_prog = main_prog.clone()
+            test_prog = main_prog
 
         exe = paddle.static.Executor(place)
         exe.run(start_prog)
@@ -696,7 +696,7 @@ def _test_static(self, python_func, paddle_api, kwarg, place):
 
             adam.minimize(loss)
             lr_var = adam._global_learning_rate()
-            test_prog = main_prog.clone()
+            test_prog = main_prog
 
         num = 0
         exe = paddle.static.Executor(place)
