昇腾社区首页
中文
注册

TIK类构造函数

功能说明

创建TIK DSL容器。

函数原型

__init__(profiling=None, disable_debug=False, err_msg_level=0)

参数说明

表1 参数说明

参数名称

输入/输出

含义

profiling

输入

昇腾AI处理器版本配置信息,支持Dprofile类型。

建议保持默认值None,昇腾AI处理器版本信息请通过set_current_compile_soc_info接口配置。

disable_debug

输入

收集调试信息禁用与使能的开关,布尔类型,默认为False(表示收集调试信息),该参数为可选参数。

err_msg_level

输入

报错信息打印级别,支持int类型,默认为0,该参数为可选参数。取值:

  • 0:表示用户级错误信息,包括错误提示语,报错文件路径及行号,报错对应代码及上下文,用户调用堆栈。
  • 1 : 表示开发级错误信息,包括错误提示语,报错文件路径及行号,报错对应代码及上下文,全部调用堆栈。

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

注意事项

  • 若在开发算子时对编译时长有严格限制,开发的过程中可以使用debug功能,最后提交代码则可以在构造TIK实例时将disable_debug参数手动置为True,禁用调试功能,可以相对减少一些编译时间。
  • 若将disable_debug参数置为True后,BuildCCE结束后若调用debug调试接口,程序会异常退出,调试失败。

返回值

输出TIK类的实例。

调用示例

  • debug调试功能使能示例:

    若不调用set_current_compile_soc_info接口设置昇腾AI处理器类型,则tik.Tik() 默认创建的是昇腾310AI处理器的容器。

    本示例中,以昇腾AI处理器的型号为“Ascend910”为例,实际调用时,请将变量soc_version的值修改为实际的昇腾AI处理器型号。

    from tbe import tik
    from tbe.common.platform import set_current_compile_soc_info
    soc_version="Ascend910"
    set_current_compile_soc_info(soc_version)
    tik_instance = tik.Tik() 
    或 
    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(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)