定义并使用 ONNX-MLIR 的命令行选项

命令行选项可以用于改变 onnx-mlir 或 onnx-mlir-opt 的默认行为，并帮助用户进行实验、调试或性能调优。我们在 ONNX-MLIR 中基于 LLVM 提供的命令行工具实现了命令行。我们没有使用 MLIR pass 类定义 Option 或 ListOption（参见讨论）。

组织选项

参考 llvm 文档 以了解如何定义选项的基本思想。在 ONNX-MLIR 中，选项被归入组 (llvm::cl::OptionCategory)。onnx-mlir 的所有命令行选项都在 OnnxMlirOptions 组中。

代码结构

命令行选项应放在 src/Compiler/CompilerOptions.cpp 中，并在 src/Compiler/CompilerOptions.hpp 中声明。

定义一个选项

• 在 src/Compiler/CompilerOptions.hpp 中添加选项的声明。
• 在 src/Compiler/CompilerOptions.cpp 中定义选项。
• 不要在新的源文件中包含 src/Compiler/CompilerOptions.hpp；它应该只在 onnx-mlir 和 onnn-mlir-opt 命令行工具中使用。
• 请创建“Pass 选项”以将信息传递给特定的 pass 和转换。

定义一个特定于某个转换的局部选项

使用 MLIR 的 Pass 选项来配置 pass。