下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

TIK类构造函数

功能说明

创建TIK DSL容器。

函数原型

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

参数说明

表1 参数说明

参数名称

输入/输出

含义

profiling

输入

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

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

disable_debug

输入

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

err_msg_level

输入

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

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

支持的型号

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) 
搜索结果
找到“0”个结果

当前产品无相关内容

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