https://github.com/razvanlupusoru approved this pull request. This is great. The only thing left for this is updating LoopOp::getDataOperand to skip the newly added operands :) Thank you! https://github.com/llvm/llvm-project/pull/67355