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 1a1d08002154fe21f7d817276bb9b80a1dee7765..671efd8c721550256c181059528bead43deb0718 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 @@ -98,7 +98,7 @@ class InplaceTestBase(unittest.TestCase): compiled_programs.append(compiled_prog) all_vars_name = self.get_all_vars(prog1) - repeated_var_names = all_vars_name * 4 + repeated_var_names = all_vars_name * 2 random.shuffle(repeated_var_names) # add some random for fetch_var in repeated_var_names: @@ -144,7 +144,7 @@ class InplaceTestBase(unittest.TestCase): places=places) compiled_programs.append(compiled_program) - repeated_var_names = self.get_all_vars(prog1) * 4 + repeated_var_names = self.get_all_vars(prog1) * 2 random.shuffle(repeated_var_names) # add some random for fetch_var in repeated_var_names: