pypto.set_pass_default_config#

产品支持情况#

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明#

修改Pass的默认配置信息。其主要功能动态修改Pass的运行时行为配置,目前可配置dump计算图开关,方便分析和调试。

函数原型#

set_pass_default_config(key: PassConfigKey, value: bool)

参数说明#

参数名

输入/输出

说明

key

输入

PassConfigKey枚举
KEY_DUMP_GRAPH Dump Pass计算图

value

输入

配置值

返回值说明#

约束说明#

  • 设置时机:必须在图编译开始前调用。

  • 作用范围:配置信息是全局性的,会影响后续所有的编译过程。

调用示例#

pypto.set_pass_default_config(pypto.PassConfigKey.KEY_DUMP_GRAPH, True)