hello I try to change the scope of a output buffer and map this operation on a loop iterator of a schedule in TE, as far as I saw it is possible with reverse_compute_at primitive in TIR, is it possible to have this functionality in tensor expression(te)?
You should use tir schedule when possible as TE is in legacy mode and likely TIR can cover what TE can do
FYI, TE can be lowered to TIR via create_prim_func, and then you can use TIR schedules.
3 Likes