apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Other] Add global SetProcLibCpuNumThreads method (#513)

Browse Source 
[FlyCV] Add global SetProcLibCpuNumThreads method
This commit is contained in:
DefTruth
2022-11-07 19:45:12 +08:00
committed by GitHub
parent 3589c0fa94
commit ff24ed6a54
4 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
fastdeploy/vision/common/processors/base.cc
View File

@@ -54,5 +54,12 @@ void DisableFlyCV() {
<< DefaultProcLib::default_lib << std::endl; << DefaultProcLib::default_lib << std::endl;
} }
void SetProcLibCpuNumThreads(int threads) {
cv::setNumThreads(threads);
#ifdef ENABLE_FLYCV
fcv::set_thread_num(threads);
#endif
}
} // namespace vision } // namespace vision
} // namespace fastdeploy } // namespace fastdeploy

5
fastdeploy/vision/common/processors/base.h
View File

@@ -31,6 +31,11 @@ FASTDEPLOY_DECL void EnableFlyCV();
/// Disable using FlyCV to process image while deploy vision models. /// Disable using FlyCV to process image while deploy vision models.
FASTDEPLOY_DECL void DisableFlyCV(); FASTDEPLOY_DECL void DisableFlyCV();
/*! @brief Set the cpu num threads of ProcLib. The cpu num threads
* of FlyCV and OpenCV is 2 by default.
*/
FASTDEPLOY_DECL void SetProcLibCpuNumThreads(int threads);
class FASTDEPLOY_DECL Processor { class FASTDEPLOY_DECL Processor {
public: public:
// default_lib has the highest priority // default_lib has the highest priority

2
java/android/fastdeploy/build.gradle
View File

@@ -46,7 +46,7 @@ dependencies {
def archives = [ def archives = [
[ [
'src' : 'https://bj.bcebos.com/fastdeploy/test/fastdeploy-android-0.5.0-shared-dev.tgz', 'src' : 'https://bj.bcebos.com/fastdeploy/test/fastdeploy-android-latest-shared-dev.tgz',
'dest': 'libs' 'dest': 'libs'
] ]
] ]

2
java/android/fastdeploy/src/main/cpp/CMakeLists.txt
View File

@@ -12,7 +12,7 @@ project("fastdeploy_jni")
# You can define multiple libraries, and CMake builds them for you. # You can define multiple libraries, and CMake builds them for you.
# Gradle automatically packages shared libraries with your APK. # Gradle automatically packages shared libraries with your APK.
set(FastDeploy_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../../../libs/fastdeploy-android-0.5.0-shared-dev") set(FastDeploy_DIR "${CMAKE_CURRENT_SOURCE_DIR}/../../../libs/fastdeploy-android-latest-shared-dev")
find_package(FastDeploy REQUIRED) find_package(FastDeploy REQUIRED)