定义和使用 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。