Browse Source

fix(mgb): print verbose logs as debug log in TensorRT logger

GitOrigin-RevId: e197d565cb
tags/v0.3.2
Megvii Engine Team 5 years ago
parent
commit
a7e28712b5
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      src/tensorrt/impl/tensorrt_opr.cpp

+ 4
- 1
src/tensorrt/impl/tensorrt_opr.cpp View File

@@ -83,8 +83,11 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity,
case Severity::kINFO:
mgb_log_debug("TRT_INFO: %s", msg);
return;
case Severity::kVERBOSE:
mgb_log_debug("TRT_VERBOSE: %s", msg);
return;
default:
mgb_log("TRT_UNKNOWN: %s", msg);
mgb_log_debug("TRT_UNKNOWN: %s", msg);
return;
}
}


Loading…
Cancel
Save