'%20fill='%23C31D20'%20fill-rule='nonzero'%3e%3cpath%20d='M7.55269822,13.7609409%20C10.8552199,9.97176134%2014.2271223,7.63452926%2017.1034483,6.5677512%20L17.0816429,0.884990191%20C17.0593112,0.508963963%2016.8188535,0.182653682%2016.4703158,0.055395956%20C16.1217781,-0.0718617705%2015.7322263,0.0244201241%2015.4799397,0.300178022%20L12.9584465,3.07097775%20L0.368821458,17.4916392%20C0.0232151224,17.8857306%20-0.0894678188,18.437454%200.0732192079,18.9389805%20C0.235906234,19.4405071%200.649247199,19.815643%201.15754056,19.9230783%20C1.66583392,20.0305135%202.19185783,19.853926%202.53746417,19.4598346%20L7.5467513,13.7690073%20L7.55269822,13.7609409%20Z'%20id='路径'%3e%3c/path%3e%3cpath%20d='M10.6896552,12.0927463%20L15.1004089,17.5535201%20C15.4511303,17.8402432%2015.9409156,17.8973222%2016.3502521,17.6991755%20C16.7595885,17.5010287%2017.0117165,17.0848129%2016.993637,16.6370685%20L17.1034483,10.0202479%20C14.5615201,9.75868982%2012.477139,10.7130773%2010.6896552,12.0927463%20Z'%20id='路径'%3e%3c/path%3e%3c/g%3e%3crect%20id='矩形'%20x='0'%20y='0'%20width='24'%20height='24'%3e%3c/rect%3e%3c/g%3e%3c/g%3e%3c/svg%3e)
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
