From a7e28712b5e2b67d609c2dce7679901df4cbe529 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 24 Mar 2020 16:49:53 +0800
Subject: [PATCH] fix(mgb): print verbose logs as debug log in TensorRT logger

GitOrigin-RevId: e197d565cbdf37d15a883329d59ade52319c338f
---
 src/tensorrt/impl/tensorrt_opr.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/tensorrt/impl/tensorrt_opr.cpp b/src/tensorrt/impl/tensorrt_opr.cpp
index 75ddac28..009b36aa 100644
--- a/src/tensorrt/impl/tensorrt_opr.cpp
+++ b/src/tensorrt/impl/tensorrt_opr.cpp
@@ -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;
     }
 }
-- 
GitLab