Hardcode84 wrote: > Another way to keep these alive is to return a token that is passed to the yield. With tokens you will still have to somehow convince `CSE` not merge two identical `reduce` ops https://github.com/llvm/llvm-project/pull/75314