Just as a info/update on this thread, MLIR now have “controlable schedule” concept that allows a separate schedule to control things on a IR. As a example, in their presentation, Halide and TVM are cited. Probably now, with this, is more easy (or at least it would be) and straightforward to represent TVM’s schedule part and the TIR parts with MLIR. This “controlable part” (or schedule) is a dialect itself, but one controlling elements of a IR.
I leave this here, just as a pure info, but I assume many of you are already being aware.
The presentation: https://youtu.be/P4gUj3QtH_Y