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 Options 来配置 passes。