為 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。