add trt verbose logs (#32459)

541d702d · Pei Yang · GitHub · 1896c777 · 541d702d
显示空白变更内容
内联并排

Showing with 4 addition and 1 deletion

paddle/fluid/inference/tensorrt/helper.h paddle/fluid/inference/tensorrt/helper.h +4 -1

未找到文件。
--- a/paddle/fluid/inference/tensorrt/helper.h
+++ b/paddle/fluid/inference/tensorrt/helper.h
@@ -67,9 +67,12 @@ class NaiveLogger : public nvinfer1::ILogger {
 public:
  void log(nvinfer1::ILogger::Severity severity, const char* msg) override {
    switch (severity) {
-      case Severity::kINFO:
+      case Severity::kVERBOSE:
        VLOG(3) << msg;
        break;
+      case Severity::kINFO:
+        VLOG(2) << msg;
+        break;
      case Severity::kWARNING:
        LOG(WARNING) << msg;
        break;