文档
注册

TIK类构造函数

功能说明

创建TIK DSL容器。

函数原型

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

参数说明

表1 参数说明

参数名称

输入/输出

含义

profiling

输入

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

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

disable_debug

输入

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

err_msg_level

输入

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

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

支持的芯片型号

昇腾310 AI处理器

昇腾910 AI处理器

昇腾310P AI处理器AI Core

昇腾310P AI处理器Vector Core

注意事项

  • 如果开启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处理器类型,则tik.Tik() 默认创建的是昇腾310AI处理器的容器。

    实际调用时,请将变量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) 
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词