Fix conv2d convert case (#36699)

* fix pool2d convert case * add pool2d convert test case for trt6
PaddlePaddle · Oct 26, 2021 · db633af · db633af
1 parent 229bae8
commit db633af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_pool2d.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_pool2d.py
@@ -33,6 +33,10 @@ def is_paddings_valid(self, program_config: ProgramConfig) -> bool:
                 for index in range(len(ksize)):
                     if ksize[index] <= paddings[index]:
                         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
 
     def is_program_valid(self, program_config: ProgramConfig) -> bool: