[SDAG] avoid vector extract/insert around binop
authorSanjay Patel <spatel@rotateright.com>
Wed, 26 Oct 2022 15:53:49 +0000 (11:53 -0400)
committerSanjay Patel <spatel@rotateright.com>
Wed, 26 Oct 2022 18:04:46 +0000 (14:04 -0400)
commit54eeadcf442df91aed0fb7244fe7885cdf1b1f3d
treedb1fd03c2ab38d858a304e1d500cf547bf81719d
parentf85ce1b236d50ebc1dd5960ce88026138506697a
[SDAG] avoid vector extract/insert around binop

scalar-to-vector (scalar binop (extractelt V, Idx), C) --> shuffle (vector binop V, C'), {Idx, -1, -1...}

We generally try to avoid ad-hoc vectorization in SDAG,
but the motivating case from issue #39482 escapes our
normal vectorization folds in IR. It seems like it should
always be a win to transform this pattern in cases where
we have the same vector type for input and output and the
target supports the vector operation. That avoids
transfers from vector to scalar and back.

In the x86 shift examples, we create the scalar-to-vector
node during legalization. I'm not sure if there's a more
general way to create the pattern for testing. (If so, I
could add tests for other targets.)

Differential Revision: https://reviews.llvm.org/D136713
llvm/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
llvm/test/CodeGen/X86/cvt16.ll
llvm/test/CodeGen/X86/half.ll
llvm/test/CodeGen/X86/urem-vector-lkk.ll
llvm/test/CodeGen/X86/vec_shift5.ll