TIK类构造函数
功能说明
创建TIK DSL容器。
函数原型
__init__(profiling=None, disable_debug=True, err_msg_level=0, block_size=32)
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
profiling |
输入 |
昇腾AI处理器版本配置信息,支持Dprofile类型。 建议保持默认值None,昇腾AI处理器版本信息请通过set_current_compile_soc_info接口配置。 |
disable_debug |
输入 |
收集调试信息禁用与使能的开关,布尔类型,默认为True(表示不收集调试信息,无法进行debug),该参数为可选参数。 |
err_msg_level |
输入 |
报错信息打印级别,支持int类型,默认为0,该参数为可选参数。取值:
|
支持的型号
Atlas 200/300/500 推理产品
Atlas 训练系列产品
Atlas推理系列产品AI Core
Atlas推理系列产品Vector Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
注意事项
- 如果开启debug,即disable_debug=False,则pass优化等级会相对降低,影响算子性能。
- 若在开发算子时对编译时长有严格限制,开发的过程中可以将disable_debug参数置为False,以使用debug功能;最后提交代码则可以在构造TIK实例时删除disable_debug参数或将disable_debug参数手动置为True,禁用调试功能,可以相对减少一些编译时间。
- 若不加disable_debug参数或将disable_debug参数置为True后,BuildCCE结束后若调用debug调试接口,程序会异常退出,调试失败。
返回值
输出TIK类的实例。
调用示例
- debug调试功能使能示例:
若不调用set_current_compile_soc_info接口设置昇腾AI处理器类型,则默认值为“Ascend310”,tik.Tik() 默认创建的是该类型的容器。
实际调用时,请将变量soc_version的值修改为实际的昇腾AI处理器型号。
from tbe import tik from tbe.common.platform import set_current_compile_soc_info soc_version="xxx" set_current_compile_soc_info(soc_version) tik_instance = tik.Tik(disable_debug=False)
- debug调试功能禁用示例:
from tbe import tik from tbe.common.platform import set_current_compile_soc_info tik_instance = tik.Tik() 或 tik_instance = tik.Tik(disable_debug=True)
- 设置err_msg_level为开发级别示例:
from tbe import tik from tbe.common.platform import set_current_compile_soc_info tik_instance = tik.Tik(err_msg_level=1)
父主题: TIK容器管理