https://github.com/fhahn commented: > LGTM. Do we have a test case somewhere for an unrolled replicate recipe w/ tail folding? I added a new test where we unmask a store with tail-folding and UF = 2, thanks https://github.com/llvm/llvm-project/pull/201676