TensorIR

TensorIR 是 Apache TVM 堆栈中的核心抽象之一，用于表示和优化原始张量函数。

• 张量程序抽象
  • 张量程序的主要元素
  • TensorIR 中的额外结构
• 理解 TensorIR 抽象
  • 函数参数和缓冲区
  • 循环迭代
  • 计算块
  • 块轴属性
  • 为什么块中需要额外信息
  • 块轴绑定的语法糖
• TensorIR 创建
  • 使用 TVMScript 创建 TensorIR
  • 使用张量表达式创建 TensorIR
• 转换
  • 初始化调度
  • 循环分块
  • 利用局部性
  • 重写归约
  • 追踪转换过程