From 541d702dda137108dcf76278cdece61fce703288 Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Sun, 25 Apr 2021 15:16:54 +0800 Subject: [PATCH] add trt verbose logs (#32459) --- paddle/fluid/inference/tensorrt/helper.h | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/inference/tensorrt/helper.h b/paddle/fluid/inference/tensorrt/helper.h index 971f99e6919..d309eb81622 100644 --- 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; -- GitLab