fix:vit_attention ut (#48884)

PaddlePaddle · Dec 9, 2022 · f9b44fe · f9b44fe
1 parent e6fdcd9
commit f9b44fe
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_multihead_matmul.py
@@ -1074,7 +1074,16 @@ def generate_trt_nodes_num():
         )
 
     def add_skip_trt_case(self):
-        pass
+        def teller1(program_config, predictor_config):
+            if self.trt_param.precision == paddle_infer.PrecisionType.Half:
+                return True
+            return False
+
+        self.add_skip_case(
+            teller1,
+            SkipReasons.TRT_NOT_IMPLEMENTED,
+            "The output has diff between gpu and trt in fp16 mode.",
+        )
 
     def test(self):
         self.add_skip_trt_case()