设置CPU核

Tengine设置CPU核有两种方式:

·通过环境变量设置:

export TENGINE_CPU_LIST=0,1,2,…,N

指定一个CPU核执行,可设置: export TENGINE_CPU_LIST=0

指定两个CPU核执行,可设置: export TENGINE_CPU_LIST=0,1

指定四个CPU核执行,可设置:export TENGINE_CPU_LIST=0,1,2,3

比如RK3399的板子有2A72 + 4A53, A53 的CPU编号为0,1,2,3, A72的CPU编号为4,5

指定1A72:export TENGINE_CPU_LIST=4

指定2A72:export TENGINE_CPU_LIST=4,5

指定1A53:export TENGINE_CPU_LIST=0

指定4A53:export TENGINE_CPU_LIST=0,1,2,3


·在源码中通过API设置

 set_working_cpu(const int* cpu_list, int cpu_number)


注意: 应该在init_tengine()前调用set_working_cpu 函数

以下是通过API设置CPU的例子:

#include "cpu_device.h"


 int cpu_list_str[]={0,1,2,3};

 set_working_cpu(cpu_list_str,4);


 init_tengine();