TIK 1.5自定义算子开发指南(Beta)-TIK类构造函数:调用示例

时间:2023-11-01 16:13:42

调用示例

  • debug调试功能使能示例:

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

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

    from tbe import tikfrom tbe.common.platform import set_current_compile_soc_infosoc_version="Ascend910"set_current_compile_soc_info(soc_version)tik_instance = tik.Tik() 或 tik_instance = tik.Tik(disable_debug=False)
  • debug调试功能禁用示例:
    from tbe import tikfrom tbe.common.platform import set_current_compile_soc_infotik_instance = tik.Tik(disable_debug=True)
  • 设置err_msg_level为开发级别示例:
    from tbe import tikfrom tbe.common.platform import set_current_compile_soc_infotik_instance = tik.Tik(err_msg_level=1) 
support.huaweicloud.com/TIKopdevgd_beta/tik1.5_10_0039.html