From 683f81907296d515eb7325c68fa8fb1a19362135 Mon Sep 17 00:00:00 2001 From: Ruibiao Chen Date: Tue, 2 Aug 2022 19:26:15 +0800 Subject: [PATCH] Do not set return_merge in test_parallel_executor_run_cinn (#44809) --- .../tests/unittests/test_parallel_executor_run_cinn.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py b/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py index 3b275a75b7d..00bacaa58e8 100644 --- a/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py +++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py @@ -99,11 +99,8 @@ def train(dot_save_dir, prefix, seed=1234): feed = rand_data(img.name, label.name, iters) loss_values = [] for step in range(iters): - loss_v = exe.run(compiled_program, - feed=feed[step], - fetch_list=[loss], - return_merged=False) - loss_values.append(loss_v[0][0][0]) + loss_v = exe.run(compiled_program, feed=feed[step], fetch_list=[loss]) + loss_values.append(loss_v[0][0]) return loss_values -- GitLab