MSConfig

查看源文件

import com.mindspore.lite.config.MSConfig;

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

公有成员函数

init

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

初始化MSConfig。

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

初始化MSConfig,enable_float16默认为false。

public boolean init(int deviceType, int threadNum)

初始化MSConfig,cpuBindMode默认为CpuBindMode.MID_CPUenable_float16默认为false。

public boolean init(int deviceType)

初始化MSConfig,cpuBindMode默认为CpuBindMode.MID_CPUthreadNum默认为2enable_float16默认为false。

public boolean init()

初始化MSConfig,deviceType默认为DeviceType.DT_CPUcpuBindMode默认为CpuBindMode.MID_CPUthreadNum默认为2enable_float16默认为false。

  • 返回值

    初始化是否成功。

free

public void free()

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

getMSConfigPtr

public long getMSConfigPtr()

获取MSConfig指针。

  • 返回值

    MSConfig指针。

DeviceType

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。

CpuBindMode

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,不绑核。