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

fix:Modify matrix latitude (#39686)

上级 12c6d06a
...@@ -172,11 +172,11 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest): ...@@ -172,11 +172,11 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest):
} }
self.dynamic_shape.max_input_shape = { self.dynamic_shape.max_input_shape = {
"input1_data": [16, 4, 4], "input1_data": [16, 4, 4],
"input2_data": [16, 4, 128] "input2_data": [16, 4, 4]
} }
self.dynamic_shape.opt_input_shape = { self.dynamic_shape.opt_input_shape = {
"input1_data": [8, 4, 4], "input1_data": [8, 4, 4],
"input2_data": [8, 4, 16] "input2_data": [8, 4, 4]
} }
attrs = [ attrs = [
...@@ -192,17 +192,7 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest): ...@@ -192,17 +192,7 @@ class TrtConvertMatmulTest_dynamic(TrtLayerAutoScanTest):
yield self.create_inference_config(), (1, 3), 1e-5 yield self.create_inference_config(), (1, 3), 1e-5
def add_skip_trt_case(self): def add_skip_trt_case(self):
def teller1(program_config, predictor_config): pass
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."
)
def test(self): def test(self):
self.add_skip_trt_case() self.add_skip_trt_case()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册