From b17e36a49851558da99b3ced7d9c6337b314639c Mon Sep 17 00:00:00 2001
From: zlsh80826 <rewang@nvidia.com>
Date: Tue, 6 Apr 2021 14:46:24 +0800
Subject: [PATCH] [PaddleTRT] Yolov3 bugfix (#32064)

* fix yolobox teller condition

* fix cuda double free bug
---
 paddle/fluid/inference/tensorrt/op_teller.cc                 | 2 +-
 paddle/fluid/inference/tensorrt/plugin/yolo_box_op_plugin.cu | 5 +----
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index f4e7c334632..b681b098c8c 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") {
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 e1b4c898d21..13d07e77403 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);
-- 
GitLab