commit f96b4ab92d6a95a0720d18b1cd948a95d0cb54e7
author Simon Pilgrim <llvm-dev@redking.me.uk> Sun May 14 20:25:42 2017 +0000
committer Simon Pilgrim <llvm-dev@redking.me.uk> Sun May 14 20:25:42 2017 +0000
tree 1fe6f6c3cb998c6df183392c3021b325754c9f1c
parent de4467b182e37392d028dfd469883ef5f0b275fb

[X86][AVX2] Fix costs for v4i64 ashr by splat

llvm-svn: 303022

llvm/lib/Target/X86/X86TargetTransformInfo.cpp

diff --git a/llvm/lib/Target/X86/X86TargetTransformInfo.cpp b/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
index 4ea66d2..4a5c7e3 100644
--- a/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
+++ b/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
@@ -471,6 +471,11 @@
   if (ST->hasSSE2() &&
       ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
        (Op2Info == TargetTransformInfo::OK_UniformValue))) {
+
+    // Handle AVX2 uniform v4i64 ISD::SRA, it's not worth a table.
+    if (ISD == ISD::SRA && LT.second == MVT::v4i64 && ST->hasAVX2())
+      return LT.first * 4; // 2*psrad + shuffle.
+
     if (const auto *Entry =
             CostTableLookup(SSE2UniformShiftCostTable, ISD, LT.second))
       return LT.first * Entry->Cost;