From c2e5f6317eae57f30db2627f93ae68b628338785 Mon Sep 17 00:00:00 2001 From: wwbitejotunn Date: Thu, 9 Feb 2023 05:51:30 +0000 Subject: [PATCH 1/2] fix paddle backend --- fastdeploy/runtime/backends/paddle/paddle_backend.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/fastdeploy/runtime/backends/paddle/paddle_backend.cc b/fastdeploy/runtime/backends/paddle/paddle_backend.cc index e0e908c36..19493f90b 100644 --- a/fastdeploy/runtime/backends/paddle/paddle_backend.cc +++ b/fastdeploy/runtime/backends/paddle/paddle_backend.cc @@ -44,6 +44,7 @@ void PaddleBackend::BuildOption(const PaddleBackendOption& option) { "file will save to the directory where paddle model saved." << std::endl; use_static = true; + config_.SetOptimCacheDir(option.trt_option.serialize_file); } config_.EnableTensorRtEngine(option.trt_option.max_workspace_size, option.trt_option.max_batch_size, 3, From 898b0632161e8384ba7178598dfb2225549d586b Mon Sep 17 00:00:00 2001 From: wwbitejotunn Date: Thu, 9 Feb 2023 12:55:05 +0000 Subject: [PATCH 2/2] get cache dir --- fastdeploy/runtime/backends/paddle/paddle_backend.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/fastdeploy/runtime/backends/paddle/paddle_backend.cc b/fastdeploy/runtime/backends/paddle/paddle_backend.cc index 19493f90b..4df109991 100644 --- a/fastdeploy/runtime/backends/paddle/paddle_backend.cc +++ b/fastdeploy/runtime/backends/paddle/paddle_backend.cc @@ -44,7 +44,10 @@ void PaddleBackend::BuildOption(const PaddleBackendOption& option) { "file will save to the directory where paddle model saved." << std::endl; use_static = true; - config_.SetOptimCacheDir(option.trt_option.serialize_file); + 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,