MSConfig

import com.mindspore.lite.config.MSConfig;

MSConfig类用于保存执行中的配置变量。

公有成员函数

function
boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16)
boolean init(int deviceType, int threadNum, int cpuBindMode)
boolean init(int deviceType, int threadNum)
boolean init(int deviceType)
boolean init()
void free()
long getMSConfigPtr()
DeviceType
CpuBindMode

public boolean init(int deviceType, int threadNum, int cpuBindMode, boolean enable_float16)

初始化MSConfig。

参数
- deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
- threadNum: 线程数。
- cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.lite.config.CpuBindMode中定义。
- enable_float16：是否优先使用float16算子。
返回值

初始化是否成功。

public boolean init(int deviceType, int threadNum, int cpuBindMode)

初始化MSConfig，enable_float16默认为false。

参数
- deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
- threadNum: 线程数。
- cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.lite.config.CpuBindMode中定义。
返回值

初始化是否成功。

public boolean init(int deviceType, int threadNum)

初始化MSConfig，cpuBindMode默认为CpuBindMode.MID_CPU，enable_float16默认为false。

参数
- deviceType: 设备类型，deviceType在com.mindspore.lite.config.DeviceType中定义。
- threadNum: 线程数。
返回值

初始化是否成功。

public boolean init(int deviceType)

初始化MSConfig，cpuBindMode默认为CpuBindMode.MID_CPU，threadNum默认为2，enable_float16默认为false。

public boolean init()

初始化MSConfig，deviceType默认为DeviceType.DT_CPU，cpuBindMode默认为CpuBindMode.MID_CPU，threadNum默认为2，enable_float16默认为false。

public void free()

释放MSConfig运行过程中动态分配的内存。LiteSession init之后即可释放。

public long getMSConfigPtr()

获取MSConfig指针。

import com.mindspore.lite.config.DeviceType;

设备类型。

public static final int DT_CPU = 0;
public static final int DT_GPU = 1;
public static final int DT_NPU = 2;

DeviceType的值为0，指定设备类型为CPU。

DeviceType的值为1，指定设备类型为GPU。

DeviceType的值为2，指定设备类型为NPU。

import com.mindspore.lite.config.CpuBindMode;

绑核策略。

public static final int MID_CPU = 2;
public static final int HIGHER_CPU = 1;
public static final int NO_BIND = 0;

CpuBindMode的值为2，优先绑定中核。

CpuBindMode的值为1，优先绑定大核。

CpuBindMode的值为0，不绑核。