提交 803bb79f 编写于 作者: M Megvii Engine Team

fix(trt/log): make trt error log visible

also, if can export RUNTIME_OVERRIDE_LOG_LEVEL=0
before run to get more debug log

GitOrigin-RevId: 15106727522907ba81d5b7996b09b5284bcf48d7
上级 97ddff4d
...@@ -51,13 +51,13 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity, const char* ...@@ -51,13 +51,13 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity, const char*
TENSORRT_NO_EXCEPT(noexcept) { TENSORRT_NO_EXCEPT(noexcept) {
switch (severity) { switch (severity) {
case Severity::kINTERNAL_ERROR: case Severity::kINTERNAL_ERROR:
mgb_log("TRT_INTERNAL_ERROR: %s", msg); mgb_log_error("TRT_INTERNAL_ERROR: %s", msg);
return; return;
case Severity::kERROR: case Severity::kERROR:
mgb_log("TRT_ERROR: %s", msg); mgb_log_error("TRT_ERROR: %s", msg);
return; return;
case Severity::kWARNING: case Severity::kWARNING:
mgb_log("TRT_WARNING: %s", msg); mgb_log_warn("TRT_WARNING: %s", msg);
return; return;
case Severity::kINFO: case Severity::kINFO:
mgb_log_debug("TRT_INFO: %s", msg); mgb_log_debug("TRT_INFO: %s", msg);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册