TensorIR

TensorIR is one of the core abstraction in Apache TVM Unity stack, which is used to represent and optimize the primitive tensor functions.

• Tensor Program Abstraction
  • Key Elements of Tensor Programs
  • Extra Structure in TensorIR
• Understand TensorIR Abstraction
  • Function Parameters and Buffers
  • Loop Iterations
  • Computational Block
  • Block Axis Properties
  • Why Extra Information in Block
  • Sugars for Block Axes Binding
• TensorIR Creation
  • Create TensorIR using TVMScript
  • Create TensorIR using Tensor Expression
• Transformation
  • Initialization Schedule
  • Loop Tiling
  • Leverage Localities
  • Rewrite Reduction
  • Trace the Transformation