arsenm wrote: > This doesn't belong in mlir. We have a separate PR adding more wave reduce intrinsics already https://github.com/llvm/llvm-project/pull/126469 and related https://github.com/llvm/llvm-project/pull/133204