aclrtGetMemInfo

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	√

功能说明

根据指定属性，获取Device上AscendCL应用可用内存的空闲大小和总大小，不包括系统预留内存大小。

函数原型

aclError aclrtGetMemInfo(aclrtMemAttr attr, size_t *free, size_t *total)

参数说明

参数名	输入/输出	说明
attr	输入	需要查询的内存的属性值。
free	输出	对应属性内存空闲大小的指针，单位Byte。
total	输出	对应属性内存总大小的指针，单位Byte。

返回值说明

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

约束说明

调用本接口前是必须先指定用于计算的Device（例如调用aclrtSetDevice接口指定用于计算的Device），因此本接口中不体现Device ID。
请根据实际硬件支持的情况，选择对应属性的内存，否则调用本接口获取到的空闲大小和总大小都为0。
该约束适用以下型号：

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 200/300/500 推理产品
由于Device上没有HBM相关的内存，若指定查询HBM相关内存，则AscendCL系统内部自动转换为查询DDR相关内存，例如，若查询ACL_HBM_MEM，则AscendCL系统内部实际查询ACL_DDR_MEM；若查询ACL_HBM_MEM_HUGE，则AscendCL系统内部实际查询ACL_DDR_MEM_HUGE。
该约束适用于以下型号：

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

父主题： 内存管理