[RFC] TensorIR: A schedulable IR for TVM

Development pre-RFC
2

It’s really a great work! @Hzfengsy

  1. Have you tried tensorize intrinsic(e.g. TensorCore schedule) using this new IR? Since I remember that to support tensorize is also one of your initial motivations.

I’m curious if this would requires a different programming logic writing such schedule. Before seeing any example, I feel hard to imagine how the schedule would be.

  1. The if_then_else case is really a strong point!

The original implementation of te.compute makes spatial iterators to follow the shape of the output Tensor, which limit us to write more flexible computations using TVM.

But seems this would break the original design of Stage, or do we still have a concept of Stage in the new TensorIR? How to schedule them?

  1. Looking forward to upgrade Ansor based on the new TensorIR.

Thanks!

1 Like