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")
父主题: TBE Platform API