昇腾社区首页
中文
注册
开发者
下载

aclrtGetMemUsageInfo

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

功能说明

查询组件的内存使用信息,包括组件名称、当前内存大小和峰值内存大小等信息。

函数原型

aclError aclrtGetMemUsageInfo(int32_t deviceId, aclrtMemUsageInfo *memUsageInfo, size_t inputNum, size_t *outputNum)

参数说明

参数名

输入/输出

说明

deviceId

输入

Device ID。

用户调用aclrtGetDeviceCount接口获取可用的Device数量后,这个Device ID的取值范围:[0, (可用的Device数量-1)]

memUsageInfo

输入&输出

内存使用信息数组。

该参数作为输入时,由用户传入aclrtMemUsageInfo结构体指针,其内存大小需确保足以存放inputNum个组件的内存使用信息。

该参数作为输出时,可以获取组件名称、当前内存大小和峰值内存大小等信息。

inputNum

输入

指定需查询的组件数量。

如果实际组件数量少于inputNum,则按实际组件数量查询。

outputNum

输出

实际查询到的组件数量。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError