文档
注册
评分
提单
论坛
小AI

get_soc_spec

函数功能

此接口用于获取硬件信息,包括SOC_VERSION、AICORE_TYPE、CORE_NUM以及各存储单元的大小等。

函数原型

def get_soc_spec(key)

参数说明

参数名

类型

说明

key

string类型

获取硬件信息,包含

"SOC_VERSION"

"AICORE_TYPE"

"CORE_NUM"

"UB_SIZE"

"L2_SIZE"

"L1_SIZE"

"CUBE_SIZE"

"L0A_SIZE"

"L0B_SIZE"

"L0C_SIZE"

"SMASK_SIZE"

返回值

根据输入的key返回对应的值:
  • "SOC_VERSION":返回标识SOC类型的字符串。
  • "AICORE_TYPE":返回Core的类型,有"AiCore"或“VectorCore”两种返回值。
  • "CORE_NUM":返回核数,int类型。
  • "UB_SIZE":返回UB大小,int类型,单位Byte。
  • "L2_SIZE":返回L2大小,int类型,单位Byte。
  • "L1_SIZE":返回L1大小,int类型,单位Byte。
  • "CUBE_SIZE":返回CUBE大小,tuple类型,如(16,16,16),单位为Byte。
  • "L0A_SIZE":返回L0A大小,int类型,单位为Byte。
  • "L0B_SIZE":返回L0B大小,int类型,单位为Byte。
  • "L0C_SIZE":返回L0C大小,int类型,单位为Byte。
  • "SMASK_SIZE":返回Smask buffer大小,int类型,单位为Byte。

约束说明

使用此接口前,需要先执行set_current_compile_soc_info接口设置昇腾AI处理器的类型。若不设置,默认使用Ascend310。

调用示例

实际调用时,请将变量soc_version的值修改为实际的昇腾AI处理器型号。

import tbe
soc_version="xxx"
tbe.common.platform.set_current_compile_soc_info(soc_version)
tbe.common.platform.get_soc_spec("CORE_NUM")
搜索结果
找到“0”个结果

当前产品无相关内容

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