From d3c9394202579ab65bedfb3cbe0cc058a410f600 Mon Sep 17 00:00:00 2001
From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com>
Date: Sun, 17 Oct 2021 22:22:30 -0500
Subject: [PATCH] Fix conv2d op_teller error (#36474)

---
 paddle/fluid/inference/tensorrt/op_teller.cc | 24 +++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index 59368a299c5..89159c0bb63 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -242,9 +242,31 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
       if (desc.HasAttr("padding_algorithm")) {
         auto padding_algorithm =
             BOOST_GET_CONST(std::string, desc.GetAttr("padding_algorithm"));
-        if (padding_algorithm == "SAME" || padding_algorithm == "VALID") {
+        if (padding_algorithm == "VALID") {
           return false;
         }
+        if (padding_algorithm == "SAME") {
+          if (desc.HasAttr("dilations")) {
+            const std::vector<int> dilations =
+                BOOST_GET_CONST(std::vector<int>, desc.GetAttr("dilations"));
+            if (dilations[0] != 1 || dilations[1] != 1) {
+              VLOG(3) << "In Same mode, Dilations must be (1, 1) for "
+                         "tensorRT, but given ("
+                      << dilations[0] << ", " << dilations[1] << ")";
+              return false;
+            }
+          }
+        }
+      }
+
+      if (use_no_calib_int8) {
+        if (desc.HasAttr("padding_algorithm")) {
+          auto padding_algorithm =
+              BOOST_GET_CONST(std::string, desc.GetAttr("padding_algorithm"));
+          if (padding_algorithm == "SAME") {
+            return false;
+          }
+        }
       }
 
       if (desc.HasAttr("enable_int8")) {
-- 
GitLab