[RFC] TensorIR: A schedulable IR for TVM

Development pre-RFC
39

When I read this RFC, I was confused because I read that the compilation flow currently goes from

TF/PyTorch/ONNX → Relay → TE (based on TE schedule) → TIR → C++/CUDA

And then I read:

TensorIR is a brand new low-level IR with full scheduling support. Here are some […]

And then see references to TIR throughout the rest of the RFC, which seem unclear to me if these references are to the new TensorIR, or the old TIR.

Can we clarify both here and in the code base moving forward whether we are referring to TensorIR or the old TIR? I think there are two ways of doing this going forward and we should explicitly pick one:

  1. Have TIR refer to the old version of TIR, and always use TensorIR when talking about the new IR.
  2. Be clear that TensorIR moving forward will be replacing the old TIR, so that any reference to TIR could be referring to the new or old version (and should be clarified if not obvious from context).

Someone more knowledgeable than me should pick one of those (or correct me and/or point out other options if I’ve gotten anything wrong here).

1 Like