From f9b44fe4b5afe9cfca7586553ed4cab41b6142ef Mon Sep 17 00:00:00 2001 From: feng_shuai Date: Fri, 9 Dec 2022 14:33:57 +0800 Subject: [PATCH] fix:vit_attention ut (#48884) --- .../ir/inference/test_trt_convert_multihead_matmul.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 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 index 074b55d5df..cca7fcddcd 100644 --- 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 @@ class TrtConvertVitToMultiHeadMatmulTest(TrtLayerAutoScanTest): ) 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() -- GitLab