news 2026/5/10 1:21:38

CANN/driver创建算力组API

张小明

前端开发工程师

1.2k 24

dcmi_create_capability_group

【免费下载链接】driver本项目是CANN提供的驱动模块，实现基础驱动和资源管理及调度等功能，使能昇腾芯片。项目地址: https://gitcode.com/cann/driver

函数原型

int dcmi_create_capability_group(int card_id, int device_id, int ts_id, struct dcmi_capability_group_info *group_info)

功能说明

创建昇腾虚拟化实例配置信息。

参数说明

参数名称	输入/输出	类型	描述
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	算力类型ID： typedef enum { DCMI_TS_AICORE = 0, DCMI_TS_AIVECTOR, } DCMI_TS_ID; 目前不支持DCMI_TS_AIVECTOR
group_info	输入	struct dcmi_capability_group_info *	算力组信息。算力信息结构体为： #define DCMI_COMPUTE_GROUP_INFO_RES_NUM - AICORE_MASK_NUM 8 #define DCMI_AICORE_MASK_NUM 2 struct 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可配置范围：0~8或255，具体参照如下说明。 aivector_number、sdma_number、aicpu_number、active_sq_number仅支持配置为255，表示配置为剩余可用值，不支持切分。 aicore_mask表示aicore的编号掩码，其中每个bit表示一个core。1：该算力组内aicore，0：非算力组内aicore。该字段作为输入时，无意义。 res表示预留参数。说明： group_info参数的约束说明： group_id取值范围为0~3，且每次创建时需要使用未使用过的group_id，否则将会返回失败。当创建不同分组时，各参数变量需要满足以下条件： ∑ g r o u p _ i d a i c o r e _ n u m b e r g r o u p _ i d < = t o t a l _ a i c o r e _ n u m b e r total_aicore_number为8。当aicore_number配置为255时，表示配置为剩余可用值，如果多个group配置255，这几个group共用剩余可用值。

返回值说明

类型	描述
int	处理结果：成功：返回0。失败：返回码请参见return_codes。

异常处理

无。

约束说明

仅Control CPU开放形态使用该接口。
调用该接口前，请先调用dcmi_get_capability_group_info查询已配置的资源信息，结合ts_id参数的说明和group_info参数的约束说明进行配置，否则可能出现接口调用失败。如果需要删除已有配置，可以调用dcmi_delete_capability_group接口进行删除。

表 1不同部署场景下的支持情况

产品形态	物理机场景（裸机）root用户	物理机场景（裸机）运行用户组（非root用户）	物理机+普通容器场景root用户
Atlas 900 A2 PoD 集群基础单元	N	N	N
Atlas 800T A2 训练服务器	N	N	N
Atlas 800I A2 推理服务器	N	N	N
Atlas 200T A2 Box16 异构子框	N	N	N
A200I A2 Box 异构组件	N	N	N
Atlas 300I A2 推理卡	N	N	N
Atlas 300T A2 训练卡	N	N	N
注：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 = {0}; group_info.group_id = 0; group_info.aivector_number = 255; group_info.sdma_number = 255; group_info.aicpu_number = 255; group_info.active_sq_number = 255; group_info.aicore_number = 8; ret = dcmi_create_capability_group(card_id, dev_id, ts_id, &group_info); if (ret != 0){ //todo：记录日志 return ret; } …

【免费下载链接】driver本项目是CANN提供的驱动模块，实现基础驱动和资源管理及调度等功能，使能昇腾芯片。项目地址: https://gitcode.com/cann/driver

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

网站建设 2026/5/10 1:19:50

CANN/catlass EVG API文档

EVG API 【免费下载链接】catlass 本项目是CANN的算子模板库，提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlass EVG（Epilogue Visitor Graph）是 CATLASS 当前用于组织 GEMM 尾处理的图…