[llvm] r274583 - [X86][AVX2] Simplified BROADCAST combining to avoid repeated matching attempts

[Simon Pilgrim via llvm-commits]

Author: rksimon
Date: Tue Jul  5 17:41:04 2016
New Revision: 274583

URL: http://llvm.org/viewvc/llvm-project?rev=274583&view=rev
Log:
[X86][AVX2] Simplified BROADCAST combining to avoid repeated matching attempts

Modified:
    llvm/trunk/lib/Target/X86/X86ISelLowering.cpp

Modified: llvm/trunk/lib/Target/X86/X86ISelLowering.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86ISelLowering.cpp?rev=274583&r1=274582&r2=274583&view=diff
==============================================================================
--- llvm/trunk/lib/Target/X86/X86ISelLowering.cpp (original)
+++ llvm/trunk/lib/Target/X86/X86ISelLowering.cpp Tue Jul  5 17:41:04 2016
@@ -24781,18 +24781,15 @@ static bool matchUnaryVectorShuffle(MVT
 
   // Attempt to match against broadcast-from-vector.
   if (Subtarget.hasAVX2()) {
-    for (MVT SVT :
-         {MVT::i8, MVT::i16, MVT::i32, MVT::i64, MVT::f32, MVT::f64}) {
-      if (FloatDomain != SVT.isFloatingPoint())
-        continue;
-
-      unsigned NumElts = SrcVT.getSizeInBits() / SVT.getSizeInBits();
-      SmallVector<int, 64> BroadcastMask(NumElts, 0);
-      if (isTargetShuffleEquivalent(Mask, BroadcastMask)) {
-        Shuffle = X86ISD::VBROADCAST;
-        ShuffleVT = MVT::getVectorVT(SVT, NumElts);
-        return true;
-      }
+    unsigned NumElts = Mask.size();
+    SmallVector<int, 64> BroadcastMask(NumElts, 0);
+    if (isTargetShuffleEquivalent(Mask, BroadcastMask)) {
+      unsigned EltSize = SrcVT.getSizeInBits() / NumElts;
+      ShuffleVT = FloatDomain ? MVT::getFloatingPointVT(EltSize)
+                              : MVT::getIntegerVT(EltSize);
+      ShuffleVT = MVT::getVectorVT(ShuffleVT, NumElts);
+      Shuffle = X86ISD::VBROADCAST;
+      return true;
     }
   }