ONNX MLIRに含まれるDialectとその変換パスを調査。 Dialectの変換はソースコードの以下に存在。

Dialect一覧

ONNX MLIRには以下のDialectが含まれている。

  • ONNX Dialect
  • Krnl Dialect
  • Handshake Dialect
  • StaticLogic Dialect
  • LLHD Dialect

その他、MLIRでデフォルトで定義されているDialectとして以下がある。

  • Affine Dialect
  • Standard Dialect
  • LLVM Dialect

Dialect変換パス

ONNXモデルをLLVM IRを生成する流れは、

  • ONNX parserでONNXモデルからONNX Dialectへ変換し、
  • ONNX DialectからKrnl Dialectへ変換、
  • Krnl DialectからLLVM Dialectへ変換し、
  • 最後にLLVM DialectからLLVM IRを生成する