context.h
context.h
概述
提供了Context相关的接口,可以配置运行时信息。
起始版本:
9
相关模块:
汇总
类型定义
类型定义名称 | 描述 |
---|---|
OH_AI_ContextHandle | Mindspore的上下文信息的指针,该指针会指向Context。 |
OH_AI_DeviceInfoHandle | Mindspore的运行设备信息的指针。 |
函数
函数名称 | 描述 |
---|---|
OH_AI_ContextCreate () | 创建一个上下文的对象。 |
OH_AI_ContextDestroy (OH_AI_ContextHandle *context) | 释放上下文对象。 |
OH_AI_ContextSetThreadNum (OH_AI_ContextHandle context, int32_t thread_num) | 设置运行时的线程数量。 |
OH_AI_ContextGetThreadNum (const OH_AI_ContextHandle context) | 获取线程数量。 |
OH_AI_ContextSetThreadAffinityMode (OH_AI_ContextHandle context, int mode) | 设置运行时线程绑定CPU核心的策略,按照CPU物理核频率分为大、中、小三种类型的核心,并且仅需绑大核或者绑中核,不需要绑小核。 |
OH_AI_ContextGetThreadAffinityMode (const OH_AI_ContextHandle context) | 获取运行时线程绑定CPU核心的策略。 |
OH_AI_ContextSetThreadAffinityCoreList (OH_AI_ContextHandle context, const int32_t *core_list, size_t core_num) | 设置运行时线程绑定CPU核心的列表。 |
OH_AI_ContextGetThreadAffinityCoreList (const OH_AI_ContextHandle context, size_t *core_num) | 获取CPU绑核列表。 |
OH_AI_ContextSetEnableParallel (OH_AI_ContextHandle context, bool is_parallel) | 设置运行时是否支持并行。 |
OH_AI_ContextGetEnableParallel (const OH_AI_ContextHandle context) | 获取是否支持算子间并行。 |
OH_AI_ContextAddDeviceInfo (OH_AI_ContextHandle context, OH_AI_DeviceInfoHandle device_info) | 添加运行设备信息。 |
OH_AI_DeviceInfoCreate (OH_AI_DeviceType device_type) | 创建一个设备信息对象。 |
OH_AI_DeviceInfoDestroy (OH_AI_DeviceInfoHandle *device_info) | 释放设备信息实例。 |
OH_AI_DeviceInfoSetProvider (OH_AI_DeviceInfoHandle device_info, const char *provider) | 设置供应商的名称。 |
OH_AI_DeviceInfoGetProvider (const OH_AI_DeviceInfoHandle device_info) | 获取生产商的名称。 |
OH_AI_DeviceInfoSetProviderDevice (OH_AI_DeviceInfoHandle device_info, const char *device) | 设置生产商设备的名称。 |
OH_AI_DeviceInfoGetProviderDevice (const OH_AI_DeviceInfoHandle device_info) | 获取生产商设备的名称。 |
OH_AI_DeviceInfoGetDeviceType (const OH_AI_DeviceInfoHandle device_info) | 获取设备的类型。 |
OH_AI_DeviceInfoSetEnableFP16 (OH_AI_DeviceInfoHandle device_info, bool is_fp16) | 设置是否开启float16推理模式,仅CPU/GPU设备可用。 |
OH_AI_DeviceInfoGetEnableFP16 (const OH_AI_DeviceInfoHandle device_info) | 获取是否开启float16推理模式, 仅CPU/GPU设备可用。 |
OH_AI_DeviceInfoSetFrequency (OH_AI_DeviceInfoHandle device_info, int frequency) | 设置NPU的频率,仅NPU设备可用。 |
OH_AI_DeviceInfoGetFrequency (const OH_AI_DeviceInfoHandle device_info) | 获取NPU的频率类型,仅NPU设备可用。 |