From 2aaea02df4e7b3f6eadfaa5c5e4ca1e7edc57f15 Mon Sep 17 00:00:00 2001
From: zhoutianzi666 <17801055074@163.com>
Date: Tue, 6 Dec 2022 05:25:25 +0000
Subject: [PATCH] forbid conv op whose weight is not a persistable weight into
 Paddle-TRT

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

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index fbe3615f4d9d6..f40ef1369d8fe 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -284,6 +284,15 @@ struct SimpleOpTypeSetTeller : public Teller {
         }
       }
 #endif
+      auto* block = desc.Block();
+      if (block) {
+        auto* filter_var_desc = block->FindVar(desc.Input("Filter")[0]);
+        if (!filter_var_desc->Persistable()) {
+          VLOG(3) << "Trt not support filter is  a intermediate tensor in "
+                     "conv2d op.";
+          return false;
+        }
+      }
     }
 
     if (op_type == "deformable_conv") {