blob: 5b0307114bbf69405d46d980b1178a1fd903a030 [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 &&
248 Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
249 Opd2PropInfo == TargetTransformInfo::OP_PowerOf2) {
250 // On X86, vector signed division by constants power-of-two are
251 // normally expanded to the sequence SRA + SRL + ADD + SRA.
252 // The OperandValue properties many not be same as that of previous
253 // operation;conservatively assume OP_None.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000254 int Cost = 2 * getArithmeticInstrCost(Instruction::AShr, Ty, Op1Info,
255 Op2Info, TargetTransformInfo::OP_None,
256 TargetTransformInfo::OP_None);
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000257 Cost += getArithmeticInstrCost(Instruction::LShr, Ty, Op1Info, Op2Info,
258 TargetTransformInfo::OP_None,
259 TargetTransformInfo::OP_None);
260 Cost += getArithmeticInstrCost(Instruction::Add, Ty, Op1Info, Op2Info,
261 TargetTransformInfo::OP_None,
262 TargetTransformInfo::OP_None);
263
264 return Cost;
265 }
266
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000267 static const CostTblEntry AVX512BWUniformConstCostTable[] = {
Simon Pilgrim9c589502017-01-08 14:14:36 +0000268 { ISD::SHL, MVT::v64i8, 2 }, // psllw + pand.
269 { ISD::SRL, MVT::v64i8, 2 }, // psrlw + pand.
270 { ISD::SRA, MVT::v64i8, 4 }, // psrlw, pand, pxor, psubb.
271
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000272 { ISD::SDIV, MVT::v32i16, 6 }, // vpmulhw sequence
273 { ISD::UDIV, MVT::v32i16, 6 }, // vpmulhuw sequence
274 };
275
276 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
277 ST->hasBWI()) {
278 if (const auto *Entry = CostTableLookup(AVX512BWUniformConstCostTable, ISD,
279 LT.second))
280 return LT.first * Entry->Cost;
281 }
282
283 static const CostTblEntry AVX512UniformConstCostTable[] = {
Simon Pilgrimd419b732017-01-14 19:24:23 +0000284 { ISD::SRA, MVT::v2i64, 1 },
285 { ISD::SRA, MVT::v4i64, 1 },
286 { ISD::SRA, MVT::v8i64, 1 },
287
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000288 { ISD::SDIV, MVT::v16i32, 15 }, // vpmuldq sequence
289 { ISD::UDIV, MVT::v16i32, 15 }, // vpmuludq sequence
290 };
291
292 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
293 ST->hasAVX512()) {
294 if (const auto *Entry = CostTableLookup(AVX512UniformConstCostTable, ISD,
295 LT.second))
296 return LT.first * Entry->Cost;
297 }
298
Craig Topper4b275762015-10-28 04:02:12 +0000299 static const CostTblEntry AVX2UniformConstCostTable[] = {
Simon Pilgrim9c589502017-01-08 14:14:36 +0000300 { ISD::SHL, MVT::v32i8, 2 }, // psllw + pand.
301 { ISD::SRL, MVT::v32i8, 2 }, // psrlw + pand.
302 { ISD::SRA, MVT::v32i8, 4 }, // psrlw, pand, pxor, psubb.
303
Simon Pilgrim8fbf1c12015-07-06 22:35:19 +0000304 { ISD::SRA, MVT::v4i64, 4 }, // 2 x psrad + shuffle.
305
Benjamin Kramer7c372272014-04-26 14:53:05 +0000306 { ISD::SDIV, MVT::v16i16, 6 }, // vpmulhw sequence
307 { ISD::UDIV, MVT::v16i16, 6 }, // vpmulhuw sequence
308 { ISD::SDIV, MVT::v8i32, 15 }, // vpmuldq sequence
309 { ISD::UDIV, MVT::v8i32, 15 }, // vpmuludq sequence
310 };
311
312 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
313 ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000314 if (const auto *Entry = CostTableLookup(AVX2UniformConstCostTable, ISD,
315 LT.second))
316 return LT.first * Entry->Cost;
Benjamin Kramer7c372272014-04-26 14:53:05 +0000317 }
318
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000319 static const CostTblEntry SSE2UniformConstCostTable[] = {
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000320 { ISD::SHL, MVT::v16i8, 2 }, // psllw + pand.
321 { ISD::SRL, MVT::v16i8, 2 }, // psrlw + pand.
322 { ISD::SRA, MVT::v16i8, 4 }, // psrlw, pand, pxor, psubb.
Simon Pilgrim9c589502017-01-08 14:14:36 +0000323
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000324 { ISD::SHL, MVT::v32i8, 4+2 }, // 2*(psllw + pand) + split.
325 { ISD::SRL, MVT::v32i8, 4+2 }, // 2*(psrlw + pand) + split.
326 { ISD::SRA, MVT::v32i8, 8+2 }, // 2*(psrlw, pand, pxor, psubb) + split.
Simon Pilgrim9c589502017-01-08 14:14:36 +0000327
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000328 { ISD::SDIV, MVT::v16i16, 12+2 }, // 2*pmulhw sequence + split.
329 { ISD::SDIV, MVT::v8i16, 6 }, // pmulhw sequence
330 { ISD::UDIV, MVT::v16i16, 12+2 }, // 2*pmulhuw sequence + split.
331 { ISD::UDIV, MVT::v8i16, 6 }, // pmulhuw sequence
332 { ISD::SDIV, MVT::v8i32, 38+2 }, // 2*pmuludq sequence + split.
333 { ISD::SDIV, MVT::v4i32, 19 }, // pmuludq sequence
334 { ISD::UDIV, MVT::v8i32, 30+2 }, // 2*pmuludq sequence + split.
335 { ISD::UDIV, MVT::v4i32, 15 }, // pmuludq sequence
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000336 };
337
338 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
339 ST->hasSSE2()) {
340 // pmuldq sequence.
341 if (ISD == ISD::SDIV && LT.second == MVT::v8i32 && ST->hasAVX())
Simon Pilgrimd3f0d032017-05-14 18:52:15 +0000342 return LT.first * 32;
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000343 if (ISD == ISD::SDIV && LT.second == MVT::v4i32 && ST->hasSSE41())
344 return LT.first * 15;
345
Simon Pilgrim5bef9c62017-05-14 17:59:46 +0000346 // XOP has faster vXi8 shifts.
347 if ((ISD != ISD::SHL && ISD != ISD::SRL && ISD != ISD::SRA) ||
348 !ST->hasXOP())
349 if (const auto *Entry =
350 CostTableLookup(SSE2UniformConstCostTable, ISD, LT.second))
351 return LT.first * Entry->Cost;
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000352 }
353
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000354 static const CostTblEntry AVX2UniformCostTable[] = {
355 // Uniform splats are cheaper for the following instructions.
356 { ISD::SHL, MVT::v16i16, 1 }, // psllw.
357 { ISD::SRL, MVT::v16i16, 1 }, // psrlw.
358 { ISD::SRA, MVT::v16i16, 1 }, // psraw.
359 };
360
361 if (ST->hasAVX2() &&
362 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
363 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
364 if (const auto *Entry =
365 CostTableLookup(AVX2UniformCostTable, ISD, LT.second))
366 return LT.first * Entry->Cost;
367 }
368
369 static const CostTblEntry SSE2UniformCostTable[] = {
370 // Uniform splats are cheaper for the following instructions.
371 { ISD::SHL, MVT::v8i16, 1 }, // psllw.
372 { ISD::SHL, MVT::v4i32, 1 }, // pslld
373 { ISD::SHL, MVT::v2i64, 1 }, // psllq.
374
375 { ISD::SRL, MVT::v8i16, 1 }, // psrlw.
376 { ISD::SRL, MVT::v4i32, 1 }, // psrld.
377 { ISD::SRL, MVT::v2i64, 1 }, // psrlq.
378
379 { ISD::SRA, MVT::v8i16, 1 }, // psraw.
380 { ISD::SRA, MVT::v4i32, 1 }, // psrad.
381 };
382
383 if (ST->hasSSE2() &&
384 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
385 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
386 if (const auto *Entry =
387 CostTableLookup(SSE2UniformCostTable, ISD, LT.second))
388 return LT.first * Entry->Cost;
389 }
390
Simon Pilgrim820e1322016-10-27 15:27:00 +0000391 static const CostTblEntry AVX512DQCostTable[] = {
392 { ISD::MUL, MVT::v2i64, 1 },
393 { ISD::MUL, MVT::v4i64, 1 },
394 { ISD::MUL, MVT::v8i64, 1 }
395 };
396
397 // Look for AVX512DQ lowering tricks for custom cases.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000398 if (ST->hasDQI())
399 if (const auto *Entry = CostTableLookup(AVX512DQCostTable, ISD, LT.second))
Simon Pilgrim820e1322016-10-27 15:27:00 +0000400 return LT.first * Entry->Cost;
Simon Pilgrim820e1322016-10-27 15:27:00 +0000401
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000402 static const CostTblEntry AVX512BWCostTable[] = {
Simon Pilgrim6ed996c2017-01-15 20:44:00 +0000403 { ISD::SHL, MVT::v8i16, 1 }, // vpsllvw
404 { ISD::SRL, MVT::v8i16, 1 }, // vpsrlvw
405 { ISD::SRA, MVT::v8i16, 1 }, // vpsravw
406
407 { ISD::SHL, MVT::v16i16, 1 }, // vpsllvw
408 { ISD::SRL, MVT::v16i16, 1 }, // vpsrlvw
409 { ISD::SRA, MVT::v16i16, 1 }, // vpsravw
410
Simon Pilgrima4109d62017-01-07 17:54:10 +0000411 { ISD::SHL, MVT::v32i16, 1 }, // vpsllvw
412 { ISD::SRL, MVT::v32i16, 1 }, // vpsrlvw
413 { ISD::SRA, MVT::v32i16, 1 }, // vpsravw
414
Simon Pilgrim5a81fef2017-01-11 10:36:51 +0000415 { ISD::SHL, MVT::v64i8, 11 }, // vpblendvb sequence.
416 { ISD::SRL, MVT::v64i8, 11 }, // vpblendvb sequence.
417 { ISD::SRA, MVT::v64i8, 24 }, // vpblendvb sequence.
418
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000419 { ISD::MUL, MVT::v64i8, 11 }, // extend/pmullw/trunc sequence.
420 { ISD::MUL, MVT::v32i8, 4 }, // extend/pmullw/trunc sequence.
421 { ISD::MUL, MVT::v16i8, 4 }, // extend/pmullw/trunc sequence.
422
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000423 // Vectorizing division is a bad idea. See the SSE2 table for more comments.
424 { ISD::SDIV, MVT::v64i8, 64*20 },
425 { ISD::SDIV, MVT::v32i16, 32*20 },
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000426 { ISD::UDIV, MVT::v64i8, 64*20 },
Simon Pilgrimd8333372017-01-06 11:12:53 +0000427 { ISD::UDIV, MVT::v32i16, 32*20 }
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000428 };
429
430 // Look for AVX512BW lowering tricks for custom cases.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000431 if (ST->hasBWI())
432 if (const auto *Entry = CostTableLookup(AVX512BWCostTable, ISD, LT.second))
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000433 return LT.first * Entry->Cost;
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000434
Craig Topper4b275762015-10-28 04:02:12 +0000435 static const CostTblEntry AVX512CostTable[] = {
Simon Pilgrimd8333372017-01-06 11:12:53 +0000436 { ISD::SHL, MVT::v16i32, 1 },
437 { ISD::SRL, MVT::v16i32, 1 },
438 { ISD::SRA, MVT::v16i32, 1 },
Simon Pilgrimd419b732017-01-14 19:24:23 +0000439
Simon Pilgrimd8333372017-01-06 11:12:53 +0000440 { ISD::SHL, MVT::v8i64, 1 },
441 { ISD::SRL, MVT::v8i64, 1 },
Simon Pilgrimd419b732017-01-14 19:24:23 +0000442
443 { ISD::SRA, MVT::v2i64, 1 },
444 { ISD::SRA, MVT::v4i64, 1 },
Simon Pilgrimd8333372017-01-06 11:12:53 +0000445 { ISD::SRA, MVT::v8i64, 1 },
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000446
Simon Pilgrimd8333372017-01-06 11:12:53 +0000447 { ISD::MUL, MVT::v32i8, 13 }, // extend/pmullw/trunc sequence.
448 { ISD::MUL, MVT::v16i8, 5 }, // extend/pmullw/trunc sequence.
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000449 { ISD::MUL, MVT::v16i32, 1 }, // pmulld (Skylake from agner.org)
450 { ISD::MUL, MVT::v8i32, 1 }, // pmulld (Skylake from agner.org)
451 { ISD::MUL, MVT::v4i32, 1 }, // pmulld (Skylake from agner.org)
Simon Pilgrimd8333372017-01-06 11:12:53 +0000452 { ISD::MUL, MVT::v8i64, 8 }, // 3*pmuludq/3*shift/2*add
453
Simon Pilgrim9929f902018-02-26 22:10:17 +0000454 { ISD::FADD, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
455 { ISD::FSUB, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
456 { ISD::FMUL, MVT::v8f64, 1 }, // Skylake from http://www.agner.org/
457
458 { ISD::FADD, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
459 { ISD::FSUB, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
460 { ISD::FMUL, MVT::v16f32, 1 }, // Skylake from http://www.agner.org/
461
Simon Pilgrimd8333372017-01-06 11:12:53 +0000462 // Vectorizing division is a bad idea. See the SSE2 table for more comments.
463 { ISD::SDIV, MVT::v16i32, 16*20 },
464 { ISD::SDIV, MVT::v8i64, 8*20 },
465 { ISD::UDIV, MVT::v16i32, 16*20 },
466 { ISD::UDIV, MVT::v8i64, 8*20 }
Elena Demikhovsky27012472014-09-16 07:57:37 +0000467 };
468
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000469 if (ST->hasAVX512())
Craig Topperee0c8592015-10-27 04:14:24 +0000470 if (const auto *Entry = CostTableLookup(AVX512CostTable, ISD, LT.second))
471 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000472
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000473 static const CostTblEntry AVX2ShiftCostTable[] = {
Michael Liao70dd7f92013-03-20 22:01:10 +0000474 // Shifts on v4i64/v8i32 on AVX2 is legal even though we declare to
475 // customize them to detect the cases where shift amount is a scalar one.
476 { ISD::SHL, MVT::v4i32, 1 },
477 { ISD::SRL, MVT::v4i32, 1 },
478 { ISD::SRA, MVT::v4i32, 1 },
479 { ISD::SHL, MVT::v8i32, 1 },
480 { ISD::SRL, MVT::v8i32, 1 },
481 { ISD::SRA, MVT::v8i32, 1 },
482 { ISD::SHL, MVT::v2i64, 1 },
483 { ISD::SRL, MVT::v2i64, 1 },
484 { ISD::SHL, MVT::v4i64, 1 },
485 { ISD::SRL, MVT::v4i64, 1 },
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000486 };
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000487
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000488 // Look for AVX2 lowering tricks.
489 if (ST->hasAVX2()) {
490 if (ISD == ISD::SHL && LT.second == MVT::v16i16 &&
491 (Op2Info == TargetTransformInfo::OK_UniformConstantValue ||
492 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue))
493 // On AVX2, a packed v16i16 shift left by a constant build_vector
494 // is lowered into a vector multiply (vpmullw).
495 return LT.first;
496
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000497 if (const auto *Entry = CostTableLookup(AVX2ShiftCostTable, ISD, LT.second))
Craig Topperee0c8592015-10-27 04:14:24 +0000498 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000499 }
500
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000501 static const CostTblEntry XOPShiftCostTable[] = {
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000502 // 128bit shifts take 1cy, but right shifts require negation beforehand.
503 { ISD::SHL, MVT::v16i8, 1 },
504 { ISD::SRL, MVT::v16i8, 2 },
505 { ISD::SRA, MVT::v16i8, 2 },
506 { ISD::SHL, MVT::v8i16, 1 },
507 { ISD::SRL, MVT::v8i16, 2 },
508 { ISD::SRA, MVT::v8i16, 2 },
509 { ISD::SHL, MVT::v4i32, 1 },
510 { ISD::SRL, MVT::v4i32, 2 },
511 { ISD::SRA, MVT::v4i32, 2 },
512 { ISD::SHL, MVT::v2i64, 1 },
513 { ISD::SRL, MVT::v2i64, 2 },
514 { ISD::SRA, MVT::v2i64, 2 },
515 // 256bit shifts require splitting if AVX2 didn't catch them above.
Simon Pilgrim4599eaa2017-05-14 13:38:53 +0000516 { ISD::SHL, MVT::v32i8, 2+2 },
517 { ISD::SRL, MVT::v32i8, 4+2 },
518 { ISD::SRA, MVT::v32i8, 4+2 },
519 { ISD::SHL, MVT::v16i16, 2+2 },
520 { ISD::SRL, MVT::v16i16, 4+2 },
521 { ISD::SRA, MVT::v16i16, 4+2 },
522 { ISD::SHL, MVT::v8i32, 2+2 },
523 { ISD::SRL, MVT::v8i32, 4+2 },
524 { ISD::SRA, MVT::v8i32, 4+2 },
525 { ISD::SHL, MVT::v4i64, 2+2 },
526 { ISD::SRL, MVT::v4i64, 4+2 },
527 { ISD::SRA, MVT::v4i64, 4+2 },
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000528 };
529
530 // Look for XOP lowering tricks.
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000531 if (ST->hasXOP())
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000532 if (const auto *Entry = CostTableLookup(XOPShiftCostTable, ISD, LT.second))
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000533 return LT.first * Entry->Cost;
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000534
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000535 static const CostTblEntry SSE2UniformShiftCostTable[] = {
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000536 // Uniform splats are cheaper for the following instructions.
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000537 { ISD::SHL, MVT::v16i16, 2+2 }, // 2*psllw + split.
538 { ISD::SHL, MVT::v8i32, 2+2 }, // 2*pslld + split.
539 { ISD::SHL, MVT::v4i64, 2+2 }, // 2*psllq + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000540
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000541 { ISD::SRL, MVT::v16i16, 2+2 }, // 2*psrlw + split.
542 { ISD::SRL, MVT::v8i32, 2+2 }, // 2*psrld + split.
543 { ISD::SRL, MVT::v4i64, 2+2 }, // 2*psrlq + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000544
Simon Pilgrimde4467b2017-05-14 20:02:34 +0000545 { ISD::SRA, MVT::v16i16, 2+2 }, // 2*psraw + split.
546 { ISD::SRA, MVT::v8i32, 2+2 }, // 2*psrad + split.
547 { ISD::SRA, MVT::v2i64, 4 }, // 2*psrad + shuffle.
548 { ISD::SRA, MVT::v4i64, 8+2 }, // 2*(2*psrad + shuffle) + split.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000549 };
550
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000551 if (ST->hasSSE2() &&
552 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
553 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
Simon Pilgrimf96b4ab2017-05-14 20:25:42 +0000554
555 // Handle AVX2 uniform v4i64 ISD::SRA, it's not worth a table.
556 if (ISD == ISD::SRA && LT.second == MVT::v4i64 && ST->hasAVX2())
557 return LT.first * 4; // 2*psrad + shuffle.
558
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000559 if (const auto *Entry =
Simon Pilgrim1fa54872017-01-08 13:12:03 +0000560 CostTableLookup(SSE2UniformShiftCostTable, ISD, LT.second))
Craig Topperee0c8592015-10-27 04:14:24 +0000561 return LT.first * Entry->Cost;
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000562 }
563
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000564 if (ISD == ISD::SHL &&
565 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue) {
Craig Toppereda02a92015-10-25 03:15:29 +0000566 MVT VT = LT.second;
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000567 // Vector shift left by non uniform constant can be lowered
Simon Pilgrime70644d2017-01-07 21:33:00 +0000568 // into vector multiply.
569 if (((VT == MVT::v8i16 || VT == MVT::v4i32) && ST->hasSSE2()) ||
570 ((VT == MVT::v16i16 || VT == MVT::v8i32) && ST->hasAVX()))
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000571 ISD = ISD::MUL;
572 }
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000573
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000574 static const CostTblEntry AVX2CostTable[] = {
575 { ISD::SHL, MVT::v32i8, 11 }, // vpblendvb sequence.
576 { ISD::SHL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
577
578 { ISD::SRL, MVT::v32i8, 11 }, // vpblendvb sequence.
579 { ISD::SRL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
580
581 { ISD::SRA, MVT::v32i8, 24 }, // vpblendvb sequence.
582 { ISD::SRA, MVT::v16i16, 10 }, // extend/vpsravd/pack sequence.
583 { ISD::SRA, MVT::v2i64, 4 }, // srl/xor/sub sequence.
584 { ISD::SRA, MVT::v4i64, 4 }, // srl/xor/sub sequence.
585
586 { ISD::SUB, MVT::v32i8, 1 }, // psubb
587 { ISD::ADD, MVT::v32i8, 1 }, // paddb
588 { ISD::SUB, MVT::v16i16, 1 }, // psubw
589 { ISD::ADD, MVT::v16i16, 1 }, // paddw
590 { ISD::SUB, MVT::v8i32, 1 }, // psubd
591 { ISD::ADD, MVT::v8i32, 1 }, // paddd
592 { ISD::SUB, MVT::v4i64, 1 }, // psubq
593 { ISD::ADD, MVT::v4i64, 1 }, // paddq
594
595 { ISD::MUL, MVT::v32i8, 17 }, // extend/pmullw/trunc sequence.
596 { ISD::MUL, MVT::v16i8, 7 }, // extend/pmullw/trunc sequence.
597 { ISD::MUL, MVT::v16i16, 1 }, // pmullw
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000598 { ISD::MUL, MVT::v8i32, 2 }, // pmulld (Haswell from agner.org)
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000599 { ISD::MUL, MVT::v4i64, 8 }, // 3*pmuludq/3*shift/2*add
600
Simon Pilgrim9929f902018-02-26 22:10:17 +0000601 { ISD::FADD, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
602 { ISD::FADD, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
603 { ISD::FSUB, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
604 { ISD::FSUB, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
605 { ISD::FMUL, MVT::v4f64, 1 }, // Haswell from http://www.agner.org/
606 { ISD::FMUL, MVT::v8f32, 1 }, // Haswell from http://www.agner.org/
607
Simon Pilgrim82e3e052017-01-07 21:47:10 +0000608 { ISD::FDIV, MVT::f32, 7 }, // Haswell from http://www.agner.org/
609 { ISD::FDIV, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
610 { ISD::FDIV, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
611 { ISD::FDIV, MVT::f64, 14 }, // Haswell from http://www.agner.org/
612 { ISD::FDIV, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
613 { ISD::FDIV, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
614 };
615
616 // Look for AVX2 lowering tricks for custom cases.
617 if (ST->hasAVX2())
618 if (const auto *Entry = CostTableLookup(AVX2CostTable, ISD, LT.second))
619 return LT.first * Entry->Cost;
620
Simon Pilgrim100eae12017-01-07 17:03:51 +0000621 static const CostTblEntry AVX1CostTable[] = {
622 // We don't have to scalarize unsupported ops. We can issue two half-sized
623 // operations and we only need to extract the upper YMM half.
624 // Two ops + 1 extract + 1 insert = 4.
Simon Pilgrim72599712017-01-07 18:19:25 +0000625 { ISD::MUL, MVT::v16i16, 4 },
626 { ISD::MUL, MVT::v8i32, 4 },
627 { ISD::SUB, MVT::v32i8, 4 },
628 { ISD::ADD, MVT::v32i8, 4 },
629 { ISD::SUB, MVT::v16i16, 4 },
630 { ISD::ADD, MVT::v16i16, 4 },
631 { ISD::SUB, MVT::v8i32, 4 },
632 { ISD::ADD, MVT::v8i32, 4 },
633 { ISD::SUB, MVT::v4i64, 4 },
634 { ISD::ADD, MVT::v4i64, 4 },
Simon Pilgrim100eae12017-01-07 17:03:51 +0000635
636 // A v4i64 multiply is custom lowered as two split v2i64 vectors that then
637 // are lowered as a series of long multiplies(3), shifts(3) and adds(2)
638 // Because we believe v4i64 to be a legal type, we must also include the
639 // extract+insert in the cost table. Therefore, the cost here is 18
640 // instead of 8.
Simon Pilgrim72599712017-01-07 18:19:25 +0000641 { ISD::MUL, MVT::v4i64, 18 },
642
643 { ISD::MUL, MVT::v32i8, 26 }, // extend/pmullw/trunc sequence.
644
645 { ISD::FDIV, MVT::f32, 14 }, // SNB from http://www.agner.org/
646 { ISD::FDIV, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
647 { ISD::FDIV, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
648 { ISD::FDIV, MVT::f64, 22 }, // SNB from http://www.agner.org/
649 { ISD::FDIV, MVT::v2f64, 22 }, // SNB from http://www.agner.org/
650 { ISD::FDIV, MVT::v4f64, 44 }, // SNB from http://www.agner.org/
651
652 // Vectorizing division is a bad idea. See the SSE2 table for more comments.
653 { ISD::SDIV, MVT::v32i8, 32*20 },
654 { ISD::SDIV, MVT::v16i16, 16*20 },
655 { ISD::SDIV, MVT::v8i32, 8*20 },
656 { ISD::SDIV, MVT::v4i64, 4*20 },
657 { ISD::UDIV, MVT::v32i8, 32*20 },
658 { ISD::UDIV, MVT::v16i16, 16*20 },
659 { ISD::UDIV, MVT::v8i32, 8*20 },
660 { ISD::UDIV, MVT::v4i64, 4*20 },
Simon Pilgrim100eae12017-01-07 17:03:51 +0000661 };
662
Simon Pilgrimdf7de7a2017-01-07 17:27:39 +0000663 if (ST->hasAVX())
Simon Pilgrim100eae12017-01-07 17:03:51 +0000664 if (const auto *Entry = CostTableLookup(AVX1CostTable, ISD, LT.second))
665 return LT.first * Entry->Cost;
666
Simon Pilgrim5b06e4d2017-01-05 19:19:39 +0000667 static const CostTblEntry SSE42CostTable[] = {
Simon Pilgrim9929f902018-02-26 22:10:17 +0000668 { ISD::FADD, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
669 { ISD::FADD, MVT::f32, 1 }, // Nehalem from http://www.agner.org/
670 { ISD::FADD, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
671 { ISD::FADD, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
672
673 { ISD::FSUB, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
674 { ISD::FSUB, MVT::f32 , 1 }, // Nehalem from http://www.agner.org/
675 { ISD::FSUB, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
676 { ISD::FSUB, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
677
678 { ISD::FMUL, MVT::f64, 1 }, // Nehalem from http://www.agner.org/
679 { ISD::FMUL, MVT::f32, 1 }, // Nehalem from http://www.agner.org/
680 { ISD::FMUL, MVT::v2f64, 1 }, // Nehalem from http://www.agner.org/
681 { ISD::FMUL, MVT::v4f32, 1 }, // Nehalem from http://www.agner.org/
682
Simon Pilgrim5b06e4d2017-01-05 19:19:39 +0000683 { ISD::FDIV, MVT::f32, 14 }, // Nehalem from http://www.agner.org/
684 { ISD::FDIV, MVT::v4f32, 14 }, // Nehalem from http://www.agner.org/
685 { ISD::FDIV, MVT::f64, 22 }, // Nehalem from http://www.agner.org/
686 { ISD::FDIV, MVT::v2f64, 22 }, // Nehalem from http://www.agner.org/
687 };
688
689 if (ST->hasSSE42())
690 if (const auto *Entry = CostTableLookup(SSE42CostTable, ISD, LT.second))
691 return LT.first * Entry->Cost;
692
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000693 static const CostTblEntry SSE41CostTable[] = {
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000694 { ISD::SHL, MVT::v16i8, 11 }, // pblendvb sequence.
695 { ISD::SHL, MVT::v32i8, 2*11+2 }, // pblendvb sequence + split.
696 { ISD::SHL, MVT::v8i16, 14 }, // pblendvb sequence.
697 { ISD::SHL, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
698 { ISD::SHL, MVT::v4i32, 4 }, // pslld/paddd/cvttps2dq/pmulld
699 { ISD::SHL, MVT::v8i32, 2*4+2 }, // pslld/paddd/cvttps2dq/pmulld + split
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000700
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000701 { ISD::SRL, MVT::v16i8, 12 }, // pblendvb sequence.
702 { ISD::SRL, MVT::v32i8, 2*12+2 }, // pblendvb sequence + split.
703 { ISD::SRL, MVT::v8i16, 14 }, // pblendvb sequence.
704 { ISD::SRL, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
705 { ISD::SRL, MVT::v4i32, 11 }, // Shift each lane + blend.
706 { ISD::SRL, MVT::v8i32, 2*11+2 }, // Shift each lane + blend + split.
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000707
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000708 { ISD::SRA, MVT::v16i8, 24 }, // pblendvb sequence.
709 { ISD::SRA, MVT::v32i8, 2*24+2 }, // pblendvb sequence + split.
710 { ISD::SRA, MVT::v8i16, 14 }, // pblendvb sequence.
711 { ISD::SRA, MVT::v16i16, 2*14+2 }, // pblendvb sequence + split.
712 { ISD::SRA, MVT::v4i32, 12 }, // Shift each lane + blend.
713 { ISD::SRA, MVT::v8i32, 2*12+2 }, // Shift each lane + blend + split.
Simon Pilgrim4c050c212017-01-05 19:42:43 +0000714
Simon Pilgrimcb9a02f62018-02-10 19:27:10 +0000715 { ISD::MUL, MVT::v4i32, 2 } // pmulld (Nehalem from agner.org)
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000716 };
717
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000718 if (ST->hasSSE41())
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000719 if (const auto *Entry = CostTableLookup(SSE41CostTable, ISD, LT.second))
720 return LT.first * Entry->Cost;
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000721
Craig Topper4b275762015-10-28 04:02:12 +0000722 static const CostTblEntry SSE2CostTable[] = {
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000723 // We don't correctly identify costs of casts because they are marked as
724 // custom.
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000725 { ISD::SHL, MVT::v16i8, 26 }, // cmpgtb sequence.
726 { ISD::SHL, MVT::v8i16, 32 }, // cmpgtb sequence.
727 { ISD::SHL, MVT::v4i32, 2*5 }, // We optimized this using mul.
728 { ISD::SHL, MVT::v2i64, 4 }, // splat+shuffle sequence.
729 { ISD::SHL, MVT::v4i64, 2*4+2 }, // splat+shuffle sequence + split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000730
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000731 { ISD::SRL, MVT::v16i8, 26 }, // cmpgtb sequence.
732 { ISD::SRL, MVT::v8i16, 32 }, // cmpgtb sequence.
733 { ISD::SRL, MVT::v4i32, 16 }, // Shift each lane + blend.
734 { ISD::SRL, MVT::v2i64, 4 }, // splat+shuffle sequence.
735 { ISD::SRL, MVT::v4i64, 2*4+2 }, // splat+shuffle sequence + split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000736
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000737 { ISD::SRA, MVT::v16i8, 54 }, // unpacked cmpgtb sequence.
738 { ISD::SRA, MVT::v8i16, 32 }, // cmpgtb sequence.
739 { ISD::SRA, MVT::v4i32, 16 }, // Shift each lane + blend.
740 { ISD::SRA, MVT::v2i64, 12 }, // srl/xor/sub sequence.
741 { ISD::SRA, MVT::v4i64, 2*12+2 }, // srl/xor/sub sequence+split.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000742
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000743 { ISD::MUL, MVT::v16i8, 12 }, // extend/pmullw/trunc sequence.
744 { ISD::MUL, MVT::v8i16, 1 }, // pmullw
745 { ISD::MUL, MVT::v4i32, 6 }, // 3*pmuludq/4*shuffle
746 { ISD::MUL, MVT::v2i64, 8 }, // 3*pmuludq/3*shift/2*add
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000747
Simon Pilgrimd0ef9d82017-05-14 20:52:11 +0000748 { ISD::FDIV, MVT::f32, 23 }, // Pentium IV from http://www.agner.org/
749 { ISD::FDIV, MVT::v4f32, 39 }, // Pentium IV from http://www.agner.org/
750 { ISD::FDIV, MVT::f64, 38 }, // Pentium IV from http://www.agner.org/
751 { ISD::FDIV, MVT::v2f64, 69 }, // Pentium IV from http://www.agner.org/
Alexey Bataevd07c7312016-10-31 12:10:53 +0000752
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000753 // It is not a good idea to vectorize division. We have to scalarize it and
Arnold Schwaighofera04b9ef2013-06-25 19:14:09 +0000754 // in the process we will often end up having to spilling regular
755 // registers. The overhead of division is going to dominate most kernels
756 // anyways so try hard to prevent vectorization of division - it is
757 // generally a bad idea. Assume somewhat arbitrarily that we have to be able
758 // to hide "20 cycles" for each lane.
759 { ISD::SDIV, MVT::v16i8, 16*20 },
Simon Pilgrime70644d2017-01-07 21:33:00 +0000760 { ISD::SDIV, MVT::v8i16, 8*20 },
761 { ISD::SDIV, MVT::v4i32, 4*20 },
762 { ISD::SDIV, MVT::v2i64, 2*20 },
Arnold Schwaighofera04b9ef2013-06-25 19:14:09 +0000763 { ISD::UDIV, MVT::v16i8, 16*20 },
Simon Pilgrime70644d2017-01-07 21:33:00 +0000764 { ISD::UDIV, MVT::v8i16, 8*20 },
765 { ISD::UDIV, MVT::v4i32, 4*20 },
766 { ISD::UDIV, MVT::v2i64, 2*20 },
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000767 };
768
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000769 if (ST->hasSSE2())
Craig Topperee0c8592015-10-27 04:14:24 +0000770 if (const auto *Entry = CostTableLookup(SSE2CostTable, ISD, LT.second))
771 return LT.first * Entry->Cost;
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000772
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000773 static const CostTblEntry SSE1CostTable[] = {
Alexey Bataevd07c7312016-10-31 12:10:53 +0000774 { ISD::FDIV, MVT::f32, 17 }, // Pentium III from http://www.agner.org/
775 { ISD::FDIV, MVT::v4f32, 34 }, // Pentium III from http://www.agner.org/
776 };
777
778 if (ST->hasSSE1())
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000779 if (const auto *Entry = CostTableLookup(SSE1CostTable, ISD, LT.second))
Alexey Bataevd07c7312016-10-31 12:10:53 +0000780 return LT.first * Entry->Cost;
Simon Pilgrimaa186c62017-01-05 22:48:02 +0000781
Chandler Carruth664e3542013-01-07 01:37:14 +0000782 // Fallback to the default implementation.
Chandler Carruth705b1852015-01-31 03:43:40 +0000783 return BaseT::getArithmeticInstrCost(Opcode, Ty, Op1Info, Op2Info);
Chandler Carruth664e3542013-01-07 01:37:14 +0000784}
785
Chandler Carruth93205eb2015-08-05 18:08:10 +0000786int X86TTIImpl::getShuffleCost(TTI::ShuffleKind Kind, Type *Tp, int Index,
787 Type *SubTp) {
Simon Pilgrima62395a2017-01-05 14:33:32 +0000788 // 64-bit packed float vectors (v2f32) are widened to type v4f32.
789 // 64-bit packed integer vectors (v2i32) are promoted to type v2i64.
790 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Tp);
Karthik Bhate03a25d2014-06-20 04:32:48 +0000791
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000792 // For Broadcasts we are splatting the first element from the first input
793 // register, so only need to reference that input and all the output
794 // registers are the same.
795 if (Kind == TTI::SK_Broadcast)
796 LT.first = 1;
Simon Pilgrimbca02f92017-01-05 15:56:08 +0000797
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000798 // We are going to permute multiple sources and the result will be in multiple
799 // destinations. Providing an accurate cost only for splits where the element
800 // type remains the same.
801 if (Kind == TTI::SK_PermuteSingleSrc && LT.first != 1) {
802 MVT LegalVT = LT.second;
Alexey Bataev771ec9f2018-01-09 19:08:22 +0000803 if (LegalVT.isVector() &&
804 LegalVT.getVectorElementType().getSizeInBits() ==
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000805 Tp->getVectorElementType()->getPrimitiveSizeInBits() &&
806 LegalVT.getVectorNumElements() < Tp->getVectorNumElements()) {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000807
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000808 unsigned VecTySize = DL.getTypeStoreSize(Tp);
809 unsigned LegalVTSize = LegalVT.getStoreSize();
810 // Number of source vectors after legalization:
811 unsigned NumOfSrcs = (VecTySize + LegalVTSize - 1) / LegalVTSize;
812 // Number of destination vectors after legalization:
813 unsigned NumOfDests = LT.first;
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000814
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000815 Type *SingleOpTy = VectorType::get(Tp->getVectorElementType(),
816 LegalVT.getVectorNumElements());
Simon Pilgrimbca02f92017-01-05 15:56:08 +0000817
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000818 unsigned NumOfShuffles = (NumOfSrcs - 1) * NumOfDests;
819 return NumOfShuffles *
820 getShuffleCost(TTI::SK_PermuteTwoSrc, SingleOpTy, 0, nullptr);
821 }
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000822
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000823 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
824 }
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000825
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000826 // For 2-input shuffles, we must account for splitting the 2 inputs into many.
827 if (Kind == TTI::SK_PermuteTwoSrc && LT.first != 1) {
Elena Demikhovsky21706cb2017-01-02 10:37:52 +0000828 // We assume that source and destination have the same vector type.
Elena Demikhovsky21706cb2017-01-02 10:37:52 +0000829 int NumOfDests = LT.first;
830 int NumOfShufflesPerDest = LT.first * 2 - 1;
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000831 LT.first = NumOfDests * NumOfShufflesPerDest;
Karthik Bhate03a25d2014-06-20 04:32:48 +0000832 }
833
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000834 static const CostTblEntry AVX512VBMIShuffleTbl[] = {
835 { TTI::SK_Reverse, MVT::v64i8, 1 }, // vpermb
836 { TTI::SK_Reverse, MVT::v32i8, 1 }, // vpermb
837
838 { TTI::SK_PermuteSingleSrc, MVT::v64i8, 1 }, // vpermb
839 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 1 }, // vpermb
840
841 { TTI::SK_PermuteTwoSrc, MVT::v64i8, 1 }, // vpermt2b
842 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 1 }, // vpermt2b
843 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 1 } // vpermt2b
844 };
845
846 if (ST->hasVBMI())
847 if (const auto *Entry =
848 CostTableLookup(AVX512VBMIShuffleTbl, Kind, LT.second))
849 return LT.first * Entry->Cost;
850
851 static const CostTblEntry AVX512BWShuffleTbl[] = {
852 { TTI::SK_Broadcast, MVT::v32i16, 1 }, // vpbroadcastw
853 { TTI::SK_Broadcast, MVT::v64i8, 1 }, // vpbroadcastb
854
855 { TTI::SK_Reverse, MVT::v32i16, 1 }, // vpermw
856 { TTI::SK_Reverse, MVT::v16i16, 1 }, // vpermw
Simon Pilgrima1b8e2c2017-01-07 15:37:50 +0000857 { TTI::SK_Reverse, MVT::v64i8, 2 }, // pshufb + vshufi64x2
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000858
859 { TTI::SK_PermuteSingleSrc, MVT::v32i16, 1 }, // vpermw
860 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 1 }, // vpermw
861 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 1 }, // vpermw
862 { TTI::SK_PermuteSingleSrc, MVT::v64i8, 8 }, // extend to v32i16
863 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 3 }, // vpermw + zext/trunc
864
865 { TTI::SK_PermuteTwoSrc, MVT::v32i16, 1 }, // vpermt2w
866 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 1 }, // vpermt2w
867 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 1 }, // vpermt2w
868 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 3 }, // zext + vpermt2w + trunc
869 { TTI::SK_PermuteTwoSrc, MVT::v64i8, 19 }, // 6 * v32i8 + 1
870 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 3 } // zext + vpermt2w + trunc
871 };
872
873 if (ST->hasBWI())
874 if (const auto *Entry =
875 CostTableLookup(AVX512BWShuffleTbl, Kind, LT.second))
876 return LT.first * Entry->Cost;
877
878 static const CostTblEntry AVX512ShuffleTbl[] = {
879 { TTI::SK_Broadcast, MVT::v8f64, 1 }, // vbroadcastpd
880 { TTI::SK_Broadcast, MVT::v16f32, 1 }, // vbroadcastps
881 { TTI::SK_Broadcast, MVT::v8i64, 1 }, // vpbroadcastq
882 { TTI::SK_Broadcast, MVT::v16i32, 1 }, // vpbroadcastd
883
884 { TTI::SK_Reverse, MVT::v8f64, 1 }, // vpermpd
885 { TTI::SK_Reverse, MVT::v16f32, 1 }, // vpermps
886 { TTI::SK_Reverse, MVT::v8i64, 1 }, // vpermq
887 { TTI::SK_Reverse, MVT::v16i32, 1 }, // vpermd
888
889 { TTI::SK_PermuteSingleSrc, MVT::v8f64, 1 }, // vpermpd
890 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 1 }, // vpermpd
891 { TTI::SK_PermuteSingleSrc, MVT::v2f64, 1 }, // vpermpd
892 { TTI::SK_PermuteSingleSrc, MVT::v16f32, 1 }, // vpermps
893 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 1 }, // vpermps
894 { TTI::SK_PermuteSingleSrc, MVT::v4f32, 1 }, // vpermps
895 { TTI::SK_PermuteSingleSrc, MVT::v8i64, 1 }, // vpermq
896 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 1 }, // vpermq
897 { TTI::SK_PermuteSingleSrc, MVT::v2i64, 1 }, // vpermq
898 { TTI::SK_PermuteSingleSrc, MVT::v16i32, 1 }, // vpermd
899 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 1 }, // vpermd
900 { TTI::SK_PermuteSingleSrc, MVT::v4i32, 1 }, // vpermd
901 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 1 }, // pshufb
902
903 { TTI::SK_PermuteTwoSrc, MVT::v8f64, 1 }, // vpermt2pd
904 { TTI::SK_PermuteTwoSrc, MVT::v16f32, 1 }, // vpermt2ps
905 { TTI::SK_PermuteTwoSrc, MVT::v8i64, 1 }, // vpermt2q
906 { TTI::SK_PermuteTwoSrc, MVT::v16i32, 1 }, // vpermt2d
907 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 1 }, // vpermt2pd
908 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 1 }, // vpermt2ps
909 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 1 }, // vpermt2q
910 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 1 }, // vpermt2d
911 { TTI::SK_PermuteTwoSrc, MVT::v2f64, 1 }, // vpermt2pd
912 { TTI::SK_PermuteTwoSrc, MVT::v4f32, 1 }, // vpermt2ps
913 { TTI::SK_PermuteTwoSrc, MVT::v2i64, 1 }, // vpermt2q
914 { TTI::SK_PermuteTwoSrc, MVT::v4i32, 1 } // vpermt2d
915 };
916
917 if (ST->hasAVX512())
918 if (const auto *Entry = CostTableLookup(AVX512ShuffleTbl, Kind, LT.second))
919 return LT.first * Entry->Cost;
920
921 static const CostTblEntry AVX2ShuffleTbl[] = {
922 { TTI::SK_Broadcast, MVT::v4f64, 1 }, // vbroadcastpd
923 { TTI::SK_Broadcast, MVT::v8f32, 1 }, // vbroadcastps
924 { TTI::SK_Broadcast, MVT::v4i64, 1 }, // vpbroadcastq
925 { TTI::SK_Broadcast, MVT::v8i32, 1 }, // vpbroadcastd
926 { TTI::SK_Broadcast, MVT::v16i16, 1 }, // vpbroadcastw
927 { TTI::SK_Broadcast, MVT::v32i8, 1 }, // vpbroadcastb
928
929 { TTI::SK_Reverse, MVT::v4f64, 1 }, // vpermpd
930 { TTI::SK_Reverse, MVT::v8f32, 1 }, // vpermps
931 { TTI::SK_Reverse, MVT::v4i64, 1 }, // vpermq
932 { TTI::SK_Reverse, MVT::v8i32, 1 }, // vpermd
933 { TTI::SK_Reverse, MVT::v16i16, 2 }, // vperm2i128 + pshufb
934 { TTI::SK_Reverse, MVT::v32i8, 2 }, // vperm2i128 + pshufb
935
936 { TTI::SK_Alternate, MVT::v16i16, 1 }, // vpblendw
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000937 { TTI::SK_Alternate, MVT::v32i8, 1 }, // vpblendvb
938
Simon Pilgrim702e5fa2017-08-10 17:27:20 +0000939 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 1 }, // vpermpd
940 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 1 }, // vpermps
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000941 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 1 }, // vpermq
942 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 1 }, // vpermd
Simon Pilgrimac2e50a2017-08-10 18:29:34 +0000943 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 4 }, // vperm2i128 + 2*vpshufb
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000944 // + vpblendvb
Simon Pilgrimac2e50a2017-08-10 18:29:34 +0000945 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 4 }, // vperm2i128 + 2*vpshufb
946 // + vpblendvb
947
948 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 3 }, // 2*vpermpd + vblendpd
949 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 3 }, // 2*vpermps + vblendps
950 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 3 }, // 2*vpermq + vpblendd
951 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 3 }, // 2*vpermd + vpblendd
952 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 7 }, // 2*vperm2i128 + 4*vpshufb
953 // + vpblendvb
954 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 7 }, // 2*vperm2i128 + 4*vpshufb
Michael Kupersteine6d59fd2017-02-02 20:27:13 +0000955 // + vpblendvb
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000956 };
957
958 if (ST->hasAVX2())
959 if (const auto *Entry = CostTableLookup(AVX2ShuffleTbl, Kind, LT.second))
960 return LT.first * Entry->Cost;
961
Simon Pilgrimc63f93a2017-08-16 13:50:20 +0000962 static const CostTblEntry XOPShuffleTbl[] = {
963 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 2 }, // vperm2f128 + vpermil2pd
964 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 2 }, // vperm2f128 + vpermil2ps
965 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 2 }, // vperm2f128 + vpermil2pd
966 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 2 }, // vperm2f128 + vpermil2ps
967 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 4 }, // vextractf128 + 2*vpperm
968 // + vinsertf128
969 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 4 }, // vextractf128 + 2*vpperm
970 // + vinsertf128
971
972 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 9 }, // 2*vextractf128 + 6*vpperm
973 // + vinsertf128
974 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 1 }, // vpperm
975 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 9 }, // 2*vextractf128 + 6*vpperm
976 // + vinsertf128
977 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 1 }, // vpperm
978 };
979
980 if (ST->hasXOP())
981 if (const auto *Entry = CostTableLookup(XOPShuffleTbl, Kind, LT.second))
982 return LT.first * Entry->Cost;
983
Simon Pilgrimf74700a2017-01-05 17:56:19 +0000984 static const CostTblEntry AVX1ShuffleTbl[] = {
985 { TTI::SK_Broadcast, MVT::v4f64, 2 }, // vperm2f128 + vpermilpd
986 { TTI::SK_Broadcast, MVT::v8f32, 2 }, // vperm2f128 + vpermilps
987 { TTI::SK_Broadcast, MVT::v4i64, 2 }, // vperm2f128 + vpermilpd
988 { TTI::SK_Broadcast, MVT::v8i32, 2 }, // vperm2f128 + vpermilps
989 { TTI::SK_Broadcast, MVT::v16i16, 3 }, // vpshuflw + vpshufd + vinsertf128
990 { TTI::SK_Broadcast, MVT::v32i8, 2 }, // vpshufb + vinsertf128
991
992 { TTI::SK_Reverse, MVT::v4f64, 2 }, // vperm2f128 + vpermilpd
993 { TTI::SK_Reverse, MVT::v8f32, 2 }, // vperm2f128 + vpermilps
994 { TTI::SK_Reverse, MVT::v4i64, 2 }, // vperm2f128 + vpermilpd
995 { TTI::SK_Reverse, MVT::v8i32, 2 }, // vperm2f128 + vpermilps
996 { TTI::SK_Reverse, MVT::v16i16, 4 }, // vextractf128 + 2*pshufb
997 // + vinsertf128
998 { TTI::SK_Reverse, MVT::v32i8, 4 }, // vextractf128 + 2*pshufb
999 // + vinsertf128
1000
1001 { TTI::SK_Alternate, MVT::v4i64, 1 }, // vblendpd
1002 { TTI::SK_Alternate, MVT::v4f64, 1 }, // vblendpd
1003 { TTI::SK_Alternate, MVT::v8i32, 1 }, // vblendps
1004 { TTI::SK_Alternate, MVT::v8f32, 1 }, // vblendps
1005 { TTI::SK_Alternate, MVT::v16i16, 3 }, // vpand + vpandn + vpor
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001006 { TTI::SK_Alternate, MVT::v32i8, 3 }, // vpand + vpandn + vpor
1007
1008 { TTI::SK_PermuteSingleSrc, MVT::v4f64, 3 }, // 2*vperm2f128 + vshufpd
1009 { TTI::SK_PermuteSingleSrc, MVT::v4i64, 3 }, // 2*vperm2f128 + vshufpd
1010 { TTI::SK_PermuteSingleSrc, MVT::v8f32, 4 }, // 2*vperm2f128 + 2*vshufps
1011 { TTI::SK_PermuteSingleSrc, MVT::v8i32, 4 }, // 2*vperm2f128 + 2*vshufps
1012 { TTI::SK_PermuteSingleSrc, MVT::v16i16, 8 }, // vextractf128 + 4*pshufb
1013 // + 2*por + vinsertf128
1014 { TTI::SK_PermuteSingleSrc, MVT::v32i8, 8 }, // vextractf128 + 4*pshufb
1015 // + 2*por + vinsertf128
Simon Pilgrim73545312017-08-10 19:02:51 +00001016
1017 { TTI::SK_PermuteTwoSrc, MVT::v4f64, 4 }, // 2*vperm2f128 + 2*vshufpd
1018 { TTI::SK_PermuteTwoSrc, MVT::v8f32, 4 }, // 2*vperm2f128 + 2*vshufps
1019 { TTI::SK_PermuteTwoSrc, MVT::v4i64, 4 }, // 2*vperm2f128 + 2*vshufpd
1020 { TTI::SK_PermuteTwoSrc, MVT::v8i32, 4 }, // 2*vperm2f128 + 2*vshufps
1021 { TTI::SK_PermuteTwoSrc, MVT::v16i16, 15 }, // 2*vextractf128 + 8*pshufb
1022 // + 4*por + vinsertf128
1023 { TTI::SK_PermuteTwoSrc, MVT::v32i8, 15 }, // 2*vextractf128 + 8*pshufb
1024 // + 4*por + vinsertf128
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001025 };
1026
1027 if (ST->hasAVX())
1028 if (const auto *Entry = CostTableLookup(AVX1ShuffleTbl, Kind, LT.second))
1029 return LT.first * Entry->Cost;
1030
1031 static const CostTblEntry SSE41ShuffleTbl[] = {
1032 { TTI::SK_Alternate, MVT::v2i64, 1 }, // pblendw
1033 { TTI::SK_Alternate, MVT::v2f64, 1 }, // movsd
1034 { TTI::SK_Alternate, MVT::v4i32, 1 }, // pblendw
1035 { TTI::SK_Alternate, MVT::v4f32, 1 }, // blendps
1036 { TTI::SK_Alternate, MVT::v8i16, 1 }, // pblendw
1037 { TTI::SK_Alternate, MVT::v16i8, 1 } // pblendvb
1038 };
1039
1040 if (ST->hasSSE41())
1041 if (const auto *Entry = CostTableLookup(SSE41ShuffleTbl, Kind, LT.second))
1042 return LT.first * Entry->Cost;
1043
1044 static const CostTblEntry SSSE3ShuffleTbl[] = {
1045 { TTI::SK_Broadcast, MVT::v8i16, 1 }, // pshufb
1046 { TTI::SK_Broadcast, MVT::v16i8, 1 }, // pshufb
1047
1048 { TTI::SK_Reverse, MVT::v8i16, 1 }, // pshufb
1049 { TTI::SK_Reverse, MVT::v16i8, 1 }, // pshufb
1050
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001051 { TTI::SK_Alternate, MVT::v8i16, 3 }, // 2*pshufb + por
1052 { TTI::SK_Alternate, MVT::v16i8, 3 }, // 2*pshufb + por
Michael Kupersteine6d59fd2017-02-02 20:27:13 +00001053
1054 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 1 }, // pshufb
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001055 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 1 }, // pshufb
1056
1057 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 3 }, // 2*pshufb + por
1058 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 3 }, // 2*pshufb + por
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001059 };
1060
1061 if (ST->hasSSSE3())
1062 if (const auto *Entry = CostTableLookup(SSSE3ShuffleTbl, Kind, LT.second))
1063 return LT.first * Entry->Cost;
1064
1065 static const CostTblEntry SSE2ShuffleTbl[] = {
1066 { TTI::SK_Broadcast, MVT::v2f64, 1 }, // shufpd
1067 { TTI::SK_Broadcast, MVT::v2i64, 1 }, // pshufd
1068 { TTI::SK_Broadcast, MVT::v4i32, 1 }, // pshufd
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001069 { TTI::SK_Broadcast, MVT::v8i16, 2 }, // pshuflw + pshufd
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001070 { TTI::SK_Broadcast, MVT::v16i8, 3 }, // unpck + pshuflw + pshufd
1071
1072 { TTI::SK_Reverse, MVT::v2f64, 1 }, // shufpd
1073 { TTI::SK_Reverse, MVT::v2i64, 1 }, // pshufd
1074 { TTI::SK_Reverse, MVT::v4i32, 1 }, // pshufd
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001075 { TTI::SK_Reverse, MVT::v8i16, 3 }, // pshuflw + pshufhw + pshufd
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001076 { TTI::SK_Reverse, MVT::v16i8, 9 }, // 2*pshuflw + 2*pshufhw
1077 // + 2*pshufd + 2*unpck + packus
1078
1079 { TTI::SK_Alternate, MVT::v2i64, 1 }, // movsd
1080 { TTI::SK_Alternate, MVT::v2f64, 1 }, // movsd
1081 { TTI::SK_Alternate, MVT::v4i32, 2 }, // 2*shufps
1082 { TTI::SK_Alternate, MVT::v8i16, 3 }, // pand + pandn + por
Michael Kupersteine6d59fd2017-02-02 20:27:13 +00001083 { TTI::SK_Alternate, MVT::v16i8, 3 }, // pand + pandn + por
1084
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001085 { TTI::SK_PermuteSingleSrc, MVT::v2f64, 1 }, // shufpd
1086 { TTI::SK_PermuteSingleSrc, MVT::v2i64, 1 }, // pshufd
1087 { TTI::SK_PermuteSingleSrc, MVT::v4i32, 1 }, // pshufd
1088 { TTI::SK_PermuteSingleSrc, MVT::v8i16, 5 }, // 2*pshuflw + 2*pshufhw
1089 // + pshufd/unpck
1090 { TTI::SK_PermuteSingleSrc, MVT::v16i8, 10 }, // 2*pshuflw + 2*pshufhw
1091 // + 2*pshufd + 2*unpck + 2*packus
1092
1093 { TTI::SK_PermuteTwoSrc, MVT::v2f64, 1 }, // shufpd
1094 { TTI::SK_PermuteTwoSrc, MVT::v2i64, 1 }, // shufpd
1095 { TTI::SK_PermuteTwoSrc, MVT::v4i32, 2 }, // 2*{unpck,movsd,pshufd}
Simon Pilgrimb59c2d92017-08-10 19:32:35 +00001096 { TTI::SK_PermuteTwoSrc, MVT::v8i16, 8 }, // blend+permute
1097 { TTI::SK_PermuteTwoSrc, MVT::v16i8, 13 }, // blend+permute
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001098 };
1099
1100 if (ST->hasSSE2())
1101 if (const auto *Entry = CostTableLookup(SSE2ShuffleTbl, Kind, LT.second))
1102 return LT.first * Entry->Cost;
1103
1104 static const CostTblEntry SSE1ShuffleTbl[] = {
Simon Pilgrim702e5fa2017-08-10 17:27:20 +00001105 { TTI::SK_Broadcast, MVT::v4f32, 1 }, // shufps
1106 { TTI::SK_Reverse, MVT::v4f32, 1 }, // shufps
1107 { TTI::SK_Alternate, MVT::v4f32, 2 }, // 2*shufps
1108 { TTI::SK_PermuteSingleSrc, MVT::v4f32, 1 }, // shufps
1109 { TTI::SK_PermuteTwoSrc, MVT::v4f32, 2 }, // 2*shufps
Simon Pilgrimf74700a2017-01-05 17:56:19 +00001110 };
1111
1112 if (ST->hasSSE1())
1113 if (const auto *Entry = CostTableLookup(SSE1ShuffleTbl, Kind, LT.second))
1114 return LT.first * Entry->Cost;
1115
Chandler Carruth705b1852015-01-31 03:43:40 +00001116 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
Chandler Carruth664e3542013-01-07 01:37:14 +00001117}
1118
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001119int X86TTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src,
1120 const Instruction *I) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001121 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1122 assert(ISD && "Invalid opcode");
1123
Cong Hou59898d82015-12-11 00:31:39 +00001124 // FIXME: Need a better design of the cost table to handle non-simple types of
1125 // potential massive combinations (elem_num x src_type x dst_type).
1126
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001127 static const TypeConversionCostTblEntry AVX512DQConversionTbl[] = {
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001128 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
1129 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001130 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
1131 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001132 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
1133 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
1134
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001135 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001136 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001137 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001138 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001139 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001140 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001141
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001142 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f32, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001143 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f32, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001144 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f32, 1 },
Simon Pilgrim841d7ca2016-11-24 14:46:55 +00001145 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +00001146 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +00001147 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f64, 1 },
1148
1149 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f32, 1 },
1150 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f32, 1 },
1151 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f32, 1 },
1152 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f64, 1 },
1153 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f64, 1 },
1154 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001155 };
1156
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001157 // TODO: For AVX512DQ + AVX512VL, we also have cheap casts for 128-bit and
1158 // 256-bit wide vectors.
1159
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001160 static const TypeConversionCostTblEntry AVX512FConversionTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +00001161 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 1 },
1162 { ISD::FP_EXTEND, MVT::v8f64, MVT::v16f32, 3 },
1163 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001164
1165 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 1 },
1166 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 1 },
1167 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i64, 1 },
1168 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001169
1170 // v16i1 -> v16i32 - load + broadcast
1171 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
1172 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001173 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
1174 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
1175 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
1176 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001177 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
1178 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001179 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
1180 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001181
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001182 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001183 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001184 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001185 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001186 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001187 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
1188 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +00001189 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001190 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 26 },
1191 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001192
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001193 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001194 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001195 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001196 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
1197 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 2 },
1198 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
1199 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001200 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i16, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001201 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
1202 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 2 },
1203 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
1204 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001205 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i32, 2 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001206 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001207 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
1208 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
1209 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
1210 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
1211 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001212 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 5 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001213 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 5 },
1214 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 12 },
1215 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
1216
1217 { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f32, 1 },
1218 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f32, 1 },
1219 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 1 },
Zvi Rackover25799d92017-09-07 07:40:34 +00001220 { ISD::FP_TO_UINT, MVT::v8i16, MVT::v8f64, 2 },
1221 { ISD::FP_TO_UINT, MVT::v8i8, MVT::v8f64, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001222 { ISD::FP_TO_UINT, MVT::v16i32, MVT::v16f32, 1 },
Zvi Rackover25799d92017-09-07 07:40:34 +00001223 { ISD::FP_TO_UINT, MVT::v16i16, MVT::v16f32, 2 },
1224 { ISD::FP_TO_UINT, MVT::v16i8, MVT::v16f32, 2 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001225 };
1226
Craig Topper4b275762015-10-28 04:02:12 +00001227 static const TypeConversionCostTblEntry AVX2ConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +00001228 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
1229 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001230 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
1231 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001232 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
1233 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001234 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
1235 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
1236 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
1237 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001238 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
1239 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001240 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
1241 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001242 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
1243 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
1244
1245 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 2 },
1246 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 2 },
1247 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 2 },
1248 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 2 },
1249 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 2 },
1250 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 4 },
Elena Demikhovsky27012472014-09-16 07:57:37 +00001251
1252 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 3 },
1253 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 3 },
Quentin Colombet360460b2014-11-11 02:23:47 +00001254
1255 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 8 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001256 };
1257
Craig Topper4b275762015-10-28 04:02:12 +00001258 static const TypeConversionCostTblEntry AVXConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +00001259 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 6 },
1260 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001261 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 7 },
1262 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001263 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 6 },
1264 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001265 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 7 },
1266 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 4 },
1267 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
1268 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001269 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 6 },
1270 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001271 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
1272 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001273 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
1274 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
1275
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001276 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 4 },
1277 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
1278 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001279 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 4 },
1280 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 4 },
1281 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +00001282 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 9 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001283
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001284 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i1, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001285 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001286 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i1, 8 },
1287 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i8, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001288 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001289 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i8, 8 },
1290 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i16, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001291 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001292 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
1293 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001294 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001295 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001296
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001297 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i1, 7 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001298 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i1, 7 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001299 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i1, 6 },
1300 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i8, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001301 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001302 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 5 },
1303 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001304 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001305 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001306 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001307 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 6 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +00001308 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001309 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 9 },
Quentin Colombet85b904d2014-03-27 22:27:41 +00001310 // The generic code to compute the scalar overhead is currently broken.
1311 // Workaround this limitation by estimating the scalarization overhead
1312 // here. We have roughly 10 instructions per scalar element.
1313 // Multiply that by the vector width.
1314 // FIXME: remove that when PR19268 is fixed.
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001315 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 10 },
1316 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 20 },
1317 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
1318 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001319
Renato Goline1fb0592013-01-20 20:57:20 +00001320 { ISD::FP_TO_SINT, MVT::v4i8, MVT::v4f32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001321 { ISD::FP_TO_SINT, MVT::v8i8, MVT::v8f32, 7 },
Adam Nemet6dafe972014-03-30 18:07:13 +00001322 // This node is expanded into scalarized operations but BasicTTI is overly
1323 // optimistic estimating its cost. It computes 3 per element (one
1324 // vector-extract, one scalar conversion and one vector-insert). The
1325 // problem is that the inserts form a read-modify-write chain so latency
1326 // should be factored in too. Inflating the cost per element by 1.
1327 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 8*4 },
Adam Nemet10c4ce22014-03-31 21:54:48 +00001328 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f64, 4*4 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001329
1330 { ISD::FP_EXTEND, MVT::v4f64, MVT::v4f32, 1 },
1331 { ISD::FP_ROUND, MVT::v4f32, MVT::v4f64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001332 };
1333
Cong Hou59898d82015-12-11 00:31:39 +00001334 static const TypeConversionCostTblEntry SSE41ConversionTbl[] = {
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001335 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
1336 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001337 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1338 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1339 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
1340 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001341
Cong Hou59898d82015-12-11 00:31:39 +00001342 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1343 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001344 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1345 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1346 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1347 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1348 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1349 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1350 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1351 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1352 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1353 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1354 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1355 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1356 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1357 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1358 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
1359 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
Cong Hou59898d82015-12-11 00:31:39 +00001360
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001361 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 2 },
1362 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 1 },
1363 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001364 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001365 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 3 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001366 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001367 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 6 },
1368
Cong Hou59898d82015-12-11 00:31:39 +00001369 };
1370
1371 static const TypeConversionCostTblEntry SSE2ConversionTbl[] = {
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001372 // These are somewhat magic numbers justified by looking at the output of
1373 // Intel's IACA, running some kernels and making sure when we take
1374 // legalization into account the throughput will be overestimated.
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001375 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001376 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1377 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1378 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
Sanjay Patel04b34962016-07-06 19:15:54 +00001379 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001380 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1381 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
1382 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
Cong Hou59898d82015-12-11 00:31:39 +00001383
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001384 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1385 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
1386 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1387 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
1388 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1389 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 8 },
1390 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
1391 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001392
Simon Pilgrim4ddc92b2016-10-18 07:42:15 +00001393 { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f64, 3 },
1394
Cong Hou59898d82015-12-11 00:31:39 +00001395 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1396 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001397 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 2 },
1398 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 3 },
1399 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
1400 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 8 },
1401 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1402 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 2 },
1403 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1404 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1405 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 3 },
1406 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
1407 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 9 },
1408 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 12 },
1409 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1410 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 2 },
1411 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
1412 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 10 },
1413 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 3 },
1414 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
1415 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 6 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001416 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001417 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 3 },
1418 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 5 },
Cong Hou59898d82015-12-11 00:31:39 +00001419
Cong Hou59898d82015-12-11 00:31:39 +00001420 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001421 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 2 },
1422 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 3 },
1423 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 3 },
1424 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 3 },
1425 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
1426 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 7 },
1427 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
1428 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 10 },
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001429 };
1430
Chandler Carruth93205eb2015-08-05 18:08:10 +00001431 std::pair<int, MVT> LTSrc = TLI->getTypeLegalizationCost(DL, Src);
1432 std::pair<int, MVT> LTDest = TLI->getTypeLegalizationCost(DL, Dst);
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001433
1434 if (ST->hasSSE2() && !ST->hasAVX()) {
Cong Hou59898d82015-12-11 00:31:39 +00001435 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
Craig Topperee0c8592015-10-27 04:14:24 +00001436 LTDest.second, LTSrc.second))
1437 return LTSrc.first * Entry->Cost;
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001438 }
1439
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001440 EVT SrcTy = TLI->getValueType(DL, Src);
1441 EVT DstTy = TLI->getValueType(DL, Dst);
1442
1443 // The function getSimpleVT only handles simple value types.
1444 if (!SrcTy.isSimple() || !DstTy.isSimple())
1445 return BaseT::getCastInstrCost(Opcode, Dst, Src);
1446
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001447 if (ST->hasDQI())
1448 if (const auto *Entry = ConvertCostTableLookup(AVX512DQConversionTbl, ISD,
1449 DstTy.getSimpleVT(),
1450 SrcTy.getSimpleVT()))
1451 return Entry->Cost;
1452
1453 if (ST->hasAVX512())
1454 if (const auto *Entry = ConvertCostTableLookup(AVX512FConversionTbl, ISD,
1455 DstTy.getSimpleVT(),
1456 SrcTy.getSimpleVT()))
1457 return Entry->Cost;
1458
Tim Northoverf0e21612014-02-06 18:18:36 +00001459 if (ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001460 if (const auto *Entry = ConvertCostTableLookup(AVX2ConversionTbl, ISD,
1461 DstTy.getSimpleVT(),
1462 SrcTy.getSimpleVT()))
1463 return Entry->Cost;
Tim Northoverf0e21612014-02-06 18:18:36 +00001464 }
1465
Chandler Carruth664e3542013-01-07 01:37:14 +00001466 if (ST->hasAVX()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001467 if (const auto *Entry = ConvertCostTableLookup(AVXConversionTbl, ISD,
1468 DstTy.getSimpleVT(),
1469 SrcTy.getSimpleVT()))
1470 return Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001471 }
1472
Cong Hou59898d82015-12-11 00:31:39 +00001473 if (ST->hasSSE41()) {
1474 if (const auto *Entry = ConvertCostTableLookup(SSE41ConversionTbl, ISD,
1475 DstTy.getSimpleVT(),
1476 SrcTy.getSimpleVT()))
1477 return Entry->Cost;
1478 }
1479
1480 if (ST->hasSSE2()) {
1481 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
1482 DstTy.getSimpleVT(),
1483 SrcTy.getSimpleVT()))
1484 return Entry->Cost;
1485 }
1486
Alexey Bataeve25a6fd2017-11-07 14:23:44 +00001487 return BaseT::getCastInstrCost(Opcode, Dst, Src, I);
Chandler Carruth664e3542013-01-07 01:37:14 +00001488}
1489
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001490int X86TTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy, Type *CondTy,
1491 const Instruction *I) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001492 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001493 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Chandler Carruth664e3542013-01-07 01:37:14 +00001494
1495 MVT MTy = LT.second;
1496
1497 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1498 assert(ISD && "Invalid opcode");
1499
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001500 static const CostTblEntry SSE2CostTbl[] = {
1501 { ISD::SETCC, MVT::v2i64, 8 },
1502 { ISD::SETCC, MVT::v4i32, 1 },
1503 { ISD::SETCC, MVT::v8i16, 1 },
1504 { ISD::SETCC, MVT::v16i8, 1 },
1505 };
1506
Craig Topper4b275762015-10-28 04:02:12 +00001507 static const CostTblEntry SSE42CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001508 { ISD::SETCC, MVT::v2f64, 1 },
1509 { ISD::SETCC, MVT::v4f32, 1 },
1510 { ISD::SETCC, MVT::v2i64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001511 };
1512
Craig Topper4b275762015-10-28 04:02:12 +00001513 static const CostTblEntry AVX1CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001514 { ISD::SETCC, MVT::v4f64, 1 },
1515 { ISD::SETCC, MVT::v8f32, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001516 // AVX1 does not support 8-wide integer compare.
Renato Goline1fb0592013-01-20 20:57:20 +00001517 { ISD::SETCC, MVT::v4i64, 4 },
1518 { ISD::SETCC, MVT::v8i32, 4 },
1519 { ISD::SETCC, MVT::v16i16, 4 },
1520 { ISD::SETCC, MVT::v32i8, 4 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001521 };
1522
Craig Topper4b275762015-10-28 04:02:12 +00001523 static const CostTblEntry AVX2CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001524 { ISD::SETCC, MVT::v4i64, 1 },
1525 { ISD::SETCC, MVT::v8i32, 1 },
1526 { ISD::SETCC, MVT::v16i16, 1 },
1527 { ISD::SETCC, MVT::v32i8, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001528 };
1529
Craig Topper4b275762015-10-28 04:02:12 +00001530 static const CostTblEntry AVX512CostTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +00001531 { ISD::SETCC, MVT::v8i64, 1 },
1532 { ISD::SETCC, MVT::v16i32, 1 },
1533 { ISD::SETCC, MVT::v8f64, 1 },
1534 { ISD::SETCC, MVT::v16f32, 1 },
1535 };
1536
Craig Topperee0c8592015-10-27 04:14:24 +00001537 if (ST->hasAVX512())
1538 if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
1539 return LT.first * Entry->Cost;
Elena Demikhovsky27012472014-09-16 07:57:37 +00001540
Craig Topperee0c8592015-10-27 04:14:24 +00001541 if (ST->hasAVX2())
1542 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1543 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001544
Craig Topperee0c8592015-10-27 04:14:24 +00001545 if (ST->hasAVX())
1546 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1547 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001548
Craig Topperee0c8592015-10-27 04:14:24 +00001549 if (ST->hasSSE42())
1550 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1551 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001552
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001553 if (ST->hasSSE2())
1554 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1555 return LT.first * Entry->Cost;
1556
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001557 return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy, I);
Chandler Carruth664e3542013-01-07 01:37:14 +00001558}
1559
Anna Thomasb2a212c2017-06-06 16:45:25 +00001560unsigned X86TTIImpl::getAtomicMemIntrinsicMaxElementSize() const { return 16; }
1561
Simon Pilgrim14000b32016-05-24 08:17:50 +00001562int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
Jonas Paulssona48ea232017-03-14 06:35:36 +00001563 ArrayRef<Type *> Tys, FastMathFlags FMF,
1564 unsigned ScalarizationCostPassed) {
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001565 // Costs should match the codegen from:
1566 // BITREVERSE: llvm\test\CodeGen\X86\vector-bitreverse.ll
1567 // BSWAP: llvm\test\CodeGen\X86\bswap-vector.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001568 // CTLZ: llvm\test\CodeGen\X86\vector-lzcnt-*.ll
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001569 // CTPOP: llvm\test\CodeGen\X86\vector-popcnt-*.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001570 // CTTZ: llvm\test\CodeGen\X86\vector-tzcnt-*.ll
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001571 static const CostTblEntry AVX512CDCostTbl[] = {
1572 { ISD::CTLZ, MVT::v8i64, 1 },
1573 { ISD::CTLZ, MVT::v16i32, 1 },
1574 { ISD::CTLZ, MVT::v32i16, 8 },
1575 { ISD::CTLZ, MVT::v64i8, 20 },
1576 { ISD::CTLZ, MVT::v4i64, 1 },
1577 { ISD::CTLZ, MVT::v8i32, 1 },
1578 { ISD::CTLZ, MVT::v16i16, 4 },
1579 { ISD::CTLZ, MVT::v32i8, 10 },
1580 { ISD::CTLZ, MVT::v2i64, 1 },
1581 { ISD::CTLZ, MVT::v4i32, 1 },
1582 { ISD::CTLZ, MVT::v8i16, 4 },
1583 { ISD::CTLZ, MVT::v16i8, 4 },
1584 };
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001585 static const CostTblEntry AVX512BWCostTbl[] = {
1586 { ISD::BITREVERSE, MVT::v8i64, 5 },
1587 { ISD::BITREVERSE, MVT::v16i32, 5 },
1588 { ISD::BITREVERSE, MVT::v32i16, 5 },
1589 { ISD::BITREVERSE, MVT::v64i8, 5 },
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001590 { ISD::CTLZ, MVT::v8i64, 23 },
1591 { ISD::CTLZ, MVT::v16i32, 22 },
1592 { ISD::CTLZ, MVT::v32i16, 18 },
1593 { ISD::CTLZ, MVT::v64i8, 17 },
Simon Pilgrim6bba6062017-05-18 10:42:34 +00001594 { ISD::CTPOP, MVT::v8i64, 7 },
1595 { ISD::CTPOP, MVT::v16i32, 11 },
1596 { ISD::CTPOP, MVT::v32i16, 9 },
1597 { ISD::CTPOP, MVT::v64i8, 6 },
Simon Pilgrimd0365962017-05-17 20:22:54 +00001598 { ISD::CTTZ, MVT::v8i64, 10 },
1599 { ISD::CTTZ, MVT::v16i32, 14 },
1600 { ISD::CTTZ, MVT::v32i16, 12 },
1601 { ISD::CTTZ, MVT::v64i8, 9 },
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001602 };
1603 static const CostTblEntry AVX512CostTbl[] = {
1604 { ISD::BITREVERSE, MVT::v8i64, 36 },
1605 { ISD::BITREVERSE, MVT::v16i32, 24 },
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001606 { ISD::CTLZ, MVT::v8i64, 29 },
1607 { ISD::CTLZ, MVT::v16i32, 35 },
Simon Pilgrim6bba6062017-05-18 10:42:34 +00001608 { ISD::CTPOP, MVT::v8i64, 16 },
1609 { ISD::CTPOP, MVT::v16i32, 24 },
Simon Pilgrimd0365962017-05-17 20:22:54 +00001610 { ISD::CTTZ, MVT::v8i64, 20 },
1611 { ISD::CTTZ, MVT::v16i32, 28 },
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001612 };
Simon Pilgrim14000b32016-05-24 08:17:50 +00001613 static const CostTblEntry XOPCostTbl[] = {
1614 { ISD::BITREVERSE, MVT::v4i64, 4 },
1615 { ISD::BITREVERSE, MVT::v8i32, 4 },
1616 { ISD::BITREVERSE, MVT::v16i16, 4 },
1617 { ISD::BITREVERSE, MVT::v32i8, 4 },
1618 { ISD::BITREVERSE, MVT::v2i64, 1 },
1619 { ISD::BITREVERSE, MVT::v4i32, 1 },
1620 { ISD::BITREVERSE, MVT::v8i16, 1 },
1621 { ISD::BITREVERSE, MVT::v16i8, 1 },
1622 { ISD::BITREVERSE, MVT::i64, 3 },
1623 { ISD::BITREVERSE, MVT::i32, 3 },
1624 { ISD::BITREVERSE, MVT::i16, 3 },
1625 { ISD::BITREVERSE, MVT::i8, 3 }
1626 };
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001627 static const CostTblEntry AVX2CostTbl[] = {
1628 { ISD::BITREVERSE, MVT::v4i64, 5 },
1629 { ISD::BITREVERSE, MVT::v8i32, 5 },
1630 { ISD::BITREVERSE, MVT::v16i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001631 { ISD::BITREVERSE, MVT::v32i8, 5 },
1632 { ISD::BSWAP, MVT::v4i64, 1 },
1633 { ISD::BSWAP, MVT::v8i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001634 { ISD::BSWAP, MVT::v16i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001635 { ISD::CTLZ, MVT::v4i64, 23 },
1636 { ISD::CTLZ, MVT::v8i32, 18 },
1637 { ISD::CTLZ, MVT::v16i16, 14 },
1638 { ISD::CTLZ, MVT::v32i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001639 { ISD::CTPOP, MVT::v4i64, 7 },
1640 { ISD::CTPOP, MVT::v8i32, 11 },
1641 { ISD::CTPOP, MVT::v16i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001642 { ISD::CTPOP, MVT::v32i8, 6 },
1643 { ISD::CTTZ, MVT::v4i64, 10 },
1644 { ISD::CTTZ, MVT::v8i32, 14 },
1645 { ISD::CTTZ, MVT::v16i16, 12 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001646 { ISD::CTTZ, MVT::v32i8, 9 },
1647 { ISD::FSQRT, MVT::f32, 7 }, // Haswell from http://www.agner.org/
1648 { ISD::FSQRT, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
1649 { ISD::FSQRT, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
1650 { ISD::FSQRT, MVT::f64, 14 }, // Haswell from http://www.agner.org/
1651 { ISD::FSQRT, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
1652 { ISD::FSQRT, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001653 };
1654 static const CostTblEntry AVX1CostTbl[] = {
Simon Pilgrim2d1c6d62017-05-07 20:58:55 +00001655 { ISD::BITREVERSE, MVT::v4i64, 12 }, // 2 x 128-bit Op + extract/insert
1656 { ISD::BITREVERSE, MVT::v8i32, 12 }, // 2 x 128-bit Op + extract/insert
1657 { ISD::BITREVERSE, MVT::v16i16, 12 }, // 2 x 128-bit Op + extract/insert
1658 { ISD::BITREVERSE, MVT::v32i8, 12 }, // 2 x 128-bit Op + extract/insert
Simon Pilgrim356e8232016-06-20 23:08:21 +00001659 { ISD::BSWAP, MVT::v4i64, 4 },
1660 { ISD::BSWAP, MVT::v8i32, 4 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001661 { ISD::BSWAP, MVT::v16i16, 4 },
Simon Pilgrim2d1c6d62017-05-07 20:58:55 +00001662 { ISD::CTLZ, MVT::v4i64, 48 }, // 2 x 128-bit Op + extract/insert
1663 { ISD::CTLZ, MVT::v8i32, 38 }, // 2 x 128-bit Op + extract/insert
1664 { ISD::CTLZ, MVT::v16i16, 30 }, // 2 x 128-bit Op + extract/insert
1665 { ISD::CTLZ, MVT::v32i8, 20 }, // 2 x 128-bit Op + extract/insert
1666 { ISD::CTPOP, MVT::v4i64, 16 }, // 2 x 128-bit Op + extract/insert
1667 { ISD::CTPOP, MVT::v8i32, 24 }, // 2 x 128-bit Op + extract/insert
1668 { ISD::CTPOP, MVT::v16i16, 20 }, // 2 x 128-bit Op + extract/insert
1669 { ISD::CTPOP, MVT::v32i8, 14 }, // 2 x 128-bit Op + extract/insert
1670 { ISD::CTTZ, MVT::v4i64, 22 }, // 2 x 128-bit Op + extract/insert
1671 { ISD::CTTZ, MVT::v8i32, 30 }, // 2 x 128-bit Op + extract/insert
1672 { ISD::CTTZ, MVT::v16i16, 26 }, // 2 x 128-bit Op + extract/insert
1673 { ISD::CTTZ, MVT::v32i8, 20 }, // 2 x 128-bit Op + extract/insert
Alexey Bataevd07c7312016-10-31 12:10:53 +00001674 { ISD::FSQRT, MVT::f32, 14 }, // SNB from http://www.agner.org/
1675 { ISD::FSQRT, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
1676 { ISD::FSQRT, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
1677 { ISD::FSQRT, MVT::f64, 21 }, // SNB from http://www.agner.org/
1678 { ISD::FSQRT, MVT::v2f64, 21 }, // SNB from http://www.agner.org/
1679 { ISD::FSQRT, MVT::v4f64, 43 }, // SNB from http://www.agner.org/
1680 };
Craig Toppera9859192018-03-25 15:58:12 +00001681 static const CostTblEntry GLMCostTbl[] = {
1682 { ISD::FSQRT, MVT::f32, 19 }, // sqrtss
1683 { ISD::FSQRT, MVT::v4f32, 37 }, // sqrtps
1684 { ISD::FSQRT, MVT::f64, 34 }, // sqrtsd
1685 { ISD::FSQRT, MVT::v2f64, 67 }, // sqrtpd
1686 };
1687 static const CostTblEntry SLMCostTbl[] = {
1688 { ISD::FSQRT, MVT::f32, 20 }, // sqrtss
1689 { ISD::FSQRT, MVT::v4f32, 40 }, // sqrtps
1690 { ISD::FSQRT, MVT::f64, 35 }, // sqrtsd
1691 { ISD::FSQRT, MVT::v2f64, 70 }, // sqrtpd
1692 };
Alexey Bataevd07c7312016-10-31 12:10:53 +00001693 static const CostTblEntry SSE42CostTbl[] = {
Simon Pilgrima0b0b742017-03-15 11:57:42 +00001694 { ISD::FSQRT, MVT::f32, 18 }, // Nehalem from http://www.agner.org/
1695 { ISD::FSQRT, MVT::v4f32, 18 }, // Nehalem from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001696 };
1697 static const CostTblEntry SSSE3CostTbl[] = {
1698 { ISD::BITREVERSE, MVT::v2i64, 5 },
1699 { ISD::BITREVERSE, MVT::v4i32, 5 },
1700 { ISD::BITREVERSE, MVT::v8i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001701 { ISD::BITREVERSE, MVT::v16i8, 5 },
1702 { ISD::BSWAP, MVT::v2i64, 1 },
1703 { ISD::BSWAP, MVT::v4i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001704 { ISD::BSWAP, MVT::v8i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001705 { ISD::CTLZ, MVT::v2i64, 23 },
1706 { ISD::CTLZ, MVT::v4i32, 18 },
1707 { ISD::CTLZ, MVT::v8i16, 14 },
1708 { ISD::CTLZ, MVT::v16i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001709 { ISD::CTPOP, MVT::v2i64, 7 },
1710 { ISD::CTPOP, MVT::v4i32, 11 },
1711 { ISD::CTPOP, MVT::v8i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001712 { ISD::CTPOP, MVT::v16i8, 6 },
1713 { ISD::CTTZ, MVT::v2i64, 10 },
1714 { ISD::CTTZ, MVT::v4i32, 14 },
1715 { ISD::CTTZ, MVT::v8i16, 12 },
1716 { ISD::CTTZ, MVT::v16i8, 9 }
Simon Pilgrim356e8232016-06-20 23:08:21 +00001717 };
1718 static const CostTblEntry SSE2CostTbl[] = {
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001719 { ISD::BITREVERSE, MVT::v2i64, 29 },
1720 { ISD::BITREVERSE, MVT::v4i32, 27 },
1721 { ISD::BITREVERSE, MVT::v8i16, 27 },
1722 { ISD::BITREVERSE, MVT::v16i8, 20 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001723 { ISD::BSWAP, MVT::v2i64, 7 },
1724 { ISD::BSWAP, MVT::v4i32, 7 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001725 { ISD::BSWAP, MVT::v8i16, 7 },
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001726 { ISD::CTLZ, MVT::v2i64, 25 },
1727 { ISD::CTLZ, MVT::v4i32, 26 },
1728 { ISD::CTLZ, MVT::v8i16, 20 },
1729 { ISD::CTLZ, MVT::v16i8, 17 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001730 { ISD::CTPOP, MVT::v2i64, 12 },
1731 { ISD::CTPOP, MVT::v4i32, 15 },
1732 { ISD::CTPOP, MVT::v8i16, 13 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001733 { ISD::CTPOP, MVT::v16i8, 10 },
1734 { ISD::CTTZ, MVT::v2i64, 14 },
1735 { ISD::CTTZ, MVT::v4i32, 18 },
1736 { ISD::CTTZ, MVT::v8i16, 16 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001737 { ISD::CTTZ, MVT::v16i8, 13 },
1738 { ISD::FSQRT, MVT::f64, 32 }, // Nehalem from http://www.agner.org/
1739 { ISD::FSQRT, MVT::v2f64, 32 }, // Nehalem from http://www.agner.org/
1740 };
1741 static const CostTblEntry SSE1CostTbl[] = {
Simon Pilgrima0b0b742017-03-15 11:57:42 +00001742 { ISD::FSQRT, MVT::f32, 28 }, // Pentium III from http://www.agner.org/
1743 { ISD::FSQRT, MVT::v4f32, 56 }, // Pentium III from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001744 };
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001745 static const CostTblEntry X64CostTbl[] = { // 64-bit targets
1746 { ISD::BITREVERSE, MVT::i64, 14 }
1747 };
1748 static const CostTblEntry X86CostTbl[] = { // 32 or 64-bit targets
1749 { ISD::BITREVERSE, MVT::i32, 14 },
1750 { ISD::BITREVERSE, MVT::i16, 14 },
1751 { ISD::BITREVERSE, MVT::i8, 11 }
1752 };
Simon Pilgrim14000b32016-05-24 08:17:50 +00001753
1754 unsigned ISD = ISD::DELETED_NODE;
1755 switch (IID) {
1756 default:
1757 break;
1758 case Intrinsic::bitreverse:
1759 ISD = ISD::BITREVERSE;
1760 break;
Simon Pilgrim356e8232016-06-20 23:08:21 +00001761 case Intrinsic::bswap:
1762 ISD = ISD::BSWAP;
1763 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001764 case Intrinsic::ctlz:
1765 ISD = ISD::CTLZ;
1766 break;
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001767 case Intrinsic::ctpop:
1768 ISD = ISD::CTPOP;
1769 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001770 case Intrinsic::cttz:
1771 ISD = ISD::CTTZ;
1772 break;
Alexey Bataevd07c7312016-10-31 12:10:53 +00001773 case Intrinsic::sqrt:
1774 ISD = ISD::FSQRT;
1775 break;
Simon Pilgrim14000b32016-05-24 08:17:50 +00001776 }
1777
1778 // Legalize the type.
1779 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, RetTy);
1780 MVT MTy = LT.second;
1781
1782 // Attempt to lookup cost.
Craig Toppera9859192018-03-25 15:58:12 +00001783 if (ST->isGLM())
1784 if (const auto *Entry = CostTableLookup(GLMCostTbl, ISD, MTy))
1785 return LT.first * Entry->Cost;
1786
1787 if (ST->isSLM())
1788 if (const auto *Entry = CostTableLookup(SLMCostTbl, ISD, MTy))
1789 return LT.first * Entry->Cost;
1790
Simon Pilgrim23ef2672017-05-17 21:02:18 +00001791 if (ST->hasCDI())
1792 if (const auto *Entry = CostTableLookup(AVX512CDCostTbl, ISD, MTy))
1793 return LT.first * Entry->Cost;
1794
Simon Pilgrima9a92a12017-05-17 19:20:20 +00001795 if (ST->hasBWI())
1796 if (const auto *Entry = CostTableLookup(AVX512BWCostTbl, ISD, MTy))
1797 return LT.first * Entry->Cost;
1798
1799 if (ST->hasAVX512())
1800 if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
1801 return LT.first * Entry->Cost;
1802
Simon Pilgrim14000b32016-05-24 08:17:50 +00001803 if (ST->hasXOP())
1804 if (const auto *Entry = CostTableLookup(XOPCostTbl, ISD, MTy))
1805 return LT.first * Entry->Cost;
1806
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001807 if (ST->hasAVX2())
1808 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1809 return LT.first * Entry->Cost;
1810
1811 if (ST->hasAVX())
1812 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1813 return LT.first * Entry->Cost;
1814
Alexey Bataevd07c7312016-10-31 12:10:53 +00001815 if (ST->hasSSE42())
1816 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1817 return LT.first * Entry->Cost;
1818
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001819 if (ST->hasSSSE3())
1820 if (const auto *Entry = CostTableLookup(SSSE3CostTbl, ISD, MTy))
1821 return LT.first * Entry->Cost;
1822
Simon Pilgrim356e8232016-06-20 23:08:21 +00001823 if (ST->hasSSE2())
1824 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1825 return LT.first * Entry->Cost;
1826
Alexey Bataevd07c7312016-10-31 12:10:53 +00001827 if (ST->hasSSE1())
1828 if (const auto *Entry = CostTableLookup(SSE1CostTbl, ISD, MTy))
1829 return LT.first * Entry->Cost;
1830
Simon Pilgrim06c70ad2017-03-15 19:34:55 +00001831 if (ST->is64Bit())
1832 if (const auto *Entry = CostTableLookup(X64CostTbl, ISD, MTy))
1833 return LT.first * Entry->Cost;
1834
1835 if (const auto *Entry = CostTableLookup(X86CostTbl, ISD, MTy))
1836 return LT.first * Entry->Cost;
1837
Jonas Paulssona48ea232017-03-14 06:35:36 +00001838 return BaseT::getIntrinsicInstrCost(IID, RetTy, Tys, FMF, ScalarizationCostPassed);
Simon Pilgrim14000b32016-05-24 08:17:50 +00001839}
1840
1841int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
Jonas Paulssona48ea232017-03-14 06:35:36 +00001842 ArrayRef<Value *> Args, FastMathFlags FMF, unsigned VF) {
1843 return BaseT::getIntrinsicInstrCost(IID, RetTy, Args, FMF, VF);
Simon Pilgrim14000b32016-05-24 08:17:50 +00001844}
1845
Chandler Carruth93205eb2015-08-05 18:08:10 +00001846int X86TTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val, unsigned Index) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001847 assert(Val->isVectorTy() && "This must be a vector type");
1848
Sanjay Patelaedc3472016-05-25 17:27:54 +00001849 Type *ScalarType = Val->getScalarType();
1850
Chandler Carruth664e3542013-01-07 01:37:14 +00001851 if (Index != -1U) {
1852 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001853 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Val);
Chandler Carruth664e3542013-01-07 01:37:14 +00001854
1855 // This type is legalized to a scalar type.
1856 if (!LT.second.isVector())
1857 return 0;
1858
1859 // The type may be split. Normalize the index to the new type.
1860 unsigned Width = LT.second.getVectorNumElements();
1861 Index = Index % Width;
1862
1863 // Floating point scalars are already located in index #0.
Sanjay Patelaedc3472016-05-25 17:27:54 +00001864 if (ScalarType->isFloatingPointTy() && Index == 0)
Chandler Carruth664e3542013-01-07 01:37:14 +00001865 return 0;
1866 }
1867
Sanjay Patelaedc3472016-05-25 17:27:54 +00001868 // Add to the base cost if we know that the extracted element of a vector is
1869 // destined to be moved to and used in the integer register file.
1870 int RegisterFileMoveCost = 0;
1871 if (Opcode == Instruction::ExtractElement && ScalarType->isPointerTy())
1872 RegisterFileMoveCost = 1;
1873
1874 return BaseT::getVectorInstrCost(Opcode, Val, Index) + RegisterFileMoveCost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001875}
1876
Chandler Carruth93205eb2015-08-05 18:08:10 +00001877int X86TTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src, unsigned Alignment,
Jonas Paulssonfccc7d62017-04-12 11:49:08 +00001878 unsigned AddressSpace, const Instruction *I) {
Alp Tokerf907b892013-12-05 05:44:44 +00001879 // Handle non-power-of-two vectors such as <3 x float>
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001880 if (VectorType *VTy = dyn_cast<VectorType>(Src)) {
1881 unsigned NumElem = VTy->getVectorNumElements();
1882
1883 // Handle a few common cases:
1884 // <3 x float>
1885 if (NumElem == 3 && VTy->getScalarSizeInBits() == 32)
1886 // Cost = 64 bit store + extract + 32 bit store.
1887 return 3;
1888
1889 // <3 x double>
1890 if (NumElem == 3 && VTy->getScalarSizeInBits() == 64)
1891 // Cost = 128 bit store + unpack + 64 bit store.
1892 return 3;
1893
Alp Tokerf907b892013-12-05 05:44:44 +00001894 // Assume that all other non-power-of-two numbers are scalarized.
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001895 if (!isPowerOf2_32(NumElem)) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00001896 int Cost = BaseT::getMemoryOpCost(Opcode, VTy->getScalarType(), Alignment,
1897 AddressSpace);
1898 int SplitCost = getScalarizationOverhead(Src, Opcode == Instruction::Load,
1899 Opcode == Instruction::Store);
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001900 return NumElem * Cost + SplitCost;
1901 }
1902 }
1903
Chandler Carruth664e3542013-01-07 01:37:14 +00001904 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001905 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src);
Chandler Carruth664e3542013-01-07 01:37:14 +00001906 assert((Opcode == Instruction::Load || Opcode == Instruction::Store) &&
1907 "Invalid Opcode");
1908
1909 // Each load/store unit costs 1.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001910 int Cost = LT.first * 1;
Chandler Carruth664e3542013-01-07 01:37:14 +00001911
Sanjay Patel9f6c4d52016-03-09 22:23:33 +00001912 // This isn't exactly right. We're using slow unaligned 32-byte accesses as a
1913 // proxy for a double-pumped AVX memory interface such as on Sandybridge.
1914 if (LT.second.getStoreSize() == 32 && ST->isUnalignedMem32Slow())
1915 Cost *= 2;
Chandler Carruth664e3542013-01-07 01:37:14 +00001916
1917 return Cost;
1918}
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001919
Chandler Carruth93205eb2015-08-05 18:08:10 +00001920int X86TTIImpl::getMaskedMemoryOpCost(unsigned Opcode, Type *SrcTy,
1921 unsigned Alignment,
1922 unsigned AddressSpace) {
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001923 VectorType *SrcVTy = dyn_cast<VectorType>(SrcTy);
1924 if (!SrcVTy)
1925 // To calculate scalar take the regular cost, without mask
1926 return getMemoryOpCost(Opcode, SrcTy, Alignment, AddressSpace);
1927
1928 unsigned NumElem = SrcVTy->getVectorNumElements();
1929 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00001930 VectorType::get(Type::getInt8Ty(SrcVTy->getContext()), NumElem);
Elena Demikhovsky20662e32015-10-19 07:43:38 +00001931 if ((Opcode == Instruction::Load && !isLegalMaskedLoad(SrcVTy)) ||
1932 (Opcode == Instruction::Store && !isLegalMaskedStore(SrcVTy)) ||
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001933 !isPowerOf2_32(NumElem)) {
1934 // Scalarization
Chandler Carruth93205eb2015-08-05 18:08:10 +00001935 int MaskSplitCost = getScalarizationOverhead(MaskTy, false, true);
1936 int ScalarCompareCost = getCmpSelInstrCost(
Mehdi Amini867e9142016-04-14 04:36:40 +00001937 Instruction::ICmp, Type::getInt8Ty(SrcVTy->getContext()), nullptr);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001938 int BranchCost = getCFInstrCost(Instruction::Br);
1939 int MaskCmpCost = NumElem * (BranchCost + ScalarCompareCost);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001940
Chandler Carruth93205eb2015-08-05 18:08:10 +00001941 int ValueSplitCost = getScalarizationOverhead(
1942 SrcVTy, Opcode == Instruction::Load, Opcode == Instruction::Store);
1943 int MemopCost =
Chandler Carruth705b1852015-01-31 03:43:40 +00001944 NumElem * BaseT::getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
1945 Alignment, AddressSpace);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001946 return MemopCost + ValueSplitCost + MaskSplitCost + MaskCmpCost;
1947 }
1948
1949 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001950 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, SrcVTy);
Cong Houda4e8ae2015-10-28 18:15:46 +00001951 auto VT = TLI->getValueType(DL, SrcVTy);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001952 int Cost = 0;
Cong Houda4e8ae2015-10-28 18:15:46 +00001953 if (VT.isSimple() && LT.second != VT.getSimpleVT() &&
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001954 LT.second.getVectorNumElements() == NumElem)
1955 // Promotion requires expand/truncate for data and a shuffle for mask.
Hans Wennborg083ca9b2015-10-06 23:24:35 +00001956 Cost += getShuffleCost(TTI::SK_Alternate, SrcVTy, 0, nullptr) +
1957 getShuffleCost(TTI::SK_Alternate, MaskTy, 0, nullptr);
Chandler Carruth705b1852015-01-31 03:43:40 +00001958
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001959 else if (LT.second.getVectorNumElements() > NumElem) {
1960 VectorType *NewMaskTy = VectorType::get(MaskTy->getVectorElementType(),
1961 LT.second.getVectorNumElements());
1962 // Expanding requires fill mask with zeroes
Chandler Carruth705b1852015-01-31 03:43:40 +00001963 Cost += getShuffleCost(TTI::SK_InsertSubvector, NewMaskTy, 0, MaskTy);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001964 }
1965 if (!ST->hasAVX512())
1966 return Cost + LT.first*4; // Each maskmov costs 4
1967
1968 // AVX-512 masked load/store is cheapper
1969 return Cost+LT.first;
1970}
1971
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001972int X86TTIImpl::getAddressComputationCost(Type *Ty, ScalarEvolution *SE,
1973 const SCEV *Ptr) {
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001974 // Address computations in vectorized code with non-consecutive addresses will
1975 // likely result in more instructions compared to scalar code where the
1976 // computation can more often be merged into the index mode. The resulting
1977 // extra micro-ops can significantly decrease throughput.
1978 unsigned NumVectorInstToHideOverhead = 10;
1979
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001980 // Cost modeling of Strided Access Computation is hidden by the indexing
1981 // modes of X86 regardless of the stride value. We dont believe that there
1982 // is a difference between constant strided access in gerenal and constant
1983 // strided value which is less than or equal to 64.
1984 // Even in the case of (loop invariant) stride whose value is not known at
1985 // compile time, the address computation will not incur more than one extra
1986 // ADD instruction.
1987 if (Ty->isVectorTy() && SE) {
1988 if (!BaseT::isStridedAccess(Ptr))
1989 return NumVectorInstToHideOverhead;
1990 if (!BaseT::getConstantStrideStep(SE, Ptr))
1991 return 1;
1992 }
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001993
Mohammed Agabaria23599ba2017-01-05 14:03:41 +00001994 return BaseT::getAddressComputationCost(Ty, SE, Ptr);
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001995}
Yi Jiang5c343de2013-09-19 17:48:48 +00001996
Alexey Bataev3e9b3eb2017-07-31 14:19:32 +00001997int X86TTIImpl::getArithmeticReductionCost(unsigned Opcode, Type *ValTy,
1998 bool IsPairwise) {
Michael Liao5bf95782014-12-04 05:20:33 +00001999
Chandler Carruth93205eb2015-08-05 18:08:10 +00002000 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Michael Liao5bf95782014-12-04 05:20:33 +00002001
Yi Jiang5c343de2013-09-19 17:48:48 +00002002 MVT MTy = LT.second;
Michael Liao5bf95782014-12-04 05:20:33 +00002003
Yi Jiang5c343de2013-09-19 17:48:48 +00002004 int ISD = TLI->InstructionOpcodeToISD(Opcode);
2005 assert(ISD && "Invalid opcode");
Michael Liao5bf95782014-12-04 05:20:33 +00002006
2007 // We use the Intel Architecture Code Analyzer(IACA) to measure the throughput
2008 // and make it as the cost.
2009
Craig Topper4b275762015-10-28 04:02:12 +00002010 static const CostTblEntry SSE42CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002011 { ISD::FADD, MVT::v2f64, 2 },
2012 { ISD::FADD, MVT::v4f32, 4 },
2013 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
2014 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
2015 { ISD::ADD, MVT::v8i16, 5 },
2016 };
Michael Liao5bf95782014-12-04 05:20:33 +00002017
Craig Topper4b275762015-10-28 04:02:12 +00002018 static const CostTblEntry AVX1CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002019 { ISD::FADD, MVT::v4f32, 4 },
2020 { ISD::FADD, MVT::v4f64, 5 },
2021 { ISD::FADD, MVT::v8f32, 7 },
2022 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
2023 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
2024 { ISD::ADD, MVT::v4i64, 5 }, // The data reported by the IACA tool is "4.8".
2025 { ISD::ADD, MVT::v8i16, 5 },
2026 { ISD::ADD, MVT::v8i32, 5 },
2027 };
2028
Craig Topper4b275762015-10-28 04:02:12 +00002029 static const CostTblEntry SSE42CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002030 { ISD::FADD, MVT::v2f64, 2 },
2031 { ISD::FADD, MVT::v4f32, 4 },
2032 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
2033 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.3".
2034 { ISD::ADD, MVT::v8i16, 4 }, // The data reported by the IACA tool is "4.3".
2035 };
Michael Liao5bf95782014-12-04 05:20:33 +00002036
Craig Topper4b275762015-10-28 04:02:12 +00002037 static const CostTblEntry AVX1CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00002038 { ISD::FADD, MVT::v4f32, 3 },
2039 { ISD::FADD, MVT::v4f64, 3 },
2040 { ISD::FADD, MVT::v8f32, 4 },
2041 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
2042 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "2.8".
2043 { ISD::ADD, MVT::v4i64, 3 },
2044 { ISD::ADD, MVT::v8i16, 4 },
2045 { ISD::ADD, MVT::v8i32, 5 },
2046 };
Michael Liao5bf95782014-12-04 05:20:33 +00002047
Yi Jiang5c343de2013-09-19 17:48:48 +00002048 if (IsPairwise) {
Craig Topperee0c8592015-10-27 04:14:24 +00002049 if (ST->hasAVX())
2050 if (const auto *Entry = CostTableLookup(AVX1CostTblPairWise, ISD, MTy))
2051 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00002052
Craig Topperee0c8592015-10-27 04:14:24 +00002053 if (ST->hasSSE42())
2054 if (const auto *Entry = CostTableLookup(SSE42CostTblPairWise, ISD, MTy))
2055 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00002056 } else {
Craig Topperee0c8592015-10-27 04:14:24 +00002057 if (ST->hasAVX())
2058 if (const auto *Entry = CostTableLookup(AVX1CostTblNoPairWise, ISD, MTy))
2059 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00002060
Craig Topperee0c8592015-10-27 04:14:24 +00002061 if (ST->hasSSE42())
2062 if (const auto *Entry = CostTableLookup(SSE42CostTblNoPairWise, ISD, MTy))
2063 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00002064 }
2065
Alexey Bataev3e9b3eb2017-07-31 14:19:32 +00002066 return BaseT::getArithmeticReductionCost(Opcode, ValTy, IsPairwise);
Yi Jiang5c343de2013-09-19 17:48:48 +00002067}
2068
Alexey Bataev6dd29fc2017-09-08 13:49:36 +00002069int X86TTIImpl::getMinMaxReductionCost(Type *ValTy, Type *CondTy,
2070 bool IsPairwise, bool IsUnsigned) {
2071 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
2072
2073 MVT MTy = LT.second;
2074
2075 int ISD;
2076 if (ValTy->isIntOrIntVectorTy()) {
2077 ISD = IsUnsigned ? ISD::UMIN : ISD::SMIN;
2078 } else {
2079 assert(ValTy->isFPOrFPVectorTy() &&
2080 "Expected float point or integer vector type.");
2081 ISD = ISD::FMINNUM;
2082 }
2083
2084 // We use the Intel Architecture Code Analyzer(IACA) to measure the throughput
2085 // and make it as the cost.
2086
2087 static const CostTblEntry SSE42CostTblPairWise[] = {
2088 {ISD::FMINNUM, MVT::v2f64, 3},
2089 {ISD::FMINNUM, MVT::v4f32, 2},
2090 {ISD::SMIN, MVT::v2i64, 7}, // The data reported by the IACA is "6.8"
2091 {ISD::UMIN, MVT::v2i64, 8}, // The data reported by the IACA is "8.6"
2092 {ISD::SMIN, MVT::v4i32, 1}, // The data reported by the IACA is "1.5"
2093 {ISD::UMIN, MVT::v4i32, 2}, // The data reported by the IACA is "1.8"
2094 {ISD::SMIN, MVT::v8i16, 2},
2095 {ISD::UMIN, MVT::v8i16, 2},
2096 };
2097
2098 static const CostTblEntry AVX1CostTblPairWise[] = {
2099 {ISD::FMINNUM, MVT::v4f32, 1},
2100 {ISD::FMINNUM, MVT::v4f64, 1},
2101 {ISD::FMINNUM, MVT::v8f32, 2},
2102 {ISD::SMIN, MVT::v2i64, 3},
2103 {ISD::UMIN, MVT::v2i64, 3},
2104 {ISD::SMIN, MVT::v4i32, 1},
2105 {ISD::UMIN, MVT::v4i32, 1},
2106 {ISD::SMIN, MVT::v8i16, 1},
2107 {ISD::UMIN, MVT::v8i16, 1},
2108 {ISD::SMIN, MVT::v8i32, 3},
2109 {ISD::UMIN, MVT::v8i32, 3},
2110 };
2111
2112 static const CostTblEntry AVX2CostTblPairWise[] = {
2113 {ISD::SMIN, MVT::v4i64, 2},
2114 {ISD::UMIN, MVT::v4i64, 2},
2115 {ISD::SMIN, MVT::v8i32, 1},
2116 {ISD::UMIN, MVT::v8i32, 1},
2117 {ISD::SMIN, MVT::v16i16, 1},
2118 {ISD::UMIN, MVT::v16i16, 1},
2119 {ISD::SMIN, MVT::v32i8, 2},
2120 {ISD::UMIN, MVT::v32i8, 2},
2121 };
2122
2123 static const CostTblEntry AVX512CostTblPairWise[] = {
2124 {ISD::FMINNUM, MVT::v8f64, 1},
2125 {ISD::FMINNUM, MVT::v16f32, 2},
2126 {ISD::SMIN, MVT::v8i64, 2},
2127 {ISD::UMIN, MVT::v8i64, 2},
2128 {ISD::SMIN, MVT::v16i32, 1},
2129 {ISD::UMIN, MVT::v16i32, 1},
2130 };
2131
2132 static const CostTblEntry SSE42CostTblNoPairWise[] = {
2133 {ISD::FMINNUM, MVT::v2f64, 3},
2134 {ISD::FMINNUM, MVT::v4f32, 3},
2135 {ISD::SMIN, MVT::v2i64, 7}, // The data reported by the IACA is "6.8"
2136 {ISD::UMIN, MVT::v2i64, 9}, // The data reported by the IACA is "8.6"
2137 {ISD::SMIN, MVT::v4i32, 1}, // The data reported by the IACA is "1.5"
2138 {ISD::UMIN, MVT::v4i32, 2}, // The data reported by the IACA is "1.8"
2139 {ISD::SMIN, MVT::v8i16, 1}, // The data reported by the IACA is "1.5"
2140 {ISD::UMIN, MVT::v8i16, 2}, // The data reported by the IACA is "1.8"
2141 };
2142
2143 static const CostTblEntry AVX1CostTblNoPairWise[] = {
2144 {ISD::FMINNUM, MVT::v4f32, 1},
2145 {ISD::FMINNUM, MVT::v4f64, 1},
2146 {ISD::FMINNUM, MVT::v8f32, 1},
2147 {ISD::SMIN, MVT::v2i64, 3},
2148 {ISD::UMIN, MVT::v2i64, 3},
2149 {ISD::SMIN, MVT::v4i32, 1},
2150 {ISD::UMIN, MVT::v4i32, 1},
2151 {ISD::SMIN, MVT::v8i16, 1},
2152 {ISD::UMIN, MVT::v8i16, 1},
2153 {ISD::SMIN, MVT::v8i32, 2},
2154 {ISD::UMIN, MVT::v8i32, 2},
2155 };
2156
2157 static const CostTblEntry AVX2CostTblNoPairWise[] = {
2158 {ISD::SMIN, MVT::v4i64, 1},
2159 {ISD::UMIN, MVT::v4i64, 1},
2160 {ISD::SMIN, MVT::v8i32, 1},
2161 {ISD::UMIN, MVT::v8i32, 1},
2162 {ISD::SMIN, MVT::v16i16, 1},
2163 {ISD::UMIN, MVT::v16i16, 1},
2164 {ISD::SMIN, MVT::v32i8, 1},
2165 {ISD::UMIN, MVT::v32i8, 1},
2166 };
2167
2168 static const CostTblEntry AVX512CostTblNoPairWise[] = {
2169 {ISD::FMINNUM, MVT::v8f64, 1},
2170 {ISD::FMINNUM, MVT::v16f32, 2},
2171 {ISD::SMIN, MVT::v8i64, 1},
2172 {ISD::UMIN, MVT::v8i64, 1},
2173 {ISD::SMIN, MVT::v16i32, 1},
2174 {ISD::UMIN, MVT::v16i32, 1},
2175 };
2176
2177 if (IsPairwise) {
2178 if (ST->hasAVX512())
2179 if (const auto *Entry = CostTableLookup(AVX512CostTblPairWise, ISD, MTy))
2180 return LT.first * Entry->Cost;
2181
2182 if (ST->hasAVX2())
2183 if (const auto *Entry = CostTableLookup(AVX2CostTblPairWise, ISD, MTy))
2184 return LT.first * Entry->Cost;
2185
2186 if (ST->hasAVX())
2187 if (const auto *Entry = CostTableLookup(AVX1CostTblPairWise, ISD, MTy))
2188 return LT.first * Entry->Cost;
2189
2190 if (ST->hasSSE42())
2191 if (const auto *Entry = CostTableLookup(SSE42CostTblPairWise, ISD, MTy))
2192 return LT.first * Entry->Cost;
2193 } else {
2194 if (ST->hasAVX512())
2195 if (const auto *Entry =
2196 CostTableLookup(AVX512CostTblNoPairWise, ISD, MTy))
2197 return LT.first * Entry->Cost;
2198
2199 if (ST->hasAVX2())
2200 if (const auto *Entry = CostTableLookup(AVX2CostTblNoPairWise, ISD, MTy))
2201 return LT.first * Entry->Cost;
2202
2203 if (ST->hasAVX())
2204 if (const auto *Entry = CostTableLookup(AVX1CostTblNoPairWise, ISD, MTy))
2205 return LT.first * Entry->Cost;
2206
2207 if (ST->hasSSE42())
2208 if (const auto *Entry = CostTableLookup(SSE42CostTblNoPairWise, ISD, MTy))
2209 return LT.first * Entry->Cost;
2210 }
2211
2212 return BaseT::getMinMaxReductionCost(ValTy, CondTy, IsPairwise, IsUnsigned);
2213}
2214
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002215/// \brief Calculate the cost of materializing a 64-bit value. This helper
2216/// method might only calculate a fraction of a larger immediate. Therefore it
2217/// is valid to return a cost of ZERO.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002218int X86TTIImpl::getIntImmCost(int64_t Val) {
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002219 if (Val == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002220 return TTI::TCC_Free;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002221
2222 if (isInt<32>(Val))
Chandler Carruth705b1852015-01-31 03:43:40 +00002223 return TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002224
Chandler Carruth705b1852015-01-31 03:43:40 +00002225 return 2 * TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002226}
2227
Chandler Carruth93205eb2015-08-05 18:08:10 +00002228int X86TTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002229 assert(Ty->isIntegerTy());
2230
2231 unsigned BitSize = Ty->getPrimitiveSizeInBits();
2232 if (BitSize == 0)
2233 return ~0U;
2234
Juergen Ributzka43176172014-05-19 21:00:53 +00002235 // Never hoist constants larger than 128bit, because this might lead to
2236 // incorrect code generation or assertions in codegen.
2237 // Fixme: Create a cost model for types larger than i128 once the codegen
2238 // issues have been fixed.
2239 if (BitSize > 128)
Chandler Carruth705b1852015-01-31 03:43:40 +00002240 return TTI::TCC_Free;
Juergen Ributzka43176172014-05-19 21:00:53 +00002241
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002242 if (Imm == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002243 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002244
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002245 // Sign-extend all constants to a multiple of 64-bit.
2246 APInt ImmVal = Imm;
2247 if (BitSize & 0x3f)
2248 ImmVal = Imm.sext((BitSize + 63) & ~0x3fU);
2249
2250 // Split the constant into 64-bit chunks and calculate the cost for each
2251 // chunk.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002252 int Cost = 0;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002253 for (unsigned ShiftVal = 0; ShiftVal < BitSize; ShiftVal += 64) {
2254 APInt Tmp = ImmVal.ashr(ShiftVal).sextOrTrunc(64);
2255 int64_t Val = Tmp.getSExtValue();
2256 Cost += getIntImmCost(Val);
2257 }
Sanjay Patel4c7d0942016-04-05 19:27:39 +00002258 // We need at least one instruction to materialize the constant.
Chandler Carruth93205eb2015-08-05 18:08:10 +00002259 return std::max(1, Cost);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002260}
2261
Chandler Carruth93205eb2015-08-05 18:08:10 +00002262int X86TTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx, const APInt &Imm,
2263 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002264 assert(Ty->isIntegerTy());
2265
2266 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00002267 // There is no cost model for constants with a bit size of 0. Return TCC_Free
2268 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002269 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002270 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002271
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002272 unsigned ImmIdx = ~0U;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002273 switch (Opcode) {
Chandler Carruth705b1852015-01-31 03:43:40 +00002274 default:
2275 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002276 case Instruction::GetElementPtr:
Juergen Ributzka27435b32014-04-02 21:45:36 +00002277 // Always hoist the base address of a GetElementPtr. This prevents the
2278 // creation of new constants for every base constant that gets constant
2279 // folded with the offset.
Juergen Ributzka631c4912014-03-25 18:01:25 +00002280 if (Idx == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002281 return 2 * TTI::TCC_Basic;
2282 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002283 case Instruction::Store:
2284 ImmIdx = 0;
2285 break;
Craig Topper074e8452015-12-20 18:41:54 +00002286 case Instruction::ICmp:
2287 // This is an imperfect hack to prevent constant hoisting of
2288 // compares that might be trying to check if a 64-bit value fits in
2289 // 32-bits. The backend can optimize these cases using a right shift by 32.
2290 // Ideally we would check the compare predicate here. There also other
2291 // similar immediates the backend can use shifts for.
2292 if (Idx == 1 && Imm.getBitWidth() == 64) {
2293 uint64_t ImmVal = Imm.getZExtValue();
2294 if (ImmVal == 0x100000000ULL || ImmVal == 0xffffffff)
2295 return TTI::TCC_Free;
2296 }
2297 ImmIdx = 1;
2298 break;
Craig Topper79dd1bf2015-10-06 02:50:24 +00002299 case Instruction::And:
2300 // We support 64-bit ANDs with immediates with 32-bits of leading zeroes
2301 // by using a 32-bit operation with implicit zero extension. Detect such
2302 // immediates here as the normal path expects bit 31 to be sign extended.
2303 if (Idx == 1 && Imm.getBitWidth() == 64 && isUInt<32>(Imm.getZExtValue()))
2304 return TTI::TCC_Free;
Justin Bognerb03fd122016-08-17 05:10:15 +00002305 LLVM_FALLTHROUGH;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002306 case Instruction::Add:
2307 case Instruction::Sub:
2308 case Instruction::Mul:
2309 case Instruction::UDiv:
2310 case Instruction::SDiv:
2311 case Instruction::URem:
2312 case Instruction::SRem:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002313 case Instruction::Or:
2314 case Instruction::Xor:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002315 ImmIdx = 1;
2316 break;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00002317 // Always return TCC_Free for the shift value of a shift instruction.
2318 case Instruction::Shl:
2319 case Instruction::LShr:
2320 case Instruction::AShr:
2321 if (Idx == 1)
Chandler Carruth705b1852015-01-31 03:43:40 +00002322 return TTI::TCC_Free;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00002323 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002324 case Instruction::Trunc:
2325 case Instruction::ZExt:
2326 case Instruction::SExt:
2327 case Instruction::IntToPtr:
2328 case Instruction::PtrToInt:
2329 case Instruction::BitCast:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002330 case Instruction::PHI:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002331 case Instruction::Call:
2332 case Instruction::Select:
2333 case Instruction::Ret:
2334 case Instruction::Load:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002335 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002336 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002337
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002338 if (Idx == ImmIdx) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00002339 int NumConstants = (BitSize + 63) / 64;
2340 int Cost = X86TTIImpl::getIntImmCost(Imm, Ty);
Chandler Carruth705b1852015-01-31 03:43:40 +00002341 return (Cost <= NumConstants * TTI::TCC_Basic)
Chandler Carruth93205eb2015-08-05 18:08:10 +00002342 ? static_cast<int>(TTI::TCC_Free)
Chandler Carruth705b1852015-01-31 03:43:40 +00002343 : Cost;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00002344 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002345
Chandler Carruth705b1852015-01-31 03:43:40 +00002346 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002347}
2348
Chandler Carruth93205eb2015-08-05 18:08:10 +00002349int X86TTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx, const APInt &Imm,
2350 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002351 assert(Ty->isIntegerTy());
2352
2353 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00002354 // There is no cost model for constants with a bit size of 0. Return TCC_Free
2355 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002356 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00002357 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002358
2359 switch (IID) {
Chandler Carruth705b1852015-01-31 03:43:40 +00002360 default:
2361 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002362 case Intrinsic::sadd_with_overflow:
2363 case Intrinsic::uadd_with_overflow:
2364 case Intrinsic::ssub_with_overflow:
2365 case Intrinsic::usub_with_overflow:
2366 case Intrinsic::smul_with_overflow:
2367 case Intrinsic::umul_with_overflow:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00002368 if ((Idx == 1) && Imm.getBitWidth() <= 64 && isInt<32>(Imm.getSExtValue()))
Chandler Carruth705b1852015-01-31 03:43:40 +00002369 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002370 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002371 case Intrinsic::experimental_stackmap:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002372 if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00002373 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002374 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002375 case Intrinsic::experimental_patchpoint_void:
2376 case Intrinsic::experimental_patchpoint_i64:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002377 if ((Idx < 4) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00002378 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00002379 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002380 }
Chandler Carruth705b1852015-01-31 03:43:40 +00002381 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00002382}
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002383
Elena Demikhovskyf58f8382017-08-20 12:34:29 +00002384unsigned X86TTIImpl::getUserCost(const User *U,
2385 ArrayRef<const Value *> Operands) {
2386 if (isa<StoreInst>(U)) {
2387 Value *Ptr = U->getOperand(1);
2388 // Store instruction with index and scale costs 2 Uops.
2389 // Check the preceding GEP to identify non-const indices.
2390 if (auto GEP = dyn_cast<GetElementPtrInst>(Ptr)) {
2391 if (!all_of(GEP->indices(), [](Value *V) { return isa<Constant>(V); }))
2392 return TTI::TCC_Basic * 2;
2393 }
2394 return TTI::TCC_Basic;
2395 }
2396 return BaseT::getUserCost(U, Operands);
2397}
2398
Elena Demikhovsky54946982015-12-28 20:10:59 +00002399// Return an average cost of Gather / Scatter instruction, maybe improved later
2400int X86TTIImpl::getGSVectorCost(unsigned Opcode, Type *SrcVTy, Value *Ptr,
2401 unsigned Alignment, unsigned AddressSpace) {
2402
2403 assert(isa<VectorType>(SrcVTy) && "Unexpected type in getGSVectorCost");
2404 unsigned VF = SrcVTy->getVectorNumElements();
2405
2406 // Try to reduce index size from 64 bit (default for GEP)
2407 // to 32. It is essential for VF 16. If the index can't be reduced to 32, the
2408 // operation will use 16 x 64 indices which do not fit in a zmm and needs
2409 // to split. Also check that the base pointer is the same for all lanes,
2410 // and that there's at most one variable index.
2411 auto getIndexSizeInBits = [](Value *Ptr, const DataLayout& DL) {
2412 unsigned IndexSize = DL.getPointerSizeInBits();
2413 GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr);
2414 if (IndexSize < 64 || !GEP)
2415 return IndexSize;
Simon Pilgrim14000b32016-05-24 08:17:50 +00002416
Elena Demikhovsky54946982015-12-28 20:10:59 +00002417 unsigned NumOfVarIndices = 0;
2418 Value *Ptrs = GEP->getPointerOperand();
2419 if (Ptrs->getType()->isVectorTy() && !getSplatValue(Ptrs))
2420 return IndexSize;
2421 for (unsigned i = 1; i < GEP->getNumOperands(); ++i) {
2422 if (isa<Constant>(GEP->getOperand(i)))
2423 continue;
2424 Type *IndxTy = GEP->getOperand(i)->getType();
2425 if (IndxTy->isVectorTy())
2426 IndxTy = IndxTy->getVectorElementType();
2427 if ((IndxTy->getPrimitiveSizeInBits() == 64 &&
2428 !isa<SExtInst>(GEP->getOperand(i))) ||
2429 ++NumOfVarIndices > 1)
2430 return IndexSize; // 64
2431 }
2432 return (unsigned)32;
2433 };
2434
2435
2436 // Trying to reduce IndexSize to 32 bits for vector 16.
2437 // By default the IndexSize is equal to pointer size.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002438 unsigned IndexSize = (ST->hasAVX512() && VF >= 16)
2439 ? getIndexSizeInBits(Ptr, DL)
2440 : DL.getPointerSizeInBits();
Elena Demikhovsky54946982015-12-28 20:10:59 +00002441
Mehdi Amini867e9142016-04-14 04:36:40 +00002442 Type *IndexVTy = VectorType::get(IntegerType::get(SrcVTy->getContext(),
Elena Demikhovsky54946982015-12-28 20:10:59 +00002443 IndexSize), VF);
2444 std::pair<int, MVT> IdxsLT = TLI->getTypeLegalizationCost(DL, IndexVTy);
2445 std::pair<int, MVT> SrcLT = TLI->getTypeLegalizationCost(DL, SrcVTy);
2446 int SplitFactor = std::max(IdxsLT.first, SrcLT.first);
2447 if (SplitFactor > 1) {
2448 // Handle splitting of vector of pointers
2449 Type *SplitSrcTy = VectorType::get(SrcVTy->getScalarType(), VF / SplitFactor);
2450 return SplitFactor * getGSVectorCost(Opcode, SplitSrcTy, Ptr, Alignment,
2451 AddressSpace);
2452 }
2453
2454 // The gather / scatter cost is given by Intel architects. It is a rough
2455 // number since we are looking at one instruction in a time.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002456 const int GSOverhead = (Opcode == Instruction::Load)
2457 ? ST->getGatherOverhead()
2458 : ST->getScatterOverhead();
Elena Demikhovsky54946982015-12-28 20:10:59 +00002459 return GSOverhead + VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
2460 Alignment, AddressSpace);
2461}
2462
2463/// Return the cost of full scalarization of gather / scatter operation.
2464///
2465/// Opcode - Load or Store instruction.
2466/// SrcVTy - The type of the data vector that should be gathered or scattered.
2467/// VariableMask - The mask is non-constant at compile time.
2468/// Alignment - Alignment for one element.
2469/// AddressSpace - pointer[s] address space.
2470///
2471int X86TTIImpl::getGSScalarCost(unsigned Opcode, Type *SrcVTy,
2472 bool VariableMask, unsigned Alignment,
2473 unsigned AddressSpace) {
2474 unsigned VF = SrcVTy->getVectorNumElements();
2475
2476 int MaskUnpackCost = 0;
2477 if (VariableMask) {
2478 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00002479 VectorType::get(Type::getInt1Ty(SrcVTy->getContext()), VF);
Elena Demikhovsky54946982015-12-28 20:10:59 +00002480 MaskUnpackCost = getScalarizationOverhead(MaskTy, false, true);
2481 int ScalarCompareCost =
Mehdi Amini867e9142016-04-14 04:36:40 +00002482 getCmpSelInstrCost(Instruction::ICmp, Type::getInt1Ty(SrcVTy->getContext()),
Elena Demikhovsky54946982015-12-28 20:10:59 +00002483 nullptr);
2484 int BranchCost = getCFInstrCost(Instruction::Br);
2485 MaskUnpackCost += VF * (BranchCost + ScalarCompareCost);
2486 }
2487
2488 // The cost of the scalar loads/stores.
2489 int MemoryOpCost = VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
2490 Alignment, AddressSpace);
2491
2492 int InsertExtractCost = 0;
2493 if (Opcode == Instruction::Load)
2494 for (unsigned i = 0; i < VF; ++i)
2495 // Add the cost of inserting each scalar load into the vector
2496 InsertExtractCost +=
2497 getVectorInstrCost(Instruction::InsertElement, SrcVTy, i);
2498 else
2499 for (unsigned i = 0; i < VF; ++i)
2500 // Add the cost of extracting each element out of the data vector
2501 InsertExtractCost +=
2502 getVectorInstrCost(Instruction::ExtractElement, SrcVTy, i);
2503
2504 return MemoryOpCost + MaskUnpackCost + InsertExtractCost;
2505}
2506
2507/// Calculate the cost of Gather / Scatter operation
2508int X86TTIImpl::getGatherScatterOpCost(unsigned Opcode, Type *SrcVTy,
2509 Value *Ptr, bool VariableMask,
2510 unsigned Alignment) {
2511 assert(SrcVTy->isVectorTy() && "Unexpected data type for Gather/Scatter");
2512 unsigned VF = SrcVTy->getVectorNumElements();
2513 PointerType *PtrTy = dyn_cast<PointerType>(Ptr->getType());
2514 if (!PtrTy && Ptr->getType()->isVectorTy())
2515 PtrTy = dyn_cast<PointerType>(Ptr->getType()->getVectorElementType());
2516 assert(PtrTy && "Unexpected type for Ptr argument");
2517 unsigned AddressSpace = PtrTy->getAddressSpace();
2518
2519 bool Scalarize = false;
2520 if ((Opcode == Instruction::Load && !isLegalMaskedGather(SrcVTy)) ||
2521 (Opcode == Instruction::Store && !isLegalMaskedScatter(SrcVTy)))
2522 Scalarize = true;
2523 // Gather / Scatter for vector 2 is not profitable on KNL / SKX
2524 // Vector-4 of gather/scatter instruction does not exist on KNL.
2525 // We can extend it to 8 elements, but zeroing upper bits of
2526 // the mask vector will add more instructions. Right now we give the scalar
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002527 // cost of vector-4 for KNL. TODO: Check, maybe the gather/scatter instruction
2528 // is better in the VariableMask case.
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002529 if (ST->hasAVX512() && (VF == 2 || (VF == 4 && !ST->hasVLX())))
Elena Demikhovsky54946982015-12-28 20:10:59 +00002530 Scalarize = true;
2531
2532 if (Scalarize)
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002533 return getGSScalarCost(Opcode, SrcVTy, VariableMask, Alignment,
2534 AddressSpace);
Elena Demikhovsky54946982015-12-28 20:10:59 +00002535
2536 return getGSVectorCost(Opcode, SrcVTy, Ptr, Alignment, AddressSpace);
2537}
2538
Evgeny Stupachenkoc6752902017-08-07 19:56:34 +00002539bool X86TTIImpl::isLSRCostLess(TargetTransformInfo::LSRCost &C1,
2540 TargetTransformInfo::LSRCost &C2) {
2541 // X86 specific here are "instruction number 1st priority".
2542 return std::tie(C1.Insns, C1.NumRegs, C1.AddRecCost,
2543 C1.NumIVMuls, C1.NumBaseAdds,
2544 C1.ScaleCost, C1.ImmCost, C1.SetupCost) <
2545 std::tie(C2.Insns, C2.NumRegs, C2.AddRecCost,
2546 C2.NumIVMuls, C2.NumBaseAdds,
2547 C2.ScaleCost, C2.ImmCost, C2.SetupCost);
2548}
2549
Sanjay Pateld7c702b2018-02-05 23:43:05 +00002550bool X86TTIImpl::canMacroFuseCmp() {
2551 return ST->hasMacroFusion();
2552}
2553
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002554bool X86TTIImpl::isLegalMaskedLoad(Type *DataTy) {
Craig Topper46a5d582017-11-16 06:02:05 +00002555 // The backend can't handle a single element vector.
2556 if (isa<VectorType>(DataTy) && DataTy->getVectorNumElements() == 1)
2557 return false;
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002558 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00002559 int DataWidth = isa<PointerType>(ScalarTy) ?
2560 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002561
Igor Bregerf44b79d2016-08-02 09:15:28 +00002562 return ((DataWidth == 32 || DataWidth == 64) && ST->hasAVX()) ||
2563 ((DataWidth == 8 || DataWidth == 16) && ST->hasBWI());
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00002564}
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00002565
Elena Demikhovsky20662e32015-10-19 07:43:38 +00002566bool X86TTIImpl::isLegalMaskedStore(Type *DataType) {
2567 return isLegalMaskedLoad(DataType);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00002568}
2569
Elena Demikhovsky09285852015-10-25 15:37:55 +00002570bool X86TTIImpl::isLegalMaskedGather(Type *DataTy) {
2571 // This function is called now in two cases: from the Loop Vectorizer
2572 // and from the Scalarizer.
2573 // When the Loop Vectorizer asks about legality of the feature,
2574 // the vectorization factor is not calculated yet. The Loop Vectorizer
2575 // sends a scalar type and the decision is based on the width of the
2576 // scalar element.
2577 // Later on, the cost model will estimate usage this intrinsic based on
2578 // the vector type.
2579 // The Scalarizer asks again about legality. It sends a vector type.
2580 // In this case we can reject non-power-of-2 vectors.
Craig Topper46a5d582017-11-16 06:02:05 +00002581 // We also reject single element vectors as the type legalizer can't
2582 // scalarize it.
2583 if (isa<VectorType>(DataTy)) {
2584 unsigned NumElts = DataTy->getVectorNumElements();
2585 if (NumElts == 1 || !isPowerOf2_32(NumElts))
2586 return false;
2587 }
Elena Demikhovsky09285852015-10-25 15:37:55 +00002588 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00002589 int DataWidth = isa<PointerType>(ScalarTy) ?
2590 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
Elena Demikhovsky09285852015-10-25 15:37:55 +00002591
Craig Topperea37e202017-11-25 18:09:37 +00002592 // Some CPUs have better gather performance than others.
2593 // TODO: Remove the explicit ST->hasAVX512()?, That would mean we would only
2594 // enable gather with a -march.
2595 return (DataWidth == 32 || DataWidth == 64) &&
Craig Topper0d797a32018-01-20 00:26:08 +00002596 (ST->hasAVX512() || (ST->hasFastGather() && ST->hasAVX2()));
Elena Demikhovsky09285852015-10-25 15:37:55 +00002597}
2598
2599bool X86TTIImpl::isLegalMaskedScatter(Type *DataType) {
Mohammed Agabaria115f68e2017-11-20 08:18:12 +00002600 // AVX2 doesn't support scatter
2601 if (!ST->hasAVX512())
2602 return false;
Elena Demikhovsky09285852015-10-25 15:37:55 +00002603 return isLegalMaskedGather(DataType);
2604}
2605
Sanjay Patel6fd43912017-09-09 13:38:18 +00002606bool X86TTIImpl::hasDivRemOp(Type *DataType, bool IsSigned) {
2607 EVT VT = TLI->getValueType(DL, DataType);
2608 return TLI->isOperationLegal(IsSigned ? ISD::SDIVREM : ISD::UDIVREM, VT);
2609}
2610
Sanjay Patel0de1a4b2017-11-27 21:15:43 +00002611bool X86TTIImpl::isFCmpOrdCheaperThanFCmpZero(Type *Ty) {
2612 return false;
2613}
2614
Eric Christopherd566fb12015-07-29 22:09:48 +00002615bool X86TTIImpl::areInlineCompatible(const Function *Caller,
2616 const Function *Callee) const {
Eric Christophere1002262015-07-02 01:11:50 +00002617 const TargetMachine &TM = getTLI()->getTargetMachine();
2618
2619 // Work this as a subsetting of subtarget features.
2620 const FeatureBitset &CallerBits =
2621 TM.getSubtargetImpl(*Caller)->getFeatureBits();
2622 const FeatureBitset &CalleeBits =
2623 TM.getSubtargetImpl(*Callee)->getFeatureBits();
2624
2625 // FIXME: This is likely too limiting as it will include subtarget features
2626 // that we might not care about for inlining, but it is conservatively
2627 // correct.
2628 return (CallerBits & CalleeBits) == CalleeBits;
2629}
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002630
Clement Courbetb2c3eb82017-10-30 14:19:33 +00002631const X86TTIImpl::TTI::MemCmpExpansionOptions *
2632X86TTIImpl::enableMemCmpExpansion(bool IsZeroCmp) const {
2633 // Only enable vector loads for equality comparison.
2634 // Right now the vector version is not as fast, see #33329.
2635 static const auto ThreeWayOptions = [this]() {
2636 TTI::MemCmpExpansionOptions Options;
2637 if (ST->is64Bit()) {
2638 Options.LoadSizes.push_back(8);
2639 }
2640 Options.LoadSizes.push_back(4);
2641 Options.LoadSizes.push_back(2);
2642 Options.LoadSizes.push_back(1);
2643 return Options;
2644 }();
2645 static const auto EqZeroOptions = [this]() {
2646 TTI::MemCmpExpansionOptions Options;
2647 // TODO: enable AVX512 when the DAG is ready.
2648 // if (ST->hasAVX512()) Options.LoadSizes.push_back(64);
2649 if (ST->hasAVX2()) Options.LoadSizes.push_back(32);
2650 if (ST->hasSSE2()) Options.LoadSizes.push_back(16);
2651 if (ST->is64Bit()) {
2652 Options.LoadSizes.push_back(8);
2653 }
2654 Options.LoadSizes.push_back(4);
2655 Options.LoadSizes.push_back(2);
2656 Options.LoadSizes.push_back(1);
2657 return Options;
2658 }();
2659 return IsZeroCmp ? &EqZeroOptions : &ThreeWayOptions;
Sanjay Patel06566292017-06-20 15:58:30 +00002660}
2661
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002662bool X86TTIImpl::enableInterleavedAccessVectorization() {
2663 // TODO: We expect this to be beneficial regardless of arch,
2664 // but there are currently some unexplained performance artifacts on Atom.
2665 // As a temporary solution, disable on Atom.
Mohammed Agabaria20caee92017-01-25 09:14:48 +00002666 return !(ST->isAtom());
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002667}
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002668
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002669// Get estimation for interleaved load/store operations for AVX2.
2670// \p Factor is the interleaved-access factor (stride) - number of
2671// (interleaved) elements in the group.
2672// \p Indices contains the indices for a strided load: when the
2673// interleaved load has gaps they indicate which elements are used.
2674// If Indices is empty (or if the number of indices is equal to the size
2675// of the interleaved-access as given in \p Factor) the access has no gaps.
2676//
2677// As opposed to AVX-512, AVX2 does not have generic shuffles that allow
2678// computing the cost using a generic formula as a function of generic
2679// shuffles. We therefore use a lookup table instead, filled according to
2680// the instruction sequences that codegen currently generates.
2681int X86TTIImpl::getInterleavedMemoryOpCostAVX2(unsigned Opcode, Type *VecTy,
2682 unsigned Factor,
2683 ArrayRef<unsigned> Indices,
2684 unsigned Alignment,
2685 unsigned AddressSpace) {
2686
2687 // We currently Support only fully-interleaved groups, with no gaps.
2688 // TODO: Support also strided loads (interleaved-groups with gaps).
2689 if (Indices.size() && Indices.size() != Factor)
2690 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2691 Alignment, AddressSpace);
2692
2693 // VecTy for interleave memop is <VF*Factor x Elt>.
2694 // So, for VF=4, Interleave Factor = 3, Element type = i32 we have
2695 // VecTy = <12 x i32>.
2696 MVT LegalVT = getTLI()->getTypeLegalizationCost(DL, VecTy).second;
2697
2698 // This function can be called with VecTy=<6xi128>, Factor=3, in which case
2699 // the VF=2, while v2i128 is an unsupported MVT vector type
2700 // (see MachineValueType.h::getVectorVT()).
2701 if (!LegalVT.isVector())
2702 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2703 Alignment, AddressSpace);
2704
2705 unsigned VF = VecTy->getVectorNumElements() / Factor;
2706 Type *ScalarTy = VecTy->getVectorElementType();
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002707
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002708 // Calculate the number of memory operations (NumOfMemOps), required
2709 // for load/store the VecTy.
2710 unsigned VecTySize = DL.getTypeStoreSize(VecTy);
2711 unsigned LegalVTSize = LegalVT.getStoreSize();
2712 unsigned NumOfMemOps = (VecTySize + LegalVTSize - 1) / LegalVTSize;
2713
2714 // Get the cost of one memory operation.
2715 Type *SingleMemOpTy = VectorType::get(VecTy->getVectorElementType(),
2716 LegalVT.getVectorNumElements());
2717 unsigned MemOpCost =
2718 getMemoryOpCost(Opcode, SingleMemOpTy, Alignment, AddressSpace);
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002719
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002720 VectorType *VT = VectorType::get(ScalarTy, VF);
2721 EVT ETy = TLI->getValueType(DL, VT);
2722 if (!ETy.isSimple())
2723 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2724 Alignment, AddressSpace);
2725
2726 // TODO: Complete for other data-types and strides.
2727 // Each combination of Stride, ElementTy and VF results in a different
2728 // sequence; The cost tables are therefore accessed with:
2729 // Factor (stride) and VectorType=VFxElemType.
2730 // The Cost accounts only for the shuffle sequence;
2731 // The cost of the loads/stores is accounted for separately.
2732 //
2733 static const CostTblEntry AVX2InterleavedLoadTbl[] = {
Mohammed Agabaria6e6d5322017-11-16 09:38:32 +00002734 { 2, MVT::v4i64, 6 }, //(load 8i64 and) deinterleave into 2 x 4i64
2735 { 2, MVT::v4f64, 6 }, //(load 8f64 and) deinterleave into 2 x 4f64
2736
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002737 { 3, MVT::v2i8, 10 }, //(load 6i8 and) deinterleave into 3 x 2i8
2738 { 3, MVT::v4i8, 4 }, //(load 12i8 and) deinterleave into 3 x 4i8
2739 { 3, MVT::v8i8, 9 }, //(load 24i8 and) deinterleave into 3 x 8i8
Michael Zuckerman49293262017-10-18 11:41:55 +00002740 { 3, MVT::v16i8, 11}, //(load 48i8 and) deinterleave into 3 x 16i8
2741 { 3, MVT::v32i8, 13}, //(load 96i8 and) deinterleave into 3 x 32i8
Mohammed Agabaria66917582017-11-06 10:56:20 +00002742 { 3, MVT::v8f32, 17 }, //(load 24f32 and)deinterleave into 3 x 8f32
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002743
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002744 { 4, MVT::v2i8, 12 }, //(load 8i8 and) deinterleave into 4 x 2i8
2745 { 4, MVT::v4i8, 4 }, //(load 16i8 and) deinterleave into 4 x 4i8
2746 { 4, MVT::v8i8, 20 }, //(load 32i8 and) deinterleave into 4 x 8i8
2747 { 4, MVT::v16i8, 39 }, //(load 64i8 and) deinterleave into 4 x 16i8
Mohammed Agabaria66917582017-11-06 10:56:20 +00002748 { 4, MVT::v32i8, 80 }, //(load 128i8 and) deinterleave into 4 x 32i8
2749
2750 { 8, MVT::v8f32, 40 } //(load 64f32 and)deinterleave into 8 x 8f32
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002751 };
2752
2753 static const CostTblEntry AVX2InterleavedStoreTbl[] = {
Mohammed Agabaria6e6d5322017-11-16 09:38:32 +00002754 { 2, MVT::v4i64, 6 }, //interleave into 2 x 4i64 into 8i64 (and store)
2755 { 2, MVT::v4f64, 6 }, //interleave into 2 x 4f64 into 8f64 (and store)
2756
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002757 { 3, MVT::v2i8, 7 }, //interleave 3 x 2i8 into 6i8 (and store)
2758 { 3, MVT::v4i8, 8 }, //interleave 3 x 4i8 into 12i8 (and store)
2759 { 3, MVT::v8i8, 11 }, //interleave 3 x 8i8 into 24i8 (and store)
Michael Zuckerman49293262017-10-18 11:41:55 +00002760 { 3, MVT::v16i8, 11 }, //interleave 3 x 16i8 into 48i8 (and store)
2761 { 3, MVT::v32i8, 13 }, //interleave 3 x 32i8 into 96i8 (and store)
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002762
2763 { 4, MVT::v2i8, 12 }, //interleave 4 x 2i8 into 8i8 (and store)
2764 { 4, MVT::v4i8, 9 }, //interleave 4 x 4i8 into 16i8 (and store)
Michael Zuckerman49293262017-10-18 11:41:55 +00002765 { 4, MVT::v8i8, 10 }, //interleave 4 x 8i8 into 32i8 (and store)
2766 { 4, MVT::v16i8, 10 }, //interleave 4 x 16i8 into 64i8 (and store)
2767 { 4, MVT::v32i8, 12 } //interleave 4 x 32i8 into 128i8 (and store)
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002768 };
2769
2770 if (Opcode == Instruction::Load) {
2771 if (const auto *Entry =
2772 CostTableLookup(AVX2InterleavedLoadTbl, Factor, ETy.getSimpleVT()))
2773 return NumOfMemOps * MemOpCost + Entry->Cost;
2774 } else {
2775 assert(Opcode == Instruction::Store &&
2776 "Expected Store Instruction at this point");
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002777 if (const auto *Entry =
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002778 CostTableLookup(AVX2InterleavedStoreTbl, Factor, ETy.getSimpleVT()))
2779 return NumOfMemOps * MemOpCost + Entry->Cost;
2780 }
2781
2782 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2783 Alignment, AddressSpace);
2784}
2785
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002786// Get estimation for interleaved load/store operations and strided load.
2787// \p Indices contains indices for strided load.
2788// \p Factor - the factor of interleaving.
2789// AVX-512 provides 3-src shuffles that significantly reduces the cost.
2790int X86TTIImpl::getInterleavedMemoryOpCostAVX512(unsigned Opcode, Type *VecTy,
2791 unsigned Factor,
2792 ArrayRef<unsigned> Indices,
2793 unsigned Alignment,
2794 unsigned AddressSpace) {
2795
2796 // VecTy for interleave memop is <VF*Factor x Elt>.
2797 // So, for VF=4, Interleave Factor = 3, Element type = i32 we have
2798 // VecTy = <12 x i32>.
2799
2800 // Calculate the number of memory operations (NumOfMemOps), required
2801 // for load/store the VecTy.
2802 MVT LegalVT = getTLI()->getTypeLegalizationCost(DL, VecTy).second;
2803 unsigned VecTySize = DL.getTypeStoreSize(VecTy);
2804 unsigned LegalVTSize = LegalVT.getStoreSize();
2805 unsigned NumOfMemOps = (VecTySize + LegalVTSize - 1) / LegalVTSize;
2806
2807 // Get the cost of one memory operation.
2808 Type *SingleMemOpTy = VectorType::get(VecTy->getVectorElementType(),
2809 LegalVT.getVectorNumElements());
2810 unsigned MemOpCost =
2811 getMemoryOpCost(Opcode, SingleMemOpTy, Alignment, AddressSpace);
2812
Michael Zuckerman49293262017-10-18 11:41:55 +00002813 unsigned VF = VecTy->getVectorNumElements() / Factor;
2814 MVT VT = MVT::getVectorVT(MVT::getVT(VecTy->getScalarType()), VF);
2815
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002816 if (Opcode == Instruction::Load) {
Michael Zuckerman49293262017-10-18 11:41:55 +00002817 // The tables (AVX512InterleavedLoadTbl and AVX512InterleavedStoreTbl)
2818 // contain the cost of the optimized shuffle sequence that the
2819 // X86InterleavedAccess pass will generate.
2820 // The cost of loads and stores are computed separately from the table.
2821
2822 // X86InterleavedAccess support only the following interleaved-access group.
2823 static const CostTblEntry AVX512InterleavedLoadTbl[] = {
2824 {3, MVT::v16i8, 12}, //(load 48i8 and) deinterleave into 3 x 16i8
2825 {3, MVT::v32i8, 14}, //(load 96i8 and) deinterleave into 3 x 32i8
2826 {3, MVT::v64i8, 22}, //(load 96i8 and) deinterleave into 3 x 32i8
2827 };
2828
2829 if (const auto *Entry =
2830 CostTableLookup(AVX512InterleavedLoadTbl, Factor, VT))
2831 return NumOfMemOps * MemOpCost + Entry->Cost;
2832 //If an entry does not exist, fallback to the default implementation.
2833
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002834 // Kind of shuffle depends on number of loaded values.
2835 // If we load the entire data in one register, we can use a 1-src shuffle.
2836 // Otherwise, we'll merge 2 sources in each operation.
2837 TTI::ShuffleKind ShuffleKind =
2838 (NumOfMemOps > 1) ? TTI::SK_PermuteTwoSrc : TTI::SK_PermuteSingleSrc;
2839
2840 unsigned ShuffleCost =
2841 getShuffleCost(ShuffleKind, SingleMemOpTy, 0, nullptr);
2842
2843 unsigned NumOfLoadsInInterleaveGrp =
2844 Indices.size() ? Indices.size() : Factor;
2845 Type *ResultTy = VectorType::get(VecTy->getVectorElementType(),
2846 VecTy->getVectorNumElements() / Factor);
2847 unsigned NumOfResults =
2848 getTLI()->getTypeLegalizationCost(DL, ResultTy).first *
2849 NumOfLoadsInInterleaveGrp;
2850
2851 // About a half of the loads may be folded in shuffles when we have only
2852 // one result. If we have more than one result, we do not fold loads at all.
2853 unsigned NumOfUnfoldedLoads =
2854 NumOfResults > 1 ? NumOfMemOps : NumOfMemOps / 2;
2855
2856 // Get a number of shuffle operations per result.
2857 unsigned NumOfShufflesPerResult =
2858 std::max((unsigned)1, (unsigned)(NumOfMemOps - 1));
2859
2860 // The SK_MergeTwoSrc shuffle clobbers one of src operands.
2861 // When we have more than one destination, we need additional instructions
2862 // to keep sources.
2863 unsigned NumOfMoves = 0;
2864 if (NumOfResults > 1 && ShuffleKind == TTI::SK_PermuteTwoSrc)
2865 NumOfMoves = NumOfResults * NumOfShufflesPerResult / 2;
2866
2867 int Cost = NumOfResults * NumOfShufflesPerResult * ShuffleCost +
2868 NumOfUnfoldedLoads * MemOpCost + NumOfMoves;
2869
2870 return Cost;
2871 }
2872
2873 // Store.
2874 assert(Opcode == Instruction::Store &&
2875 "Expected Store Instruction at this point");
Michael Zuckerman49293262017-10-18 11:41:55 +00002876 // X86InterleavedAccess support only the following interleaved-access group.
2877 static const CostTblEntry AVX512InterleavedStoreTbl[] = {
2878 {3, MVT::v16i8, 12}, // interleave 3 x 16i8 into 48i8 (and store)
2879 {3, MVT::v32i8, 14}, // interleave 3 x 32i8 into 96i8 (and store)
2880 {3, MVT::v64i8, 26}, // interleave 3 x 64i8 into 96i8 (and store)
2881
2882 {4, MVT::v8i8, 10}, // interleave 4 x 8i8 into 32i8 (and store)
2883 {4, MVT::v16i8, 11}, // interleave 4 x 16i8 into 64i8 (and store)
2884 {4, MVT::v32i8, 14}, // interleave 4 x 32i8 into 128i8 (and store)
2885 {4, MVT::v64i8, 24} // interleave 4 x 32i8 into 256i8 (and store)
2886 };
2887
2888 if (const auto *Entry =
2889 CostTableLookup(AVX512InterleavedStoreTbl, Factor, VT))
2890 return NumOfMemOps * MemOpCost + Entry->Cost;
2891 //If an entry does not exist, fallback to the default implementation.
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002892
2893 // There is no strided stores meanwhile. And store can't be folded in
2894 // shuffle.
2895 unsigned NumOfSources = Factor; // The number of values to be merged.
2896 unsigned ShuffleCost =
2897 getShuffleCost(TTI::SK_PermuteTwoSrc, SingleMemOpTy, 0, nullptr);
2898 unsigned NumOfShufflesPerStore = NumOfSources - 1;
2899
2900 // The SK_MergeTwoSrc shuffle clobbers one of src operands.
2901 // We need additional instructions to keep sources.
2902 unsigned NumOfMoves = NumOfMemOps * NumOfShufflesPerStore / 2;
2903 int Cost = NumOfMemOps * (MemOpCost + NumOfShufflesPerStore * ShuffleCost) +
2904 NumOfMoves;
2905 return Cost;
2906}
2907
2908int X86TTIImpl::getInterleavedMemoryOpCost(unsigned Opcode, Type *VecTy,
2909 unsigned Factor,
2910 ArrayRef<unsigned> Indices,
2911 unsigned Alignment,
2912 unsigned AddressSpace) {
Craig Topper8b0f1852017-12-06 18:40:46 +00002913 auto isSupportedOnAVX512 = [](Type *VecTy, bool HasBW) {
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002914 Type *EltTy = VecTy->getVectorElementType();
2915 if (EltTy->isFloatTy() || EltTy->isDoubleTy() || EltTy->isIntegerTy(64) ||
2916 EltTy->isIntegerTy(32) || EltTy->isPointerTy())
2917 return true;
Craig Topper8b0f1852017-12-06 18:40:46 +00002918 if (EltTy->isIntegerTy(16) || EltTy->isIntegerTy(8))
2919 return HasBW;
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002920 return false;
2921 };
Craig Topper8b0f1852017-12-06 18:40:46 +00002922 if (ST->hasAVX512() && isSupportedOnAVX512(VecTy, ST->hasBWI()))
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002923 return getInterleavedMemoryOpCostAVX512(Opcode, VecTy, Factor, Indices,
2924 Alignment, AddressSpace);
Dorit Nuzmane0e0f1d2017-06-25 08:26:25 +00002925 if (ST->hasAVX2())
2926 return getInterleavedMemoryOpCostAVX2(Opcode, VecTy, Factor, Indices,
2927 Alignment, AddressSpace);
Simon Pilgrim7b89ab52017-07-31 17:09:27 +00002928
Elena Demikhovsky21706cb2017-01-02 10:37:52 +00002929 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
2930 Alignment, AddressSpace);
2931}