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
}