matthias-springer wrote: I think it would be OK to CSE two identical `reduce` ops. The same `reduce` op token would be passed to the `yield` twice, indicating that there are two reductions. https://github.com/llvm/llvm-project/pull/75314