reset context clear profile_num (#49188)

d5366c47 · JingZhuangzhuang · GitHub · cdb6b746 · d5366c47
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/inference/tensorrt/engine.h paddle/fluid/inference/tensorrt/engine.h +1 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/engine.h
+++ b/paddle/fluid/inference/tensorrt/engine.h
@@ -326,6 +326,7 @@ class TensorRTEngine {
    std::unique_lock<std::mutex> lock(mutex_);
    infer_context_[predictor_id_per_thread].reset(nullptr);
    infer_context_.erase(predictor_id_per_thread);
+    cur_profile_num_ = 0;
  }

  nvinfer1::IHostMemory* Serialize() {