函数:get_mem_info

C函数原型

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

Python函数

free, total, ret = acl.rt.get_mem_info(attr)

函数功能

获取指定属性的内存的空闲大小和总大小。同步接口。

请根据实际硬件支持的情况,选择对应属性的内存,否则调用本接口获取到的空闲大小和总大小都为0。

输入说明

attr:int,

  • 0:ACL_DDR_MEM, DDR内存,DDR上所有大页内存+普通内存。
  • 1:ACL_HBM_MEM, HBM内存,HBM上所有大页内存+普通内存。
  • 2:ACL_DDR_MEM_HUGE,DDR大页内存。
  • 3:ACL_DDR_MEM_NORMAL,DDR普通内存。
  • 4:ACL_HBM_MEM_HUGE,HBM大页内存。
  • 5:ACL_HBM_MEM_NORMAL,HBM普通内存。
  • 6:ACL_DDR_MEM_P2P_HUGE,DDR中用于Device间数据复制的大页内存。预留选项。
  • 7:ACL_DDR_MEM_P2P_NORMAL,DDR中用于Device间数据复制的普通内存。预留选项。
  • 8:ACL_HBM_MEM_P2P_HUGE,HBM中用于Device间数据复制的大页内存。预留选项。
  • 9:ACL_HBM_MEM_P2P_NORMAL,HBM中用于Device间数据复制的普通内存。预留选项。

返回值说明

free:int,对应属性的内存的空闲大小,单位Byte。

total:int,对应属性的内存的总大小,单位Byte。

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

调用本接口前是必须先指定用于计算的Device,因此本接口中不体现Device ID。

注意事项