From 079cc02d15eadc11a0efef658743d5b168d0fd2b Mon Sep 17 00:00:00 2001
From: Reese Wang <rewang@nvidia.com>
Date: Sun, 4 Apr 2021 20:15:03 +0800
Subject: [PATCH 1/2] fix yolobox teller condition

---
 paddle/fluid/inference/tensorrt/op_teller.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index f4e7c33463240f..b681b098c8c765 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -206,7 +206,7 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
           (desc.HasAttr("class_num") && desc.HasAttr("anchors") &&
            desc.HasAttr("downsample_ratio") && desc.HasAttr("conf_thresh") &&
            desc.HasAttr("clip_bbox") && desc.HasAttr("scale_x_y"));
-      return has_attrs;
+      if (!has_attrs) return false;
     }
 
     if (op_type == "affine_channel") {

From e9e10f404bef607cee21f6cd9d15ca352fe0689f Mon Sep 17 00:00:00 2001
From: Reese Wang <rewang@nvidia.com>
Date: Sun, 4 Apr 2021 20:15:24 +0800
Subject: [PATCH 2/2] fix cuda double free bug

---
 paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu
index e1b4c898d212ff..13d07e774036a4 100644
--- a/paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu
+++ b/paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu
@@ -283,10 +283,7 @@ void YoloBoxPlugin::serialize(void* buffer) const {
   SerializeValue(&buffer, input_w_);
 }
 
-void YoloBoxPlugin::destroy() {
-  cudaFree(anchors_device_);
-  delete this;
-}
+void YoloBoxPlugin::destroy() {}
 
 void YoloBoxPlugin::setPluginNamespace(const char* lib_namespace) {
   namespace_ = std::string(lib_namespace);