fix trt convert conv2d skip

PaddlePaddle · Jan 18, 2022 · 62b382d · 62b382d · paddle-bot-old · Jan 18, 2022
1 parent 2c47678
commit 62b382d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_conv2d.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_conv2d.py
@@ -34,6 +34,12 @@ def is_program_valid(self, program_config: ProgramConfig) -> bool:
                 1] * attrs[0]['groups']:
             return False
 
+        ver = paddle_infer.get_trt_compile_version()
+        if ver[0] * 1000 + ver[1] * 100 + ver[0] * 10 < 7000:
+            if attrs[0]['padding_algorithm'] == 'SAME' and (
+                    attrs[0]['strides'][0] > 1 or attrs[0]['strides'][1] > 1):
+                return False
+
         return True
 
     def sample_program_configs(self):

diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_conv2d_fusion.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_conv2d_fusion.py
@@ -37,6 +37,12 @@ def is_program_valid(self, program_config: ProgramConfig) -> bool:
         if attrs[0]['groups'] <= 1:
             return False
 
+        ver = paddle_infer.get_trt_compile_version()
+        if ver[0] * 1000 + ver[1] * 100 + ver[0] * 10 < 7000:
+            if attrs[0]['padding_algorithm'] == 'SAME' and (
+                    attrs[0]['strides'][0] > 1 or attrs[0]['strides'][1] > 1):
+                return False
+
         return True
 
     def sample_program_configs(self):