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
