fix the matmul_v2 test for cuda11 (#28635)

5 years ago · 912a5c30b4
parent 82f0b5ea5c
commit 912a5c30b4
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py
+++ b/python/paddle/fluid/tests/unittests/test_matmul_v2_op.py
@ -148,8 +148,8 @@ class TestMatMuklOp6(TestMatMulV2Op):
    """

    def config(self):
-        self.x_shape = (1, 2, 100, 1)
-        self.y_shape = (100, )
+        self.x_shape = (1, 2, 102, 1)
+        self.y_shape = (102, )
        self.trans_x = True
        self.trans_y = False