reduce size for overtime again

modified: ../test/legacy_test/test_baddbmm_op.py
PaddlePaddle · Jan 11, 2025 · 85bb2f5 · 85bb2f5
1 parent da9467b
commit 85bb2f5
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/test/legacy_test/test_baddbmm_op.py b/test/legacy_test/test_baddbmm_op.py
@@ -31,8 +31,8 @@ def setUp(self):
         self.public_python_api = paddle.baddbmm
         self.init_dtype_type()
         self.inputs = {
-            'Input': np.random.random((2, 20, 5)).astype(self.dtype),
-            'X': np.random.random((2, 20, 10)).astype(self.dtype),
+            'Input': np.random.random((2, 10, 5)).astype(self.dtype),
+            'X': np.random.random((2, 10, 10)).astype(self.dtype),
             'Y': np.random.random((2, 10, 5)).astype(self.dtype),
         }
         self.outputs = {
@@ -101,9 +101,9 @@ def setUp(self):
         self.python_api = paddle.baddbmm
         self.init_dtype_type()
         self.inputs = {
-            'Input': np.random.random((3, 40, 1)).astype(self.dtype),
-            'X': np.random.random((3, 40, 5)).astype(self.dtype),
-            'Y': np.random.random((3, 5, 10)).astype(self.dtype),
+            'Input': np.random.random((2, 50, 1)).astype(self.dtype),
+            'X': np.random.random((2, 50, 5)).astype(self.dtype),
+            'Y': np.random.random((2, 5, 10)).astype(self.dtype),
         }
         self.outputs = {
             'Out': self.inputs['Input']
@@ -258,8 +258,8 @@ def setUp(self):
         self.dtype = np.float64
         self.init_dtype_type()
         self.inputs = {
-            'Input': np.random.random((2, 20, 5)).astype(self.dtype),
-            'X': np.random.random((2, 20, 10)).astype(self.dtype),
+            'Input': np.random.random((2, 10, 5)).astype(self.dtype),
+            'X': np.random.random((2, 10, 10)).astype(self.dtype),
             'Y': np.random.random((2, 10, 5)).astype(self.dtype),
         }
         self.attrs = {
@@ -282,8 +282,8 @@ def setUp(self):
         self.dtype = np.float64
         self.init_dtype_type()
         self.inputs = {
-            'Input': np.random.random((2, 20, 5)).astype(self.dtype),
-            'X': np.random.random((2, 20, 10)).astype(self.dtype),
+            'Input': np.random.random((2, 10, 5)).astype(self.dtype),
+            'X': np.random.random((2, 10, 10)).astype(self.dtype),
             'Y': np.random.random((2, 10, 5)).astype(self.dtype),
         }
         self.attrs = {