From b8dbffb772b4539c1fd3c8a613e5067cd238746d Mon Sep 17 00:00:00 2001
From: feng_shuai <fengshuai03@baidu.com>
Date: Tue, 22 Feb 2022 10:48:56 +0800
Subject: [PATCH] fix:Modify matrix latitude (#39686)

---
 .../ir/inference/test_trt_convert_matmul.py      | 16 +++-------------
 1 file changed, 3 insertions(+), 13 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_matmul.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_matmul.py
index 8913159b2c4..c6f2fa205c7 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_matmul.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_matmul.py
@@ -172,11 +172,11 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest):
             }
             self.dynamic_shape.max_input_shape = {
                 "input1_data": [16, 4, 4],
-                "input2_data": [16, 4, 128]
+                "input2_data": [16, 4, 4]
             }
             self.dynamic_shape.opt_input_shape = {
                 "input1_data": [8, 4, 4],
-                "input2_data": [8, 4, 16]
+                "input2_data": [8, 4, 4]
             }
 
         attrs = [
@@ -192,17 +192,7 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest):
         yield self.create_inference_config(), (1, 3), 1e-5
 
     def add_skip_trt_case(self):
-        def teller1(program_config, predictor_config):
-            if len(
-                    self.dynamic_shape.min_input_shape
-            ) != 0 and self.trt_param.precision == paddle_infer.PrecisionType.Half:
-                return True
-            return False
-
-        self.add_skip_case(
-            teller1, SkipReasons.TRT_NOT_IMPLEMENTED,
-            "Tensorrt MatrixMultiply layer will get error when dynamic shape fp16 mode."
-        )
+        pass
 
     def test(self):
         self.add_skip_trt_case()
-- 
GitLab