MSContext

import com.mindspore.config.MSContext;

MSContext类用于配置运行时的上下文配置。

公有成员函数

function
boolean init()
boolean init(int threadNum, int cpuBindMode)
boolean init(int threadNum, int cpuBindMode, boolean isEnableParallel)
boolean addDeviceInfo(int deviceType, boolean isEnableFloat16)
boolean addDeviceInfo(int deviceType, boolean isEnableFloat16, int npuFreq)
void free()
long getMSContextPtr()
DeviceType
CpuBindMode

public boolean init()

使用默认参数初始化MSContext，2线程，不绑核，不开启异构并行。

public boolean init(int threadNum, int cpuBindMode)

使用线程数和绑＆模式初始化MSContext。

参数
- threadNum: 线程数。
- cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.config.CpuBindMode中定义。
返回值

初始化是否成功。

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

初始化MSContext。

参数
- threadNum: 线程数。
- cpuBindMode: CPU绑定模式，cpuBindMode在com.mindspore.config.CpuBindMode中定义。
- isEnableParallel: 是否开启异构并行。
返回值

初始化是否成功。

boolean addDeviceInfo(int deviceType, boolean isEnableFloat16)

添加运行设备信息。

参数
- deviceType: 设备类型，deviceType在com.mindspore.config.DeviceType中定义。
- isEnableFloat16: 是否开启fp16。
返回值

设备添加是否成功。

boolean addDeviceInfo(int deviceType, boolean isEnableFloat16, int npuFreq)

添加运行设备信息。

参数
- deviceType: 设备类型，deviceType在com.mindspore.config.DeviceType中定义。
- isEnableFloat16: 是否开启fp16。
- npuFreq: NPU运行频率，仅当deviceType为npu才需要。
返回值

设备添加是否成功。

public long getMSContextPtr()

获取MSContext底层运行指针。

public void free()

释放MSContext运行过程中动态分配的内存。

import com.mindspore.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.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，不绑核。