https://github.com/dcaballe commented: Thanks for fixing this! Left a few comments. I would also revisit the commit description as this seems to be a reduction problem, not a broadcast problem? Or maybe both broadcast and reduction? https://github.com/llvm/llvm-project/pull/99015