https://github.com/MaheshRavishankar commented: Instead of this you could just use `affine.delinearize`. I think I want even the upstream methods to use affine.delinearize https://github.com/llvm/llvm-project/pull/138784