From be00a42f028b38eb9108e954890fb2d11a07d3b1 Mon Sep 17 00:00:00 2001
From: weishengying <63448337+weishengying@users.noreply.github.com>
Date: Fri, 16 Sep 2022 16:15:55 +0800
Subject: [PATCH] Fix bugs in "test_custom_plugin_creater" unit test (#46075)

---
 paddle/fluid/inference/tensorrt/convert/CMakeLists.txt    | 8 +-------
 .../tensorrt/convert/test_custom_plugin_creater.cc        | 4 ++++
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt b/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt
index aebfa3d983d..60a5d0f2825 100644
--- a/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt
+++ b/paddle/fluid/inference/tensorrt/convert/CMakeLists.txt
@@ -103,13 +103,7 @@ nv_test(
 nv_test(
   test_custom_plugin_creater
   SRCS test_custom_plugin_creater.cc
-  DEPS paddle_framework
-       ${GLOB_OPERATOR_DEPS}
-       tensorrt_engine
-       tensorrt_plugin
-       tensorrt_converter
-       op_meta_info
-       custom_operator)
+  DEPS paddle_framework tensorrt_converter op_meta_info custom_operator)
 
 if(WITH_ONNXRUNTIME AND WIN32)
   # Copy onnxruntime for some c++ test in Windows, since the test will
diff --git a/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc b/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc
index a5c4627c407..2a3ead9c8e6 100644
--- a/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc
+++ b/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc
@@ -102,6 +102,8 @@ TEST(CustomPluginCreater, StaticShapePlugin) {
 
   framework::Scope scope;
 
+  tensorrt::plugin::TrtPluginRegistry::Global()->RegistToTrt();
+
   auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
 
   framework::OpDesc custom_op(*op_desc, nullptr);
@@ -186,6 +188,8 @@ TEST(CustomPluginCreater, DynamicShapePlugin) {
 
   framework::Scope scope;
 
+  tensorrt::plugin::TrtPluginRegistry::Global()->RegistToTrt();
+
   auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
 
   framework::OpDesc custom_op(*op_desc, nullptr);
-- 
GitLab