From 1f2c38a0a662c09474c7c89b0c4d89b357e213c8 Mon Sep 17 00:00:00 2001 From: Ruibiao Chen Date: Fri, 2 Sep 2022 14:29:56 +0800 Subject: [PATCH] Reduce test cases for test_buffer_shared_memory_reuse_pass (#45662) --- .../unittests/test_buffer_shared_memory_reuse_pass.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_buffer_shared_memory_reuse_pass.py b/python/paddle/fluid/tests/unittests/test_buffer_shared_memory_reuse_pass.py index b52b6d4803..43fb6b7674 100644 --- a/python/paddle/fluid/tests/unittests/test_buffer_shared_memory_reuse_pass.py +++ b/python/paddle/fluid/tests/unittests/test_buffer_shared_memory_reuse_pass.py @@ -106,8 +106,8 @@ class InplaceTestBase(unittest.TestCase): repeated_var_names = all_vars_name random.shuffle(repeated_var_names) # add some random - for fetch_var in repeated_var_names: - for _ in range(4): + for fetch_var in repeated_var_names[:4]: + for _ in range(2): with fluid.scope_guard(scope1): fetch_val1, = exe.run(prog1, feed=feed_dict, @@ -158,8 +158,8 @@ class InplaceTestBase(unittest.TestCase): repeated_var_names = self.get_all_vars(prog1) random.shuffle(repeated_var_names) # add some random - for fetch_var in repeated_var_names: - for _ in range(4): + for fetch_var in repeated_var_names[:4]: + for _ in range(2): fetch_vals = [] for scope, compiled_prog in zip(scopes, compiled_programs): with fluid.scope_guard(scope): -- GitLab