CANN/driver获取能力组信息API

dcmi_get_capability_group_info【免费下载链接】driver本项目是CANN提供的驱动模块实现基础驱动和资源管理及调度等功能使能昇腾芯片。项目地址: https://gitcode.com/cann/driver函数原型int dcmi_get_capability_group_info(int card_id, int device_id, int ts_id, int group_id, struct dcmi_capability_group_info *group_info, int group_count)功能说明获取昇腾虚拟化实例配置信息。参数说明参数名称输入/输出类型描述card_id输入int设备ID当前实际支持的ID通过dcmi_get_card_list接口获取。device_id输入int芯片ID通过dcmi_get_device_id_in_card接口获取。取值范围如下NPU芯片[0, device_id_max-1]。说明device_id_max值为1当device_id为0时表示NPU芯片当device_id为1时表示MCU芯片。ts_id输入int算力类型IDtypedef enum {DCMI_TS_AICORE 0,DCMI_TS_AIVECTOR,} DCMI_TS_ID;目前不支持DCMI_TS_AIVECTORgroup_id输入int算力组ID范围[0, 3]或者-1。当group_id为-1时获取所有的group信息。group_info输出struct dcmi_capability_group_info*算力组信息。算力信息结构体为#define DCMI_COMPUTE_GROUP_INFO_RES_NUM#define DCMI_AICORE_MASK_NUM 2struct dcmi_capability_group_info {unsigned int group_id;unsigned int state;unsigned int extend_attribute;unsigned int aicore_number;unsigned int aivector_number;unsigned int sdma_number;unsigned int aicpu_number;unsigned int active_sq_number;unsigned int aicore_mask [DCMI_AICORE_MASK_NUM];unsigned int res[DCMI_COMPUTE_GROUP_INFO_RES_NUM - DCMI_AICORE_MASK_NUM];};group_id表示算力组ID全局唯一。可配置范围0~3。state表示算力组创建状态。0: not created, 1: created。extend_attribute表示默认昇腾虚拟化实例标志1系统默认使用该昇腾虚拟化实例其他值表示不使用该昇腾虚拟化实例为默认组缺省值0。aicore_number表示算力组的aicore数量。aivector_number、sdma_number、aicpu_number、active_sq_number分别表示aivector、sdma、aicpu、active资源数量目前会返回255表示当前资源为共享资源不支持切分。aicore_mask表示aicore的编号掩码其中每个bit表示一个core。1该算力组内aicore0非算力组内aicore。res表示预留参数。group_count输入int参数group_info的数组长度。在group_id为-1时group_count需要不小于4。返回值说明类型描述int处理结果成功返回0。失败返回码请参见return_codes。异常处理无。约束说明仅Control CPU开放形态使用该接口。表 1不同部署场景下的支持情况产品形态物理机场景裸机root用户物理机场景裸机运行用户组非root用户物理机普通容器场景root用户Atlas 900 A2 PoD 集群基础单元NNNAtlas 800T A2 训练服务器NNNAtlas 800I A2 推理服务器NNNAtlas 200T A2 Box16 异构子框NNNA200I A2 Box 异构组件NNNAtlas 300I A2 推理卡NNNAtlas 300T A2 训练卡NNN注Y表示支持N表示不支持NA表示不涉及当前未规划此场景。调用示例… int ret 0; int card_id 0; int dev_id 0; int ts_id 0; struct dcmi_capability_group_info group_info[4] {0}; ret dcmi_get_capability_group_info(card_id, dev_id, ts_id, -1, group_info, 4); if (ret ! 0){ //todo记录日志 return ret; } …【免费下载链接】driver本项目是CANN提供的驱动模块实现基础驱动和资源管理及调度等功能使能昇腾芯片。项目地址: https://gitcode.com/cann/driver创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考