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

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

GitOrigin-RevId: e197d565cbdf37d15a883329d59ade52319c338f
上级 e6d7c5e9
......@@ -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;
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册