From 912a5c30b44bd6d254093ee815c0596dbb94e1cd Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Tue, 17 Nov 2020 14:36:52 +0800 Subject: [PATCH] fix the matmul_v2 test for cuda11 (#28635) --- python/paddle/fluid/tests/unittests/test_matmul_v2_op.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 640771df23..a6667db622 100644 --- 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 -- GitLab