未验证 提交 f9b44fe4 编写于 作者: F feng_shuai 提交者: GitHub

fix:vit_attention ut (#48884)

上级 e6fdcd90
......@@ -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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册