update vlog output (#49541)

bbc6dd94 · Yuanle Liu · GitHub · f17b2de8 · bbc6dd94
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

paddle/fluid/inference/tensorrt/engine.h paddle/fluid/inference/tensorrt/engine.h +3 -2

未找到文件。
--- 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};

    {