CANN/driver:获取设备内存信息API
2026/5/9 14:13:29 网站建设 项目流程

dcmi_get_device_memory_info_v2

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

函数原型

int dcmi_get_device_memory_info_v2(int card_id, int device_id, struct dcmi_memory_info *memory_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]。

memory_info

输出

struct dcmi_memory_info*

返回内存信息,内存信息结构体:

struct dcmi_memory_info {

unsigned long long memory_size; //内存大小,单位MB

unsigned int freq; //频率MHz

unsigned int utiliza; //利用率

};

返回值说明

类型

描述

int

处理结果:

  • 成功:返回0。
  • 失败:返回码请参见return_codes。

异常处理

无。

约束说明

该接口在后续版本将会删除,推荐使用dcmi_get_device_hbm_info。

表 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 device_id = 0; struct dcmi_memory_info memory_info = {0}; ret = dcmi_get_device_memory_info_v2(card_id, device_id, &memory_info); if (ret != 0){ //todo:记录日志 return ret; } …

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

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询