https://github.com/razvanlupusoru commented: Thank you Susan! Would you mind figuring out if there is any missing functionality since it seems something is different compared to scf.for? https://github.com/llvm/llvm-project/pull/172940