[X86] Add more extract subvector cost model tests for smaller element sizes and smaller than 128-bit vectors.

With the switch to widening legalization, we need to a better
job of costing extractions of less than 128-bits.

llvm-svn: 368081

1 files changed, 9 insertions, 0 deletions

diff --git a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
index dd28b18258f..42371a5beb6 100644
--- a/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
+++ b/llvm/lib/Transforms/Vectorize/LoopVectorize.cpp
@@ -5135,6 +5135,15 @@ unsigned LoopVectorizationCostModel::selectInterleaveCount(unsigned VF,
   else if (IC < 1)
     IC = 1;
 
+  // If the trip count is constant, clamp the calculated IC to be between 1 and
+  // the trip count divided by VF.
+  if (TC > 0) {
+    if ((TC / VF) < IC)
+      IC = (TC / VF);
+    if (IC < 1)
+      IC = 1;
+  }
+
   // Interleave if we vectorized this loop and there is a reduction that could
   // benefit from interleaving.
   if (VF > 1 && !Legal->getReductionVars()->empty()) {