aclrtGetMemInfo
函数功能
根据指定属性,获取Device上AscendCL应用可用内存的空闲大小和总大小,不包括系统预留内存大小。
约束说明
- 调用本接口前是必须先指定用于计算的Device(例如调用aclrtSetDevice接口指定用于计算的Device),因此本接口中不体现Device ID。
- 请根据实际硬件支持的情况,选择对应属性的内存,否则调用本接口获取到的空闲大小和总大小都为0。
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
- 由于Device上没有HBM相关的内存,若指定查询HBM相关内存,则AscendCL系统内部自动转换为查询DDR相关内存,例如,若查询ACL_HBM_MEM,则AscendCL系统内部实际查询ACL_DDR_MEM;若查询ACL_HBM_MEM_HUGE,则AscendCL系统内部实际查询ACL_DDR_MEM_HUGE。
Atlas 200/500 A2推理产品
Atlas 推理系列产品
函数原型
aclError aclrtGetMemInfo(aclrtMemAttr attr, size_t *free, size_t *total)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
attr |
输入 |
需要查询的内存的属性值。 |
free |
输出 |
对应属性内存空闲大小的指针,单位Byte。 |
total |
输出 |
对应属性内存总大小的指针,单位Byte。 |
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: 内存管理