Skip to content

Commit a59a091

Browse files
committed
refactor: update log message
Signed-off-by: ktro2828 <kotaro.uetake@tier4.jp>
1 parent ffd0b40 commit a59a091

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

common/tensorrt_common/include/tensorrt_common/logger.hpp

+5-5
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ namespace
493493
//!
494494
inline LogStreamConsumer LOG_VERBOSE(const Logger & logger)
495495
{
496-
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kVERBOSE);
496+
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kVERBOSE) << "[TRT] ";
497497
}
498498

499499
//!
@@ -505,7 +505,7 @@ inline LogStreamConsumer LOG_VERBOSE(const Logger & logger)
505505
//!
506506
inline LogStreamConsumer LOG_INFO(const Logger & logger)
507507
{
508-
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kINFO);
508+
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kINFO) << "[TRT] ";
509509
}
510510

511511
//!
@@ -517,7 +517,7 @@ inline LogStreamConsumer LOG_INFO(const Logger & logger)
517517
//!
518518
inline LogStreamConsumer LOG_WARN(const Logger & logger)
519519
{
520-
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kWARNING);
520+
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kWARNING) << "[TRT] ";
521521
}
522522

523523
//!
@@ -529,7 +529,7 @@ inline LogStreamConsumer LOG_WARN(const Logger & logger)
529529
//!
530530
inline LogStreamConsumer LOG_ERROR(const Logger & logger)
531531
{
532-
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kERROR);
532+
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kERROR) << "[TRT] ";
533533
}
534534

535535
//!
@@ -543,7 +543,7 @@ inline LogStreamConsumer LOG_ERROR(const Logger & logger)
543543
//!
544544
inline LogStreamConsumer LOG_FATAL(const Logger & logger)
545545
{
546-
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kINTERNAL_ERROR);
546+
return LogStreamConsumer(logger.getReportableSeverity(), Severity::kINTERNAL_ERROR) << "[TRT] ";
547547
}
548548

549549
} // anonymous namespace

common/tensorrt_common/src/tensorrt_common.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ void TrtCommon::setup()
188188
logger_.log(nvinfer1::ILogger::Severity::kINFO, "Start build engine");
189189
auto log_thread = logger_.log_throttle(
190190
nvinfer1::ILogger::Severity::kINFO,
191-
"Building engine from onnx. Please wait for a few minutes...", 5);
191+
"Applying optimizations and building TRT CUDA engine. Please wait for a few minutes...", 5);
192192
buildEngineFromOnnx(model_file_path_, cache_engine_path);
193193
logger_.stop_throttle(log_thread);
194194
logger_.log(nvinfer1::ILogger::Severity::kINFO, "End build engine");

0 commit comments

Comments
 (0)