Custom plugin registration calls moved

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

core/plugins/impl/interpolate_plugin.cpp

@@ -417,6 +417,8 @@ const nvinfer1::PluginFieldCollection* InterpolatePluginCreator::getFieldNames()
   return nullptr;
 }
 
+REGISTER_TRTORCH_PLUGIN(InterpolatePluginCreator);
+
 } // namespace impl
 } // namespace plugins
 } // namespace core

core/plugins/impl/normalize_plugin.cpp

@@ -286,6 +286,8 @@ const nvinfer1::PluginFieldCollection* NormalizePluginCreator::getFieldNames() {
   return nullptr;
 }
 
+REGISTER_TRTORCH_PLUGIN(NormalizePluginCreator);
+
 } // namespace impl
 } // namespace plugins
 } // namespace core

core/plugins/register_plugins.cpp

@@ -14,15 +14,33 @@ namespace impl {
 class TRTorchPluginRegistry {
  public:
   TRTorchPluginRegistry() {
-    auto trtorch_logger = util::logging::TRTorchLogger(
-        "[TRTorch Plugins Context] - ",
-        util::logging::get_logger().get_reportable_severity(),
-        util::logging::get_logger().get_is_colored_output_on());
+    trtorch_logger.log(util::logging::LogLevel::kINFO, "Instatiated the TRTorch plugin registry class");
     // register libNvInferPlugins and TRTorch plugins
+    // trtorch_logger logging level is set to kERROR and reset back to kDEBUG.
+    // This is because initLibNvInferPlugins initializes only a subset of plugins and logs them.
+    // Plugins outside this subset in TensorRT are not being logged in this. So temporarily we disable this to prevent multiple logging of same plugins.
+    // To provide a clear list of all plugins, we iterate through getPluginRegistry() where it prints the
+    // list of all the plugins registered in TensorRT with their namespaces.
+    trtorch_logger.set_reportable_log_level(util::logging::LogLevel::kERROR);
     initLibNvInferPlugins(&trtorch_logger, "");
-    REGISTER_TRTORCH_PLUGIN(InterpolatePluginCreator);
-    REGISTER_TRTORCH_PLUGIN(NormalizePluginCreator);
+    trtorch_logger.set_reportable_log_level(util::logging::LogLevel::kDEBUG);
+
+    int numCreators = 0;
+    auto pluginsList = getPluginRegistry()->getPluginCreatorList(&numCreators);
+    for (int k = 0; k < numCreators; ++k) {
+      if (!pluginsList[k]) {
+        trtorch_logger.log(util::logging::LogLevel::kDEBUG, "Plugin creator for plugin " + str(k) + " is a nullptr");
+        continue;
+      }
+      std::string pluginNamespace = pluginsList[k]->getPluginNamespace();
+      trtorch_logger.log(util::logging::LogLevel::kDEBUG, "Registered plugin creator - " + std::string(pluginsList[k]->getPluginName()) + ", Namespace: " + pluginNamespace);
+    }
+    trtorch_logger.log(util::logging::LogLevel::kDEBUG, "Total number of plugins registered: " + str(numCreators));
   }
+public:
+  util::logging::TRTorchLogger trtorch_logger = util::logging::TRTorchLogger("[TRTorch Plugins Context] - ",
+                                                                              util::logging::LogLevel::kDEBUG,
+                                                                              true);
 };
 
 namespace {

examples/sample_rt_app/BUILD

@@ -4,7 +4,7 @@ package(default_visibility = ["//visibility:public"])
 # It has include, lib, bin directory and LICENSE file
 cc_library(
     name = "trtorch_runtime",
-    srcs = ["trtorch/lib/libtrtorchrt.so", "trtorch/lib/libtrtorch_plugins.so"],
+    srcs = ["trtorch/lib/libtrtorchrt.so"], # , "trtorch/lib/libtrtorch_plugins.so"
 )
 
 cc_binary(