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。
调用示例
本示例中，以昇腾AI处理器的型号为“Ascend910”为例，实际调用时，请将变量soc_version的值修改为实际的昇腾AI处理器型号。
import tbe
soc_version="Ascend910"
tbe.common.platform.set_current_compile_soc_info(soc_version)
tbe.common.platform.get_soc_spec("CORE_NUM")
父主题： TBE Platform API
