From bbc6dd944f9df90bd912f88180b1bc5cb87d35cf Mon Sep 17 00:00:00 2001 From: Yuanle Liu Date: Wed, 4 Jan 2023 20:39:44 +0800 Subject: [PATCH] update vlog output (#49541) --- paddle/fluid/inference/tensorrt/engine.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/engine.h b/paddle/fluid/inference/tensorrt/engine.h index 908e3729d5..8a4afa3255 100644 --- a/paddle/fluid/inference/tensorrt/engine.h +++ b/paddle/fluid/inference/tensorrt/engine.h @@ -802,8 +802,9 @@ class TRTEngineManager { } void updateContextMemorySize(size_t mem_size, PredictorID predictor_id) { - VLOG(3) << "TensorRT engine context memory size is " << mem_size - << " in predictor id " << predictor_id; + VLOG(3) << "TensorRT engine context memory size is " + << mem_size / 1024.0 / 1024.0 << "MiB in predictor id " + << predictor_id; bool size_updated{false}; { -- GitLab