https://github.com/david-arm approved this pull request. LGTM! I think this is a nice incremental improvement that doesn't regress the scalar cases, which can be fixed in a follow-on PR. https://github.com/llvm/llvm-project/pull/193268