@@ -18,9 +18,9 @@ class TRTorchPluginRegistry {
18
18
// register libNvInferPlugins and TRTorch plugins
19
19
// trtorch_logger logging level is set to kERROR and reset back to kDEBUG.
20
20
// This is because initLibNvInferPlugins initializes only a subset of plugins and logs them.
21
- // Plugins outside this subset in TensorRT are not being logged in this. So temporarily we disable this to prevent multiple logging of same plugins.
22
- // To provide a clear list of all plugins, we iterate through getPluginRegistry() where it prints the
23
- // list of all the plugins registered in TensorRT with their namespaces.
21
+ // Plugins outside this subset in TensorRT are not being logged in this. So temporarily we disable this to prevent
22
+ // multiple logging of same plugins. To provide a clear list of all plugins, we iterate through getPluginRegistry()
23
+ // where it prints the list of all the plugins registered in TensorRT with their namespaces.
24
24
trtorch_logger.set_reportable_log_level (util::logging::LogLevel::kERROR );
25
25
initLibNvInferPlugins (&trtorch_logger, " " );
26
26
trtorch_logger.set_reportable_log_level (util::logging::LogLevel::kDEBUG );
@@ -33,14 +33,17 @@ class TRTorchPluginRegistry {
33
33
continue ;
34
34
}
35
35
std::string pluginNamespace = pluginsList[k]->getPluginNamespace ();
36
- trtorch_logger.log (util::logging::LogLevel::kDEBUG , " Registered plugin creator - " + std::string (pluginsList[k]->getPluginName ()) + " , Namespace: " + pluginNamespace);
36
+ trtorch_logger.log (
37
+ util::logging::LogLevel::kDEBUG ,
38
+ " Registered plugin creator - " + std::string (pluginsList[k]->getPluginName ()) +
39
+ " , Namespace: " + pluginNamespace);
37
40
}
38
41
trtorch_logger.log (util::logging::LogLevel::kDEBUG , " Total number of plugins registered: " + str (numCreators));
39
42
}
40
- public:
41
- util::logging::TRTorchLogger trtorch_logger = util::logging::TRTorchLogger( " [TRTorch Plugins Context] - " ,
42
- util::logging::LogLevel:: kDEBUG ,
43
- true );
43
+
44
+ public:
45
+ util::logging::TRTorchLogger trtorch_logger =
46
+ util::logging::TRTorchLogger ( " [TRTorch Plugins Context] - " , util::logging::LogLevel:: kDEBUG , true );
44
47
};
45
48
46
49
namespace {
0 commit comments