未验证 提交 db633aff 编写于 作者: J JingZhuangzhuang 提交者: GitHub

Fix conv2d convert case (#36699)

* fix pool2d convert case

* add pool2d convert test case for trt6
上级 229bae81
...@@ -33,6 +33,10 @@ class TrtConvertPool2dTest(TrtLayerAutoScanTest): ...@@ -33,6 +33,10 @@ class TrtConvertPool2dTest(TrtLayerAutoScanTest):
for index in range(len(ksize)): for index in range(len(ksize)):
if ksize[index] <= paddings[index]: if ksize[index] <= paddings[index]:
return False return False
ver = paddle_infer.get_trt_compile_version()
if ver[0] * 1000 + ver[1] * 100 + ver[0] * 10 < 7000:
if program_config.ops[0].attrs['pooling_type'] == 'avg':
return False
return True return True
def is_program_valid(self, program_config: ProgramConfig) -> bool: def is_program_valid(self, program_config: ProgramConfig) -> bool:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册