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"：返回如"Ascend310"，"Ascend710"等标识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。

调用示例

本示例中，以昇腾AI处理器的型号为“Ascend910”为例，实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。

soc_version="Ascend910"
tbe.common.platform.set_current_compile_soc_info(soc_version)
tbe.common.platform.get_soc_spec("CORE_NUM")
父主题： TBE Platform Info API