commit 80ce1dde44ec7909173e1a6ca546d1db81549f5f
author Simon Pilgrim <llvm-dev@redking.me.uk> Sat Apr 07 13:24:33 2018 +0000
committer Simon Pilgrim <llvm-dev@redking.me.uk> Sat Apr 07 13:24:33 2018 +0000
tree 1a41b8a3228741b80c8cdb88c381101c9f4222f6
parent a49a1b9cccbc632fa2763dd4efe49fb22c4ef899

[CostModel][X86] Fix v32i16/v64i8 SETCC costs on AVX512BW targets

llvm-svn: 329498

llvm/lib/Target/X86/X86TargetTransformInfo.cpp

diff --git a/llvm/lib/Target/X86/X86TargetTransformInfo.cpp b/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
index 5b03071..4813ab3d 100644
--- a/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
+++ b/llvm/lib/Target/X86/X86TargetTransformInfo.cpp
@@ -1534,6 +1534,15 @@
     { ISD::SETCC,   MVT::v16f32,  1 },
   };
 
+  static const CostTblEntry AVX512BWCostTbl[] = {
+    { ISD::SETCC,   MVT::v32i16,  1 },
+    { ISD::SETCC,   MVT::v64i8,   1 },
+  };
+
+  if (ST->hasBWI())
+    if (const auto *Entry = CostTableLookup(AVX512BWCostTbl, ISD, MTy))
+      return LT.first * Entry->Cost;
+
   if (ST->hasAVX512())
     if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
       return LT.first * Entry->Cost;