下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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表示成功,返回其他值表示失败。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词