未验证 提交 541d702d 编写于 作者: P Pei Yang 提交者: GitHub

add trt verbose logs (#32459)

上级 1896c777
...@@ -67,9 +67,12 @@ class NaiveLogger : public nvinfer1::ILogger { ...@@ -67,9 +67,12 @@ class NaiveLogger : public nvinfer1::ILogger {
public: public:
void log(nvinfer1::ILogger::Severity severity, const char* msg) override { void log(nvinfer1::ILogger::Severity severity, const char* msg) override {
switch (severity) { switch (severity) {
case Severity::kINFO: case Severity::kVERBOSE:
VLOG(3) << msg; VLOG(3) << msg;
break; break;
case Severity::kINFO:
VLOG(2) << msg;
break;
case Severity::kWARNING: case Severity::kWARNING:
LOG(WARNING) << msg; LOG(WARNING) << msg;
break; break;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册