fixed paddle inference cpu_num_threads setting (PaddlePaddle#64)

felixhjh · Aug 2, 2022 · 59c374e · 59c374e
1 parent 90e0f53
commit 59c374e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/csrcs/fastdeploy/fastdeploy_runtime.cc b/csrcs/fastdeploy/fastdeploy_runtime.cc
@@ -212,7 +212,6 @@ void RuntimeOption::SetTrtCacheFile(const std::string& cache_file_path) {
   trt_serialize_file = cache_file_path;
 }
 
-
 bool Runtime::Init(const RuntimeOption& _option) {
   option = _option;
   if (option.model_format == Frontend::AUTOREC) {
@@ -274,6 +273,7 @@ void Runtime::CreatePaddleBackend() {
   pd_option.mkldnn_cache_size = option.pd_mkldnn_cache_size;
   pd_option.use_gpu = (option.device == Device::GPU) ? true : false;
   pd_option.gpu_id = option.device_id;
+  pd_option.cpu_thread_num = option.cpu_thread_num;
   FDASSERT(option.model_format == Frontend::PADDLE,
            "PaddleBackend only support model format of Frontend::PADDLE.");
   backend_ = new PaddleBackend();

diff --git a/examples/resources/.gitignore b/examples/resources/.gitignore
@@ -6,6 +6,10 @@ models/*.pd*
 models/*.engine
 models/*.trt
 models/*.nb
+models/*param*
+models/*model*
 outputs/*.jpg
 outputs/*.jpeg
-outputs/*.png
+outputs/*.png
+outputs/*.txt
+outputs/*.json