718c41634f
1.项目后端整体迁移至PaddleOCR-NCNN算法，已通过基本的兼容性测试 2.工程改为使用CMake组织，后续为了更好地兼容第三方库，不再提供QMake工程 3.重整权利声明文件，重整代码工程，确保最小化侵权风险 Log: 切换后端至PaddleOCR-NCNN，切换工程为CMake Change-Id: I4d5d2c5d37505a4a24b389b1a4c5d12f17bfa38c
131 lines
2.2 KiB
C++
131 lines
2.2 KiB
C++
#include <stdio.h>
|
|
|
|
#include "cpu.h"
|
|
|
|
#if defined __ANDROID__ || defined __linux__ || defined __APPLE__
|
|
|
|
static int test_cpu_set()
|
|
{
|
|
ncnn::CpuSet set;
|
|
|
|
if (set.num_enabled() != 0)
|
|
{
|
|
fprintf(stderr, "By default all cpus should be disabled\n");
|
|
return 1;
|
|
}
|
|
|
|
set.enable(0);
|
|
if (!set.is_enabled(0))
|
|
{
|
|
fprintf(stderr, "CpuSet enable doesn't work\n");
|
|
return 1;
|
|
}
|
|
|
|
if (set.num_enabled() != 1)
|
|
{
|
|
fprintf(stderr, "Only one cpu should be enabled\n");
|
|
return 1;
|
|
}
|
|
|
|
set.disable(0);
|
|
if (set.is_enabled(0))
|
|
{
|
|
fprintf(stderr, "CpuSet disable doesn't work\n");
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#else
|
|
|
|
static int test_cpu_set()
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
#endif
|
|
|
|
#if defined __ANDROID__ || defined __linux__
|
|
|
|
static int test_cpu_info()
|
|
{
|
|
if (ncnn::get_cpu_count() >= 0 && ncnn::get_little_cpu_count() >= 0 && ncnn::get_big_cpu_count() >= 0)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "The system cannot have a negative number of processors\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
static int test_cpu_omp()
|
|
{
|
|
if (ncnn::get_omp_num_threads() >= 0 && ncnn::get_omp_thread_num() >= 0 && ncnn::get_omp_dynamic() >= 0)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "The OMP cannot have a negative number of processors\n");
|
|
return 1;
|
|
}
|
|
|
|
ncnn::set_omp_num_threads(1);
|
|
|
|
ncnn::set_omp_dynamic(1);
|
|
}
|
|
|
|
static int test_cpu_powersave()
|
|
{
|
|
if (ncnn::get_cpu_powersave() >= 0)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "By default powersave must be zero\n");
|
|
return 1;
|
|
}
|
|
|
|
if (ncnn::set_cpu_powersave(-1) == -1 && ncnn::set_cpu_powersave(3) == -1)
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
fprintf(stderr, "Set cpu powersave for `-1 < argument < 2` works incorrectly.\n");
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
#else
|
|
|
|
static int test_cpu_info()
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static int test_cpu_omp()
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static int test_cpu_powersave()
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
#endif
|
|
|
|
int main()
|
|
{
|
|
return 0
|
|
|| test_cpu_set()
|
|
|| test_cpu_info()
|
|
|| test_cpu_omp()
|
|
|| test_cpu_powersave();
|
|
}
|