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 b52b6d4803f07dabdca46bca3fb36b8db0f081b8..43fb6b7674b778146274410e650c347b7fc16bd8 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):