diff --git a/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py b/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py
index bcf5078fc932416b3efbb2491c35f0b907ab14b4..75d82d270240ca9ff75282798901dc3c9ee33113 100644
--- a/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py
+++ b/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py
@@ -232,8 +232,8 @@ class TestMatMuklOp13(TestMatMulV2Op):
     """
 
     def config(self):
-        self.x_shape = (2, 2, 2, 50)
-        self.y_shape = (2, 2, 2, 50)
+        self.x_shape = (2, 2, 10, 10)
+        self.y_shape = (2, 2, 10, 10)
         self.trans_x = True
         self.trans_y = False
 
@@ -256,8 +256,8 @@ class TestMatMuklOp15(TestMatMulV2Op):
     """
 
     def config(self):
-        self.x_shape = (3, 1, 1, 2, 100)
-        self.y_shape = (1, 2, 2, 100, 1)
+        self.x_shape = (3, 1, 1, 10, 10)
+        self.y_shape = (1, 2, 2, 10, 10)
         self.trans_x = False
         self.trans_y = False