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 }