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处理器型号。昇腾AI处理器的型号可通过如下方式查看：单击如下手册中的链接并进入该手册《ToolBox 用户指南》。完成“使用前准备”，然后进入“使用工具>设备实时状态查询”章节。使用相关命令查看芯片的详细信息，例如使用ascend-dmi -i -dt命令查看芯片的详细信息，返回信息中“Chip Name”对应取值（将大写字母转换为小写，并去掉中间空格）即为具体使用的${soc_version}。
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