ONNX 模型在 MLIR 编译器基础设施中的表示和参考降低
本项目由 onnx 维护
托管于 GitHub Pages — 主题作者:orderedlist
命令行选项可用于更改 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 命令行工具中使用。使用 MLIR 的 Pass 选项来配置 Pass。