blob: 2fbba6a44f245c8f1135cf6141e7d490702977c9 [file] [log] [blame]
Chandler Carruth664e3542013-01-07 01:37:14 +00001//===-- X86TargetTransformInfo.cpp - X86 specific TTI pass ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9/// \file
10/// This file implements a TargetTransformInfo analysis pass specific to the
11/// X86 target machine. It uses the target's detailed information to provide
12/// more precise answers to certain TTI queries, while letting the target
13/// independent and default TTI implementations handle the rest.
14///
15//===----------------------------------------------------------------------===//
Alexey Bataevb271a582016-10-12 13:24:13 +000016/// About Cost Model numbers used below it's necessary to say the following:
17/// the numbers correspond to some "generic" X86 CPU instead of usage of
18/// concrete CPU model. Usually the numbers correspond to CPU where the feature
19/// apeared at the first time. For example, if we do Subtarget.hasSSE42() in
20/// the lookups below the cost is based on Nehalem as that was the first CPU
21/// to support that feature level and thus has most likely the worst case cost.
22/// Some examples of other technologies/CPUs:
23/// SSE 3 - Pentium4 / Athlon64
24/// SSE 4.1 - Penryn
25/// SSE 4.2 - Nehalem
26/// AVX - Sandy Bridge
27/// AVX2 - Haswell
28/// AVX-512 - Xeon Phi / Skylake
29/// And some examples of instruction target dependent costs (latency)
30/// divss sqrtss rsqrtss
31/// AMD K7 11-16 19 3
32/// Piledriver 9-24 13-15 5
33/// Jaguar 14 16 2
34/// Pentium II,III 18 30 2
35/// Nehalem 7-14 7-18 3
36/// Haswell 10-13 11 5
37/// TODO: Develop and implement the target dependent cost model and
38/// specialize cost numbers for different Cost Model Targets such as throughput,
39/// code size, latency and uop count.
40//===----------------------------------------------------------------------===//
Chandler Carruth664e3542013-01-07 01:37:14 +000041
Chandler Carruth93dcdc42015-01-31 11:17:59 +000042#include "X86TargetTransformInfo.h"
Chandler Carruthd3e73552013-01-07 03:08:10 +000043#include "llvm/Analysis/TargetTransformInfo.h"
Chandler Carruth705b1852015-01-31 03:43:40 +000044#include "llvm/CodeGen/BasicTTIImpl.h"
David Blaikieb3bde2e2017-11-17 01:07:10 +000045#include "llvm/CodeGen/CostTable.h"
46#include "llvm/CodeGen/TargetLowering.h"
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000047#include "llvm/IR/IntrinsicInst.h"
Chandler Carruth664e3542013-01-07 01:37:14 +000048#include "llvm/Support/Debug.h"
Hans Wennborg083ca9b2015-10-06 23:24:35 +000049
Chandler Carruth664e3542013-01-07 01:37:14 +000050using namespace llvm;
51
Chandler Carruth84e68b22014-04-22 02:41:26 +000052#define DEBUG_TYPE "x86tti"
53
Chandler Carruth664e3542013-01-07 01:37:14 +000054//===----------------------------------------------------------------------===//
55//
56// X86 cost model.
57//
58//===----------------------------------------------------------------------===//
59
Chandler Carruth705b1852015-01-31 03:43:40 +000060TargetTransformInfo::PopcntSupportKind
61X86TTIImpl::getPopcntSupport(unsigned TyWidth) {
Chandler Carruth664e3542013-01-07 01:37:14 +000062 assert(isPowerOf2_32(TyWidth) && "Ty width must be power of 2");
63 // TODO: Currently the __builtin_popcount() implementation using SSE3
64 // instructions is inefficient. Once the problem is fixed, we should
Craig Topper0a63e1d2013-09-08 00:47:31 +000065 // call ST->hasSSE3() instead of ST->hasPOPCNT().
Chandler Carruth705b1852015-01-31 03:43:40 +000066 return ST->hasPOPCNT() ? TTI::PSK_FastHardware : TTI::PSK_Software;
Chandler Carruth664e3542013-01-07 01:37:14 +000067}
68
Tobias Grosserd7eb6192017-08-24 09:46:25 +000069llvm::Optional<unsigned> X86TTIImpl::getCacheSize(
70 TargetTransformInfo::CacheLevel Level) const {
71 switch (Level) {
72 case TargetTransformInfo::CacheLevel::L1D:
Craig Topperd5b5bbe2017-11-22 18:23:40 +000073 // - Penryn
Tobias Grosserd7eb6192017-08-24 09:46:25 +000074 // - Nehalem
75 // - Westmere
76 // - Sandy Bridge
77 // - Ivy Bridge
78 // - Haswell
79 // - Broadwell
80 // - Skylake
81 // - Kabylake
82 return 32 * 1024; // 32 KByte
83 case TargetTransformInfo::CacheLevel::L2D:
Craig Topperd5b5bbe2017-11-22 18:23:40 +000084 // - Penryn
Tobias Grosserd7eb6192017-08-24 09:46:25 +000085 // - Nehalem
86 // - Westmere
87 // - Sandy Bridge
88 // - Ivy Bridge
89 // - Haswell
90 // - Broadwell
91 // - Skylake
92 // - Kabylake
93 return 256 * 1024; // 256 KByte
94 }
95
96 llvm_unreachable("Unknown TargetTransformInfo::CacheLevel");
97}
98
99llvm::Optional<unsigned> X86TTIImpl::getCacheAssociativity(
100 TargetTransformInfo::CacheLevel Level) const {
Craig Topperd5b5bbe2017-11-22 18:23:40 +0000101 // - Penryn
Tobias Grosserd7eb6192017-08-24 09:46:25 +0000102 // - Nehalem
103 // - Westmere
104 // - Sandy Bridge
105 // - Ivy Bridge
106 // - Haswell
107 // - Broadwell
108 // - Skylake
109 // - Kabylake
110 switch (Level) {
111 case TargetTransformInfo::CacheLevel::L1D:
112 LLVM_FALLTHROUGH;
113 case TargetTransformInfo::CacheLevel::L2D:
114 return 8;
115 }
116
117 llvm_unreachable("Unknown TargetTransformInfo::CacheLevel");
118}
119
Chandler Carruth705b1852015-01-31 03:43:40 +0000120unsigned X86TTIImpl::getNumberOfRegisters(bool Vector) {
Nadav Rotemb1791a72013-01-09 22:29:00 +0000121 if (Vector && !ST->hasSSE1())
122 return 0;
123
Adam Nemet2820a5b2014-07-09 18:22:33 +0000124 if (ST->is64Bit()) {
125 if (Vector && ST->hasAVX512())
126 return 32;
Chandler Carruth664e3542013-01-07 01:37:14 +0000127 return 16;
Adam Nemet2820a5b2014-07-09 18:22:33 +0000128 }
Chandler Carruth664e3542013-01-07 01:37:14 +0000129 return 8;
130}
131
Keno Fischer1ec5dd82017-04-05 20:51:38 +0000132unsigned X86TTIImpl::getRegisterBitWidth(bool Vector) const {
Craig Topper0d797a32018-01-20 00:26:08 +0000133 unsigned PreferVectorWidth = ST->getPreferVectorWidth();
Nadav Rotemb1791a72013-01-09 22:29:00 +0000134 if (Vector) {
Craig Topper0d797a32018-01-20 00:26:08 +0000135 if (ST->hasAVX512() && PreferVectorWidth >= 512)
Mohammed Agabaria189e2d22017-01-05 09:51:02 +0000136 return 512;
Craig Topper0d797a32018-01-20 00:26:08 +0000137 if (ST->hasAVX() && PreferVectorWidth >= 256)
Mohammed Agabaria189e2d22017-01-05 09:51:02 +0000138 return 256;
Craig Topper0d797a32018-01-20 00:26:08 +0000139 if (ST->hasSSE1() && PreferVectorWidth >= 128)
Mohammed Agabaria189e2d22017-01-05 09:51:02 +0000140 return 128;
Nadav Rotemb1791a72013-01-09 22:29:00 +0000141 return 0;
142 }
143
144 if (ST->is64Bit())
145 return 64;
Nadav Rotemb1791a72013-01-09 22:29:00 +0000146
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000147 return 32;
Nadav Rotemb1791a72013-01-09 22:29:00 +0000148}
149
Keno Fischer1ec5dd82017-04-05 20:51:38 +0000150unsigned X86TTIImpl::getLoadStoreVecRegBitWidth(unsigned) const {
151 return getRegisterBitWidth(true);
152}
153
Wei Mi062c7442015-05-06 17:12:25 +0000154unsigned X86TTIImpl::getMaxInterleaveFactor(unsigned VF) {
155 // If the loop will not be vectorized, don't interleave the loop.
156 // Let regular unroll to unroll the loop, which saves the overflow
157 // check and memory check cost.
158 if (VF == 1)
159 return 1;
160
Nadav Rotemb696c362013-01-09 01:15:42 +0000161 if (ST->isAtom())
162 return 1;
163
164 // Sandybridge and Haswell have multiple execution ports and pipelined
165 // vector units.
166 if (ST->hasAVX())
167 return 4;
168
169 return 2;
170}
171
Chandler Carruth93205eb2015-08-05 18:08:10 +0000172int X86TTIImpl::getArithmeticInstrCost(
Simon Pilgrim3e5b5252017-01-20 15:15:59 +0000173 unsigned Opcode, Type *Ty,
Mohammed Agabaria2c96c432017-01-11 08:23:37 +0000174 TTI::OperandValueKind Op1Info, TTI::OperandValueKind Op2Info,
175 TTI::OperandValueProperties Opd1PropInfo,
176 TTI::OperandValueProperties Opd2PropInfo,
177 ArrayRef<const Value *> Args) {
Chandler Carruth664e3542013-01-07 01:37:14 +0000178 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000179 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Ty);
Chandler Carruth664e3542013-01-07 01:37:14 +0000180
181 int ISD = TLI->InstructionOpcodeToISD(Opcode);
182 assert(ISD && "Invalid opcode");
183
Craig Toppera9859192018-03-25 15:58:12 +0000184 static const CostTblEntry GLMCostTable[] = {
185 { ISD::FDIV, MVT::f32, 18 }, // divss
186 { ISD::FDIV, MVT::v4f32, 35 }, // divps
187 { ISD::FDIV, MVT::f64, 33 }, // divsd
188 { ISD::FDIV, MVT::v2f64, 65 }, // divpd
189 };
190
191 if (ST->isGLM())
192 if (const auto *Entry = CostTableLookup(GLMCostTable, ISD,
193 LT.second))
194 return LT.first * Entry->Cost;
195
Mohammed Agabaria2c96c432017-01-11 08:23:37 +0000196 static const CostTblEntry SLMCostTable[] = {
Craig Toppera9859192018-03-25 15:58:12 +0000197 { ISD::MUL, MVT::v4i32, 11 }, // pmulld
198 { ISD::MUL, MVT::v8i16, 2 }, // pmullw
199 { ISD::MUL, MVT::v16i8, 14 }, // extend/pmullw/trunc sequence.
200 { ISD::FMUL, MVT::f64, 2 }, // mulsd
201 { ISD::FMUL, MVT::v2f64, 4 }, // mulpd
202 { ISD::FMUL, MVT::v4f32, 2 }, // mulps
203 { ISD::FDIV, MVT::f32, 17 }, // divss
204 { ISD::FDIV, MVT::v4f32, 39 }, // divps
205 { ISD::FDIV, MVT::f64, 32 }, // divsd
206 { ISD::FDIV, MVT::v2f64, 69 }, // divpd
207 { ISD::FADD, MVT::v2f64, 2 }, // addpd
208 { ISD::FSUB, MVT::v2f64, 2 }, // subpd
Mohammed Agabariaeb09a812017-07-02 12:16:15 +0000209 // v2i64/v4i64 mul is custom lowered as a series of long:
210 // multiplies(3), shifts(3) and adds(2)
Simon Pilgrim7b89ab52017-07-31 17:09:27 +0000211 // slm muldq version throughput is 2 and addq throughput 4
Simon Pilgrimeb070162018-01-30 12:18:51 +0000212 // thus: 3X2 (muldq throughput) + 3X1 (shift throughput) +
Simon Pilgrim7b89ab52017-07-31 17:09:27 +0000213 // 3X4 (addq throughput) = 17
Craig Toppera9859192018-03-25 15:58:12 +0000214 { ISD::MUL, MVT::v2i64, 17 },
Mohammed Agabariaeb09a812017-07-02 12:16:15 +0000215 // slm addq\subq throughput is 4
Craig Toppera9859192018-03-25 15:58:12 +0000216 { ISD::ADD, MVT::v2i64, 4 },
217 { ISD::SUB, MVT::v2i64, 4 },
Mohammed Agabaria2c96c432017-01-11 08:23:37 +0000218 };
219
220 if (ST->isSLM()) {
221 if (Args.size() == 2 && ISD == ISD::MUL && LT.second == MVT::v4i32) {
222 // Check if the operands can be shrinked into a smaller datatype.
223 bool Op1Signed = false;
224 unsigned Op1MinSize = BaseT::minRequiredElementSize(Args[0], Op1Signed);
225 bool Op2Signed = false;
226 unsigned Op2MinSize = BaseT::minRequiredElementSize(Args[1], Op2Signed);
227
228 bool signedMode = Op1Signed | Op2Signed;
229 unsigned OpMinSize = std::max(Op1MinSize, Op2MinSize);
230
231 if (OpMinSize <= 7)
232 return LT.first * 3; // pmullw/sext
233 if (!signedMode && OpMinSize <= 8)
234 return LT.first * 3; // pmullw/zext
235 if (OpMinSize <= 15)
236 return LT.first * 5; // pmullw/pmulhw/pshuf
237 if (!signedMode && OpMinSize <= 16)
238 return LT.first * 5; // pmullw/pmulhw/pshuf
239 }
Craig Toppera9859192018-03-25 15:58:12 +0000240
Mohammed Agabaria2c96c432017-01-11 08:23:37 +0000241 if (const auto *Entry = CostTableLookup(SLMCostTable, ISD,
242 LT.second)) {
243 return LT.first * Entry->Cost;
244 }
245 }
246
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000247 if (ISD == ISD::SDIV &&
Simon Pilgrim4162d772018-05-22 10:40:09 +0000248 (Op2Info == TargetTransformInfo::OK_UniformConstantValue ||
249 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue) &&
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000250 Opd2PropInfo == TargetTransformInfo::OP_PowerOf2) {
251 // On X86, vector signed division by constants power-of-two are
252 // normally expanded to the sequence SRA + SRL + ADD + SRA.
Simon Pilgrim4162d772018-05-22 10:40:09 +0000253 // The OperandValue properties may not be the same as that of the previous
254 // operation; conservatively assume OP_None.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000255 int Cost = 2 * getArithmeticInstrCost(Instruction::AShr, Ty, Op1Info,
256 Op2Info, TargetTransformInfo::OP_None,
257 TargetTransformInfo::OP_None);
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000258 Cost += getArithmeticInstrCost(Instruction::LShr, Ty, Op1Info, Op2Info,
259 TargetTransformInfo::OP_None,
260 TargetTransformInfo::OP_None);
261 Cost += getArithmeticInstrCost(Instruction::Add, Ty, Op1Info, Op2Info,
262 TargetTransformInfo::OP_None,
263 TargetTransformInfo::OP_None);
264
265 return Cost;
266 }
267
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000268 static const CostTblEntry AVX512BWUniformConstCostTable[] = {
Simon Pilgrim9c589502017-01-08 14:14:36 +0000269 { ISD::SHL, MVT::v64i8, 2 }, // psllw + pand.
270 { ISD::SRL, MVT::v64i8, 2 }, // psrlw + pand.
271 { ISD::SRA, MVT::v64i8, 4 }, // psrlw, pand, pxor, psubb.
272
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000273 { ISD::SDIV, MVT::v32i16, 6 }, // vpmulhw sequence
274 { ISD::UDIV, MVT::v32i16, 6 }, // vpmulhuw sequence
275 };
276
277 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
278 ST->hasBWI()) {
279 if (const auto *Entry = CostTableLookup(AVX512BWUniformConstCostTable, ISD,
280 LT.second))
281 return LT.first * Entry->Cost;
282 }
283
284 static const CostTblEntry AVX512UniformConstCostTable[] = {
Simon Pilgrimd419b732017-01-14 19:24:23 +0000285 { ISD::SRA, MVT::v2i64, 1 },
286 { ISD::SRA, MVT::v4i64, 1 },
287 { ISD::SRA, MVT::v8i64, 1 },
288
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000289 { ISD::SDIV, MVT::v16i32, 15 }, // vpmuldq sequence
290 { ISD::UDIV, MVT::v16i32, 15 }, // vpmuludq sequence
291 };
292
293 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
294 ST->hasAVX512()) {
295 if (const auto *Entry = CostTableLookup(AVX512UniformConstCostTable, ISD,
296 LT.second))
297 return LT.first * Entry->Cost;
298 }
299
Craig Topper4b275762015-10-28 04:02:12 +0000300 static const CostTblEntry AVX2UniformConstCostTable[] = {
Simon Pilgrim9c589502017-01-08 14:14:36 +0000301 { ISD::SHL, MVT::v32i8, 2 }, // psllw + pand.
302 { ISD::SRL, MVT::v32i8, 2 }, // psrlw + pand.
303 { ISD::SRA, MVT::v32i8, 4 }, // psrlw, pand, pxor, psubb.
304
Simon Pilgrim8fbf1c12015-07-06 22:35:19 +0000305 { ISD::SRA, MVT::v4i64, 4 }, // 2 x psrad + shuffle.
306
Benjamin Kramer7c372272014-04-26 14:53:05 +0000307 { ISD::SDIV, MVT::v16i16, 6 }, // vpmulhw sequence
308 { ISD::UDIV, MVT::v16i16, 6 }, // vpmulhuw sequence
309 { ISD::SDIV, MVT::v8i32, 15 }, // vpmuldq sequence
310 { ISD::UDIV, MVT::v8i32, 15 }, // vpmuludq sequence
311 };
312
313 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
314 ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000315 if (const auto *Entry = CostTableLookup(AVX2UniformConstCostTable, ISD,
316 LT.second))
317 return LT.first * Entry->Cost;
Benjamin Kramer7c372272014-04-26 14:53:05 +0000318 }
319
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000320 static const CostTblEntry SSE2UniformConstCostTable[] = {
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000321 { ISD::SHL, MVT::v16i8, 2 }, // psllw + pand.
322 { ISD::SRL, MVT::v16i8, 2 }, // psrlw + pand.
323 { ISD::SRA, MVT::v16i8, 4 }, // psrlw, pand, pxor, psubb.
Simon Pilgrim9c589502017-01-08 14:14:36 +0000324
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000325 { ISD::SHL, MVT::v32i8, 4+2 }, // 2*(psllw + pand) + split.
326 { ISD::SRL, MVT::v32i8, 4+2 }, // 2*(psrlw + pand) + split.
327 { ISD::SRA, MVT::v32i8, 8+2 }, // 2*(psrlw, pand, pxor, psubb) + split.
Simon Pilgrim9c589502017-01-08 14:14:36 +0000328
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000329 { ISD::SDIV, MVT::v16i16, 12+2 }, // 2*pmulhw sequence + split.
330 { ISD::SDIV, MVT::v8i16, 6 }, // pmulhw sequence
331 { ISD::UDIV, MVT::v16i16, 12+2 }, // 2*pmulhuw sequence + split.
332 { ISD::UDIV, MVT::v8i16, 6 }, // pmulhuw sequence
333 { ISD::SDIV, MVT::v8i32, 38+2 }, // 2*pmuludq sequence + split.
334 { ISD::SDIV, MVT::v4i32, 19 }, // pmuludq sequence
335 { ISD::UDIV, MVT::v8i32, 30+2 }, // 2*pmuludq sequence + split.
336 { ISD::UDIV, MVT::v4i32, 15 }, // pmuludq sequence
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000337 };
338
339 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
340 ST->hasSSE2()) {
341 // pmuldq sequence.
342 if (ISD == ISD::SDIV && LT.second == MVT::v8i32 && ST->hasAVX())
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000343 return LT.first * 32;
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000344 if (ISD == ISD::SDIV && LT.second == MVT::v4i32 && ST->hasSSE41())
345 return LT.first * 15;
346
Simon Pilgrim5bef9c62017-05-14 17:59:46 +0000347 // XOP has faster vXi8 shifts.
348 if ((ISD != ISD::SHL && ISD != ISD::SRL && ISD != ISD::SRA) ||
349 !ST->hasXOP())
350 if (const auto *Entry =
351 CostTableLookup(SSE2UniformConstCostTable, ISD, LT.second))
352 return LT.first * Entry->Cost;
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000353 }
354
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000355 static const CostTblEntry AVX2UniformCostTable[] = {
356 // Uniform splats are cheaper for the following instructions.
357 { ISD::SHL, MVT::v16i16, 1 }, // psllw.
358 { ISD::SRL, MVT::v16i16, 1 }, // psrlw.
359 { ISD::SRA, MVT::v16i16, 1 }, // psraw.
360 };
361
362 if (ST->hasAVX2() &&
363 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
364 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
365 if (const auto *Entry =
366 CostTableLookup(AVX2UniformCostTable, ISD, LT.second))
367 return LT.first * Entry->Cost;
368 }
369
370 static const CostTblEntry SSE2UniformCostTable[] = {
371 // Uniform splats are cheaper for the following instructions.
372 { ISD::SHL, MVT::v8i16, 1 }, // psllw.
373 { ISD::SHL, MVT::v4i32, 1 }, // pslld
374 { ISD::SHL, MVT::v2i64, 1 }, // psllq.
375
376 { ISD::SRL, MVT::v8i16, 1 }, // psrlw.
377 { ISD::SRL, MVT::v4i32, 1 }, // psrld.
378 { ISD::SRL, MVT::v2i64, 1 }, // psrlq.
379
380 { ISD::SRA, MVT::v8i16, 1 }, // psraw.
381 { ISD::SRA, MVT::v4i32, 1 }, // psrad.
382 };
383
384 if (ST->hasSSE2() &&
385 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
386 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
387 if (const auto *Entry =
388 CostTableLookup(SSE2UniformCostTable, ISD, LT.second))
389 return LT.first * Entry->Cost;
390 }
391
Simon Pilgrim820e1322016-10-27 15:27:00 +0000392 static const CostTblEntry AVX512DQCostTable[] = {
393 { ISD::MUL, MVT::v2i64, 1 },
394 { ISD::MUL, MVT::v4i64, 1 },
395 { ISD::MUL, MVT::v8i64, 1 }
396 };
397
398 // Look for AVX512DQ lowering tricks for custom cases.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000399 if (ST->hasDQI())
400 if (const auto *Entry = CostTableLookup(AVX512DQCostTable, ISD, LT.second))
Simon Pilgrim820e1322016-10-27 15:27:00 +0000401 return LT.first * Entry->Cost;
Simon Pilgrim820e1322016-10-27 15:27:00 +0000402
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000403 static const CostTblEntry AVX512BWCostTable[] = {
Simon Pilgrim6ed996c2017-01-15 20:44:00 +0000404 { ISD::SHL, MVT::v8i16, 1 }, // vpsllvw
405 { ISD::SRL, MVT::v8i16, 1 }, // vpsrlvw
406 { ISD::SRA, MVT::v8i16, 1 }, // vpsravw
407
408 { ISD::SHL, MVT::v16i16, 1 }, // vpsllvw
409 { ISD::SRL, MVT::v16i16, 1 }, // vpsrlvw
410 { ISD::SRA, MVT::v16i16, 1 }, // vpsravw
411
Simon Pilgrima4109d62017-01-07 17:54:10 +0000412 { ISD::SHL, MVT::v32i16, 1 }, // vpsllvw
413 { ISD::SRL, MVT::v32i16, 1 }, // vpsrlvw
414 { ISD::SRA, MVT::v32i16, 1 }, // vpsravw
415
Simon Pilgrim5a81fef2017-01-11 10:36:51 +0000416 { ISD::SHL, MVT::v64i8, 11 }, // vpblendvb sequence.
417 { ISD::SRL, MVT::v64i8, 11 }, // vpblendvb sequence.
418 { ISD::SRA, MVT::v64i8, 24 }, // vpblendvb sequence.
419
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000420 { ISD::MUL, MVT::v64i8, 11 }, // extend/pmullw/trunc sequence.
421 { ISD::MUL, MVT::v32i8, 4 }, // extend/pmullw/trunc sequence.
422 { ISD::MUL, MVT::v16i8, 4 }, // extend/pmullw/trunc sequence.
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000423 };
424
425 // Look for AVX512BW lowering tricks for custom cases.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000426 if (ST->hasBWI())
427 if (const auto *Entry = CostTableLookup(AVX512BWCostTable, ISD, LT.second))
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000428 return LT.first * Entry->Cost;
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000429
Craig Topper4b275762015-10-28 04:02:12 +0000430 static const CostTblEntry AVX512CostTable[] = {
Simon Pilgrimd8333372017-01-06 11:12:53 +0000431 { ISD::SHL, MVT::v16i32, 1 },
432 { ISD::SRL, MVT::v16i32, 1 },
433 { ISD::SRA, MVT::v16i32, 1 },
Simon Pilgrimd419b732017-01-14 19:24:23 +0000434
Simon Pilgrimd8333372017-01-06 11:12:53 +0000435 { ISD::SHL, MVT::v8i64, 1 },
436 { ISD::SRL, MVT::v8i64, 1 },
Simon Pilgrimd419b732017-01-14 19:24:23 +0000437
438 { ISD::SRA, MVT::v2i64, 1 },
439 { ISD::SRA, MVT::v4i64, 1 },
Simon Pilgrimd8333372017-01-06 11:12:53 +0000440 { ISD::SRA, MVT::v8i64, 1 },
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000441
Simon Pilgrimd8333372017-01-06 11:12:53 +0000442 { ISD::MUL, MVT::v32i8, 13 }, // extend/pmullw/trunc sequence.
443 { ISD::MUL, MVT::v16i8, 5 }, // extend/pmullw/trunc sequence.
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000444 { ISD::MUL, MVT::v16i32, 1 }, // pmulld (Skylake from agner.org)
445 { ISD::MUL, MVT::v8i32, 1 }, // pmulld (Skylake from agner.org)
446 { ISD::MUL, MVT::v4i32, 1 }, // pmulld (Skylake from agner.org)
Simon Pilgrimd8333372017-01-06 11:12:53 +0000447 { ISD::MUL, MVT::v8i64, 8 }, // 3*pmuludq/3*shift/2*add
448
Simon Pilgrim9929f902018-02-26 22:10:17 +0000449 { ISD::FADD, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
450 { ISD::FSUB, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
451 { ISD::FMUL, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
452
453 { ISD::FADD, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
454 { ISD::FSUB, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
455 { ISD::FMUL, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
Elena Demikhovsky27012472014-09-16 07:57:37 +0000456 };
457
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000458 if (ST->hasAVX512())
Craig Topperee0c8592015-10-27 04:14:24 +0000459 if (const auto *Entry = CostTableLookup(AVX512CostTable, ISD, LT.second))
460 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000461
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000462 static const CostTblEntry AVX2ShiftCostTable[] = {
Michael Liao70dd7f92013-03-20 22:01:10 +0000463 // Shifts on v4i64/v8i32 on AVX2 is legal even though we declare to
464 // customize them to detect the cases where shift amount is a scalar one.
465 { ISD::SHL, MVT::v4i32, 1 },
466 { ISD::SRL, MVT::v4i32, 1 },
467 { ISD::SRA, MVT::v4i32, 1 },
468 { ISD::SHL, MVT::v8i32, 1 },
469 { ISD::SRL, MVT::v8i32, 1 },
470 { ISD::SRA, MVT::v8i32, 1 },
471 { ISD::SHL, MVT::v2i64, 1 },
472 { ISD::SRL, MVT::v2i64, 1 },
473 { ISD::SHL, MVT::v4i64, 1 },
474 { ISD::SRL, MVT::v4i64, 1 },
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000475 };
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000476
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000477 // Look for AVX2 lowering tricks.
478 if (ST->hasAVX2()) {
479 if (ISD == ISD::SHL && LT.second == MVT::v16i16 &&
480 (Op2Info == TargetTransformInfo::OK_UniformConstantValue ||
481 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue))
482 // On AVX2, a packed v16i16 shift left by a constant build_vector
483 // is lowered into a vector multiply (vpmullw).
Simon Pilgrim58e03a02018-04-25 15:22:03 +0000484 return getArithmeticInstrCost(Instruction::Mul, Ty, Op1Info, Op2Info,
485 TargetTransformInfo::OP_None,
486 TargetTransformInfo::OP_None);
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000487
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000488 if (const auto *Entry = CostTableLookup(AVX2ShiftCostTable, ISD, LT.second))
Craig Topperee0c8592015-10-27 04:14:24 +0000489 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000490 }
491
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000492 static const CostTblEntry XOPShiftCostTable[] = {
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000493 // 128bit shifts take 1cy, but right shifts require negation beforehand.
494 { ISD::SHL, MVT::v16i8, 1 },
495 { ISD::SRL, MVT::v16i8, 2 },
496 { ISD::SRA, MVT::v16i8, 2 },
497 { ISD::SHL, MVT::v8i16, 1 },
498 { ISD::SRL, MVT::v8i16, 2 },
499 { ISD::SRA, MVT::v8i16, 2 },
500 { ISD::SHL, MVT::v4i32, 1 },
501 { ISD::SRL, MVT::v4i32, 2 },
502 { ISD::SRA, MVT::v4i32, 2 },
503 { ISD::SHL, MVT::v2i64, 1 },
504 { ISD::SRL, MVT::v2i64, 2 },
505 { ISD::SRA, MVT::v2i64, 2 },
506 // 256bit shifts require splitting if AVX2 didn't catch them above.
Simon Pilgrim4599eaa2017-05-14 13:38:53 +0000507 { ISD::SHL, MVT::v32i8, 2+2 },
508 { ISD::SRL, MVT::v32i8, 4+2 },
509 { ISD::SRA, MVT::v32i8, 4+2 },
510 { ISD::SHL, MVT::v16i16, 2+2 },
511 { ISD::SRL, MVT::v16i16, 4+2 },
512 { ISD::SRA, MVT::v16i16, 4+2 },
513 { ISD::SHL, MVT::v8i32, 2+2 },
514 { ISD::SRL, MVT::v8i32, 4+2 },
515 { ISD::SRA, MVT::v8i32, 4+2 },
516 { ISD::SHL, MVT::v4i64, 2+2 },
517 { ISD::SRL, MVT::v4i64, 4+2 },
518 { ISD::SRA, MVT::v4i64, 4+2 },
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000519 };
520
521 // Look for XOP lowering tricks.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000522 if (ST->hasXOP())
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000523 if (const auto *Entry = CostTableLookup(XOPShiftCostTable, ISD, LT.second))
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000524 return LT.first * Entry->Cost;
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000525
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000526 static const CostTblEntry SSE2UniformShiftCostTable[] = {
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000527 // Uniform splats are cheaper for the following instructions.
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000528 { ISD::SHL, MVT::v16i16, 2+2 }, // 2*psllw + split.
529 { ISD::SHL, MVT::v8i32, 2+2 }, // 2*pslld + split.
530 { ISD::SHL, MVT::v4i64, 2+2 }, // 2*psllq + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000531
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000532 { ISD::SRL, MVT::v16i16, 2+2 }, // 2*psrlw + split.
533 { ISD::SRL, MVT::v8i32, 2+2 }, // 2*psrld + split.
534 { ISD::SRL, MVT::v4i64, 2+2 }, // 2*psrlq + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000535
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000536 { ISD::SRA, MVT::v16i16, 2+2 }, // 2*psraw + split.
537 { ISD::SRA, MVT::v8i32, 2+2 }, // 2*psrad + split.
538 { ISD::SRA, MVT::v2i64, 4 }, // 2*psrad + shuffle.
539 { ISD::SRA, MVT::v4i64, 8+2 }, // 2*(2*psrad + shuffle) + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000540 };
541
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000542 if (ST->hasSSE2() &&
543 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
544 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
Simon Pilgrimf96b4ab2017-05-14 20:25:42 +0000545
546 // Handle AVX2 uniform v4i64 ISD::SRA, it's not worth a table.
547 if (ISD == ISD::SRA && LT.second == MVT::v4i64 && ST->hasAVX2())
548 return LT.first * 4; // 2*psrad + shuffle.
549
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000550 if (const auto *Entry =
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000551 CostTableLookup(SSE2UniformShiftCostTable, ISD, LT.second))
Craig Topperee0c8592015-10-27 04:14:24 +0000552 return LT.first * Entry->Cost;
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000553 }
554
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000555 if (ISD == ISD::SHL &&
556 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue) {
Craig Toppereda02a92015-10-25 03:15:29 +0000557 MVT VT = LT.second;
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000558 // Vector shift left by non uniform constant can be lowered
Simon Pilgrime70644d2017-01-07 21:33:00 +0000559 // into vector multiply.
560 if (((VT == MVT::v8i16 || VT == MVT::v4i32) && ST->hasSSE2()) ||
561 ((VT == MVT::v16i16 || VT == MVT::v8i32) && ST->hasAVX()))
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000562 ISD = ISD::MUL;
563 }
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000564
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000565 static const CostTblEntry AVX2CostTable[] = {
566 { ISD::SHL, MVT::v32i8, 11 }, // vpblendvb sequence.
567 { ISD::SHL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
568
569 { ISD::SRL, MVT::v32i8, 11 }, // vpblendvb sequence.
570 { ISD::SRL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
571
572 { ISD::SRA, MVT::v32i8, 24 }, // vpblendvb sequence.
573 { ISD::SRA, MVT::v16i16, 10 }, // extend/vpsravd/pack sequence.
574 { ISD::SRA, MVT::v2i64, 4 }, // srl/xor/sub sequence.
575 { ISD::SRA, MVT::v4i64, 4 }, // srl/xor/sub sequence.
576
577 { ISD::SUB, MVT::v32i8, 1 }, // psubb
578 { ISD::ADD, MVT::v32i8, 1 }, // paddb
579 { ISD::SUB, MVT::v16i16, 1 }, // psubw
580 { ISD::ADD, MVT::v16i16, 1 }, // paddw
581 { ISD::SUB, MVT::v8i32, 1 }, // psubd
582 { ISD::ADD, MVT::v8i32, 1 }, // paddd
583 { ISD::SUB, MVT::v4i64, 1 }, // psubq
584 { ISD::ADD, MVT::v4i64, 1 }, // paddq
585
586 { ISD::MUL, MVT::v32i8, 17 }, // extend/pmullw/trunc sequence.
587 { ISD::MUL, MVT::v16i8, 7 }, // extend/pmullw/trunc sequence.
588 { ISD::MUL, MVT::v16i16, 1 }, // pmullw
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000589 { ISD::MUL, MVT::v8i32, 2 }, // pmulld (Haswell from agner.org)
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000590 { ISD::MUL, MVT::v4i64, 8 }, // 3*pmuludq/3*shift/2*add
591
Simon Pilgrim9929f902018-02-26 22:10:17 +0000592 { ISD::FADD, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
593 { ISD::FADD, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
594 { ISD::FSUB, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
595 { ISD::FSUB, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
596 { ISD::FMUL, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
597 { ISD::FMUL, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
598
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000599 { ISD::FDIV, MVT::f32, 7 }, // Haswell from http://www.agner.org/
600 { ISD::FDIV, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
601 { ISD::FDIV, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
602 { ISD::FDIV, MVT::f64, 14 }, // Haswell from http://www.agner.org/
603 { ISD::FDIV, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
604 { ISD::FDIV, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
605 };
606
607 // Look for AVX2 lowering tricks for custom cases.
608 if (ST->hasAVX2())
609 if (const auto *Entry = CostTableLookup(AVX2CostTable, ISD, LT.second))
610 return LT.first * Entry->Cost;
611
Simon Pilgrim100eae12017-01-07 17:03:51 +0000612 static const CostTblEntry AVX1CostTable[] = {
613 // We don't have to scalarize unsupported ops. We can issue two half-sized
614 // operations and we only need to extract the upper YMM half.
615 // Two ops + 1 extract + 1 insert = 4.
Simon Pilgrim72599712017-01-07 18:19:25 +0000616 { ISD::MUL, MVT::v16i16, 4 },
617 { ISD::MUL, MVT::v8i32, 4 },
618 { ISD::SUB, MVT::v32i8, 4 },
619 { ISD::ADD, MVT::v32i8, 4 },
620 { ISD::SUB, MVT::v16i16, 4 },
621 { ISD::ADD, MVT::v16i16, 4 },
622 { ISD::SUB, MVT::v8i32, 4 },
623 { ISD::ADD, MVT::v8i32, 4 },
624 { ISD::SUB, MVT::v4i64, 4 },
625 { ISD::ADD, MVT::v4i64, 4 },
Simon Pilgrim100eae12017-01-07 17:03:51 +0000626
627 // A v4i64 multiply is custom lowered as two split v2i64 vectors that then
628 // are lowered as a series of long multiplies(3), shifts(3) and adds(2)
629 // Because we believe v4i64 to be a legal type, we must also include the
630 // extract+insert in the cost table. Therefore, the cost here is 18
631 // instead of 8.
Simon Pilgrim72599712017-01-07 18:19:25 +0000632 { ISD::MUL, MVT::v4i64, 18 },
633
634 { ISD::MUL, MVT::v32i8, 26 }, // extend/pmullw/trunc sequence.
635
636 { ISD::FDIV, MVT::f32, 14 }, // SNB from http://www.agner.org/
637 { ISD::FDIV, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
638 { ISD::FDIV, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
639 { ISD::FDIV, MVT::f64, 22 }, // SNB from http://www.agner.org/
640 { ISD::FDIV, MVT::v2f64, 22 }, // SNB from http://www.agner.org/
641 { ISD::FDIV, MVT::v4f64, 44 }, // SNB from http://www.agner.org/
Simon Pilgrim100eae12017-01-07 17:03:51 +0000642 };
643
Simon Pilgrimdf7de7a2017-01-07 17:27:39 +0000644 if (ST->hasAVX())
Simon Pilgrim100eae12017-01-07 17:03:51 +0000645 if (const auto *Entry = CostTableLookup(AVX1CostTable, ISD, LT.second))
646 return LT.first * Entry->Cost;
647
Simon Pilgrim5b06e4d2017-01-05 19:19:39 +0000648 static const CostTblEntry SSE42CostTable[] = {
Simon Pilgrim9929f902018-02-26 22:10:17 +0000649 { ISD::FADD, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
650 { ISD::FADD, MVT::f32, 1 }, // Nehalem from http://www.agner.org/
651 { ISD::FADD, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
652 { ISD::FADD, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
653
654 { ISD::FSUB, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
655 { ISD::FSUB, MVT::f32 , 1 }, // Nehalem from http://www.agner.org/
656 { ISD::FSUB, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
657 { ISD::FSUB, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
658
659 { ISD::FMUL, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
660 { ISD::FMUL, MVT::f32, 1 }, // Nehalem from http://www.agner.org/
661 { ISD::FMUL, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
662 { ISD::FMUL, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
663
Simon Pilgrim5b06e4d2017-01-05 19:19:39 +0000664 { ISD::FDIV, MVT::f32, 14 }, // Nehalem from http://www.agner.org/
665 { ISD::FDIV, MVT::v4f32, 14 }, // Nehalem from http://www.agner.org/
666 { ISD::FDIV, MVT::f64, 22 }, // Nehalem from http://www.agner.org/
667 { ISD::FDIV, MVT::v2f64, 22 }, // Nehalem from http://www.agner.org/
668 };
669
670 if (ST->hasSSE42())
671 if (const auto *Entry = CostTableLookup(SSE42CostTable, ISD, LT.second))
672 return LT.first * Entry->Cost;
673
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000674 static const CostTblEntry SSE41CostTable[] = {
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000675 { ISD::SHL, MVT::v16i8, 11 }, // pblendvb sequence.
676 { ISD::SHL, MVT::v32i8, 2*11+2 }, // pblendvb sequence + split.
677 { ISD::SHL, MVT::v8i16, 14 }, // pblendvb sequence.
678 { ISD::SHL, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
679 { ISD::SHL, MVT::v4i32, 4 }, // pslld/paddd/cvttps2dq/pmulld
680 { ISD::SHL, MVT::v8i32, 2*4+2 }, // pslld/paddd/cvttps2dq/pmulld + split
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000681
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000682 { ISD::SRL, MVT::v16i8, 12 }, // pblendvb sequence.
683 { ISD::SRL, MVT::v32i8, 2*12+2 }, // pblendvb sequence + split.
684 { ISD::SRL, MVT::v8i16, 14 }, // pblendvb sequence.
685 { ISD::SRL, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
686 { ISD::SRL, MVT::v4i32, 11 }, // Shift each lane + blend.
687 { ISD::SRL, MVT::v8i32, 2*11+2 }, // Shift each lane + blend + split.
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000688
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000689 { ISD::SRA, MVT::v16i8, 24 }, // pblendvb sequence.
690 { ISD::SRA, MVT::v32i8, 2*24+2 }, // pblendvb sequence + split.
691 { ISD::SRA, MVT::v8i16, 14 }, // pblendvb sequence.
692 { ISD::SRA, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
693 { ISD::SRA, MVT::v4i32, 12 }, // Shift each lane + blend.
694 { ISD::SRA, MVT::v8i32, 2*12+2 }, // Shift each lane + blend + split.
Simon Pilgrim4c050c212017-01-05 19:42:43 +0000695
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000696 { ISD::MUL, MVT::v4i32, 2 } // pmulld (Nehalem from agner.org)
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000697 };
698
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000699 if (ST->hasSSE41())
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000700 if (const auto *Entry = CostTableLookup(SSE41CostTable, ISD, LT.second))
701 return LT.first * Entry->Cost;
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000702
Craig Topper4b275762015-10-28 04:02:12 +0000703 static const CostTblEntry SSE2CostTable[] = {
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000704 // We don't correctly identify costs of casts because they are marked as
705 // custom.
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000706 { ISD::SHL, MVT::v16i8, 26 }, // cmpgtb sequence.
707 { ISD::SHL, MVT::v8i16, 32 }, // cmpgtb sequence.
708 { ISD::SHL, MVT::v4i32, 2*5 }, // We optimized this using mul.
709 { ISD::SHL, MVT::v2i64, 4 }, // splat+shuffle sequence.
710 { ISD::SHL, MVT::v4i64, 2*4+2 }, // splat+shuffle sequence + split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000711
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000712 { ISD::SRL, MVT::v16i8, 26 }, // cmpgtb sequence.
713 { ISD::SRL, MVT::v8i16, 32 }, // cmpgtb sequence.
714 { ISD::SRL, MVT::v4i32, 16 }, // Shift each lane + blend.
715 { ISD::SRL, MVT::v2i64, 4 }, // splat+shuffle sequence.
716 { ISD::SRL, MVT::v4i64, 2*4+2 }, // splat+shuffle sequence + split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000717
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000718 { ISD::SRA, MVT::v16i8, 54 }, // unpacked cmpgtb sequence.
719 { ISD::SRA, MVT::v8i16, 32 }, // cmpgtb sequence.
720 { ISD::SRA, MVT::v4i32, 16 }, // Shift each lane + blend.
721 { ISD::SRA, MVT::v2i64, 12 }, // srl/xor/sub sequence.
722 { ISD::SRA, MVT::v4i64, 2*12+2 }, // srl/xor/sub sequence+split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000723
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000724 { ISD::MUL, MVT::v16i8, 12 }, // extend/pmullw/trunc sequence.
725 { ISD::MUL, MVT::v8i16, 1 }, // pmullw
726 { ISD::MUL, MVT::v4i32, 6 }, // 3*pmuludq/4*shuffle
727 { ISD::MUL, MVT::v2i64, 8 }, // 3*pmuludq/3*shift/2*add
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000728
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000729 { ISD::FDIV, MVT::f32, 23 }, // Pentium IV from http://www.agner.org/
730 { ISD::FDIV, MVT::v4f32, 39 }, // Pentium IV from http://www.agner.org/
731 { ISD::FDIV, MVT::f64, 38 }, // Pentium IV from http://www.agner.org/
732 { ISD::FDIV, MVT::v2f64, 69 }, // Pentium IV from http://www.agner.org/
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000733 };
734
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000735 if (ST->hasSSE2())
Craig Topperee0c8592015-10-27 04:14:24 +0000736 if (const auto *Entry = CostTableLookup(SSE2CostTable, ISD, LT.second))
737 return LT.first * Entry->Cost;
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000738
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000739 static const CostTblEntry SSE1CostTable[] = {
Alexey Bataevd07c7312016-10-31 12:10:53 +0000740 { ISD::FDIV, MVT::f32, 17 }, // Pentium III from http://www.agner.org/
741 { ISD::FDIV, MVT::v4f32, 34 }, // Pentium III from http://www.agner.org/
742 };
743
744 if (ST->hasSSE1())
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000745 if (const auto *Entry = CostTableLookup(SSE1CostTable, ISD, LT.second))
Alexey Bataevd07c7312016-10-31 12:10:53 +0000746 return LT.first * Entry->Cost;
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000747
Simon Pilgrim2faf6062018-04-25 20:59:16 +0000748 // It is not a good idea to vectorize division. We have to scalarize it and
749 // in the process we will often end up having to spilling regular
750 // registers. The overhead of division is going to dominate most kernels
751 // anyways so try hard to prevent vectorization of division - it is
752 // generally a bad idea. Assume somewhat arbitrarily that we have to be able
753 // to hide "20 cycles" for each lane.
754 if ((ISD == ISD::SDIV || ISD == ISD::UDIV) && LT.second.isVector()) {
755 int ScalarCost = getArithmeticInstrCost(
756 Opcode, Ty->getScalarType(), Op1Info, Op2Info,
757 TargetTransformInfo::OP_None, TargetTransformInfo::OP_None);
758 return 20 * LT.first * LT.second.getVectorNumElements() * ScalarCost;
759 }
760
Chandler Carruth664e3542013-01-07 01:37:14 +0000761 // Fallback to the default implementation.
Chandler Carruth705b1852015-01-31 03:43:40 +0000762 return BaseT::getArithmeticInstrCost(Opcode, Ty, Op1Info, Op2Info);
Chandler Carruth664e3542013-01-07 01:37:14 +0000763}
764
Chandler Carruth93205eb2015-08-05 18:08:10 +0000765int X86TTIImpl::getShuffleCost(TTI::ShuffleKind Kind, Type *Tp, int Index,
766 Type *SubTp) {
Simon Pilgrima62395a2017-01-05 14:33:32 +0000767 // 64-bit packed float vectors (v2f32) are widened to type v4f32.
768 // 64-bit packed integer vectors (v2i32) are promoted to type v2i64.
769 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Tp);
Karthik Bhate03a25d2014-06-20 04:32:48 +0000770
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000771 // For Broadcasts we are splatting the first element from the first input
772 // register, so only need to reference that input and all the output
773 // registers are the same.
774 if (Kind == TTI::SK_Broadcast)
775 LT.first = 1;
Simon Pilgrimbca02f92017-01-05 15:56:08 +0000776
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000777 // We are going to permute multiple sources and the result will be in multiple
778 // destinations. Providing an accurate cost only for splits where the element
779 // type remains the same.
780 if (Kind == TTI::SK_PermuteSingleSrc && LT.first != 1) {
781 MVT LegalVT = LT.second;
Alexey Bataev771ec9f2018-01-09 19:08:22 +0000782 if (LegalVT.isVector() &&
783 LegalVT.getVectorElementType().getSizeInBits() ==
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000784 Tp->getVectorElementType()->getPrimitiveSizeInBits() &&
785 LegalVT.getVectorNumElements() < Tp->getVectorNumElements()) {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000786
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000787 unsigned VecTySize = DL.getTypeStoreSize(Tp);
788 unsigned LegalVTSize = LegalVT.getStoreSize();
789 // Number of source vectors after legalization:
790 unsigned NumOfSrcs = (VecTySize + LegalVTSize - 1) / LegalVTSize;
791 // Number of destination vectors after legalization:
792 unsigned NumOfDests = LT.first;
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000793
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000794 Type *SingleOpTy = VectorType::get(Tp->getVectorElementType(),
795 LegalVT.getVectorNumElements());
Simon Pilgrimbca02f92017-01-05 15:56:08 +0000796
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000797 unsigned NumOfShuffles = (NumOfSrcs - 1) * NumOfDests;
798 return NumOfShuffles *
799 getShuffleCost(TTI::SK_PermuteTwoSrc, SingleOpTy, 0, nullptr);
800 }
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000801
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000802 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
803 }
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000804
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000805 // For 2-input shuffles, we must account for splitting the 2 inputs into many.
806 if (Kind == TTI::SK_PermuteTwoSrc && LT.first != 1) {
Elena Demikhovsky21706cb2017-01-02 10:37:52 +0000807 // We assume that source and destination have the same vector type.
Elena Demikhovsky21706cb2017-01-02 10:37:52 +0000808 int NumOfDests = LT.first;
809 int NumOfShufflesPerDest = LT.first * 2 - 1;
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000810 LT.first = NumOfDests * NumOfShufflesPerDest;
Karthik Bhate03a25d2014-06-20 04:32:48 +0000811 }
812
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000813 static const CostTblEntry AVX512VBMIShuffleTbl[] = {
814 { TTI::SK_Reverse, MVT::v64i8, 1 }, // vpermb
815 { TTI::SK_Reverse, MVT::v32i8, 1 }, // vpermb
816
817 { TTI::SK_PermuteSingleSrc, MVT::v64i8, 1 }, // vpermb
818 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 1 }, // vpermb
819
820 { TTI::SK_PermuteTwoSrc, MVT::v64i8, 1 }, // vpermt2b
821 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 1 }, // vpermt2b
822 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 1 } // vpermt2b
823 };
824
825 if (ST->hasVBMI())
826 if (const auto *Entry =
827 CostTableLookup(AVX512VBMIShuffleTbl, Kind, LT.second))
828 return LT.first * Entry->Cost;
829
830 static const CostTblEntry AVX512BWShuffleTbl[] = {
831 { TTI::SK_Broadcast, MVT::v32i16, 1 }, // vpbroadcastw
832 { TTI::SK_Broadcast, MVT::v64i8, 1 }, // vpbroadcastb
833
834 { TTI::SK_Reverse, MVT::v32i16, 1 }, // vpermw
835 { TTI::SK_Reverse, MVT::v16i16, 1 }, // vpermw
Simon Pilgrima1b8e2c2017-01-07 15:37:50 +0000836 { TTI::SK_Reverse, MVT::v64i8, 2 }, // pshufb + vshufi64x2
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000837
838 { TTI::SK_PermuteSingleSrc, MVT::v32i16, 1 }, // vpermw
839 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 1 }, // vpermw
840 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 1 }, // vpermw
841 { TTI::SK_PermuteSingleSrc, MVT::v64i8, 8 }, // extend to v32i16
842 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 3 }, // vpermw + zext/trunc
843
844 { TTI::SK_PermuteTwoSrc, MVT::v32i16, 1 }, // vpermt2w
845 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 1 }, // vpermt2w
846 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 1 }, // vpermt2w
847 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 3 }, // zext + vpermt2w + trunc
848 { TTI::SK_PermuteTwoSrc, MVT::v64i8, 19 }, // 6 * v32i8 + 1
849 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 3 } // zext + vpermt2w + trunc
850 };
851
852 if (ST->hasBWI())
853 if (const auto *Entry =
854 CostTableLookup(AVX512BWShuffleTbl, Kind, LT.second))
855 return LT.first * Entry->Cost;
856
857 static const CostTblEntry AVX512ShuffleTbl[] = {
858 { TTI::SK_Broadcast, MVT::v8f64, 1 }, // vbroadcastpd
859 { TTI::SK_Broadcast, MVT::v16f32, 1 }, // vbroadcastps
860 { TTI::SK_Broadcast, MVT::v8i64, 1 }, // vpbroadcastq
861 { TTI::SK_Broadcast, MVT::v16i32, 1 }, // vpbroadcastd
862
863 { TTI::SK_Reverse, MVT::v8f64, 1 }, // vpermpd
864 { TTI::SK_Reverse, MVT::v16f32, 1 }, // vpermps
865 { TTI::SK_Reverse, MVT::v8i64, 1 }, // vpermq
866 { TTI::SK_Reverse, MVT::v16i32, 1 }, // vpermd
867
868 { TTI::SK_PermuteSingleSrc, MVT::v8f64, 1 }, // vpermpd
869 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 1 }, // vpermpd
870 { TTI::SK_PermuteSingleSrc, MVT::v2f64, 1 }, // vpermpd
871 { TTI::SK_PermuteSingleSrc, MVT::v16f32, 1 }, // vpermps
872 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 1 }, // vpermps
873 { TTI::SK_PermuteSingleSrc, MVT::v4f32, 1 }, // vpermps
874 { TTI::SK_PermuteSingleSrc, MVT::v8i64, 1 }, // vpermq
875 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 1 }, // vpermq
876 { TTI::SK_PermuteSingleSrc, MVT::v2i64, 1 }, // vpermq
877 { TTI::SK_PermuteSingleSrc, MVT::v16i32, 1 }, // vpermd
878 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 1 }, // vpermd
879 { TTI::SK_PermuteSingleSrc, MVT::v4i32, 1 }, // vpermd
880 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 1 }, // pshufb
881
882 { TTI::SK_PermuteTwoSrc, MVT::v8f64, 1 }, // vpermt2pd
883 { TTI::SK_PermuteTwoSrc, MVT::v16f32, 1 }, // vpermt2ps
884 { TTI::SK_PermuteTwoSrc, MVT::v8i64, 1 }, // vpermt2q
885 { TTI::SK_PermuteTwoSrc, MVT::v16i32, 1 }, // vpermt2d
886 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 1 }, // vpermt2pd
887 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 1 }, // vpermt2ps
888 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 1 }, // vpermt2q
889 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 1 }, // vpermt2d
890 { TTI::SK_PermuteTwoSrc, MVT::v2f64, 1 }, // vpermt2pd
891 { TTI::SK_PermuteTwoSrc, MVT::v4f32, 1 }, // vpermt2ps
892 { TTI::SK_PermuteTwoSrc, MVT::v2i64, 1 }, // vpermt2q
893 { TTI::SK_PermuteTwoSrc, MVT::v4i32, 1 } // vpermt2d
894 };
895
896 if (ST->hasAVX512())
897 if (const auto *Entry = CostTableLookup(AVX512ShuffleTbl, Kind, LT.second))
898 return LT.first * Entry->Cost;
899
900 static const CostTblEntry AVX2ShuffleTbl[] = {
901 { TTI::SK_Broadcast, MVT::v4f64, 1 }, // vbroadcastpd
902 { TTI::SK_Broadcast, MVT::v8f32, 1 }, // vbroadcastps
903 { TTI::SK_Broadcast, MVT::v4i64, 1 }, // vpbroadcastq
904 { TTI::SK_Broadcast, MVT::v8i32, 1 }, // vpbroadcastd
905 { TTI::SK_Broadcast, MVT::v16i16, 1 }, // vpbroadcastw
906 { TTI::SK_Broadcast, MVT::v32i8, 1 }, // vpbroadcastb
907
908 { TTI::SK_Reverse, MVT::v4f64, 1 }, // vpermpd
909 { TTI::SK_Reverse, MVT::v8f32, 1 }, // vpermps
910 { TTI::SK_Reverse, MVT::v4i64, 1 }, // vpermq
911 { TTI::SK_Reverse, MVT::v8i32, 1 }, // vpermd
912 { TTI::SK_Reverse, MVT::v16i16, 2 }, // vperm2i128 + pshufb
913 { TTI::SK_Reverse, MVT::v32i8, 2 }, // vperm2i128 + pshufb
914
Simon Pilgrime39fa6c2018-06-12 16:12:29 +0000915 { TTI::SK_Select, MVT::v16i16, 1 }, // vpblendvb
916 { TTI::SK_Select, MVT::v32i8, 1 }, // vpblendvb
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000917
Simon Pilgrim702e5fa2017-08-10 17:27:20 +0000918 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 1 }, // vpermpd
919 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 1 }, // vpermps
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000920 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 1 }, // vpermq
921 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 1 }, // vpermd
Simon Pilgrimac2e50a2017-08-10 18:29:34 +0000922 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 4 }, // vperm2i128 + 2*vpshufb
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000923 // + vpblendvb
Simon Pilgrimac2e50a2017-08-10 18:29:34 +0000924 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 4 }, // vperm2i128 + 2*vpshufb
925 // + vpblendvb
926
927 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 3 }, // 2*vpermpd + vblendpd
928 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 3 }, // 2*vpermps + vblendps
929 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 3 }, // 2*vpermq + vpblendd
930 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 3 }, // 2*vpermd + vpblendd
931 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 7 }, // 2*vperm2i128 + 4*vpshufb
932 // + vpblendvb
933 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 7 }, // 2*vperm2i128 + 4*vpshufb
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000934 // + vpblendvb
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000935 };
936
937 if (ST->hasAVX2())
938 if (const auto *Entry = CostTableLookup(AVX2ShuffleTbl, Kind, LT.second))
939 return LT.first * Entry->Cost;
940
Simon Pilgrimc63f93a2017-08-16 13:50:20 +0000941 static const CostTblEntry XOPShuffleTbl[] = {
942 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 2 }, // vperm2f128 + vpermil2pd
943 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 2 }, // vperm2f128 + vpermil2ps
944 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 2 }, // vperm2f128 + vpermil2pd
945 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 2 }, // vperm2f128 + vpermil2ps
946 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 4 }, // vextractf128 + 2*vpperm
947 // + vinsertf128
948 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 4 }, // vextractf128 + 2*vpperm
949 // + vinsertf128
950
951 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 9 }, // 2*vextractf128 + 6*vpperm
952 // + vinsertf128
953 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 1 }, // vpperm
954 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 9 }, // 2*vextractf128 + 6*vpperm
955 // + vinsertf128
956 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 1 }, // vpperm
957 };
958
959 if (ST->hasXOP())
960 if (const auto *Entry = CostTableLookup(XOPShuffleTbl, Kind, LT.second))
961 return LT.first * Entry->Cost;
962
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000963 static const CostTblEntry AVX1ShuffleTbl[] = {
964 { TTI::SK_Broadcast, MVT::v4f64, 2 }, // vperm2f128 + vpermilpd
965 { TTI::SK_Broadcast, MVT::v8f32, 2 }, // vperm2f128 + vpermilps
966 { TTI::SK_Broadcast, MVT::v4i64, 2 }, // vperm2f128 + vpermilpd
967 { TTI::SK_Broadcast, MVT::v8i32, 2 }, // vperm2f128 + vpermilps
968 { TTI::SK_Broadcast, MVT::v16i16, 3 }, // vpshuflw + vpshufd + vinsertf128
969 { TTI::SK_Broadcast, MVT::v32i8, 2 }, // vpshufb + vinsertf128
970
971 { TTI::SK_Reverse, MVT::v4f64, 2 }, // vperm2f128 + vpermilpd
972 { TTI::SK_Reverse, MVT::v8f32, 2 }, // vperm2f128 + vpermilps
973 { TTI::SK_Reverse, MVT::v4i64, 2 }, // vperm2f128 + vpermilpd
974 { TTI::SK_Reverse, MVT::v8i32, 2 }, // vperm2f128 + vpermilps
975 { TTI::SK_Reverse, MVT::v16i16, 4 }, // vextractf128 + 2*pshufb
976 // + vinsertf128
977 { TTI::SK_Reverse, MVT::v32i8, 4 }, // vextractf128 + 2*pshufb
978 // + vinsertf128
979
Simon Pilgrime39fa6c2018-06-12 16:12:29 +0000980 { TTI::SK_Select, MVT::v4i64, 1 }, // vblendpd
981 { TTI::SK_Select, MVT::v4f64, 1 }, // vblendpd
982 { TTI::SK_Select, MVT::v8i32, 1 }, // vblendps
983 { TTI::SK_Select, MVT::v8f32, 1 }, // vblendps
984 { TTI::SK_Select, MVT::v16i16, 3 }, // vpand + vpandn + vpor
985 { TTI::SK_Select, MVT::v32i8, 3 }, // vpand + vpandn + vpor
Simon Pilgrim702e5fa2017-08-10 17:27:20 +0000986
987 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 3 }, // 2*vperm2f128 + vshufpd
988 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 3 }, // 2*vperm2f128 + vshufpd
989 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 4 }, // 2*vperm2f128 + 2*vshufps
990 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 4 }, // 2*vperm2f128 + 2*vshufps
991 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 8 }, // vextractf128 + 4*pshufb
992 // + 2*por + vinsertf128
993 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 8 }, // vextractf128 + 4*pshufb
994 // + 2*por + vinsertf128
Simon Pilgrim73545312017-08-10 19:02:51 +0000995
996 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 4 }, // 2*vperm2f128 + 2*vshufpd
997 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 4 }, // 2*vperm2f128 + 2*vshufps
998 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 4 }, // 2*vperm2f128 + 2*vshufpd
999 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 4 }, // 2*vperm2f128 + 2*vshufps
1000 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 15 }, // 2*vextractf128 + 8*pshufb
1001 // + 4*por + vinsertf128
1002 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 15 }, // 2*vextractf128 + 8*pshufb
1003 // + 4*por + vinsertf128
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001004 };
1005
1006 if (ST->hasAVX())
1007 if (const auto *Entry = CostTableLookup(AVX1ShuffleTbl, Kind, LT.second))
1008 return LT.first * Entry->Cost;
1009
1010 static const CostTblEntry SSE41ShuffleTbl[] = {
Simon Pilgrime39fa6c2018-06-12 16:12:29 +00001011 { TTI::SK_Select, MVT::v2i64, 1 }, // pblendw
1012 { TTI::SK_Select, MVT::v2f64, 1 }, // movsd
1013 { TTI::SK_Select, MVT::v4i32, 1 }, // pblendw
1014 { TTI::SK_Select, MVT::v4f32, 1 }, // blendps
1015 { TTI::SK_Select, MVT::v8i16, 1 }, // pblendw
1016 { TTI::SK_Select, MVT::v16i8, 1 } // pblendvb
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001017 };
1018
1019 if (ST->hasSSE41())
1020 if (const auto *Entry = CostTableLookup(SSE41ShuffleTbl, Kind, LT.second))
1021 return LT.first * Entry->Cost;
1022
1023 static const CostTblEntry SSSE3ShuffleTbl[] = {
1024 { TTI::SK_Broadcast, MVT::v8i16, 1 }, // pshufb
1025 { TTI::SK_Broadcast, MVT::v16i8, 1 }, // pshufb
1026
1027 { TTI::SK_Reverse, MVT::v8i16, 1 }, // pshufb
1028 { TTI::SK_Reverse, MVT::v16i8, 1 }, // pshufb
1029
Simon Pilgrime39fa6c2018-06-12 16:12:29 +00001030 { TTI::SK_Select, MVT::v8i16, 3 }, // 2*pshufb + por
1031 { TTI::SK_Select, MVT::v16i8, 3 }, // 2*pshufb + por
Michael Kupersteine6d59fd2017-02-02 20:27:13 +00001032
1033 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 1 }, // pshufb
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001034 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 1 }, // pshufb
1035
1036 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 3 }, // 2*pshufb + por
1037 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 3 }, // 2*pshufb + por
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001038 };
1039
1040 if (ST->hasSSSE3())
1041 if (const auto *Entry = CostTableLookup(SSSE3ShuffleTbl, Kind, LT.second))
1042 return LT.first * Entry->Cost;
1043
1044 static const CostTblEntry SSE2ShuffleTbl[] = {
1045 { TTI::SK_Broadcast, MVT::v2f64, 1 }, // shufpd
1046 { TTI::SK_Broadcast, MVT::v2i64, 1 }, // pshufd
1047 { TTI::SK_Broadcast, MVT::v4i32, 1 }, // pshufd
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001048 { TTI::SK_Broadcast, MVT::v8i16, 2 }, // pshuflw + pshufd
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001049 { TTI::SK_Broadcast, MVT::v16i8, 3 }, // unpck + pshuflw + pshufd
1050
1051 { TTI::SK_Reverse, MVT::v2f64, 1 }, // shufpd
1052 { TTI::SK_Reverse, MVT::v2i64, 1 }, // pshufd
1053 { TTI::SK_Reverse, MVT::v4i32, 1 }, // pshufd
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001054 { TTI::SK_Reverse, MVT::v8i16, 3 }, // pshuflw + pshufhw + pshufd
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001055 { TTI::SK_Reverse, MVT::v16i8, 9 }, // 2*pshuflw + 2*pshufhw
1056 // + 2*pshufd + 2*unpck + packus
1057
Simon Pilgrime39fa6c2018-06-12 16:12:29 +00001058 { TTI::SK_Select, MVT::v2i64, 1 }, // movsd
1059 { TTI::SK_Select, MVT::v2f64, 1 }, // movsd
1060 { TTI::SK_Select, MVT::v4i32, 2 }, // 2*shufps
1061 { TTI::SK_Select, MVT::v8i16, 3 }, // pand + pandn + por
1062 { TTI::SK_Select, MVT::v16i8, 3 }, // pand + pandn + por
Michael Kupersteine6d59fd2017-02-02 20:27:13 +00001063
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001064 { TTI::SK_PermuteSingleSrc, MVT::v2f64, 1 }, // shufpd
1065 { TTI::SK_PermuteSingleSrc, MVT::v2i64, 1 }, // pshufd
1066 { TTI::SK_PermuteSingleSrc, MVT::v4i32, 1 }, // pshufd
1067 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 5 }, // 2*pshuflw + 2*pshufhw
1068 // + pshufd/unpck
1069 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 10 }, // 2*pshuflw + 2*pshufhw
1070 // + 2*pshufd + 2*unpck + 2*packus
1071
1072 { TTI::SK_PermuteTwoSrc, MVT::v2f64, 1 }, // shufpd
1073 { TTI::SK_PermuteTwoSrc, MVT::v2i64, 1 }, // shufpd
1074 { TTI::SK_PermuteTwoSrc, MVT::v4i32, 2 }, // 2*{unpck,movsd,pshufd}
Simon Pilgrimb59c2d92017-08-10 19:32:35 +00001075 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 8 }, // blend+permute
1076 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 13 }, // blend+permute
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001077 };
1078
1079 if (ST->hasSSE2())
1080 if (const auto *Entry = CostTableLookup(SSE2ShuffleTbl, Kind, LT.second))
1081 return LT.first * Entry->Cost;
1082
1083 static const CostTblEntry SSE1ShuffleTbl[] = {
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001084 { TTI::SK_Broadcast, MVT::v4f32, 1 }, // shufps
1085 { TTI::SK_Reverse, MVT::v4f32, 1 }, // shufps
Simon Pilgrime39fa6c2018-06-12 16:12:29 +00001086 { TTI::SK_Select, MVT::v4f32, 2 }, // 2*shufps
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001087 { TTI::SK_PermuteSingleSrc, MVT::v4f32, 1 }, // shufps
1088 { TTI::SK_PermuteTwoSrc, MVT::v4f32, 2 }, // 2*shufps
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001089 };
1090
1091 if (ST->hasSSE1())
1092 if (const auto *Entry = CostTableLookup(SSE1ShuffleTbl, Kind, LT.second))
1093 return LT.first * Entry->Cost;
1094
Chandler Carruth705b1852015-01-31 03:43:40 +00001095 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
Chandler Carruth664e3542013-01-07 01:37:14 +00001096}
1097
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001098int X86TTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src,
1099 const Instruction *I) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001100 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1101 assert(ISD && "Invalid opcode");
1102
Cong Hou59898d82015-12-11 00:31:39 +00001103 // FIXME: Need a better design of the cost table to handle non-simple types of
1104 // potential massive combinations (elem_num x src_type x dst_type).
1105
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001106 static const TypeConversionCostTblEntry AVX512DQConversionTbl[] = {
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001107 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
1108 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001109 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
1110 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001111 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
1112 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
1113
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001114 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001115 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001116 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001117 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001118 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001119 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001120
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001121 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f32, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001122 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f32, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001123 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f32, 1 },
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001124 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001125 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001126 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f64, 1 },
1127
1128 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f32, 1 },
1129 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f32, 1 },
1130 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f32, 1 },
1131 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f64, 1 },
1132 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f64, 1 },
1133 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001134 };
1135
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001136 // TODO: For AVX512DQ + AVX512VL, we also have cheap casts for 128-bit and
1137 // 256-bit wide vectors.
1138
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001139 static const TypeConversionCostTblEntry AVX512FConversionTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +00001140 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 1 },
1141 { ISD::FP_EXTEND, MVT::v8f64, MVT::v16f32, 3 },
1142 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001143
1144 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 1 },
1145 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 1 },
1146 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i64, 1 },
1147 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001148
1149 // v16i1 -> v16i32 - load + broadcast
1150 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
1151 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001152 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
1153 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
1154 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
1155 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001156 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
1157 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001158 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
1159 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001160
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001161 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001162 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001163 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001164 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001165 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001166 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
1167 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001168 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001169 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 26 },
1170 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001171
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001172 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001173 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001174 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001175 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
1176 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 2 },
1177 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
1178 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001179 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i16, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001180 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
1181 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 2 },
1182 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
1183 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001184 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i32, 2 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001185 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001186 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
1187 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
1188 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
1189 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
1190 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001191 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 5 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001192 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 5 },
1193 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 12 },
1194 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
1195
1196 { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f32, 1 },
1197 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f32, 1 },
1198 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 1 },
Zvi Rackover25799d92017-09-07 07:40:34 +00001199 { ISD::FP_TO_UINT, MVT::v8i16, MVT::v8f64, 2 },
1200 { ISD::FP_TO_UINT, MVT::v8i8, MVT::v8f64, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001201 { ISD::FP_TO_UINT, MVT::v16i32, MVT::v16f32, 1 },
Zvi Rackover25799d92017-09-07 07:40:34 +00001202 { ISD::FP_TO_UINT, MVT::v16i16, MVT::v16f32, 2 },
1203 { ISD::FP_TO_UINT, MVT::v16i8, MVT::v16f32, 2 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001204 };
1205
Craig Topper4b275762015-10-28 04:02:12 +00001206 static const TypeConversionCostTblEntry AVX2ConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +00001207 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
1208 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001209 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
1210 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001211 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
1212 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001213 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
1214 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
1215 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
1216 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001217 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
1218 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001219 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
1220 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001221 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
1222 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
1223
1224 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 2 },
1225 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 2 },
1226 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 2 },
1227 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 2 },
1228 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 2 },
1229 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 4 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001230
1231 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 3 },
1232 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 3 },
Quentin Colombet360460b2014-11-11 02:23:47 +00001233
1234 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 8 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001235 };
1236
Craig Topper4b275762015-10-28 04:02:12 +00001237 static const TypeConversionCostTblEntry AVXConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +00001238 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 6 },
1239 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001240 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 7 },
1241 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001242 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 6 },
1243 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001244 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 7 },
1245 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 4 },
1246 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
1247 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001248 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 6 },
1249 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001250 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
1251 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001252 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
1253 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
1254
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001255 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 4 },
1256 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
1257 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001258 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 4 },
1259 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 4 },
1260 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001261 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 9 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001262
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001263 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i1, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001264 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001265 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i1, 8 },
1266 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i8, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001267 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001268 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i8, 8 },
1269 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i16, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001270 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001271 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
1272 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001273 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001274 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001275
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001276 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i1, 7 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001277 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i1, 7 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001278 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i1, 6 },
1279 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i8, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001280 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001281 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 5 },
1282 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001283 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001284 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001285 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001286 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 6 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001287 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001288 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 9 },
Quentin Colombet85b904d2014-03-27 22:27:41 +00001289 // The generic code to compute the scalar overhead is currently broken.
1290 // Workaround this limitation by estimating the scalarization overhead
1291 // here. We have roughly 10 instructions per scalar element.
1292 // Multiply that by the vector width.
1293 // FIXME: remove that when PR19268 is fixed.
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001294 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 10 },
1295 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 20 },
1296 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
1297 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001298
Renato Goline1fb0592013-01-20 20:57:20 +00001299 { ISD::FP_TO_SINT, MVT::v4i8, MVT::v4f32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001300 { ISD::FP_TO_SINT, MVT::v8i8, MVT::v8f32, 7 },
Adam Nemet6dafe972014-03-30 18:07:13 +00001301 // This node is expanded into scalarized operations but BasicTTI is overly
1302 // optimistic estimating its cost. It computes 3 per element (one
1303 // vector-extract, one scalar conversion and one vector-insert). The
1304 // problem is that the inserts form a read-modify-write chain so latency
1305 // should be factored in too. Inflating the cost per element by 1.
1306 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 8*4 },
Adam Nemet10c4ce22014-03-31 21:54:48 +00001307 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f64, 4*4 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001308
1309 { ISD::FP_EXTEND, MVT::v4f64, MVT::v4f32, 1 },
1310 { ISD::FP_ROUND, MVT::v4f32, MVT::v4f64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001311 };
1312
Cong Hou59898d82015-12-11 00:31:39 +00001313 static const TypeConversionCostTblEntry SSE41ConversionTbl[] = {
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001314 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
1315 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001316 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1317 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1318 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
1319 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001320
Cong Hou59898d82015-12-11 00:31:39 +00001321 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1322 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001323 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1324 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1325 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1326 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1327 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1328 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1329 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1330 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1331 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1332 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1333 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1334 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1335 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1336 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1337 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
1338 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
Cong Hou59898d82015-12-11 00:31:39 +00001339
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001340 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 2 },
1341 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 1 },
1342 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001343 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001344 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 3 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001345 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001346 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 6 },
1347
Cong Hou59898d82015-12-11 00:31:39 +00001348 };
1349
1350 static const TypeConversionCostTblEntry SSE2ConversionTbl[] = {
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001351 // These are somewhat magic numbers justified by looking at the output of
1352 // Intel's IACA, running some kernels and making sure when we take
1353 // legalization into account the throughput will be overestimated.
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001354 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001355 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1356 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1357 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
Sanjay Patel04b34962016-07-06 19:15:54 +00001358 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001359 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1360 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
1361 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
Cong Hou59898d82015-12-11 00:31:39 +00001362
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001363 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1364 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
1365 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1366 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
1367 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1368 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 8 },
1369 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
1370 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001371
Simon Pilgrim4ddc92b2016-10-18 07:42:15 +00001372 { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f64, 3 },
1373
Cong Hou59898d82015-12-11 00:31:39 +00001374 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1375 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001376 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 2 },
1377 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 3 },
1378 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
1379 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 8 },
1380 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1381 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 2 },
1382 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1383 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1384 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 3 },
1385 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
1386 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 9 },
1387 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 12 },
1388 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1389 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 2 },
1390 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
1391 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 10 },
1392 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 3 },
1393 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
1394 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 6 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001395 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001396 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 3 },
1397 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 5 },
Cong Hou59898d82015-12-11 00:31:39 +00001398
Cong Hou59898d82015-12-11 00:31:39 +00001399 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001400 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 2 },
1401 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 3 },
1402 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 3 },
1403 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 3 },
1404 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
1405 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 7 },
1406 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
1407 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 10 },
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001408 };
1409
Chandler Carruth93205eb2015-08-05 18:08:10 +00001410 std::pair<int, MVT> LTSrc = TLI->getTypeLegalizationCost(DL, Src);
1411 std::pair<int, MVT> LTDest = TLI->getTypeLegalizationCost(DL, Dst);
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001412
1413 if (ST->hasSSE2() && !ST->hasAVX()) {
Cong Hou59898d82015-12-11 00:31:39 +00001414 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
Craig Topperee0c8592015-10-27 04:14:24 +00001415 LTDest.second, LTSrc.second))
1416 return LTSrc.first * Entry->Cost;
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001417 }
1418
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001419 EVT SrcTy = TLI->getValueType(DL, Src);
1420 EVT DstTy = TLI->getValueType(DL, Dst);
1421
1422 // The function getSimpleVT only handles simple value types.
1423 if (!SrcTy.isSimple() || !DstTy.isSimple())
1424 return BaseT::getCastInstrCost(Opcode, Dst, Src);
1425
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001426 if (ST->hasDQI())
1427 if (const auto *Entry = ConvertCostTableLookup(AVX512DQConversionTbl, ISD,
1428 DstTy.getSimpleVT(),
1429 SrcTy.getSimpleVT()))
1430 return Entry->Cost;
1431
1432 if (ST->hasAVX512())
1433 if (const auto *Entry = ConvertCostTableLookup(AVX512FConversionTbl, ISD,
1434 DstTy.getSimpleVT(),
1435 SrcTy.getSimpleVT()))
1436 return Entry->Cost;
1437
Tim Northoverf0e21612014-02-06 18:18:36 +00001438 if (ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001439 if (const auto *Entry = ConvertCostTableLookup(AVX2ConversionTbl, ISD,
1440 DstTy.getSimpleVT(),
1441 SrcTy.getSimpleVT()))
1442 return Entry->Cost;
Tim Northoverf0e21612014-02-06 18:18:36 +00001443 }
1444
Chandler Carruth664e3542013-01-07 01:37:14 +00001445 if (ST->hasAVX()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001446 if (const auto *Entry = ConvertCostTableLookup(AVXConversionTbl, ISD,
1447 DstTy.getSimpleVT(),
1448 SrcTy.getSimpleVT()))
1449 return Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001450 }
1451
Cong Hou59898d82015-12-11 00:31:39 +00001452 if (ST->hasSSE41()) {
1453 if (const auto *Entry = ConvertCostTableLookup(SSE41ConversionTbl, ISD,
1454 DstTy.getSimpleVT(),
1455 SrcTy.getSimpleVT()))
1456 return Entry->Cost;
1457 }
1458
1459 if (ST->hasSSE2()) {
1460 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
1461 DstTy.getSimpleVT(),
1462 SrcTy.getSimpleVT()))
1463 return Entry->Cost;
1464 }
1465
Alexey Bataeve25a6fd2017-11-07 14:23:44 +00001466 return BaseT::getCastInstrCost(Opcode, Dst, Src, I);
Chandler Carruth664e3542013-01-07 01:37:14 +00001467}
1468
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001469int X86TTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy, Type *CondTy,
1470 const Instruction *I) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001471 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001472 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Chandler Carruth664e3542013-01-07 01:37:14 +00001473
1474 MVT MTy = LT.second;
1475
1476 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1477 assert(ISD && "Invalid opcode");
1478
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001479 static const CostTblEntry SSE2CostTbl[] = {
1480 { ISD::SETCC, MVT::v2i64, 8 },
1481 { ISD::SETCC, MVT::v4i32, 1 },
1482 { ISD::SETCC, MVT::v8i16, 1 },
1483 { ISD::SETCC, MVT::v16i8, 1 },
1484 };
1485
Craig Topper4b275762015-10-28 04:02:12 +00001486 static const CostTblEntry SSE42CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001487 { ISD::SETCC, MVT::v2f64, 1 },
1488 { ISD::SETCC, MVT::v4f32, 1 },
1489 { ISD::SETCC, MVT::v2i64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001490 };
1491
Craig Topper4b275762015-10-28 04:02:12 +00001492 static const CostTblEntry AVX1CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001493 { ISD::SETCC, MVT::v4f64, 1 },
1494 { ISD::SETCC, MVT::v8f32, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001495 // AVX1 does not support 8-wide integer compare.
Renato Goline1fb0592013-01-20 20:57:20 +00001496 { ISD::SETCC, MVT::v4i64, 4 },
1497 { ISD::SETCC, MVT::v8i32, 4 },
1498 { ISD::SETCC, MVT::v16i16, 4 },
1499 { ISD::SETCC, MVT::v32i8, 4 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001500 };
1501
Craig Topper4b275762015-10-28 04:02:12 +00001502 static const CostTblEntry AVX2CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001503 { ISD::SETCC, MVT::v4i64, 1 },
1504 { ISD::SETCC, MVT::v8i32, 1 },
1505 { ISD::SETCC, MVT::v16i16, 1 },
1506 { ISD::SETCC, MVT::v32i8, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001507 };
1508
Craig Topper4b275762015-10-28 04:02:12 +00001509 static const CostTblEntry AVX512CostTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +00001510 { ISD::SETCC, MVT::v8i64, 1 },
1511 { ISD::SETCC, MVT::v16i32, 1 },
1512 { ISD::SETCC, MVT::v8f64, 1 },
1513 { ISD::SETCC, MVT::v16f32, 1 },
1514 };
1515
Simon Pilgrim80ce1dd2018-04-07 13:24:33 +00001516 static const CostTblEntry AVX512BWCostTbl[] = {
1517 { ISD::SETCC, MVT::v32i16, 1 },
1518 { ISD::SETCC, MVT::v64i8, 1 },
1519 };
1520
1521 if (ST->hasBWI())
1522 if (const auto *Entry = CostTableLookup(AVX512BWCostTbl, ISD, MTy))
1523 return LT.first * Entry->Cost;
1524
Craig Topperee0c8592015-10-27 04:14:24 +00001525 if (ST->hasAVX512())
1526 if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
1527 return LT.first * Entry->Cost;
Elena Demikhovsky27012472014-09-16 07:57:37 +00001528
Craig Topperee0c8592015-10-27 04:14:24 +00001529 if (ST->hasAVX2())
1530 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1531 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001532
Craig Topperee0c8592015-10-27 04:14:24 +00001533 if (ST->hasAVX())
1534 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1535 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001536
Craig Topperee0c8592015-10-27 04:14:24 +00001537 if (ST->hasSSE42())
1538 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1539 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001540
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001541 if (ST->hasSSE2())
1542 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1543 return LT.first * Entry->Cost;
1544
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001545 return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy, I);
Chandler Carruth664e3542013-01-07 01:37:14 +00001546}
1547
Anna Thomasb2a212c2017-06-06 16:45:25 +00001548unsigned X86TTIImpl::getAtomicMemIntrinsicMaxElementSize() const { return 16; }
1549
Simon Pilgrim14000b32016-05-24 08:17:50 +00001550int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
Jonas Paulssona48ea232017-03-14 06:35:36 +00001551 ArrayRef<Type *> Tys, FastMathFlags FMF,
1552 unsigned ScalarizationCostPassed) {
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001553 // Costs should match the codegen from:
1554 // BITREVERSE: llvm\test\CodeGen\X86\vector-bitreverse.ll
1555 // BSWAP: llvm\test\CodeGen\X86\bswap-vector.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001556 // CTLZ: llvm\test\CodeGen\X86\vector-lzcnt-*.ll
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001557 // CTPOP: llvm\test\CodeGen\X86\vector-popcnt-*.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001558 // CTTZ: llvm\test\CodeGen\X86\vector-tzcnt-*.ll
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001559 static const CostTblEntry AVX512CDCostTbl[] = {
1560 { ISD::CTLZ, MVT::v8i64, 1 },
1561 { ISD::CTLZ, MVT::v16i32, 1 },
1562 { ISD::CTLZ, MVT::v32i16, 8 },
1563 { ISD::CTLZ, MVT::v64i8, 20 },
1564 { ISD::CTLZ, MVT::v4i64, 1 },
1565 { ISD::CTLZ, MVT::v8i32, 1 },
1566 { ISD::CTLZ, MVT::v16i16, 4 },
1567 { ISD::CTLZ, MVT::v32i8, 10 },
1568 { ISD::CTLZ, MVT::v2i64, 1 },
1569 { ISD::CTLZ, MVT::v4i32, 1 },
1570 { ISD::CTLZ, MVT::v8i16, 4 },
1571 { ISD::CTLZ, MVT::v16i8, 4 },
1572 };
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001573 static const CostTblEntry AVX512BWCostTbl[] = {
1574 { ISD::BITREVERSE, MVT::v8i64, 5 },
1575 { ISD::BITREVERSE, MVT::v16i32, 5 },
1576 { ISD::BITREVERSE, MVT::v32i16, 5 },
1577 { ISD::BITREVERSE, MVT::v64i8, 5 },
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001578 { ISD::CTLZ, MVT::v8i64, 23 },
1579 { ISD::CTLZ, MVT::v16i32, 22 },
1580 { ISD::CTLZ, MVT::v32i16, 18 },
1581 { ISD::CTLZ, MVT::v64i8, 17 },
Simon Pilgrim6bba6062017-05-18 10:42:34 +00001582 { ISD::CTPOP, MVT::v8i64, 7 },
1583 { ISD::CTPOP, MVT::v16i32, 11 },
1584 { ISD::CTPOP, MVT::v32i16, 9 },
1585 { ISD::CTPOP, MVT::v64i8, 6 },
Simon Pilgrimd0365962017-05-17 20:22:54 +00001586 { ISD::CTTZ, MVT::v8i64, 10 },
1587 { ISD::CTTZ, MVT::v16i32, 14 },
1588 { ISD::CTTZ, MVT::v32i16, 12 },
1589 { ISD::CTTZ, MVT::v64i8, 9 },
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001590 };
1591 static const CostTblEntry AVX512CostTbl[] = {
1592 { ISD::BITREVERSE, MVT::v8i64, 36 },
1593 { ISD::BITREVERSE, MVT::v16i32, 24 },
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001594 { ISD::CTLZ, MVT::v8i64, 29 },
1595 { ISD::CTLZ, MVT::v16i32, 35 },
Simon Pilgrim6bba6062017-05-18 10:42:34 +00001596 { ISD::CTPOP, MVT::v8i64, 16 },
1597 { ISD::CTPOP, MVT::v16i32, 24 },
Simon Pilgrimd0365962017-05-17 20:22:54 +00001598 { ISD::CTTZ, MVT::v8i64, 20 },
1599 { ISD::CTTZ, MVT::v16i32, 28 },
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001600 };
Simon Pilgrim14000b32016-05-24 08:17:50 +00001601 static const CostTblEntry XOPCostTbl[] = {
1602 { ISD::BITREVERSE, MVT::v4i64, 4 },
1603 { ISD::BITREVERSE, MVT::v8i32, 4 },
1604 { ISD::BITREVERSE, MVT::v16i16, 4 },
1605 { ISD::BITREVERSE, MVT::v32i8, 4 },
1606 { ISD::BITREVERSE, MVT::v2i64, 1 },
1607 { ISD::BITREVERSE, MVT::v4i32, 1 },
1608 { ISD::BITREVERSE, MVT::v8i16, 1 },
1609 { ISD::BITREVERSE, MVT::v16i8, 1 },
1610 { ISD::BITREVERSE, MVT::i64, 3 },
1611 { ISD::BITREVERSE, MVT::i32, 3 },
1612 { ISD::BITREVERSE, MVT::i16, 3 },
1613 { ISD::BITREVERSE, MVT::i8, 3 }
1614 };
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001615 static const CostTblEntry AVX2CostTbl[] = {
1616 { ISD::BITREVERSE, MVT::v4i64, 5 },
1617 { ISD::BITREVERSE, MVT::v8i32, 5 },
1618 { ISD::BITREVERSE, MVT::v16i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001619 { ISD::BITREVERSE, MVT::v32i8, 5 },
1620 { ISD::BSWAP, MVT::v4i64, 1 },
1621 { ISD::BSWAP, MVT::v8i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001622 { ISD::BSWAP, MVT::v16i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001623 { ISD::CTLZ, MVT::v4i64, 23 },
1624 { ISD::CTLZ, MVT::v8i32, 18 },
1625 { ISD::CTLZ, MVT::v16i16, 14 },
1626 { ISD::CTLZ, MVT::v32i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001627 { ISD::CTPOP, MVT::v4i64, 7 },
1628 { ISD::CTPOP, MVT::v8i32, 11 },
1629 { ISD::CTPOP, MVT::v16i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001630 { ISD::CTPOP, MVT::v32i8, 6 },
1631 { ISD::CTTZ, MVT::v4i64, 10 },
1632 { ISD::CTTZ, MVT::v8i32, 14 },
1633 { ISD::CTTZ, MVT::v16i16, 12 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001634 { ISD::CTTZ, MVT::v32i8, 9 },
1635 { ISD::FSQRT, MVT::f32, 7 }, // Haswell from http://www.agner.org/
1636 { ISD::FSQRT, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
1637 { ISD::FSQRT, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
1638 { ISD::FSQRT, MVT::f64, 14 }, // Haswell from http://www.agner.org/
1639 { ISD::FSQRT, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
1640 { ISD::FSQRT, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001641 };
1642 static const CostTblEntry AVX1CostTbl[] = {
Simon Pilgrim2d1c6d62017-05-07 20:58:55 +00001643 { ISD::BITREVERSE, MVT::v4i64, 12 }, // 2 x 128-bit Op + extract/insert
1644 { ISD::BITREVERSE, MVT::v8i32, 12 }, // 2 x 128-bit Op + extract/insert
1645 { ISD::BITREVERSE, MVT::v16i16, 12 }, // 2 x 128-bit Op + extract/insert
1646 { ISD::BITREVERSE, MVT::v32i8, 12 }, // 2 x 128-bit Op + extract/insert
Simon Pilgrim356e8232016-06-20 23:08:21 +00001647 { ISD::BSWAP, MVT::v4i64, 4 },
1648 { ISD::BSWAP, MVT::v8i32, 4 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001649 { ISD::BSWAP, MVT::v16i16, 4 },
Simon Pilgrim2d1c6d62017-05-07 20:58:55 +00001650 { ISD::CTLZ, MVT::v4i64, 48 }, // 2 x 128-bit Op + extract/insert
1651 { ISD::CTLZ, MVT::v8i32, 38 }, // 2 x 128-bit Op + extract/insert
1652 { ISD::CTLZ, MVT::v16i16, 30 }, // 2 x 128-bit Op + extract/insert
1653 { ISD::CTLZ, MVT::v32i8, 20 }, // 2 x 128-bit Op + extract/insert
1654 { ISD::CTPOP, MVT::v4i64, 16 }, // 2 x 128-bit Op + extract/insert
1655 { ISD::CTPOP, MVT::v8i32, 24 }, // 2 x 128-bit Op + extract/insert
1656 { ISD::CTPOP, MVT::v16i16, 20 }, // 2 x 128-bit Op + extract/insert
1657 { ISD::CTPOP, MVT::v32i8, 14 }, // 2 x 128-bit Op + extract/insert
1658 { ISD::CTTZ, MVT::v4i64, 22 }, // 2 x 128-bit Op + extract/insert
1659 { ISD::CTTZ, MVT::v8i32, 30 }, // 2 x 128-bit Op + extract/insert
1660 { ISD::CTTZ, MVT::v16i16, 26 }, // 2 x 128-bit Op + extract/insert
1661 { ISD::CTTZ, MVT::v32i8, 20 }, // 2 x 128-bit Op + extract/insert
Alexey Bataevd07c7312016-10-31 12:10:53 +00001662 { ISD::FSQRT, MVT::f32, 14 }, // SNB from http://www.agner.org/
1663 { ISD::FSQRT, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
1664 { ISD::FSQRT, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
1665 { ISD::FSQRT, MVT::f64, 21 }, // SNB from http://www.agner.org/
1666 { ISD::FSQRT, MVT::v2f64, 21 }, // SNB from http://www.agner.org/
1667 { ISD::FSQRT, MVT::v4f64, 43 }, // SNB from http://www.agner.org/
1668 };
Craig Toppera9859192018-03-25 15:58:12 +00001669 static const CostTblEntry GLMCostTbl[] = {
1670 { ISD::FSQRT, MVT::f32, 19 }, // sqrtss
1671 { ISD::FSQRT, MVT::v4f32, 37 }, // sqrtps
1672 { ISD::FSQRT, MVT::f64, 34 }, // sqrtsd
1673 { ISD::FSQRT, MVT::v2f64, 67 }, // sqrtpd
1674 };
1675 static const CostTblEntry SLMCostTbl[] = {
1676 { ISD::FSQRT, MVT::f32, 20 }, // sqrtss
1677 { ISD::FSQRT, MVT::v4f32, 40 }, // sqrtps
1678 { ISD::FSQRT, MVT::f64, 35 }, // sqrtsd
1679 { ISD::FSQRT, MVT::v2f64, 70 }, // sqrtpd
1680 };
Alexey Bataevd07c7312016-10-31 12:10:53 +00001681 static const CostTblEntry SSE42CostTbl[] = {
Simon Pilgrima0b0b742017-03-15 11:57:42 +00001682 { ISD::FSQRT, MVT::f32, 18 }, // Nehalem from http://www.agner.org/
1683 { ISD::FSQRT, MVT::v4f32, 18 }, // Nehalem from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001684 };
1685 static const CostTblEntry SSSE3CostTbl[] = {
1686 { ISD::BITREVERSE, MVT::v2i64, 5 },
1687 { ISD::BITREVERSE, MVT::v4i32, 5 },
1688 { ISD::BITREVERSE, MVT::v8i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001689 { ISD::BITREVERSE, MVT::v16i8, 5 },
1690 { ISD::BSWAP, MVT::v2i64, 1 },
1691 { ISD::BSWAP, MVT::v4i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001692 { ISD::BSWAP, MVT::v8i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001693 { ISD::CTLZ, MVT::v2i64, 23 },
1694 { ISD::CTLZ, MVT::v4i32, 18 },
1695 { ISD::CTLZ, MVT::v8i16, 14 },
1696 { ISD::CTLZ, MVT::v16i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001697 { ISD::CTPOP, MVT::v2i64, 7 },
1698 { ISD::CTPOP, MVT::v4i32, 11 },
1699 { ISD::CTPOP, MVT::v8i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001700 { ISD::CTPOP, MVT::v16i8, 6 },
1701 { ISD::CTTZ, MVT::v2i64, 10 },
1702 { ISD::CTTZ, MVT::v4i32, 14 },
1703 { ISD::CTTZ, MVT::v8i16, 12 },
1704 { ISD::CTTZ, MVT::v16i8, 9 }
Simon Pilgrim356e8232016-06-20 23:08:21 +00001705 };
1706 static const CostTblEntry SSE2CostTbl[] = {
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001707 { ISD::BITREVERSE, MVT::v2i64, 29 },
1708 { ISD::BITREVERSE, MVT::v4i32, 27 },
1709 { ISD::BITREVERSE, MVT::v8i16, 27 },
1710 { ISD::BITREVERSE, MVT::v16i8, 20 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001711 { ISD::BSWAP, MVT::v2i64, 7 },
1712 { ISD::BSWAP, MVT::v4i32, 7 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001713 { ISD::BSWAP, MVT::v8i16, 7 },
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001714 { ISD::CTLZ, MVT::v2i64, 25 },
1715 { ISD::CTLZ, MVT::v4i32, 26 },
1716 { ISD::CTLZ, MVT::v8i16, 20 },
1717 { ISD::CTLZ, MVT::v16i8, 17 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001718 { ISD::CTPOP, MVT::v2i64, 12 },
1719 { ISD::CTPOP, MVT::v4i32, 15 },
1720 { ISD::CTPOP, MVT::v8i16, 13 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001721 { ISD::CTPOP, MVT::v16i8, 10 },
1722 { ISD::CTTZ, MVT::v2i64, 14 },
1723 { ISD::CTTZ, MVT::v4i32, 18 },
1724 { ISD::CTTZ, MVT::v8i16, 16 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001725 { ISD::CTTZ, MVT::v16i8, 13 },
1726 { ISD::FSQRT, MVT::f64, 32 }, // Nehalem from http://www.agner.org/
1727 { ISD::FSQRT, MVT::v2f64, 32 }, // Nehalem from http://www.agner.org/
1728 };
1729 static const CostTblEntry SSE1CostTbl[] = {
Simon Pilgrima0b0b742017-03-15 11:57:42 +00001730 { ISD::FSQRT, MVT::f32, 28 }, // Pentium III from http://www.agner.org/
1731 { ISD::FSQRT, MVT::v4f32, 56 }, // Pentium III from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001732 };
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001733 static const CostTblEntry X64CostTbl[] = { // 64-bit targets
1734 { ISD::BITREVERSE, MVT::i64, 14 }
1735 };
1736 static const CostTblEntry X86CostTbl[] = { // 32 or 64-bit targets
1737 { ISD::BITREVERSE, MVT::i32, 14 },
1738 { ISD::BITREVERSE, MVT::i16, 14 },
1739 { ISD::BITREVERSE, MVT::i8, 11 }
1740 };
Simon Pilgrim14000b32016-05-24 08:17:50 +00001741
1742 unsigned ISD = ISD::DELETED_NODE;
1743 switch (IID) {
1744 default:
1745 break;
1746 case Intrinsic::bitreverse:
1747 ISD = ISD::BITREVERSE;
1748 break;
Simon Pilgrim356e8232016-06-20 23:08:21 +00001749 case Intrinsic::bswap:
1750 ISD = ISD::BSWAP;
1751 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001752 case Intrinsic::ctlz:
1753 ISD = ISD::CTLZ;
1754 break;
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001755 case Intrinsic::ctpop:
1756 ISD = ISD::CTPOP;
1757 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001758 case Intrinsic::cttz:
1759 ISD = ISD::CTTZ;
1760 break;
Alexey Bataevd07c7312016-10-31 12:10:53 +00001761 case Intrinsic::sqrt:
1762 ISD = ISD::FSQRT;
1763 break;
Simon Pilgrim14000b32016-05-24 08:17:50 +00001764 }
1765
1766 // Legalize the type.
1767 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, RetTy);
1768 MVT MTy = LT.second;
1769
1770 // Attempt to lookup cost.
Craig Toppera9859192018-03-25 15:58:12 +00001771 if (ST->isGLM())
1772 if (const auto *Entry = CostTableLookup(GLMCostTbl, ISD, MTy))
1773 return LT.first * Entry->Cost;
1774
1775 if (ST->isSLM())
1776 if (const auto *Entry = CostTableLookup(SLMCostTbl, ISD, MTy))
1777 return LT.first * Entry->Cost;
1778
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001779 if (ST->hasCDI())
1780 if (const auto *Entry = CostTableLookup(AVX512CDCostTbl, ISD, MTy))
1781 return LT.first * Entry->Cost;
1782
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001783 if (ST->hasBWI())
1784 if (const auto *Entry = CostTableLookup(AVX512BWCostTbl, ISD, MTy))
1785 return LT.first * Entry->Cost;
1786
1787 if (ST->hasAVX512())
1788 if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
1789 return LT.first * Entry->Cost;
1790
Simon Pilgrim14000b32016-05-24 08:17:50 +00001791 if (ST->hasXOP())
1792 if (const auto *Entry = CostTableLookup(XOPCostTbl, ISD, MTy))
1793 return LT.first * Entry->Cost;
1794
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001795 if (ST->hasAVX2())
1796 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1797 return LT.first * Entry->Cost;
1798
1799 if (ST->hasAVX())
1800 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1801 return LT.first * Entry->Cost;
1802
Alexey Bataevd07c7312016-10-31 12:10:53 +00001803 if (ST->hasSSE42())
1804 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1805 return LT.first * Entry->Cost;
1806
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001807 if (ST->hasSSSE3())
1808 if (const auto *Entry = CostTableLookup(SSSE3CostTbl, ISD, MTy))
1809 return LT.first * Entry->Cost;
1810
Simon Pilgrim356e8232016-06-20 23:08:21 +00001811 if (ST->hasSSE2())
1812 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1813 return LT.first * Entry->Cost;
1814
Alexey Bataevd07c7312016-10-31 12:10:53 +00001815 if (ST->hasSSE1())
1816 if (const auto *Entry = CostTableLookup(SSE1CostTbl, ISD, MTy))
1817 return LT.first * Entry->Cost;
1818
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001819 if (ST->is64Bit())
1820 if (const auto *Entry = CostTableLookup(X64CostTbl, ISD, MTy))
1821 return LT.first * Entry->Cost;
1822
1823 if (const auto *Entry = CostTableLookup(X86CostTbl, ISD, MTy))
1824 return LT.first * Entry->Cost;
1825
Jonas Paulssona48ea232017-03-14 06:35:36 +00001826 return BaseT::getIntrinsicInstrCost(IID, RetTy, Tys, FMF, ScalarizationCostPassed);
Simon Pilgrim14000b32016-05-24 08:17:50 +00001827}
1828
1829int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
Jonas Paulssona48ea232017-03-14 06:35:36 +00001830 ArrayRef<Value *> Args, FastMathFlags FMF, unsigned VF) {
1831 return BaseT::getIntrinsicInstrCost(IID, RetTy, Args, FMF, VF);
Simon Pilgrim14000b32016-05-24 08:17:50 +00001832}
1833
Chandler Carruth93205eb2015-08-05 18:08:10 +00001834int X86TTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val, unsigned Index) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001835 assert(Val->isVectorTy() && "This must be a vector type");
1836
Sanjay Patelaedc3472016-05-25 17:27:54 +00001837 Type *ScalarType = Val->getScalarType();
1838
Chandler Carruth664e3542013-01-07 01:37:14 +00001839 if (Index != -1U) {
1840 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001841 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Val);
Chandler Carruth664e3542013-01-07 01:37:14 +00001842
1843 // This type is legalized to a scalar type.
1844 if (!LT.second.isVector())
1845 return 0;
1846
1847 // The type may be split. Normalize the index to the new type.
1848 unsigned Width = LT.second.getVectorNumElements();
1849 Index = Index % Width;
1850
1851 // Floating point scalars are already located in index #0.
Sanjay Patelaedc3472016-05-25 17:27:54 +00001852 if (ScalarType->isFloatingPointTy() && Index == 0)
Chandler Carruth664e3542013-01-07 01:37:14 +00001853 return 0;
1854 }
1855
Sanjay Patelaedc3472016-05-25 17:27:54 +00001856 // Add to the base cost if we know that the extracted element of a vector is
1857 // destined to be moved to and used in the integer register file.
1858 int RegisterFileMoveCost = 0;
1859 if (Opcode == Instruction::ExtractElement && ScalarType->isPointerTy())
1860 RegisterFileMoveCost = 1;
1861
1862 return BaseT::getVectorInstrCost(Opcode, Val, Index) + RegisterFileMoveCost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001863}
1864
Chandler Carruth93205eb2015-08-05 18:08:10 +00001865int X86TTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src, unsigned Alignment,
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001866 unsigned AddressSpace, const Instruction *I) {
Alp Tokerf907b892013-12-05 05:44:44 +00001867 // Handle non-power-of-two vectors such as <3 x float>
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001868 if (VectorType *VTy = dyn_cast<VectorType>(Src)) {
1869 unsigned NumElem = VTy->getVectorNumElements();
1870
1871 // Handle a few common cases:
1872 // <3 x float>
1873 if (NumElem == 3 && VTy->getScalarSizeInBits() == 32)
1874 // Cost = 64 bit store + extract + 32 bit store.
1875 return 3;
1876
1877 // <3 x double>
1878 if (NumElem == 3 && VTy->getScalarSizeInBits() == 64)
1879 // Cost = 128 bit store + unpack + 64 bit store.
1880 return 3;
1881
Alp Tokerf907b892013-12-05 05:44:44 +00001882 // Assume that all other non-power-of-two numbers are scalarized.
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001883 if (!isPowerOf2_32(NumElem)) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00001884 int Cost = BaseT::getMemoryOpCost(Opcode, VTy->getScalarType(), Alignment,
1885 AddressSpace);
1886 int SplitCost = getScalarizationOverhead(Src, Opcode == Instruction::Load,
1887 Opcode == Instruction::Store);
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001888 return NumElem * Cost + SplitCost;
1889 }
1890 }
1891
Chandler Carruth664e3542013-01-07 01:37:14 +00001892 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001893 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src);
Chandler Carruth664e3542013-01-07 01:37:14 +00001894 assert((Opcode == Instruction::Load || Opcode == Instruction::Store) &&
1895 "Invalid Opcode");
1896
1897 // Each load/store unit costs 1.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001898 int Cost = LT.first * 1;
Chandler Carruth664e3542013-01-07 01:37:14 +00001899
Sanjay Patel9f6c4d52016-03-09 22:23:33 +00001900 // This isn't exactly right. We're using slow unaligned 32-byte accesses as a
1901 // proxy for a double-pumped AVX memory interface such as on Sandybridge.
1902 if (LT.second.getStoreSize() == 32 && ST->isUnalignedMem32Slow())
1903 Cost *= 2;
Chandler Carruth664e3542013-01-07 01:37:14 +00001904
1905 return Cost;
1906}
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001907
Chandler Carruth93205eb2015-08-05 18:08:10 +00001908int X86TTIImpl::getMaskedMemoryOpCost(unsigned Opcode, Type *SrcTy,
1909 unsigned Alignment,
1910 unsigned AddressSpace) {
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001911 VectorType *SrcVTy = dyn_cast<VectorType>(SrcTy);
1912 if (!SrcVTy)
1913 // To calculate scalar take the regular cost, without mask
1914 return getMemoryOpCost(Opcode, SrcTy, Alignment, AddressSpace);
1915
1916 unsigned NumElem = SrcVTy->getVectorNumElements();
1917 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00001918 VectorType::get(Type::getInt8Ty(SrcVTy->getContext()), NumElem);
Elena Demikhovsky20662e32015-10-19 07:43:38 +00001919 if ((Opcode == Instruction::Load && !isLegalMaskedLoad(SrcVTy)) ||
1920 (Opcode == Instruction::Store && !isLegalMaskedStore(SrcVTy)) ||
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001921 !isPowerOf2_32(NumElem)) {
1922 // Scalarization
Chandler Carruth93205eb2015-08-05 18:08:10 +00001923 int MaskSplitCost = getScalarizationOverhead(MaskTy, false, true);
1924 int ScalarCompareCost = getCmpSelInstrCost(
Mehdi Amini867e9142016-04-14 04:36:40 +00001925 Instruction::ICmp, Type::getInt8Ty(SrcVTy->getContext()), nullptr);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001926 int BranchCost = getCFInstrCost(Instruction::Br);
1927 int MaskCmpCost = NumElem * (BranchCost + ScalarCompareCost);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001928
Chandler Carruth93205eb2015-08-05 18:08:10 +00001929 int ValueSplitCost = getScalarizationOverhead(
1930 SrcVTy, Opcode == Instruction::Load, Opcode == Instruction::Store);
1931 int MemopCost =
Chandler Carruth705b1852015-01-31 03:43:40 +00001932 NumElem * BaseT::getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
1933 Alignment, AddressSpace);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001934 return MemopCost + ValueSplitCost + MaskSplitCost + MaskCmpCost;
1935 }
1936
1937 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001938 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, SrcVTy);
Cong Houda4e8ae2015-10-28 18:15:46 +00001939 auto VT = TLI->getValueType(DL, SrcVTy);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001940 int Cost = 0;
Cong Houda4e8ae2015-10-28 18:15:46 +00001941 if (VT.isSimple() && LT.second != VT.getSimpleVT() &&
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001942 LT.second.getVectorNumElements() == NumElem)
1943 // Promotion requires expand/truncate for data and a shuffle for mask.
Simon Pilgrime39fa6c2018-06-12 16:12:29 +00001944 Cost += getShuffleCost(TTI::SK_Select, SrcVTy, 0, nullptr) +
1945 getShuffleCost(TTI::SK_Select, MaskTy, 0, nullptr);
Chandler Carruth705b1852015-01-31 03:43:40 +00001946
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001947 else if (LT.second.getVectorNumElements() > NumElem) {
1948 VectorType *NewMaskTy = VectorType::get(MaskTy->getVectorElementType(),
1949 LT.second.getVectorNumElements());
1950 // Expanding requires fill mask with zeroes
Chandler Carruth705b1852015-01-31 03:43:40 +00001951 Cost += getShuffleCost(TTI::SK_InsertSubvector, NewMaskTy, 0, MaskTy);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001952 }
1953 if (!ST->hasAVX512())
1954 return Cost + LT.first*4; // Each maskmov costs 4
1955
1956 // AVX-512 masked load/store is cheapper
1957 return Cost+LT.first;
1958}
1959
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001960int X86TTIImpl::getAddressComputationCost(Type *Ty, ScalarEvolution *SE,
1961 const SCEV *Ptr) {
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001962 // Address computations in vectorized code with non-consecutive addresses will
1963 // likely result in more instructions compared to scalar code where the
1964 // computation can more often be merged into the index mode. The resulting
1965 // extra micro-ops can significantly decrease throughput.
1966 unsigned NumVectorInstToHideOverhead = 10;
1967
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001968 // Cost modeling of Strided Access Computation is hidden by the indexing
1969 // modes of X86 regardless of the stride value. We dont believe that there
1970 // is a difference between constant strided access in gerenal and constant
1971 // strided value which is less than or equal to 64.
1972 // Even in the case of (loop invariant) stride whose value is not known at
1973 // compile time, the address computation will not incur more than one extra
1974 // ADD instruction.
1975 if (Ty->isVectorTy() && SE) {
1976 if (!BaseT::isStridedAccess(Ptr))
1977 return NumVectorInstToHideOverhead;
1978 if (!BaseT::getConstantStrideStep(SE, Ptr))
1979 return 1;
1980 }
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001981
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001982 return BaseT::getAddressComputationCost(Ty, SE, Ptr);
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001983}
Yi Jiang5c343de2013-09-19 17:48:48 +00001984
Alexey Bataev3e9b3eb2017-07-31 14:19:32 +00001985int X86TTIImpl::getArithmeticReductionCost(unsigned Opcode, Type *ValTy,
1986 bool IsPairwise) {
Michael Liao5bf95782014-12-04 05:20:33 +00001987
Chandler Carruth93205eb2015-08-05 18:08:10 +00001988 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Michael Liao5bf95782014-12-04 05:20:33 +00001989
Yi Jiang5c343de2013-09-19 17:48:48 +00001990 MVT MTy = LT.second;
Michael Liao5bf95782014-12-04 05:20:33 +00001991
Yi Jiang5c343de2013-09-19 17:48:48 +00001992 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1993 assert(ISD && "Invalid opcode");
Michael Liao5bf95782014-12-04 05:20:33 +00001994
1995 // We use the Intel Architecture Code Analyzer(IACA) to measure the throughput
1996 // and make it as the cost.
1997
Craig Topper4b275762015-10-28 04:02:12 +00001998 static const CostTblEntry SSE42CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00001999 { ISD::FADD, MVT::v2f64, 2 },
2000 { ISD::FADD, MVT::v4f32, 4 },
2001 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
2002 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
2003 { ISD::ADD, MVT::v8i16, 5 },
2004 };
Michael Liao5bf95782014-12-04 05:20:33 +00002005
Craig Topper4b275762015-10-28 04:02:12 +00002006 static const CostTblEntry AVX1CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002007 { ISD::FADD, MVT::v4f32, 4 },
2008 { ISD::FADD, MVT::v4f64, 5 },
2009 { ISD::FADD, MVT::v8f32, 7 },
2010 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
2011 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
2012 { ISD::ADD, MVT::v4i64, 5 }, // The data reported by the IACA tool is "4.8".
2013 { ISD::ADD, MVT::v8i16, 5 },
2014 { ISD::ADD, MVT::v8i32, 5 },
2015 };
2016
Craig Topper4b275762015-10-28 04:02:12 +00002017 static const CostTblEntry SSE42CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002018 { ISD::FADD, MVT::v2f64, 2 },
2019 { ISD::FADD, MVT::v4f32, 4 },
2020 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
2021 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.3".
2022 { ISD::ADD, MVT::v8i16, 4 }, // The data reported by the IACA tool is "4.3".
2023 };
Michael Liao5bf95782014-12-04 05:20:33 +00002024
Craig Topper4b275762015-10-28 04:02:12 +00002025 static const CostTblEntry AVX1CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002026 { ISD::FADD, MVT::v4f32, 3 },
2027 { ISD::FADD, MVT::v4f64, 3 },
2028 { ISD::FADD, MVT::v8f32, 4 },
2029 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
2030 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "2.8".
2031 { ISD::ADD, MVT::v4i64, 3 },
2032 { ISD::ADD, MVT::v8i16, 4 },
2033 { ISD::ADD, MVT::v8i32, 5 },
2034 };
Michael Liao5bf95782014-12-04 05:20:33 +00002035
Yi Jiang5c343de2013-09-19 17:48:48 +00002036 if (IsPairwise) {
Craig Topperee0c8592015-10-27 04:14:24 +00002037 if (ST->hasAVX())
2038 if (const auto *Entry = CostTableLookup(AVX1CostTblPairWise, ISD, MTy))
2039 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00002040
Craig Topperee0c8592015-10-27 04:14:24 +00002041 if (ST->hasSSE42())
2042 if (const auto *Entry = CostTableLookup(SSE42CostTblPairWise, ISD, MTy))
2043 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00002044 } else {
Craig Topperee0c8592015-10-27 04:14:24 +00002045 if (ST->hasAVX())
2046 if (const auto *Entry = CostTableLookup(AVX1CostTblNoPairWise, ISD, MTy))
2047 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00002048
Craig Topperee0c8592015-10-27 04:14:24 +00002049 if (ST->hasSSE42())
2050 if (const auto *Entry = CostTableLookup(SSE42CostTblNoPairWise, ISD, MTy))
2051 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00002052 }
2053
Alexey Bataev3e9b3eb2017-07-31 14:19:32 +00002054 return BaseT::getArithmeticReductionCost(Opcode, ValTy, IsPairwise);
Yi Jiang5c343de2013-09-19 17:48:48 +00002055}
2056
Alexey Bataev6dd29fc2017-09-08 13:49:36 +00002057int X86TTIImpl::getMinMaxReductionCost(Type *ValTy, Type *CondTy,
2058 bool IsPairwise, bool IsUnsigned) {
2059 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
2060
2061 MVT MTy = LT.second;
2062
2063 int ISD;
2064 if (ValTy->isIntOrIntVectorTy()) {
2065 ISD = IsUnsigned ? ISD::UMIN : ISD::SMIN;
2066 } else {
2067 assert(ValTy->isFPOrFPVectorTy() &&
2068 "Expected float point or integer vector type.");
2069 ISD = ISD::FMINNUM;
2070 }
2071
2072 // We use the Intel Architecture Code Analyzer(IACA) to measure the throughput
2073 // and make it as the cost.
2074
2075 static const CostTblEntry SSE42CostTblPairWise[] = {
2076 {ISD::FMINNUM, MVT::v2f64, 3},
2077 {ISD::FMINNUM, MVT::v4f32, 2},
2078 {ISD::SMIN, MVT::v2i64, 7}, // The data reported by the IACA is "6.8"
2079 {ISD::UMIN, MVT::v2i64, 8}, // The data reported by the IACA is "8.6"
2080 {ISD::SMIN, MVT::v4i32, 1}, // The data reported by the IACA is "1.5"
2081 {ISD::UMIN, MVT::v4i32, 2}, // The data reported by the IACA is "1.8"
2082 {ISD::SMIN, MVT::v8i16, 2},
2083 {ISD::UMIN, MVT::v8i16, 2},
2084 };
2085
2086 static const CostTblEntry AVX1CostTblPairWise[] = {
2087 {ISD::FMINNUM, MVT::v4f32, 1},
2088 {ISD::FMINNUM, MVT::v4f64, 1},
2089 {ISD::FMINNUM, MVT::v8f32, 2},
2090 {ISD::SMIN, MVT::v2i64, 3},
2091 {ISD::UMIN, MVT::v2i64, 3},
2092 {ISD::SMIN, MVT::v4i32, 1},
2093 {ISD::UMIN, MVT::v4i32, 1},
2094 {ISD::SMIN, MVT::v8i16, 1},
2095 {ISD::UMIN, MVT::v8i16, 1},
2096 {ISD::SMIN, MVT::v8i32, 3},
2097 {ISD::UMIN, MVT::v8i32, 3},
2098 };
2099
2100 static const CostTblEntry AVX2CostTblPairWise[] = {
2101 {ISD::SMIN, MVT::v4i64, 2},
2102 {ISD::UMIN, MVT::v4i64, 2},
2103 {ISD::SMIN, MVT::v8i32, 1},
2104 {ISD::UMIN, MVT::v8i32, 1},
2105 {ISD::SMIN, MVT::v16i16, 1},
2106 {ISD::UMIN, MVT::v16i16, 1},
2107 {ISD::SMIN, MVT::v32i8, 2},
2108 {ISD::UMIN, MVT::v32i8, 2},
2109 };
2110
2111 static const CostTblEntry AVX512CostTblPairWise[] = {
2112 {ISD::FMINNUM, MVT::v8f64, 1},
2113 {ISD::FMINNUM, MVT::v16f32, 2},
2114 {ISD::SMIN, MVT::v8i64, 2},
2115 {ISD::UMIN, MVT::v8i64, 2},
2116 {ISD::SMIN, MVT::v16i32, 1},
2117 {ISD::UMIN, MVT::v16i32, 1},
2118 };
2119
2120 static const CostTblEntry SSE42CostTblNoPairWise[] = {
2121 {ISD::FMINNUM, MVT::v2f64, 3},
2122 {ISD::FMINNUM, MVT::v4f32, 3},
2123 {ISD::SMIN, MVT::v2i64, 7}, // The data reported by the IACA is "6.8"
2124 {ISD::UMIN, MVT::v2i64, 9}, // The data reported by the IACA is "8.6"
2125 {ISD::SMIN, MVT::v4i32, 1}, // The data reported by the IACA is "1.5"
2126 {ISD::UMIN, MVT::v4i32, 2}, // The data reported by the IACA is "1.8"
2127 {ISD::SMIN, MVT::v8i16, 1}, // The data reported by the IACA is "1.5"
2128 {ISD::UMIN, MVT::v8i16, 2}, // The data reported by the IACA is "1.8"
2129 };
2130
2131 static const CostTblEntry AVX1CostTblNoPairWise[] = {
2132 {ISD::FMINNUM, MVT::v4f32, 1},
2133 {ISD::FMINNUM, MVT::v4f64, 1},
2134 {ISD::FMINNUM, MVT::v8f32, 1},
2135 {ISD::SMIN, MVT::v2i64, 3},
2136 {ISD::UMIN, MVT::v2i64, 3},
2137 {ISD::SMIN, MVT::v4i32, 1},
2138 {ISD::UMIN, MVT::v4i32, 1},
2139 {ISD::SMIN, MVT::v8i16, 1},
2140 {ISD::UMIN, MVT::v8i16, 1},
2141 {ISD::SMIN, MVT::v8i32, 2},
2142 {ISD::UMIN, MVT::v8i32, 2},
2143 };
2144
2145 static const CostTblEntry AVX2CostTblNoPairWise[] = {
2146 {ISD::SMIN, MVT::v4i64, 1},
2147 {ISD::UMIN, MVT::v4i64, 1},
2148 {ISD::SMIN, MVT::v8i32, 1},
2149 {ISD::UMIN, MVT::v8i32, 1},
2150 {ISD::SMIN, MVT::v16i16, 1},
2151 {ISD::UMIN, MVT::v16i16, 1},
2152 {ISD::SMIN, MVT::v32i8, 1},
2153 {ISD::UMIN, MVT::v32i8, 1},
2154 };
2155
2156 static const CostTblEntry AVX512CostTblNoPairWise[] = {
2157 {ISD::FMINNUM, MVT::v8f64, 1},
2158 {ISD::FMINNUM, MVT::v16f32, 2},
2159 {ISD::SMIN, MVT::v8i64, 1},
2160 {ISD::UMIN, MVT::v8i64, 1},
2161 {ISD::SMIN, MVT::v16i32, 1},
2162 {ISD::UMIN, MVT::v16i32, 1},
2163 };
2164
2165 if (IsPairwise) {
2166 if (ST->hasAVX512())
2167 if (const auto *Entry = CostTableLookup(AVX512CostTblPairWise, ISD, MTy))
2168 return LT.first * Entry->Cost;
2169
2170 if (ST->hasAVX2())
2171 if (const auto *Entry = CostTableLookup(AVX2CostTblPairWise, ISD, MTy))
2172 return LT.first * Entry->Cost;
2173
2174 if (ST->hasAVX())
2175 if (const auto *Entry = CostTableLookup(AVX1CostTblPairWise, ISD, MTy))
2176 return LT.first * Entry->Cost;
2177
2178 if (ST->hasSSE42())
2179 if (const auto *Entry = CostTableLookup(SSE42CostTblPairWise, ISD, MTy))
2180 return LT.first * Entry->Cost;
2181 } else {
2182 if (ST->hasAVX512())
2183 if (const auto *Entry =
2184 CostTableLookup(AVX512CostTblNoPairWise, ISD, MTy))
2185 return LT.first * Entry->Cost;
2186
2187 if (ST->hasAVX2())
2188 if (const auto *Entry = CostTableLookup(AVX2CostTblNoPairWise, ISD, MTy))
2189 return LT.first * Entry->Cost;
2190
2191 if (ST->hasAVX())
2192 if (const auto *Entry = CostTableLookup(AVX1CostTblNoPairWise, ISD, MTy))
2193 return LT.first * Entry->Cost;
2194
2195 if (ST->hasSSE42())
2196 if (const auto *Entry = CostTableLookup(SSE42CostTblNoPairWise, ISD, MTy))
2197 return LT.first * Entry->Cost;
2198 }
2199
2200 return BaseT::getMinMaxReductionCost(ValTy, CondTy, IsPairwise, IsUnsigned);
2201}
2202
Adrian Prantl5f8f34e42018-05-01 15:54:18 +00002203/// Calculate the cost of materializing a 64-bit value. This helper
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002204/// method might only calculate a fraction of a larger immediate. Therefore it
2205/// is valid to return a cost of ZERO.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002206int X86TTIImpl::getIntImmCost(int64_t Val) {
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002207 if (Val == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002208 return TTI::TCC_Free;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002209
2210 if (isInt<32>(Val))
Chandler Carruth705b1852015-01-31 03:43:40 +00002211 return TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002212
Chandler Carruth705b1852015-01-31 03:43:40 +00002213 return 2 * TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002214}
2215
Chandler Carruth93205eb2015-08-05 18:08:10 +00002216int X86TTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002217 assert(Ty->isIntegerTy());
2218
2219 unsigned BitSize = Ty->getPrimitiveSizeInBits();
2220 if (BitSize == 0)
2221 return ~0U;
2222
Juergen Ributzka43176172014-05-19 21:00:53 +00002223 // Never hoist constants larger than 128bit, because this might lead to
2224 // incorrect code generation or assertions in codegen.
2225 // Fixme: Create a cost model for types larger than i128 once the codegen
2226 // issues have been fixed.
2227 if (BitSize > 128)
Chandler Carruth705b1852015-01-31 03:43:40 +00002228 return TTI::TCC_Free;
Juergen Ributzka43176172014-05-19 21:00:53 +00002229
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002230 if (Imm == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002231 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002232
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002233 // Sign-extend all constants to a multiple of 64-bit.
2234 APInt ImmVal = Imm;
2235 if (BitSize & 0x3f)
2236 ImmVal = Imm.sext((BitSize + 63) & ~0x3fU);
2237
2238 // Split the constant into 64-bit chunks and calculate the cost for each
2239 // chunk.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002240 int Cost = 0;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002241 for (unsigned ShiftVal = 0; ShiftVal < BitSize; ShiftVal += 64) {
2242 APInt Tmp = ImmVal.ashr(ShiftVal).sextOrTrunc(64);
2243 int64_t Val = Tmp.getSExtValue();
2244 Cost += getIntImmCost(Val);
2245 }
Sanjay Patel4c7d0942016-04-05 19:27:39 +00002246 // We need at least one instruction to materialize the constant.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002247 return std::max(1, Cost);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002248}
2249
Chandler Carruth93205eb2015-08-05 18:08:10 +00002250int X86TTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx, const APInt &Imm,
2251 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002252 assert(Ty->isIntegerTy());
2253
2254 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00002255 // There is no cost model for constants with a bit size of 0. Return TCC_Free
2256 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002257 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002258 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002259
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002260 unsigned ImmIdx = ~0U;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002261 switch (Opcode) {
Chandler Carruth705b1852015-01-31 03:43:40 +00002262 default:
2263 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002264 case Instruction::GetElementPtr:
Juergen Ributzka27435b32014-04-02 21:45:36 +00002265 // Always hoist the base address of a GetElementPtr. This prevents the
2266 // creation of new constants for every base constant that gets constant
2267 // folded with the offset.
Juergen Ributzka631c4912014-03-25 18:01:25 +00002268 if (Idx == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002269 return 2 * TTI::TCC_Basic;
2270 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002271 case Instruction::Store:
2272 ImmIdx = 0;
2273 break;
Craig Topper074e8452015-12-20 18:41:54 +00002274 case Instruction::ICmp:
2275 // This is an imperfect hack to prevent constant hoisting of
2276 // compares that might be trying to check if a 64-bit value fits in
2277 // 32-bits. The backend can optimize these cases using a right shift by 32.
2278 // Ideally we would check the compare predicate here. There also other
2279 // similar immediates the backend can use shifts for.
2280 if (Idx == 1 && Imm.getBitWidth() == 64) {
2281 uint64_t ImmVal = Imm.getZExtValue();
2282 if (ImmVal == 0x100000000ULL || ImmVal == 0xffffffff)
2283 return TTI::TCC_Free;
2284 }
2285 ImmIdx = 1;
2286 break;
Craig Topper79dd1bf2015-10-06 02:50:24 +00002287 case Instruction::And:
2288 // We support 64-bit ANDs with immediates with 32-bits of leading zeroes
2289 // by using a 32-bit operation with implicit zero extension. Detect such
2290 // immediates here as the normal path expects bit 31 to be sign extended.
2291 if (Idx == 1 && Imm.getBitWidth() == 64 && isUInt<32>(Imm.getZExtValue()))
2292 return TTI::TCC_Free;
Justin Bognerb03fd122016-08-17 05:10:15 +00002293 LLVM_FALLTHROUGH;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002294 case Instruction::Add:
2295 case Instruction::Sub:
2296 case Instruction::Mul:
2297 case Instruction::UDiv:
2298 case Instruction::SDiv:
2299 case Instruction::URem:
2300 case Instruction::SRem:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002301 case Instruction::Or:
2302 case Instruction::Xor:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002303 ImmIdx = 1;
2304 break;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00002305 // Always return TCC_Free for the shift value of a shift instruction.
2306 case Instruction::Shl:
2307 case Instruction::LShr:
2308 case Instruction::AShr:
2309 if (Idx == 1)
Chandler Carruth705b1852015-01-31 03:43:40 +00002310 return TTI::TCC_Free;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00002311 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002312 case Instruction::Trunc:
2313 case Instruction::ZExt:
2314 case Instruction::SExt:
2315 case Instruction::IntToPtr:
2316 case Instruction::PtrToInt:
2317 case Instruction::BitCast:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002318 case Instruction::PHI:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002319 case Instruction::Call:
2320 case Instruction::Select:
2321 case Instruction::Ret:
2322 case Instruction::Load:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002323 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002324 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002325
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002326 if (Idx == ImmIdx) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00002327 int NumConstants = (BitSize + 63) / 64;
2328 int Cost = X86TTIImpl::getIntImmCost(Imm, Ty);
Chandler Carruth705b1852015-01-31 03:43:40 +00002329 return (Cost <= NumConstants * TTI::TCC_Basic)
Chandler Carruth93205eb2015-08-05 18:08:10 +00002330 ? static_cast<int>(TTI::TCC_Free)
Chandler Carruth705b1852015-01-31 03:43:40 +00002331 : Cost;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002332 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002333
Chandler Carruth705b1852015-01-31 03:43:40 +00002334 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002335}
2336
Chandler Carruth93205eb2015-08-05 18:08:10 +00002337int X86TTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx, const APInt &Imm,
2338 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002339 assert(Ty->isIntegerTy());
2340
2341 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00002342 // There is no cost model for constants with a bit size of 0. Return TCC_Free
2343 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002344 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002345 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002346
2347 switch (IID) {
Chandler Carruth705b1852015-01-31 03:43:40 +00002348 default:
2349 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002350 case Intrinsic::sadd_with_overflow:
2351 case Intrinsic::uadd_with_overflow:
2352 case Intrinsic::ssub_with_overflow:
2353 case Intrinsic::usub_with_overflow:
2354 case Intrinsic::smul_with_overflow:
2355 case Intrinsic::umul_with_overflow:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002356 if ((Idx == 1) && Imm.getBitWidth() <= 64 && isInt<32>(Imm.getSExtValue()))
Chandler Carruth705b1852015-01-31 03:43:40 +00002357 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002358 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002359 case Intrinsic::experimental_stackmap:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002360 if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00002361 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002362 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002363 case Intrinsic::experimental_patchpoint_void:
2364 case Intrinsic::experimental_patchpoint_i64:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002365 if ((Idx < 4) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00002366 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002367 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002368 }
Chandler Carruth705b1852015-01-31 03:43:40 +00002369 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002370}
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002371
Elena Demikhovskyf58f8382017-08-20 12:34:29 +00002372unsigned X86TTIImpl::getUserCost(const User *U,
2373 ArrayRef<const Value *> Operands) {
2374 if (isa<StoreInst>(U)) {
2375 Value *Ptr = U->getOperand(1);
2376 // Store instruction with index and scale costs 2 Uops.
2377 // Check the preceding GEP to identify non-const indices.
2378 if (auto GEP = dyn_cast<GetElementPtrInst>(Ptr)) {
2379 if (!all_of(GEP->indices(), [](Value *V) { return isa<Constant>(V); }))
2380 return TTI::TCC_Basic * 2;
2381 }
2382 return TTI::TCC_Basic;
2383 }
2384 return BaseT::getUserCost(U, Operands);
2385}
2386
Elena Demikhovsky54946982015-12-28 20:10:59 +00002387// Return an average cost of Gather / Scatter instruction, maybe improved later
2388int X86TTIImpl::getGSVectorCost(unsigned Opcode, Type *SrcVTy, Value *Ptr,
2389 unsigned Alignment, unsigned AddressSpace) {
2390
2391 assert(isa<VectorType>(SrcVTy) && "Unexpected type in getGSVectorCost");
2392 unsigned VF = SrcVTy->getVectorNumElements();
2393
2394 // Try to reduce index size from 64 bit (default for GEP)
2395 // to 32. It is essential for VF 16. If the index can't be reduced to 32, the
2396 // operation will use 16 x 64 indices which do not fit in a zmm and needs
2397 // to split. Also check that the base pointer is the same for all lanes,
2398 // and that there's at most one variable index.
2399 auto getIndexSizeInBits = [](Value *Ptr, const DataLayout& DL) {
2400 unsigned IndexSize = DL.getPointerSizeInBits();
2401 GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr);
2402 if (IndexSize < 64 || !GEP)
2403 return IndexSize;
Simon Pilgrim14000b32016-05-24 08:17:50 +00002404
Elena Demikhovsky54946982015-12-28 20:10:59 +00002405 unsigned NumOfVarIndices = 0;
2406 Value *Ptrs = GEP->getPointerOperand();
2407 if (Ptrs->getType()->isVectorTy() && !getSplatValue(Ptrs))
2408 return IndexSize;
2409 for (unsigned i = 1; i < GEP->getNumOperands(); ++i) {
2410 if (isa<Constant>(GEP->getOperand(i)))
2411 continue;
2412 Type *IndxTy = GEP->getOperand(i)->getType();
2413 if (IndxTy->isVectorTy())
2414 IndxTy = IndxTy->getVectorElementType();
2415 if ((IndxTy->getPrimitiveSizeInBits() == 64 &&
2416 !isa<SExtInst>(GEP->getOperand(i))) ||
2417 ++NumOfVarIndices > 1)
2418 return IndexSize; // 64
2419 }
2420 return (unsigned)32;
2421 };
2422
2423
2424 // Trying to reduce IndexSize to 32 bits for vector 16.
2425 // By default the IndexSize is equal to pointer size.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002426 unsigned IndexSize = (ST->hasAVX512() && VF >= 16)
2427 ? getIndexSizeInBits(Ptr, DL)
2428 : DL.getPointerSizeInBits();
Elena Demikhovsky54946982015-12-28 20:10:59 +00002429
Mehdi Amini867e9142016-04-14 04:36:40 +00002430 Type *IndexVTy = VectorType::get(IntegerType::get(SrcVTy->getContext(),
Elena Demikhovsky54946982015-12-28 20:10:59 +00002431 IndexSize), VF);
2432 std::pair<int, MVT> IdxsLT = TLI->getTypeLegalizationCost(DL, IndexVTy);
2433 std::pair<int, MVT> SrcLT = TLI->getTypeLegalizationCost(DL, SrcVTy);
2434 int SplitFactor = std::max(IdxsLT.first, SrcLT.first);
2435 if (SplitFactor > 1) {
2436 // Handle splitting of vector of pointers
2437 Type *SplitSrcTy = VectorType::get(SrcVTy->getScalarType(), VF / SplitFactor);
2438 return SplitFactor * getGSVectorCost(Opcode, SplitSrcTy, Ptr, Alignment,
2439 AddressSpace);
2440 }
2441
2442 // The gather / scatter cost is given by Intel architects. It is a rough
2443 // number since we are looking at one instruction in a time.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002444 const int GSOverhead = (Opcode == Instruction::Load)
2445 ? ST->getGatherOverhead()
2446 : ST->getScatterOverhead();
Elena Demikhovsky54946982015-12-28 20:10:59 +00002447 return GSOverhead + VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
2448 Alignment, AddressSpace);
2449}
2450
2451/// Return the cost of full scalarization of gather / scatter operation.
2452///
2453/// Opcode - Load or Store instruction.
2454/// SrcVTy - The type of the data vector that should be gathered or scattered.
2455/// VariableMask - The mask is non-constant at compile time.
2456/// Alignment - Alignment for one element.
2457/// AddressSpace - pointer[s] address space.
2458///
2459int X86TTIImpl::getGSScalarCost(unsigned Opcode, Type *SrcVTy,
2460 bool VariableMask, unsigned Alignment,
2461 unsigned AddressSpace) {
2462 unsigned VF = SrcVTy->getVectorNumElements();
2463
2464 int MaskUnpackCost = 0;
2465 if (VariableMask) {
2466 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00002467 VectorType::get(Type::getInt1Ty(SrcVTy->getContext()), VF);
Elena Demikhovsky54946982015-12-28 20:10:59 +00002468 MaskUnpackCost = getScalarizationOverhead(MaskTy, false, true);
2469 int ScalarCompareCost =
Mehdi Amini867e9142016-04-14 04:36:40 +00002470 getCmpSelInstrCost(Instruction::ICmp, Type::getInt1Ty(SrcVTy->getContext()),
Elena Demikhovsky54946982015-12-28 20:10:59 +00002471 nullptr);
2472 int BranchCost = getCFInstrCost(Instruction::Br);
2473 MaskUnpackCost += VF * (BranchCost + ScalarCompareCost);
2474 }
2475
2476 // The cost of the scalar loads/stores.
2477 int MemoryOpCost = VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
2478 Alignment, AddressSpace);
2479
2480 int InsertExtractCost = 0;
2481 if (Opcode == Instruction::Load)
2482 for (unsigned i = 0; i < VF; ++i)
2483 // Add the cost of inserting each scalar load into the vector
2484 InsertExtractCost +=
2485 getVectorInstrCost(Instruction::InsertElement, SrcVTy, i);
2486 else
2487 for (unsigned i = 0; i < VF; ++i)
2488 // Add the cost of extracting each element out of the data vector
2489 InsertExtractCost +=
2490 getVectorInstrCost(Instruction::ExtractElement, SrcVTy, i);
2491
2492 return MemoryOpCost + MaskUnpackCost + InsertExtractCost;
2493}
2494
2495/// Calculate the cost of Gather / Scatter operation
2496int X86TTIImpl::getGatherScatterOpCost(unsigned Opcode, Type *SrcVTy,
2497 Value *Ptr, bool VariableMask,
2498 unsigned Alignment) {
2499 assert(SrcVTy->isVectorTy() && "Unexpected data type for Gather/Scatter");
2500 unsigned VF = SrcVTy->getVectorNumElements();
2501 PointerType *PtrTy = dyn_cast<PointerType>(Ptr->getType());
2502 if (!PtrTy && Ptr->getType()->isVectorTy())
2503 PtrTy = dyn_cast<PointerType>(Ptr->getType()->getVectorElementType());
2504 assert(PtrTy && "Unexpected type for Ptr argument");
2505 unsigned AddressSpace = PtrTy->getAddressSpace();
2506
2507 bool Scalarize = false;
2508 if ((Opcode == Instruction::Load && !isLegalMaskedGather(SrcVTy)) ||
2509 (Opcode == Instruction::Store && !isLegalMaskedScatter(SrcVTy)))
2510 Scalarize = true;
2511 // Gather / Scatter for vector 2 is not profitable on KNL / SKX
2512 // Vector-4 of gather/scatter instruction does not exist on KNL.
2513 // We can extend it to 8 elements, but zeroing upper bits of
2514 // the mask vector will add more instructions. Right now we give the scalar
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002515 // cost of vector-4 for KNL. TODO: Check, maybe the gather/scatter instruction
2516 // is better in the VariableMask case.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002517 if (ST->hasAVX512() && (VF == 2 || (VF == 4 && !ST->hasVLX())))
Elena Demikhovsky54946982015-12-28 20:10:59 +00002518 Scalarize = true;
2519
2520 if (Scalarize)
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002521 return getGSScalarCost(Opcode, SrcVTy, VariableMask, Alignment,
2522 AddressSpace);
Elena Demikhovsky54946982015-12-28 20:10:59 +00002523
2524 return getGSVectorCost(Opcode, SrcVTy, Ptr, Alignment, AddressSpace);
2525}
2526
Evgeny Stupachenkoc6752902017-08-07 19:56:34 +00002527bool X86TTIImpl::isLSRCostLess(TargetTransformInfo::LSRCost &C1,
2528 TargetTransformInfo::LSRCost &C2) {
2529 // X86 specific here are "instruction number 1st priority".
2530 return std::tie(C1.Insns, C1.NumRegs, C1.AddRecCost,
2531 C1.NumIVMuls, C1.NumBaseAdds,
2532 C1.ScaleCost, C1.ImmCost, C1.SetupCost) <
2533 std::tie(C2.Insns, C2.NumRegs, C2.AddRecCost,
2534 C2.NumIVMuls, C2.NumBaseAdds,
2535 C2.ScaleCost, C2.ImmCost, C2.SetupCost);
2536}
2537
Sanjay Pateld7c702b2018-02-05 23:43:05 +00002538bool X86TTIImpl::canMacroFuseCmp() {
2539 return ST->hasMacroFusion();
2540}
2541
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002542bool X86TTIImpl::isLegalMaskedLoad(Type *DataTy) {
Craig Topper46a5d582017-11-16 06:02:05 +00002543 // The backend can't handle a single element vector.
2544 if (isa<VectorType>(DataTy) && DataTy->getVectorNumElements() == 1)
2545 return false;
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002546 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00002547 int DataWidth = isa<PointerType>(ScalarTy) ?
2548 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002549
Igor Bregerf44b79d2016-08-02 09:15:28 +00002550 return ((DataWidth == 32 || DataWidth == 64) && ST->hasAVX()) ||
2551 ((DataWidth == 8 || DataWidth == 16) && ST->hasBWI());
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002552}
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00002553
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002554bool X86TTIImpl::isLegalMaskedStore(Type *DataType) {
2555 return isLegalMaskedLoad(DataType);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00002556}
2557
Elena Demikhovsky09285852015-10-25 15:37:55 +00002558bool X86TTIImpl::isLegalMaskedGather(Type *DataTy) {
2559 // This function is called now in two cases: from the Loop Vectorizer
2560 // and from the Scalarizer.
2561 // When the Loop Vectorizer asks about legality of the feature,
2562 // the vectorization factor is not calculated yet. The Loop Vectorizer
2563 // sends a scalar type and the decision is based on the width of the
2564 // scalar element.
2565 // Later on, the cost model will estimate usage this intrinsic based on
2566 // the vector type.
2567 // The Scalarizer asks again about legality. It sends a vector type.
2568 // In this case we can reject non-power-of-2 vectors.
Craig Topper46a5d582017-11-16 06:02:05 +00002569 // We also reject single element vectors as the type legalizer can't
2570 // scalarize it.
2571 if (isa<VectorType>(DataTy)) {
2572 unsigned NumElts = DataTy->getVectorNumElements();
2573 if (NumElts == 1 || !isPowerOf2_32(NumElts))
2574 return false;
2575 }
Elena Demikhovsky09285852015-10-25 15:37:55 +00002576 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00002577 int DataWidth = isa<PointerType>(ScalarTy) ?
2578 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
Elena Demikhovsky09285852015-10-25 15:37:55 +00002579
Craig Topperea37e202017-11-25 18:09:37 +00002580 // Some CPUs have better gather performance than others.
2581 // TODO: Remove the explicit ST->hasAVX512()?, That would mean we would only
2582 // enable gather with a -march.
2583 return (DataWidth == 32 || DataWidth == 64) &&
Craig Topper0d797a32018-01-20 00:26:08 +00002584 (ST->hasAVX512() || (ST->hasFastGather() && ST->hasAVX2()));
Elena Demikhovsky09285852015-10-25 15:37:55 +00002585}
2586
2587bool X86TTIImpl::isLegalMaskedScatter(Type *DataType) {
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002588 // AVX2 doesn't support scatter
2589 if (!ST->hasAVX512())
2590 return false;
Elena Demikhovsky09285852015-10-25 15:37:55 +00002591 return isLegalMaskedGather(DataType);
2592}
2593
Sanjay Patel6fd43912017-09-09 13:38:18 +00002594bool X86TTIImpl::hasDivRemOp(Type *DataType, bool IsSigned) {
2595 EVT VT = TLI->getValueType(DL, DataType);
2596 return TLI->isOperationLegal(IsSigned ? ISD::SDIVREM : ISD::UDIVREM, VT);
2597}
2598
Sanjay Patel0de1a4b2017-11-27 21:15:43 +00002599bool X86TTIImpl::isFCmpOrdCheaperThanFCmpZero(Type *Ty) {
2600 return false;
2601}
2602
Eric Christopherd566fb12015-07-29 22:09:48 +00002603bool X86TTIImpl::areInlineCompatible(const Function *Caller,
2604 const Function *Callee) const {
Eric Christophere1002262015-07-02 01:11:50 +00002605 const TargetMachine &TM = getTLI()->getTargetMachine();
2606
2607 // Work this as a subsetting of subtarget features.
2608 const FeatureBitset &CallerBits =
2609 TM.getSubtargetImpl(*Caller)->getFeatureBits();
2610 const FeatureBitset &CalleeBits =
2611 TM.getSubtargetImpl(*Callee)->getFeatureBits();
2612
2613 // FIXME: This is likely too limiting as it will include subtarget features
2614 // that we might not care about for inlining, but it is conservatively
2615 // correct.
2616 return (CallerBits & CalleeBits) == CalleeBits;
2617}
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002618
Clement Courbetb2c3eb82017-10-30 14:19:33 +00002619const X86TTIImpl::TTI::MemCmpExpansionOptions *
2620X86TTIImpl::enableMemCmpExpansion(bool IsZeroCmp) const {
2621 // Only enable vector loads for equality comparison.
2622 // Right now the vector version is not as fast, see #33329.
2623 static const auto ThreeWayOptions = [this]() {
2624 TTI::MemCmpExpansionOptions Options;
2625 if (ST->is64Bit()) {
2626 Options.LoadSizes.push_back(8);
2627 }
2628 Options.LoadSizes.push_back(4);
2629 Options.LoadSizes.push_back(2);
2630 Options.LoadSizes.push_back(1);
2631 return Options;
2632 }();
2633 static const auto EqZeroOptions = [this]() {
2634 TTI::MemCmpExpansionOptions Options;
2635 // TODO: enable AVX512 when the DAG is ready.
2636 // if (ST->hasAVX512()) Options.LoadSizes.push_back(64);
2637 if (ST->hasAVX2()) Options.LoadSizes.push_back(32);
2638 if (ST->hasSSE2()) Options.LoadSizes.push_back(16);
2639 if (ST->is64Bit()) {
2640 Options.LoadSizes.push_back(8);
2641 }
2642 Options.LoadSizes.push_back(4);
2643 Options.LoadSizes.push_back(2);
2644 Options.LoadSizes.push_back(1);
2645 return Options;
2646 }();
2647 return IsZeroCmp ? &EqZeroOptions : &ThreeWayOptions;
Sanjay Patel06566292017-06-20 15:58:30 +00002648}
2649
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002650bool X86TTIImpl::enableInterleavedAccessVectorization() {
2651 // TODO: We expect this to be beneficial regardless of arch,
2652 // but there are currently some unexplained performance artifacts on Atom.
2653 // As a temporary solution, disable on Atom.
Mohammed Agabaria20caee92017-01-25 09:14:48 +00002654 return !(ST->isAtom());
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002655}
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002656
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002657// Get estimation for interleaved load/store operations for AVX2.
2658// \p Factor is the interleaved-access factor (stride) - number of
2659// (interleaved) elements in the group.
2660// \p Indices contains the indices for a strided load: when the
2661// interleaved load has gaps they indicate which elements are used.
2662// If Indices is empty (or if the number of indices is equal to the size
2663// of the interleaved-access as given in \p Factor) the access has no gaps.
2664//
2665// As opposed to AVX-512, AVX2 does not have generic shuffles that allow
2666// computing the cost using a generic formula as a function of generic
2667// shuffles. We therefore use a lookup table instead, filled according to
2668// the instruction sequences that codegen currently generates.
2669int X86TTIImpl::getInterleavedMemoryOpCostAVX2(unsigned Opcode, Type *VecTy,
2670 unsigned Factor,
2671 ArrayRef<unsigned> Indices,
2672 unsigned Alignment,
2673 unsigned AddressSpace) {
2674
2675 // We currently Support only fully-interleaved groups, with no gaps.
2676 // TODO: Support also strided loads (interleaved-groups with gaps).
2677 if (Indices.size() && Indices.size() != Factor)
2678 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2679 Alignment, AddressSpace);
2680
2681 // VecTy for interleave memop is <VF*Factor x Elt>.
2682 // So, for VF=4, Interleave Factor = 3, Element type = i32 we have
2683 // VecTy = <12 x i32>.
2684 MVT LegalVT = getTLI()->getTypeLegalizationCost(DL, VecTy).second;
2685
2686 // This function can be called with VecTy=<6xi128>, Factor=3, in which case
2687 // the VF=2, while v2i128 is an unsupported MVT vector type
2688 // (see MachineValueType.h::getVectorVT()).
2689 if (!LegalVT.isVector())
2690 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2691 Alignment, AddressSpace);
2692
2693 unsigned VF = VecTy->getVectorNumElements() / Factor;
2694 Type *ScalarTy = VecTy->getVectorElementType();
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002695
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002696 // Calculate the number of memory operations (NumOfMemOps), required
2697 // for load/store the VecTy.
2698 unsigned VecTySize = DL.getTypeStoreSize(VecTy);
2699 unsigned LegalVTSize = LegalVT.getStoreSize();
2700 unsigned NumOfMemOps = (VecTySize + LegalVTSize - 1) / LegalVTSize;
2701
2702 // Get the cost of one memory operation.
2703 Type *SingleMemOpTy = VectorType::get(VecTy->getVectorElementType(),
2704 LegalVT.getVectorNumElements());
2705 unsigned MemOpCost =
2706 getMemoryOpCost(Opcode, SingleMemOpTy, Alignment, AddressSpace);
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002707
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002708 VectorType *VT = VectorType::get(ScalarTy, VF);
2709 EVT ETy = TLI->getValueType(DL, VT);
2710 if (!ETy.isSimple())
2711 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2712 Alignment, AddressSpace);
2713
2714 // TODO: Complete for other data-types and strides.
2715 // Each combination of Stride, ElementTy and VF results in a different
2716 // sequence; The cost tables are therefore accessed with:
2717 // Factor (stride) and VectorType=VFxElemType.
2718 // The Cost accounts only for the shuffle sequence;
2719 // The cost of the loads/stores is accounted for separately.
2720 //
2721 static const CostTblEntry AVX2InterleavedLoadTbl[] = {
Mohammed Agabaria6e6d5322017-11-16 09:38:32 +00002722 { 2, MVT::v4i64, 6 }, //(load 8i64 and) deinterleave into 2 x 4i64
2723 { 2, MVT::v4f64, 6 }, //(load 8f64 and) deinterleave into 2 x 4f64
2724
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002725 { 3, MVT::v2i8, 10 }, //(load 6i8 and) deinterleave into 3 x 2i8
2726 { 3, MVT::v4i8, 4 }, //(load 12i8 and) deinterleave into 3 x 4i8
2727 { 3, MVT::v8i8, 9 }, //(load 24i8 and) deinterleave into 3 x 8i8
Michael Zuckerman49293262017-10-18 11:41:55 +00002728 { 3, MVT::v16i8, 11}, //(load 48i8 and) deinterleave into 3 x 16i8
2729 { 3, MVT::v32i8, 13}, //(load 96i8 and) deinterleave into 3 x 32i8
Mohammed Agabaria66917582017-11-06 10:56:20 +00002730 { 3, MVT::v8f32, 17 }, //(load 24f32 and)deinterleave into 3 x 8f32
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002731
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002732 { 4, MVT::v2i8, 12 }, //(load 8i8 and) deinterleave into 4 x 2i8
2733 { 4, MVT::v4i8, 4 }, //(load 16i8 and) deinterleave into 4 x 4i8
2734 { 4, MVT::v8i8, 20 }, //(load 32i8 and) deinterleave into 4 x 8i8
2735 { 4, MVT::v16i8, 39 }, //(load 64i8 and) deinterleave into 4 x 16i8
Mohammed Agabaria66917582017-11-06 10:56:20 +00002736 { 4, MVT::v32i8, 80 }, //(load 128i8 and) deinterleave into 4 x 32i8
2737
2738 { 8, MVT::v8f32, 40 } //(load 64f32 and)deinterleave into 8 x 8f32
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002739 };
2740
2741 static const CostTblEntry AVX2InterleavedStoreTbl[] = {
Mohammed Agabaria6e6d5322017-11-16 09:38:32 +00002742 { 2, MVT::v4i64, 6 }, //interleave into 2 x 4i64 into 8i64 (and store)
2743 { 2, MVT::v4f64, 6 }, //interleave into 2 x 4f64 into 8f64 (and store)
2744
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002745 { 3, MVT::v2i8, 7 }, //interleave 3 x 2i8 into 6i8 (and store)
2746 { 3, MVT::v4i8, 8 }, //interleave 3 x 4i8 into 12i8 (and store)
2747 { 3, MVT::v8i8, 11 }, //interleave 3 x 8i8 into 24i8 (and store)
Michael Zuckerman49293262017-10-18 11:41:55 +00002748 { 3, MVT::v16i8, 11 }, //interleave 3 x 16i8 into 48i8 (and store)
2749 { 3, MVT::v32i8, 13 }, //interleave 3 x 32i8 into 96i8 (and store)
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002750
2751 { 4, MVT::v2i8, 12 }, //interleave 4 x 2i8 into 8i8 (and store)
2752 { 4, MVT::v4i8, 9 }, //interleave 4 x 4i8 into 16i8 (and store)
Michael Zuckerman49293262017-10-18 11:41:55 +00002753 { 4, MVT::v8i8, 10 }, //interleave 4 x 8i8 into 32i8 (and store)
2754 { 4, MVT::v16i8, 10 }, //interleave 4 x 16i8 into 64i8 (and store)
2755 { 4, MVT::v32i8, 12 } //interleave 4 x 32i8 into 128i8 (and store)
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002756 };
2757
2758 if (Opcode == Instruction::Load) {
2759 if (const auto *Entry =
2760 CostTableLookup(AVX2InterleavedLoadTbl, Factor, ETy.getSimpleVT()))
2761 return NumOfMemOps * MemOpCost + Entry->Cost;
2762 } else {
2763 assert(Opcode == Instruction::Store &&
2764 "Expected Store Instruction at this point");
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002765 if (const auto *Entry =
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002766 CostTableLookup(AVX2InterleavedStoreTbl, Factor, ETy.getSimpleVT()))
2767 return NumOfMemOps * MemOpCost + Entry->Cost;
2768 }
2769
2770 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2771 Alignment, AddressSpace);
2772}
2773
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002774// Get estimation for interleaved load/store operations and strided load.
2775// \p Indices contains indices for strided load.
2776// \p Factor - the factor of interleaving.
2777// AVX-512 provides 3-src shuffles that significantly reduces the cost.
2778int X86TTIImpl::getInterleavedMemoryOpCostAVX512(unsigned Opcode, Type *VecTy,
2779 unsigned Factor,
2780 ArrayRef<unsigned> Indices,
2781 unsigned Alignment,
2782 unsigned AddressSpace) {
2783
2784 // VecTy for interleave memop is <VF*Factor x Elt>.
2785 // So, for VF=4, Interleave Factor = 3, Element type = i32 we have
2786 // VecTy = <12 x i32>.
2787
2788 // Calculate the number of memory operations (NumOfMemOps), required
2789 // for load/store the VecTy.
2790 MVT LegalVT = getTLI()->getTypeLegalizationCost(DL, VecTy).second;
2791 unsigned VecTySize = DL.getTypeStoreSize(VecTy);
2792 unsigned LegalVTSize = LegalVT.getStoreSize();
2793 unsigned NumOfMemOps = (VecTySize + LegalVTSize - 1) / LegalVTSize;
2794
2795 // Get the cost of one memory operation.
2796 Type *SingleMemOpTy = VectorType::get(VecTy->getVectorElementType(),
2797 LegalVT.getVectorNumElements());
2798 unsigned MemOpCost =
2799 getMemoryOpCost(Opcode, SingleMemOpTy, Alignment, AddressSpace);
2800
Michael Zuckerman49293262017-10-18 11:41:55 +00002801 unsigned VF = VecTy->getVectorNumElements() / Factor;
2802 MVT VT = MVT::getVectorVT(MVT::getVT(VecTy->getScalarType()), VF);
2803
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002804 if (Opcode == Instruction::Load) {
Michael Zuckerman49293262017-10-18 11:41:55 +00002805 // The tables (AVX512InterleavedLoadTbl and AVX512InterleavedStoreTbl)
2806 // contain the cost of the optimized shuffle sequence that the
2807 // X86InterleavedAccess pass will generate.
2808 // The cost of loads and stores are computed separately from the table.
2809
2810 // X86InterleavedAccess support only the following interleaved-access group.
2811 static const CostTblEntry AVX512InterleavedLoadTbl[] = {
2812 {3, MVT::v16i8, 12}, //(load 48i8 and) deinterleave into 3 x 16i8
2813 {3, MVT::v32i8, 14}, //(load 96i8 and) deinterleave into 3 x 32i8
2814 {3, MVT::v64i8, 22}, //(load 96i8 and) deinterleave into 3 x 32i8
2815 };
2816
2817 if (const auto *Entry =
2818 CostTableLookup(AVX512InterleavedLoadTbl, Factor, VT))
2819 return NumOfMemOps * MemOpCost + Entry->Cost;
2820 //If an entry does not exist, fallback to the default implementation.
2821
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002822 // Kind of shuffle depends on number of loaded values.
2823 // If we load the entire data in one register, we can use a 1-src shuffle.
2824 // Otherwise, we'll merge 2 sources in each operation.
2825 TTI::ShuffleKind ShuffleKind =
2826 (NumOfMemOps > 1) ? TTI::SK_PermuteTwoSrc : TTI::SK_PermuteSingleSrc;
2827
2828 unsigned ShuffleCost =
2829 getShuffleCost(ShuffleKind, SingleMemOpTy, 0, nullptr);
2830
2831 unsigned NumOfLoadsInInterleaveGrp =
2832 Indices.size() ? Indices.size() : Factor;
2833 Type *ResultTy = VectorType::get(VecTy->getVectorElementType(),
2834 VecTy->getVectorNumElements() / Factor);
2835 unsigned NumOfResults =
2836 getTLI()->getTypeLegalizationCost(DL, ResultTy).first *
2837 NumOfLoadsInInterleaveGrp;
2838
2839 // About a half of the loads may be folded in shuffles when we have only
2840 // one result. If we have more than one result, we do not fold loads at all.
2841 unsigned NumOfUnfoldedLoads =
2842 NumOfResults > 1 ? NumOfMemOps : NumOfMemOps / 2;
2843
2844 // Get a number of shuffle operations per result.
2845 unsigned NumOfShufflesPerResult =
2846 std::max((unsigned)1, (unsigned)(NumOfMemOps - 1));
2847
2848 // The SK_MergeTwoSrc shuffle clobbers one of src operands.
2849 // When we have more than one destination, we need additional instructions
2850 // to keep sources.
2851 unsigned NumOfMoves = 0;
2852 if (NumOfResults > 1 && ShuffleKind == TTI::SK_PermuteTwoSrc)
2853 NumOfMoves = NumOfResults * NumOfShufflesPerResult / 2;
2854
2855 int Cost = NumOfResults * NumOfShufflesPerResult * ShuffleCost +
2856 NumOfUnfoldedLoads * MemOpCost + NumOfMoves;
2857
2858 return Cost;
2859 }
2860
2861 // Store.
2862 assert(Opcode == Instruction::Store &&
2863 "Expected Store Instruction at this point");
Michael Zuckerman49293262017-10-18 11:41:55 +00002864 // X86InterleavedAccess support only the following interleaved-access group.
2865 static const CostTblEntry AVX512InterleavedStoreTbl[] = {
2866 {3, MVT::v16i8, 12}, // interleave 3 x 16i8 into 48i8 (and store)
2867 {3, MVT::v32i8, 14}, // interleave 3 x 32i8 into 96i8 (and store)
2868 {3, MVT::v64i8, 26}, // interleave 3 x 64i8 into 96i8 (and store)
2869
2870 {4, MVT::v8i8, 10}, // interleave 4 x 8i8 into 32i8 (and store)
2871 {4, MVT::v16i8, 11}, // interleave 4 x 16i8 into 64i8 (and store)
2872 {4, MVT::v32i8, 14}, // interleave 4 x 32i8 into 128i8 (and store)
2873 {4, MVT::v64i8, 24} // interleave 4 x 32i8 into 256i8 (and store)
2874 };
2875
2876 if (const auto *Entry =
2877 CostTableLookup(AVX512InterleavedStoreTbl, Factor, VT))
2878 return NumOfMemOps * MemOpCost + Entry->Cost;
2879 //If an entry does not exist, fallback to the default implementation.
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002880
2881 // There is no strided stores meanwhile. And store can't be folded in
2882 // shuffle.
2883 unsigned NumOfSources = Factor; // The number of values to be merged.
2884 unsigned ShuffleCost =
2885 getShuffleCost(TTI::SK_PermuteTwoSrc, SingleMemOpTy, 0, nullptr);
2886 unsigned NumOfShufflesPerStore = NumOfSources - 1;
2887
2888 // The SK_MergeTwoSrc shuffle clobbers one of src operands.
2889 // We need additional instructions to keep sources.
2890 unsigned NumOfMoves = NumOfMemOps * NumOfShufflesPerStore / 2;
2891 int Cost = NumOfMemOps * (MemOpCost + NumOfShufflesPerStore * ShuffleCost) +
2892 NumOfMoves;
2893 return Cost;
2894}
2895
2896int X86TTIImpl::getInterleavedMemoryOpCost(unsigned Opcode, Type *VecTy,
2897 unsigned Factor,
2898 ArrayRef<unsigned> Indices,
2899 unsigned Alignment,
2900 unsigned AddressSpace) {
Craig Topper8b0f1852017-12-06 18:40:46 +00002901 auto isSupportedOnAVX512 = [](Type *VecTy, bool HasBW) {
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002902 Type *EltTy = VecTy->getVectorElementType();
2903 if (EltTy->isFloatTy() || EltTy->isDoubleTy() || EltTy->isIntegerTy(64) ||
2904 EltTy->isIntegerTy(32) || EltTy->isPointerTy())
2905 return true;
Craig Topper8b0f1852017-12-06 18:40:46 +00002906 if (EltTy->isIntegerTy(16) || EltTy->isIntegerTy(8))
2907 return HasBW;
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002908 return false;
2909 };
Craig Topper8b0f1852017-12-06 18:40:46 +00002910 if (ST->hasAVX512() && isSupportedOnAVX512(VecTy, ST->hasBWI()))
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002911 return getInterleavedMemoryOpCostAVX512(Opcode, VecTy, Factor, Indices,
2912 Alignment, AddressSpace);
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002913 if (ST->hasAVX2())
2914 return getInterleavedMemoryOpCostAVX2(Opcode, VecTy, Factor, Indices,
2915 Alignment, AddressSpace);
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002916
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002917 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2918 Alignment, AddressSpace);
2919}