Groverkss wrote: @banach-space @dcaballe After a rethink, I found out I was wrong, and we can also extend the folder to dim-1 broadcasting cases. This should be a much simpler change now. Thanks for the reviews so far! https://github.com/llvm/llvm-project/pull/116234