From db633affe1b04a880935eeb20d405ff3466a0841 Mon Sep 17 00:00:00 2001
From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com>
Date: Mon, 25 Oct 2021 21:13:53 -0500
Subject: [PATCH] Fix conv2d convert case (#36699)

* fix pool2d convert case

* add pool2d convert test case for trt6
---
 .../tests/unittests/ir/inference/test_trt_convert_pool2d.py   | 4 ++++
 1 file changed, 4 insertions(+)

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
index 05545f0b0e9..ddb96c37db7 100644
--- 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 @@ class TrtConvertPool2dTest(TrtLayerAutoScanTest):
                 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:
-- 
GitLab