diff --git a/paddle/fluid/inference/tensorrt/engine.h b/paddle/fluid/inference/tensorrt/engine.h index 908e3729d57f3cfdd8ede7e1342ffc4fec92e8e3..8a4afa325559d042a66c7ed25d85242841d91612 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}; {