Merge pull request PaddlePaddle#1283 from wwbitejotunn/ppinfer_trt_bu…

…gfix [Bug Fix] Paddle backend bug when tensorrt enable
lishicheng1996 · Feb 9, 2023 · 1476f15 · 1476f15
2 parents 4742f97 + 898b063
commit 1476f15
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fastdeploy/runtime/backends/paddle/paddle_backend.cc b/fastdeploy/runtime/backends/paddle/paddle_backend.cc
@@ -44,6 +44,10 @@ void PaddleBackend::BuildOption(const PaddleBackendOption& option) {
                "file will save to the directory where paddle model saved."
             << std::endl;
         use_static = true;
+        std::string opt_cache_dir =
+            GetDirFromPath(option.trt_option.serialize_file);
+
+        config_.SetOptimCacheDir(opt_cache_dir);
       }
       config_.EnableTensorRtEngine(option.trt_option.max_workspace_size,
                                    option.trt_option.max_batch_size, 3,