IREE项目中插件管道参数传递的正确使用方式

在IREE编译器工具链的使用过程中，开发者经常会遇到需要自定义优化管道(pipeline)参数的情况。本文将通过一个实际案例，深入分析如何在iree-opt工具中正确传递管道参数，帮助开发者避免常见的配置错误。

问题背景

在IREE编译流程中，torch-to-iree转换管道提供了多个可配置选项，如iree-torch-decompose-complex-ops用于控制是否分解复杂操作。开发者最初尝试通过直接命令行参数的方式配置这些选项：

iree-opt input.mlir --torch-to-iree --iree-torch-decompose-complex-ops=0

然而发现无论参数值如何变化，输出结果都没有差异，这表明参数未能正确生效。

原因分析

这种问题的根源在于对IREE参数传递机制的理解不足。IREE中的管道参数不是作为独立命令行参数传递的，而是需要嵌入到pass-pipeline的配置字符串中。

正确配置方法

正确的参数传递方式是将所有管道选项作为pass-pipeline字符串的一部分：

iree-opt input.mlir --pass-pipeline='builtin.module(torch-to-iree{decompose=false})'

这种格式明确指定了：

1. 在builtin.module上应用转换
2. 使用torch-to-iree管道
3. 设置decompose参数为false

技术细节

IREE的管道参数系统基于MLIR的PassPipelineCLParser机制，它要求：

• 管道名称后跟花括号{}包裹的参数列表
• 参数格式为key=value，多个参数用逗号分隔
• 整个管道定义作为pass-pipeline参数的值传递

实际应用建议

1. 对于复杂管道，建议先测试单个管道的参数效果
2. 使用--print-ir-after-all参数验证参数是否生效
3. 参数名称可能与源码中的定义略有不同，需参考具体管道实现

总结

理解IREE中管道参数的传递机制对于有效使用编译器工具链至关重要。通过正确的pass-pipeline语法，开发者可以精确控制各个优化阶段的参数，实现所需的编译行为。这种配置方式不仅适用于torch-to-iree管道，也适用于IREE中的所有其他优化管道。