onnx-mlir

Logo

MLIR コンパイラインフラストラクチャにおける ONNX モデルの表現や参照ローワーリング

GitHub でプロジェクトを見る onnx/onnx-mlir

方法

Python を使用した推論
C/C++ を使用した推論
Java を使用した推論

リファレンス

ONNX 方言
OMTensor C99 ランタイム API
OMTensorList C99 ランタイム API
OMTensor Java ランタイム API
OMTensorList Java ランタイム API
ONNX 方言を生成
ドキュメンテーションについて

開発

操作を追加
テストガイドライン
エラー処理
コマンドラインオプション
計装
定数の伝搬
アクセラレータを追加

ツール

ツール

RunONNXModel.py
DocCheck

このプロジェクトは onnx によってメンテナンスされます。

GitHub Pages でホスト、テーマは orderedlist

ONNX-MLIR のコマンドラインオプションを定義および使用

コマンドラインオプションは、onnx-mlir または onnx-mlir-opt のデフォルトの動作を変更するために使用したり、ユーザーのテスト、デバッグ、またはパフォーマンス調整に役立てることができます。LLVM によって提供されているコマンドラインユーティリティに基づいて、ONNX-MLIR でコマンドラインを実装しました。OptionListOption は、MLIR パス クラスと共に定義されていません(議論を参照)。

オプションを整理

オプションを定義する方法の基本的な概念については、llvm ドキュメント を参照してください。ONNX-MLIR では、オプションはグループ(llvm::cl::OptionCategory)に分けられます。onnx-mlir のすべてのコマンドラインオプションは、OnnxMlirOptions グループにあります。

コード構造

コマンドラインオプションは、src/Compiler/CompilerOptions.cpp に配置し、src/Compiler/CompilerOptions.hpp で宣言する必要があります。

オプションの定義

変換に固有のオプションを定義します。

MLIR のパスオプションを使用してパスを設定します。