MLIR 統一了 TensorFlow 中高效能 ML 模型的基礎架構。

MLIR 專案定義了一種通用的中繼表示法 (IR),統一了在 TensorFlow 和類似 ML 架構中執行高效能機器學習模型所需的基础架構。此專案將包含 HPC 技術的應用,以及強化學習等搜尋演算法的整合。MLIR 旨在降低啟動新硬體的成本,並提升現有 TensorFlow 使用者的可用性。
// Syntactically similar to LLVM:
func @testFunction(%arg0: i32) {
  %x = call @thingToCall(%arg0) : (i32) -> i32
  br ^bb1
^bb1:
  %y = arith.addi %x, %x : i32
  return %y : i32
}