pypto.set_pass_config#
产品支持情况#
产品 |
是否支持 |
|---|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品 |
√ |
Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
√ |
功能说明#
修改指定Pass的配置信息。
函数原型#
set_pass_config(strategy: str, identifier: str, key: PassConfigKey, value: bool)
参数说明#
参数名 |
输入/输出 |
说明 |
|---|---|---|
strategy |
输入 |
Pass策略名称，如”PVC2_OOO” |
identifier |
输入 |
Pass名称，如”ExpandFunction” |
key |
输入 |
PassConfigKey枚举 |
value |
输入 |
配置值 |
返回值说明#
无
约束说明#
设置时机：必须在图编译开始前调用。
作用范围：配置信息是全局性的，会影响后续所有的编译过程。
调用示例#
pypto.set_pass_config("PVC2_OOO", "ExpandFunction", pypto.PassConfigKey.KEY_DUMP_GRAPH, True)