set_current_compile_soc_info
函数功能
此接口用于设置相应的昇腾AI处理器版本。
函数原型
def set_current_compile_soc_info(soc_version, core_type="AiCore", aicore_num=None, l1_fusion=None)
参数说明
参数名 |
类型 |
说明 |
---|---|---|
soc_version |
string类型,必选 |
昇腾AI处理器型号。 |
core_type |
string类型,可选 |
核的类型,取值范围:"AiCore"、 "VectorCore"。 不填时默认为"AiCore"。 |
aicore_num |
int类型,可选 |
配置要使用的核数,取值范围[1~实际核数],不填时默认为实际核数。 |
l1_fusion |
bool类型,可选 |
是否做l1 fusion,取值范围:True, False。 若不设置,默认使用False。 |
返回值
设置成功时返回字符串"success",否则返回"error"。
约束说明
无
调用示例
实际调用时,请将变量soc_version的值修改为实际的昇腾AI处理器型号。
import tbe soc_version="xxx" tbe.common.platform.set_current_compile_soc_info(soc_version)
父主题: TBE Platform API