blob: db563c08632d072eee38943aea50a0bc7b350428 [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"
Juergen Ributzkaf26beda2014-01-25 02:02:55 +000045#include "llvm/IR/IntrinsicInst.h"
Chandler Carruth664e3542013-01-07 01:37:14 +000046#include "llvm/Support/Debug.h"
Renato Golind4c392e2013-01-24 23:01:00 +000047#include "llvm/Target/CostTable.h"
Chandler Carruth8a8cd2b2014-01-07 11:48:04 +000048#include "llvm/Target/TargetLowering.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
Chandler Carruth705b1852015-01-31 03:43:40 +000069unsigned X86TTIImpl::getNumberOfRegisters(bool Vector) {
Nadav Rotemb1791a72013-01-09 22:29:00 +000070 if (Vector && !ST->hasSSE1())
71 return 0;
72
Adam Nemet2820a5b2014-07-09 18:22:33 +000073 if (ST->is64Bit()) {
74 if (Vector && ST->hasAVX512())
75 return 32;
Chandler Carruth664e3542013-01-07 01:37:14 +000076 return 16;
Adam Nemet2820a5b2014-07-09 18:22:33 +000077 }
Chandler Carruth664e3542013-01-07 01:37:14 +000078 return 8;
79}
80
Chandler Carruth705b1852015-01-31 03:43:40 +000081unsigned X86TTIImpl::getRegisterBitWidth(bool Vector) {
Nadav Rotemb1791a72013-01-09 22:29:00 +000082 if (Vector) {
Adam Nemet2820a5b2014-07-09 18:22:33 +000083 if (ST->hasAVX512()) return 512;
Nadav Rotemb1791a72013-01-09 22:29:00 +000084 if (ST->hasAVX()) return 256;
85 if (ST->hasSSE1()) return 128;
86 return 0;
87 }
88
89 if (ST->is64Bit())
90 return 64;
Nadav Rotemb1791a72013-01-09 22:29:00 +000091
Hans Wennborg083ca9b2015-10-06 23:24:35 +000092 return 32;
Nadav Rotemb1791a72013-01-09 22:29:00 +000093}
94
Wei Mi062c7442015-05-06 17:12:25 +000095unsigned X86TTIImpl::getMaxInterleaveFactor(unsigned VF) {
96 // If the loop will not be vectorized, don't interleave the loop.
97 // Let regular unroll to unroll the loop, which saves the overflow
98 // check and memory check cost.
99 if (VF == 1)
100 return 1;
101
Nadav Rotemb696c362013-01-09 01:15:42 +0000102 if (ST->isAtom())
103 return 1;
104
105 // Sandybridge and Haswell have multiple execution ports and pipelined
106 // vector units.
107 if (ST->hasAVX())
108 return 4;
109
110 return 2;
111}
112
Chandler Carruth93205eb2015-08-05 18:08:10 +0000113int X86TTIImpl::getArithmeticInstrCost(
Chandler Carruth705b1852015-01-31 03:43:40 +0000114 unsigned Opcode, Type *Ty, TTI::OperandValueKind Op1Info,
115 TTI::OperandValueKind Op2Info, TTI::OperandValueProperties Opd1PropInfo,
116 TTI::OperandValueProperties Opd2PropInfo) {
Chandler Carruth664e3542013-01-07 01:37:14 +0000117 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000118 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Ty);
Chandler Carruth664e3542013-01-07 01:37:14 +0000119
120 int ISD = TLI->InstructionOpcodeToISD(Opcode);
121 assert(ISD && "Invalid opcode");
122
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000123 if (ISD == ISD::SDIV &&
124 Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
125 Opd2PropInfo == TargetTransformInfo::OP_PowerOf2) {
126 // On X86, vector signed division by constants power-of-two are
127 // normally expanded to the sequence SRA + SRL + ADD + SRA.
128 // The OperandValue properties many not be same as that of previous
129 // operation;conservatively assume OP_None.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000130 int Cost = 2 * getArithmeticInstrCost(Instruction::AShr, Ty, Op1Info,
131 Op2Info, TargetTransformInfo::OP_None,
132 TargetTransformInfo::OP_None);
Karthik Bhat7f33ff72014-08-25 04:56:54 +0000133 Cost += getArithmeticInstrCost(Instruction::LShr, Ty, Op1Info, Op2Info,
134 TargetTransformInfo::OP_None,
135 TargetTransformInfo::OP_None);
136 Cost += getArithmeticInstrCost(Instruction::Add, Ty, Op1Info, Op2Info,
137 TargetTransformInfo::OP_None,
138 TargetTransformInfo::OP_None);
139
140 return Cost;
141 }
142
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000143 static const CostTblEntry AVX512BWUniformConstCostTable[] = {
144 { ISD::SDIV, MVT::v32i16, 6 }, // vpmulhw sequence
145 { ISD::UDIV, MVT::v32i16, 6 }, // vpmulhuw sequence
146 };
147
148 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
149 ST->hasBWI()) {
150 if (const auto *Entry = CostTableLookup(AVX512BWUniformConstCostTable, ISD,
151 LT.second))
152 return LT.first * Entry->Cost;
153 }
154
155 static const CostTblEntry AVX512UniformConstCostTable[] = {
156 { ISD::SDIV, MVT::v16i32, 15 }, // vpmuldq sequence
157 { ISD::UDIV, MVT::v16i32, 15 }, // vpmuludq sequence
158 };
159
160 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
161 ST->hasAVX512()) {
162 if (const auto *Entry = CostTableLookup(AVX512UniformConstCostTable, ISD,
163 LT.second))
164 return LT.first * Entry->Cost;
165 }
166
Craig Topper4b275762015-10-28 04:02:12 +0000167 static const CostTblEntry AVX2UniformConstCostTable[] = {
Simon Pilgrim8fbf1c12015-07-06 22:35:19 +0000168 { ISD::SRA, MVT::v4i64, 4 }, // 2 x psrad + shuffle.
169
Benjamin Kramer7c372272014-04-26 14:53:05 +0000170 { ISD::SDIV, MVT::v16i16, 6 }, // vpmulhw sequence
171 { ISD::UDIV, MVT::v16i16, 6 }, // vpmulhuw sequence
172 { ISD::SDIV, MVT::v8i32, 15 }, // vpmuldq sequence
173 { ISD::UDIV, MVT::v8i32, 15 }, // vpmuludq sequence
174 };
175
176 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
177 ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000178 if (const auto *Entry = CostTableLookup(AVX2UniformConstCostTable, ISD,
179 LT.second))
180 return LT.first * Entry->Cost;
Benjamin Kramer7c372272014-04-26 14:53:05 +0000181 }
182
Simon Pilgrim365be4f2016-10-20 18:00:35 +0000183 static const CostTblEntry SSE2UniformConstCostTable[] = {
184 { ISD::SDIV, MVT::v16i16, 12 }, // pmulhw sequence
185 { ISD::SDIV, MVT::v8i16, 6 }, // pmulhw sequence
186 { ISD::UDIV, MVT::v16i16, 12 }, // pmulhuw sequence
187 { ISD::UDIV, MVT::v8i16, 6 }, // pmulhuw sequence
188 { ISD::SDIV, MVT::v8i32, 38 }, // pmuludq sequence
189 { ISD::SDIV, MVT::v4i32, 19 }, // pmuludq sequence
190 { ISD::UDIV, MVT::v8i32, 30 }, // pmuludq sequence
191 { ISD::UDIV, MVT::v4i32, 15 }, // pmuludq sequence
192 };
193
194 if (Op2Info == TargetTransformInfo::OK_UniformConstantValue &&
195 ST->hasSSE2()) {
196 // pmuldq sequence.
197 if (ISD == ISD::SDIV && LT.second == MVT::v8i32 && ST->hasAVX())
198 return LT.first * 30;
199 if (ISD == ISD::SDIV && LT.second == MVT::v4i32 && ST->hasSSE41())
200 return LT.first * 15;
201
202 if (const auto *Entry = CostTableLookup(SSE2UniformConstCostTable, ISD,
203 LT.second))
204 return LT.first * Entry->Cost;
205 }
206
Simon Pilgrim820e1322016-10-27 15:27:00 +0000207 static const CostTblEntry AVX512DQCostTable[] = {
208 { ISD::MUL, MVT::v2i64, 1 },
209 { ISD::MUL, MVT::v4i64, 1 },
210 { ISD::MUL, MVT::v8i64, 1 }
211 };
212
213 // Look for AVX512DQ lowering tricks for custom cases.
214 if (ST->hasDQI()) {
215 if (const auto *Entry = CostTableLookup(AVX512DQCostTable, ISD,
216 LT.second))
217 return LT.first * Entry->Cost;
218 }
219
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000220 static const CostTblEntry AVX512BWCostTable[] = {
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000221 { ISD::MUL, MVT::v64i8, 11 }, // extend/pmullw/trunc sequence.
222 { ISD::MUL, MVT::v32i8, 4 }, // extend/pmullw/trunc sequence.
223 { ISD::MUL, MVT::v16i8, 4 }, // extend/pmullw/trunc sequence.
224
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000225 // Vectorizing division is a bad idea. See the SSE2 table for more comments.
226 { ISD::SDIV, MVT::v64i8, 64*20 },
227 { ISD::SDIV, MVT::v32i16, 32*20 },
228 { ISD::SDIV, MVT::v16i32, 16*20 },
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000229 { ISD::SDIV, MVT::v8i64, 8*20 },
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000230 { ISD::UDIV, MVT::v64i8, 64*20 },
231 { ISD::UDIV, MVT::v32i16, 32*20 },
232 { ISD::UDIV, MVT::v16i32, 16*20 },
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000233 { ISD::UDIV, MVT::v8i64, 8*20 },
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000234 };
235
236 // Look for AVX512BW lowering tricks for custom cases.
237 if (ST->hasBWI()) {
238 if (const auto *Entry = CostTableLookup(AVX512BWCostTable, ISD,
239 LT.second))
240 return LT.first * Entry->Cost;
241 }
242
Craig Topper4b275762015-10-28 04:02:12 +0000243 static const CostTblEntry AVX512CostTable[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +0000244 { ISD::SHL, MVT::v16i32, 1 },
245 { ISD::SRL, MVT::v16i32, 1 },
246 { ISD::SRA, MVT::v16i32, 1 },
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000247 { ISD::SHL, MVT::v8i64, 1 },
248 { ISD::SRL, MVT::v8i64, 1 },
249 { ISD::SRA, MVT::v8i64, 1 },
250
251 { ISD::MUL, MVT::v32i8, 13 }, // extend/pmullw/trunc sequence.
252 { ISD::MUL, MVT::v16i8, 5 }, // extend/pmullw/trunc sequence.
Elena Demikhovsky27012472014-09-16 07:57:37 +0000253 };
254
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000255 if (ST->hasAVX512()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000256 if (const auto *Entry = CostTableLookup(AVX512CostTable, ISD, LT.second))
257 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000258 }
259
Craig Topper4b275762015-10-28 04:02:12 +0000260 static const CostTblEntry AVX2CostTable[] = {
Michael Liao70dd7f92013-03-20 22:01:10 +0000261 // Shifts on v4i64/v8i32 on AVX2 is legal even though we declare to
262 // customize them to detect the cases where shift amount is a scalar one.
263 { ISD::SHL, MVT::v4i32, 1 },
264 { ISD::SRL, MVT::v4i32, 1 },
265 { ISD::SRA, MVT::v4i32, 1 },
266 { ISD::SHL, MVT::v8i32, 1 },
267 { ISD::SRL, MVT::v8i32, 1 },
268 { ISD::SRA, MVT::v8i32, 1 },
269 { ISD::SHL, MVT::v2i64, 1 },
270 { ISD::SRL, MVT::v2i64, 1 },
271 { ISD::SHL, MVT::v4i64, 1 },
272 { ISD::SRL, MVT::v4i64, 1 },
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000273 };
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000274
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000275 // Look for AVX2 lowering tricks.
276 if (ST->hasAVX2()) {
277 if (ISD == ISD::SHL && LT.second == MVT::v16i16 &&
278 (Op2Info == TargetTransformInfo::OK_UniformConstantValue ||
279 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue))
280 // On AVX2, a packed v16i16 shift left by a constant build_vector
281 // is lowered into a vector multiply (vpmullw).
282 return LT.first;
283
Craig Topperee0c8592015-10-27 04:14:24 +0000284 if (const auto *Entry = CostTableLookup(AVX2CostTable, ISD, LT.second))
285 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000286 }
287
Craig Topper4b275762015-10-28 04:02:12 +0000288 static const CostTblEntry XOPCostTable[] = {
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000289 // 128bit shifts take 1cy, but right shifts require negation beforehand.
290 { ISD::SHL, MVT::v16i8, 1 },
291 { ISD::SRL, MVT::v16i8, 2 },
292 { ISD::SRA, MVT::v16i8, 2 },
293 { ISD::SHL, MVT::v8i16, 1 },
294 { ISD::SRL, MVT::v8i16, 2 },
295 { ISD::SRA, MVT::v8i16, 2 },
296 { ISD::SHL, MVT::v4i32, 1 },
297 { ISD::SRL, MVT::v4i32, 2 },
298 { ISD::SRA, MVT::v4i32, 2 },
299 { ISD::SHL, MVT::v2i64, 1 },
300 { ISD::SRL, MVT::v2i64, 2 },
301 { ISD::SRA, MVT::v2i64, 2 },
302 // 256bit shifts require splitting if AVX2 didn't catch them above.
303 { ISD::SHL, MVT::v32i8, 2 },
304 { ISD::SRL, MVT::v32i8, 4 },
305 { ISD::SRA, MVT::v32i8, 4 },
306 { ISD::SHL, MVT::v16i16, 2 },
307 { ISD::SRL, MVT::v16i16, 4 },
308 { ISD::SRA, MVT::v16i16, 4 },
309 { ISD::SHL, MVT::v8i32, 2 },
310 { ISD::SRL, MVT::v8i32, 4 },
311 { ISD::SRA, MVT::v8i32, 4 },
312 { ISD::SHL, MVT::v4i64, 2 },
313 { ISD::SRL, MVT::v4i64, 4 },
314 { ISD::SRA, MVT::v4i64, 4 },
315 };
316
317 // Look for XOP lowering tricks.
318 if (ST->hasXOP()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000319 if (const auto *Entry = CostTableLookup(XOPCostTable, ISD, LT.second))
320 return LT.first * Entry->Cost;
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000321 }
322
Craig Topper4b275762015-10-28 04:02:12 +0000323 static const CostTblEntry AVX2CustomCostTable[] = {
Simon Pilgrim59656802015-06-11 07:46:37 +0000324 { ISD::SHL, MVT::v32i8, 11 }, // vpblendvb sequence.
Simon Pilgrim0be4fa72015-05-25 17:49:13 +0000325 { ISD::SHL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000326
Simon Pilgrim59656802015-06-11 07:46:37 +0000327 { ISD::SRL, MVT::v32i8, 11 }, // vpblendvb sequence.
Simon Pilgrim0be4fa72015-05-25 17:49:13 +0000328 { ISD::SRL, MVT::v16i16, 10 }, // extend/vpsrlvd/pack sequence.
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000329
Simon Pilgrim59656802015-06-11 07:46:37 +0000330 { ISD::SRA, MVT::v32i8, 24 }, // vpblendvb sequence.
Simon Pilgrim0be4fa72015-05-25 17:49:13 +0000331 { ISD::SRA, MVT::v16i16, 10 }, // extend/vpsravd/pack sequence.
Simon Pilgrim86478c62015-07-29 20:31:45 +0000332 { ISD::SRA, MVT::v2i64, 4 }, // srl/xor/sub sequence.
333 { ISD::SRA, MVT::v4i64, 4 }, // srl/xor/sub sequence.
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000334
335 { ISD::MUL, MVT::v32i8, 17 }, // extend/pmullw/trunc sequence.
336 { ISD::MUL, MVT::v16i8, 7 }, // extend/pmullw/trunc sequence.
337
Alexey Bataevd07c7312016-10-31 12:10:53 +0000338 { ISD::FDIV, MVT::f32, 7 }, // Haswell from http://www.agner.org/
339 { ISD::FDIV, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
340 { ISD::FDIV, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
341 { ISD::FDIV, MVT::f64, 14 }, // Haswell from http://www.agner.org/
342 { ISD::FDIV, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
343 { ISD::FDIV, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000344 };
Arnold Schwaighofera04b9ef2013-06-25 19:14:09 +0000345
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000346 // Look for AVX2 lowering tricks for custom cases.
347 if (ST->hasAVX2()) {
348 if (const auto *Entry = CostTableLookup(AVX2CustomCostTable, ISD,
349 LT.second))
350 return LT.first * Entry->Cost;
351 }
352
353 static const CostTblEntry AVXCustomCostTable[] = {
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000354 { ISD::MUL, MVT::v32i8, 26 }, // extend/pmullw/trunc sequence.
355
Alexey Bataevd07c7312016-10-31 12:10:53 +0000356 { ISD::FDIV, MVT::f32, 14 }, // SNB from http://www.agner.org/
357 { ISD::FDIV, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
358 { ISD::FDIV, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
359 { ISD::FDIV, MVT::f64, 22 }, // SNB from http://www.agner.org/
360 { ISD::FDIV, MVT::v2f64, 22 }, // SNB from http://www.agner.org/
361 { ISD::FDIV, MVT::v4f64, 44 }, // SNB from http://www.agner.org/
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000362
Arnold Schwaighofera04b9ef2013-06-25 19:14:09 +0000363 // Vectorizing division is a bad idea. See the SSE2 table for more comments.
364 { ISD::SDIV, MVT::v32i8, 32*20 },
365 { ISD::SDIV, MVT::v16i16, 16*20 },
366 { ISD::SDIV, MVT::v8i32, 8*20 },
367 { ISD::SDIV, MVT::v4i64, 4*20 },
368 { ISD::UDIV, MVT::v32i8, 32*20 },
369 { ISD::UDIV, MVT::v16i16, 16*20 },
370 { ISD::UDIV, MVT::v8i32, 8*20 },
371 { ISD::UDIV, MVT::v4i64, 4*20 },
Michael Liao70dd7f92013-03-20 22:01:10 +0000372 };
373
Simon Pilgrim3d11c992015-09-30 08:17:50 +0000374 // Look for AVX2 lowering tricks for custom cases.
Simon Pilgrim025e26d2016-10-20 16:39:11 +0000375 if (ST->hasAVX()) {
376 if (const auto *Entry = CostTableLookup(AVXCustomCostTable, ISD,
Craig Topperee0c8592015-10-27 04:14:24 +0000377 LT.second))
378 return LT.first * Entry->Cost;
Michael Liao70dd7f92013-03-20 22:01:10 +0000379 }
380
Alexey Bataevd07c7312016-10-31 12:10:53 +0000381 static const CostTblEntry SSE42FloatCostTable[] = {
382 { ISD::FDIV, MVT::f32, 14 }, // Nehalem from http://www.agner.org/
383 { ISD::FDIV, MVT::v4f32, 14 }, // Nehalem from http://www.agner.org/
384 { ISD::FDIV, MVT::f64, 22 }, // Nehalem from http://www.agner.org/
385 { ISD::FDIV, MVT::v2f64, 22 }, // Nehalem from http://www.agner.org/
386 };
387
388 if (ST->hasSSE42()) {
389 if (const auto *Entry = CostTableLookup(SSE42FloatCostTable, ISD,
390 LT.second))
391 return LT.first * Entry->Cost;
392 }
393
Craig Topper4b275762015-10-28 04:02:12 +0000394 static const CostTblEntry
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000395 SSE2UniformCostTable[] = {
396 // Uniform splats are cheaper for the following instructions.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000397 { ISD::SHL, MVT::v16i8, 1 }, // psllw.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000398 { ISD::SHL, MVT::v32i8, 2 }, // psllw.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000399 { ISD::SHL, MVT::v8i16, 1 }, // psllw.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000400 { ISD::SHL, MVT::v16i16, 2 }, // psllw.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000401 { ISD::SHL, MVT::v4i32, 1 }, // pslld
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000402 { ISD::SHL, MVT::v8i32, 2 }, // pslld
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000403 { ISD::SHL, MVT::v2i64, 1 }, // psllq.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000404 { ISD::SHL, MVT::v4i64, 2 }, // psllq.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000405
406 { ISD::SRL, MVT::v16i8, 1 }, // psrlw.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000407 { ISD::SRL, MVT::v32i8, 2 }, // psrlw.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000408 { ISD::SRL, MVT::v8i16, 1 }, // psrlw.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000409 { ISD::SRL, MVT::v16i16, 2 }, // psrlw.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000410 { ISD::SRL, MVT::v4i32, 1 }, // psrld.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000411 { ISD::SRL, MVT::v8i32, 2 }, // psrld.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000412 { ISD::SRL, MVT::v2i64, 1 }, // psrlq.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000413 { ISD::SRL, MVT::v4i64, 2 }, // psrlq.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000414
415 { ISD::SRA, MVT::v16i8, 4 }, // psrlw, pand, pxor, psubb.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000416 { ISD::SRA, MVT::v32i8, 8 }, // psrlw, pand, pxor, psubb.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000417 { ISD::SRA, MVT::v8i16, 1 }, // psraw.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000418 { ISD::SRA, MVT::v16i16, 2 }, // psraw.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000419 { ISD::SRA, MVT::v4i32, 1 }, // psrad.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000420 { ISD::SRA, MVT::v8i32, 2 }, // psrad.
Simon Pilgrim8fbf1c12015-07-06 22:35:19 +0000421 { ISD::SRA, MVT::v2i64, 4 }, // 2 x psrad + shuffle.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000422 { ISD::SRA, MVT::v4i64, 8 }, // 2 x psrad + shuffle.
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000423 };
424
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000425 if (ST->hasSSE2() &&
426 ((Op2Info == TargetTransformInfo::OK_UniformConstantValue) ||
427 (Op2Info == TargetTransformInfo::OK_UniformValue))) {
Michael Kuperstein3ceac2b2016-08-04 22:48:03 +0000428 if (const auto *Entry =
429 CostTableLookup(SSE2UniformCostTable, ISD, LT.second))
Craig Topperee0c8592015-10-27 04:14:24 +0000430 return LT.first * Entry->Cost;
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000431 }
432
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000433 if (ISD == ISD::SHL &&
434 Op2Info == TargetTransformInfo::OK_NonUniformConstantValue) {
Craig Toppereda02a92015-10-25 03:15:29 +0000435 MVT VT = LT.second;
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000436 // Vector shift left by non uniform constant can be lowered
437 // into vector multiply (pmullw/pmulld).
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000438 if ((VT == MVT::v8i16 && ST->hasSSE2()) ||
439 (VT == MVT::v4i32 && ST->hasSSE41()))
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000440 return LT.first;
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000441
442 // v16i16 and v8i32 shifts by non-uniform constants are lowered into a
443 // sequence of extract + two vector multiply + insert.
444 if ((VT == MVT::v8i32 || VT == MVT::v16i16) &&
445 (ST->hasAVX() && !ST->hasAVX2()))
446 ISD = ISD::MUL;
447
448 // A vector shift left by non uniform constant is converted
449 // into a vector multiply; the new multiply is eventually
450 // lowered into a sequence of shuffles and 2 x pmuludq.
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000451 if (VT == MVT::v4i32 && ST->hasSSE2())
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000452 ISD = ISD::MUL;
453 }
Arnold Schwaighofer44f902e2013-04-04 23:26:24 +0000454
Simon Pilgrim6ac1e982016-10-23 16:49:04 +0000455 static const CostTblEntry SSE41CostTable[] = {
456 { ISD::SHL, MVT::v16i8, 11 }, // pblendvb sequence.
457 { ISD::SHL, MVT::v32i8, 2*11 }, // pblendvb sequence.
458 { ISD::SHL, MVT::v8i16, 14 }, // pblendvb sequence.
459 { ISD::SHL, MVT::v16i16, 2*14 }, // pblendvb sequence.
460
461 { ISD::SRL, MVT::v16i8, 12 }, // pblendvb sequence.
462 { ISD::SRL, MVT::v32i8, 2*12 }, // pblendvb sequence.
463 { ISD::SRL, MVT::v8i16, 14 }, // pblendvb sequence.
464 { ISD::SRL, MVT::v16i16, 2*14 }, // pblendvb sequence.
465 { ISD::SRL, MVT::v4i32, 11 }, // Shift each lane + blend.
466 { ISD::SRL, MVT::v8i32, 2*11 }, // Shift each lane + blend.
467
468 { ISD::SRA, MVT::v16i8, 24 }, // pblendvb sequence.
469 { ISD::SRA, MVT::v32i8, 2*24 }, // pblendvb sequence.
470 { ISD::SRA, MVT::v8i16, 14 }, // pblendvb sequence.
471 { ISD::SRA, MVT::v16i16, 2*14 }, // pblendvb sequence.
472 { ISD::SRA, MVT::v4i32, 12 }, // Shift each lane + blend.
473 { ISD::SRA, MVT::v8i32, 2*12 }, // Shift each lane + blend.
474 };
475
476 if (ST->hasSSE41()) {
477 if (const auto *Entry = CostTableLookup(SSE41CostTable, ISD, LT.second))
478 return LT.first * Entry->Cost;
479 }
480
Craig Topper4b275762015-10-28 04:02:12 +0000481 static const CostTblEntry SSE2CostTable[] = {
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000482 // We don't correctly identify costs of casts because they are marked as
483 // custom.
Simon Pilgrim59656802015-06-11 07:46:37 +0000484 { ISD::SHL, MVT::v16i8, 26 }, // cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000485 { ISD::SHL, MVT::v32i8, 2*26 }, // cmpgtb sequence.
Simon Pilgrim59656802015-06-11 07:46:37 +0000486 { ISD::SHL, MVT::v8i16, 32 }, // cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000487 { ISD::SHL, MVT::v16i16, 2*32 }, // cmpgtb sequence.
Simon Pilgrim59656802015-06-11 07:46:37 +0000488 { ISD::SHL, MVT::v4i32, 2*5 }, // We optimized this using mul.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000489 { ISD::SHL, MVT::v8i32, 2*2*5 }, // We optimized this using mul.
Simon Pilgrim59764dc2015-07-18 20:06:30 +0000490 { ISD::SHL, MVT::v2i64, 4 }, // splat+shuffle sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000491 { ISD::SHL, MVT::v4i64, 2*4 }, // splat+shuffle sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000492
493 { ISD::SRL, MVT::v16i8, 26 }, // cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000494 { ISD::SRL, MVT::v32i8, 2*26 }, // cmpgtb sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000495 { ISD::SRL, MVT::v8i16, 32 }, // cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000496 { ISD::SRL, MVT::v16i16, 2*32 }, // cmpgtb sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000497 { ISD::SRL, MVT::v4i32, 16 }, // Shift each lane + blend.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000498 { ISD::SRL, MVT::v8i32, 2*16 }, // Shift each lane + blend.
Simon Pilgrim59764dc2015-07-18 20:06:30 +0000499 { ISD::SRL, MVT::v2i64, 4 }, // splat+shuffle sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000500 { ISD::SRL, MVT::v4i64, 2*4 }, // splat+shuffle sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000501
502 { ISD::SRA, MVT::v16i8, 54 }, // unpacked cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000503 { ISD::SRA, MVT::v32i8, 2*54 }, // unpacked cmpgtb sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000504 { ISD::SRA, MVT::v8i16, 32 }, // cmpgtb sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000505 { ISD::SRA, MVT::v16i16, 2*32 }, // cmpgtb sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000506 { ISD::SRA, MVT::v4i32, 16 }, // Shift each lane + blend.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000507 { ISD::SRA, MVT::v8i32, 2*16 }, // Shift each lane + blend.
Simon Pilgrim86478c62015-07-29 20:31:45 +0000508 { ISD::SRA, MVT::v2i64, 12 }, // srl/xor/sub sequence.
Simon Pilgrima18ae9b2015-10-17 13:23:38 +0000509 { ISD::SRA, MVT::v4i64, 2*12 }, // srl/xor/sub sequence.
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000510
Simon Pilgrim779da8e2016-11-14 15:54:24 +0000511 { ISD::MUL, MVT::v16i8, 12 }, // extend/pmullw/trunc sequence.
512
Alexey Bataevd07c7312016-10-31 12:10:53 +0000513 { ISD::FDIV, MVT::f32, 23 }, // Pentium IV from http://www.agner.org/
514 { ISD::FDIV, MVT::v4f32, 39 }, // Pentium IV from http://www.agner.org/
515 { ISD::FDIV, MVT::f64, 38 }, // Pentium IV from http://www.agner.org/
516 { ISD::FDIV, MVT::v2f64, 69 }, // Pentium IV from http://www.agner.org/
517
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +0000518 // It is not a good idea to vectorize division. We have to scalarize it and
Arnold Schwaighofera04b9ef2013-06-25 19:14:09 +0000519 // in the process we will often end up having to spilling regular
520 // registers. The overhead of division is going to dominate most kernels
521 // anyways so try hard to prevent vectorization of division - it is
522 // generally a bad idea. Assume somewhat arbitrarily that we have to be able
523 // to hide "20 cycles" for each lane.
524 { ISD::SDIV, MVT::v16i8, 16*20 },
525 { ISD::SDIV, MVT::v8i16, 8*20 },
526 { ISD::SDIV, MVT::v4i32, 4*20 },
527 { ISD::SDIV, MVT::v2i64, 2*20 },
528 { ISD::UDIV, MVT::v16i8, 16*20 },
529 { ISD::UDIV, MVT::v8i16, 8*20 },
530 { ISD::UDIV, MVT::v4i32, 4*20 },
531 { ISD::UDIV, MVT::v2i64, 2*20 },
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000532 };
533
534 if (ST->hasSSE2()) {
Craig Topperee0c8592015-10-27 04:14:24 +0000535 if (const auto *Entry = CostTableLookup(SSE2CostTable, ISD, LT.second))
536 return LT.first * Entry->Cost;
Arnold Schwaighofere9b50162013-04-03 21:46:05 +0000537 }
538
Craig Topper4b275762015-10-28 04:02:12 +0000539 static const CostTblEntry AVX1CostTable[] = {
Renato Goline1fb0592013-01-20 20:57:20 +0000540 // We don't have to scalarize unsupported ops. We can issue two half-sized
541 // operations and we only need to extract the upper YMM half.
542 // Two ops + 1 extract + 1 insert = 4.
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000543 { ISD::MUL, MVT::v16i16, 4 },
Renato Goline1fb0592013-01-20 20:57:20 +0000544 { ISD::MUL, MVT::v8i32, 4 },
Simon Pilgrim27fed8e2016-11-14 14:45:16 +0000545 { ISD::SUB, MVT::v32i8, 4 },
546 { ISD::ADD, MVT::v32i8, 4 },
547 { ISD::SUB, MVT::v16i16, 4 },
548 { ISD::ADD, MVT::v16i16, 4 },
Renato Goline1fb0592013-01-20 20:57:20 +0000549 { ISD::SUB, MVT::v8i32, 4 },
550 { ISD::ADD, MVT::v8i32, 4 },
Renato Goline1fb0592013-01-20 20:57:20 +0000551 { ISD::SUB, MVT::v4i64, 4 },
552 { ISD::ADD, MVT::v4i64, 4 },
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000553 // A v4i64 multiply is custom lowered as two split v2i64 vectors that then
Simon Pilgrim081abbb2016-12-21 20:00:10 +0000554 // are lowered as a series of long multiplies(3), shifts(3) and adds(2)
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000555 // Because we believe v4i64 to be a legal type, we must also include the
Simon Pilgrim081abbb2016-12-21 20:00:10 +0000556 // split factor of two in the cost table. Therefore, the cost here is 16
557 // instead of 8.
558 { ISD::MUL, MVT::v4i64, 16 },
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000559 };
Chandler Carruth664e3542013-01-07 01:37:14 +0000560
561 // Look for AVX1 lowering tricks.
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000562 if (ST->hasAVX() && !ST->hasAVX2()) {
Craig Toppereda02a92015-10-25 03:15:29 +0000563 MVT VT = LT.second;
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000564
Craig Topperee0c8592015-10-27 04:14:24 +0000565 if (const auto *Entry = CostTableLookup(AVX1CostTable, ISD, VT))
566 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +0000567 }
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000568
569 // Custom lowering of vectors.
Craig Topper4b275762015-10-28 04:02:12 +0000570 static const CostTblEntry CustomLowered[] = {
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000571 // A v2i64/v4i64 and multiply is custom lowered as a series of long
Simon Pilgrim081abbb2016-12-21 20:00:10 +0000572 // multiplies(3), shifts(3) and adds(2).
573 { ISD::MUL, MVT::v2i64, 8 },
574 { ISD::MUL, MVT::v4i64, 8 },
575 { ISD::MUL, MVT::v8i64, 8 }
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000576 };
Craig Topperee0c8592015-10-27 04:14:24 +0000577 if (const auto *Entry = CostTableLookup(CustomLowered, ISD, LT.second))
578 return LT.first * Entry->Cost;
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000579
580 // Special lowering of v4i32 mul on sse2, sse3: Lower v4i32 mul as 2x shuffle,
581 // 2x pmuludq, 2x shuffle.
582 if (ISD == ISD::MUL && LT.second == MVT::v4i32 && ST->hasSSE2() &&
583 !ST->hasSSE41())
Andrea Di Biagiob7882b32014-02-12 23:43:47 +0000584 return LT.first * 6;
Arnold Schwaighofer20ef54f2013-03-02 04:02:52 +0000585
Alexey Bataevd07c7312016-10-31 12:10:53 +0000586 static const CostTblEntry SSE1FloatCostTable[] = {
587 { ISD::FDIV, MVT::f32, 17 }, // Pentium III from http://www.agner.org/
588 { ISD::FDIV, MVT::v4f32, 34 }, // Pentium III from http://www.agner.org/
589 };
590
591 if (ST->hasSSE1())
592 if (const auto *Entry = CostTableLookup(SSE1FloatCostTable, ISD,
593 LT.second))
594 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +0000595 // Fallback to the default implementation.
Chandler Carruth705b1852015-01-31 03:43:40 +0000596 return BaseT::getArithmeticInstrCost(Opcode, Ty, Op1Info, Op2Info);
Chandler Carruth664e3542013-01-07 01:37:14 +0000597}
598
Chandler Carruth93205eb2015-08-05 18:08:10 +0000599int X86TTIImpl::getShuffleCost(TTI::ShuffleKind Kind, Type *Tp, int Index,
600 Type *SubTp) {
Karthik Bhate03a25d2014-06-20 04:32:48 +0000601 // We only estimate the cost of reverse and alternate shuffles.
Chandler Carruth705b1852015-01-31 03:43:40 +0000602 if (Kind != TTI::SK_Reverse && Kind != TTI::SK_Alternate)
603 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
Chandler Carruth664e3542013-01-07 01:37:14 +0000604
Chandler Carruth705b1852015-01-31 03:43:40 +0000605 if (Kind == TTI::SK_Reverse) {
Chandler Carruth93205eb2015-08-05 18:08:10 +0000606 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Tp);
Chandler Carruth664e3542013-01-07 01:37:14 +0000607
Simon Pilgrim2f7f0e72016-12-15 14:24:07 +0000608 static const CostTblEntry AVX512VBMIShuffleTbl[] = {
609 { ISD::VECTOR_SHUFFLE, MVT::v64i8, 1 }, // vpermb
610 { ISD::VECTOR_SHUFFLE, MVT::v32i8, 1 } // vpermb
611 };
612
613 if (ST->hasVBMI())
614 if (const auto *Entry = CostTableLookup(AVX512VBMIShuffleTbl,
615 ISD::VECTOR_SHUFFLE, LT.second))
616 return LT.first * Entry->Cost;
617
618 static const CostTblEntry AVX512BWShuffleTbl[] = {
619 { ISD::VECTOR_SHUFFLE, MVT::v32i16, 1 }, // vpermw
620 { ISD::VECTOR_SHUFFLE, MVT::v64i8, 6 } // vextracti64x4 + 2*vperm2i128
621 // + 2*pshufb + vinserti64x4
622 };
623
624 if (ST->hasBWI())
625 if (const auto *Entry = CostTableLookup(AVX512BWShuffleTbl,
626 ISD::VECTOR_SHUFFLE, LT.second))
627 return LT.first * Entry->Cost;
628
629 static const CostTblEntry AVX512ShuffleTbl[] = {
630 { ISD::VECTOR_SHUFFLE, MVT::v8f64, 1 }, // vpermpd
631 { ISD::VECTOR_SHUFFLE, MVT::v16f32, 1 }, // vpermps
632 { ISD::VECTOR_SHUFFLE, MVT::v8i64, 1 }, // vpermq
633 { ISD::VECTOR_SHUFFLE, MVT::v16i32, 1 }, // vpermd
634 };
635
636 if (ST->hasAVX512())
637 if (const auto *Entry =
638 CostTableLookup(AVX512ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
639 return LT.first * Entry->Cost;
640
641 static const CostTblEntry AVX2ShuffleTbl[] = {
642 { ISD::VECTOR_SHUFFLE, MVT::v4f64, 1 }, // vpermpd
643 { ISD::VECTOR_SHUFFLE, MVT::v8f32, 1 }, // vpermps
644 { ISD::VECTOR_SHUFFLE, MVT::v4i64, 1 }, // vpermq
645 { ISD::VECTOR_SHUFFLE, MVT::v8i32, 1 }, // vpermd
646 { ISD::VECTOR_SHUFFLE, MVT::v16i16, 2 }, // vperm2i128 + pshufb
647 { ISD::VECTOR_SHUFFLE, MVT::v32i8, 2 } // vperm2i128 + pshufb
648 };
649
650 if (ST->hasAVX2())
651 if (const auto *Entry =
652 CostTableLookup(AVX2ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
653 return LT.first * Entry->Cost;
654
655 static const CostTblEntry AVX1ShuffleTbl[] = {
656 { ISD::VECTOR_SHUFFLE, MVT::v4f64, 2 }, // vperm2f128 + vpermilpd
657 { ISD::VECTOR_SHUFFLE, MVT::v8f32, 2 }, // vperm2f128 + vpermilps
658 { ISD::VECTOR_SHUFFLE, MVT::v4i64, 2 }, // vperm2f128 + vpermilpd
659 { ISD::VECTOR_SHUFFLE, MVT::v8i32, 2 }, // vperm2f128 + vpermilps
660 { ISD::VECTOR_SHUFFLE, MVT::v16i16, 4 }, // vextractf128 + 2*pshufb
661 // + vinsertf128
662 { ISD::VECTOR_SHUFFLE, MVT::v32i8, 4 } // vextractf128 + 2*pshufb
663 // + vinsertf128
664 };
665
666 if (ST->hasAVX())
667 if (const auto *Entry =
668 CostTableLookup(AVX1ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
669 return LT.first * Entry->Cost;
670
671 static const CostTblEntry SSSE3ShuffleTbl[] = {
672 { ISD::VECTOR_SHUFFLE, MVT::v8i16, 1 }, // pshufb
673 { ISD::VECTOR_SHUFFLE, MVT::v16i8, 1 } // pshufb
674 };
675
676 if (ST->hasSSSE3())
677 if (const auto *Entry =
678 CostTableLookup(SSSE3ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
679 return LT.first * Entry->Cost;
680
681 static const CostTblEntry SSE2ShuffleTbl[] = {
682 { ISD::VECTOR_SHUFFLE, MVT::v2f64, 1 }, // shufpd
683 { ISD::VECTOR_SHUFFLE, MVT::v2i64, 1 }, // pshufd
684 { ISD::VECTOR_SHUFFLE, MVT::v4i32, 1 }, // pshufd
685 { ISD::VECTOR_SHUFFLE, MVT::v8i16, 3 }, // pshuflw + pshufhw + pshufd
686 { ISD::VECTOR_SHUFFLE, MVT::v16i8, 9 } // 2*pshuflw + 2*pshufhw
687 // + 2*pshufd + 2*unpck + packus
688 };
689
690 if (ST->hasSSE2())
691 if (const auto *Entry =
692 CostTableLookup(SSE2ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
693 return LT.first * Entry->Cost;
694
695 static const CostTblEntry SSE1ShuffleTbl[] = {
696 { ISD::VECTOR_SHUFFLE, MVT::v4f32, 1 }, // shufps
697 };
698
699 if (ST->hasSSE1())
700 if (const auto *Entry =
701 CostTableLookup(SSE1ShuffleTbl, ISD::VECTOR_SHUFFLE, LT.second))
702 return LT.first * Entry->Cost;
Karthik Bhate03a25d2014-06-20 04:32:48 +0000703 }
704
Chandler Carruth705b1852015-01-31 03:43:40 +0000705 if (Kind == TTI::SK_Alternate) {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000706 // 64-bit packed float vectors (v2f32) are widened to type v4f32.
707 // 64-bit packed integer vectors (v2i32) are promoted to type v2i64.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000708 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Tp);
Karthik Bhate03a25d2014-06-20 04:32:48 +0000709
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000710 // The backend knows how to generate a single VEX.256 version of
711 // instruction VPBLENDW if the target supports AVX2.
712 if (ST->hasAVX2() && LT.second == MVT::v16i16)
713 return LT.first;
714
Craig Topper4b275762015-10-28 04:02:12 +0000715 static const CostTblEntry AVXAltShuffleTbl[] = {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000716 {ISD::VECTOR_SHUFFLE, MVT::v4i64, 1}, // vblendpd
717 {ISD::VECTOR_SHUFFLE, MVT::v4f64, 1}, // vblendpd
718
719 {ISD::VECTOR_SHUFFLE, MVT::v8i32, 1}, // vblendps
720 {ISD::VECTOR_SHUFFLE, MVT::v8f32, 1}, // vblendps
721
722 // This shuffle is custom lowered into a sequence of:
723 // 2x vextractf128 , 2x vpblendw , 1x vinsertf128
724 {ISD::VECTOR_SHUFFLE, MVT::v16i16, 5},
725
726 // This shuffle is custom lowered into a long sequence of:
727 // 2x vextractf128 , 4x vpshufb , 2x vpor , 1x vinsertf128
728 {ISD::VECTOR_SHUFFLE, MVT::v32i8, 9}
729 };
730
Craig Topperee0c8592015-10-27 04:14:24 +0000731 if (ST->hasAVX())
732 if (const auto *Entry = CostTableLookup(AVXAltShuffleTbl,
733 ISD::VECTOR_SHUFFLE, LT.second))
734 return LT.first * Entry->Cost;
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000735
Craig Topper4b275762015-10-28 04:02:12 +0000736 static const CostTblEntry SSE41AltShuffleTbl[] = {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000737 // These are lowered into movsd.
738 {ISD::VECTOR_SHUFFLE, MVT::v2i64, 1},
739 {ISD::VECTOR_SHUFFLE, MVT::v2f64, 1},
740
741 // packed float vectors with four elements are lowered into BLENDI dag
742 // nodes. A v4i32/v4f32 BLENDI generates a single 'blendps'/'blendpd'.
743 {ISD::VECTOR_SHUFFLE, MVT::v4i32, 1},
744 {ISD::VECTOR_SHUFFLE, MVT::v4f32, 1},
745
746 // This shuffle generates a single pshufw.
747 {ISD::VECTOR_SHUFFLE, MVT::v8i16, 1},
748
749 // There is no instruction that matches a v16i8 alternate shuffle.
750 // The backend will expand it into the sequence 'pshufb + pshufb + or'.
751 {ISD::VECTOR_SHUFFLE, MVT::v16i8, 3}
752 };
753
Craig Topperee0c8592015-10-27 04:14:24 +0000754 if (ST->hasSSE41())
755 if (const auto *Entry = CostTableLookup(SSE41AltShuffleTbl, ISD::VECTOR_SHUFFLE,
756 LT.second))
757 return LT.first * Entry->Cost;
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000758
Craig Topper4b275762015-10-28 04:02:12 +0000759 static const CostTblEntry SSSE3AltShuffleTbl[] = {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000760 {ISD::VECTOR_SHUFFLE, MVT::v2i64, 1}, // movsd
761 {ISD::VECTOR_SHUFFLE, MVT::v2f64, 1}, // movsd
762
763 // SSE3 doesn't have 'blendps'. The following shuffles are expanded into
764 // the sequence 'shufps + pshufd'
765 {ISD::VECTOR_SHUFFLE, MVT::v4i32, 2},
766 {ISD::VECTOR_SHUFFLE, MVT::v4f32, 2},
767
768 {ISD::VECTOR_SHUFFLE, MVT::v8i16, 3}, // pshufb + pshufb + or
769 {ISD::VECTOR_SHUFFLE, MVT::v16i8, 3} // pshufb + pshufb + or
770 };
Michael Liao5bf95782014-12-04 05:20:33 +0000771
Craig Topperee0c8592015-10-27 04:14:24 +0000772 if (ST->hasSSSE3())
773 if (const auto *Entry = CostTableLookup(SSSE3AltShuffleTbl,
774 ISD::VECTOR_SHUFFLE, LT.second))
775 return LT.first * Entry->Cost;
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000776
Craig Topper4b275762015-10-28 04:02:12 +0000777 static const CostTblEntry SSEAltShuffleTbl[] = {
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000778 {ISD::VECTOR_SHUFFLE, MVT::v2i64, 1}, // movsd
779 {ISD::VECTOR_SHUFFLE, MVT::v2f64, 1}, // movsd
780
781 {ISD::VECTOR_SHUFFLE, MVT::v4i32, 2}, // shufps + pshufd
782 {ISD::VECTOR_SHUFFLE, MVT::v4f32, 2}, // shufps + pshufd
Michael Liao5bf95782014-12-04 05:20:33 +0000783
Andrea Di Biagioc8e8bda2014-07-03 22:24:18 +0000784 // This is expanded into a long sequence of four extract + four insert.
785 {ISD::VECTOR_SHUFFLE, MVT::v8i16, 8}, // 4 x pextrw + 4 pinsrw.
786
787 // 8 x (pinsrw + pextrw + and + movb + movzb + or)
788 {ISD::VECTOR_SHUFFLE, MVT::v16i8, 48}
789 };
790
Michael Liao5bf95782014-12-04 05:20:33 +0000791 // Fall-back (SSE3 and SSE2).
Craig Topperee0c8592015-10-27 04:14:24 +0000792 if (const auto *Entry = CostTableLookup(SSEAltShuffleTbl,
793 ISD::VECTOR_SHUFFLE, LT.second))
794 return LT.first * Entry->Cost;
Chandler Carruth705b1852015-01-31 03:43:40 +0000795 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
Karthik Bhate03a25d2014-06-20 04:32:48 +0000796 }
797
Chandler Carruth705b1852015-01-31 03:43:40 +0000798 return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
Chandler Carruth664e3542013-01-07 01:37:14 +0000799}
800
Chandler Carruth93205eb2015-08-05 18:08:10 +0000801int X86TTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src) {
Chandler Carruth664e3542013-01-07 01:37:14 +0000802 int ISD = TLI->InstructionOpcodeToISD(Opcode);
803 assert(ISD && "Invalid opcode");
804
Cong Hou59898d82015-12-11 00:31:39 +0000805 // FIXME: Need a better design of the cost table to handle non-simple types of
806 // potential massive combinations (elem_num x src_type x dst_type).
807
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000808 static const TypeConversionCostTblEntry AVX512DQConversionTbl[] = {
Simon Pilgrim841d7ca2016-11-24 14:46:55 +0000809 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
810 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +0000811 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
812 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +0000813 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
814 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
815
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000816 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000817 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000818 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000819 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +0000820 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000821 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000822
Simon Pilgrim841d7ca2016-11-24 14:46:55 +0000823 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f32, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +0000824 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f32, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +0000825 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f32, 1 },
Simon Pilgrim841d7ca2016-11-24 14:46:55 +0000826 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f64, 1 },
Simon Pilgrim4e9b9cb2016-11-23 14:01:18 +0000827 { ISD::FP_TO_SINT, MVT::v4i64, MVT::v4f64, 1 },
Simon Pilgrim03cd8f82016-11-23 13:42:09 +0000828 { ISD::FP_TO_SINT, MVT::v8i64, MVT::v8f64, 1 },
829
830 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f32, 1 },
831 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f32, 1 },
832 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f32, 1 },
833 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f64, 1 },
834 { ISD::FP_TO_UINT, MVT::v4i64, MVT::v4f64, 1 },
835 { ISD::FP_TO_UINT, MVT::v8i64, MVT::v8f64, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000836 };
837
Michael Kupersteinf0c59332016-07-11 21:39:44 +0000838 // TODO: For AVX512DQ + AVX512VL, we also have cheap casts for 128-bit and
839 // 256-bit wide vectors.
840
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000841 static const TypeConversionCostTblEntry AVX512FConversionTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +0000842 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 1 },
843 { ISD::FP_EXTEND, MVT::v8f64, MVT::v16f32, 3 },
844 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000845
846 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 1 },
847 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 1 },
848 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i64, 1 },
849 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000850
851 // v16i1 -> v16i32 - load + broadcast
852 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
853 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i1, 2 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000854 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
855 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 1 },
856 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
857 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 1 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000858 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
859 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i16, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000860 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
861 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i32, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000862
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000863 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +0000864 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000865 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +0000866 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000867 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +0000868 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
869 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Elena Demikhovskyd5e95b52014-11-13 11:46:16 +0000870 { ISD::SINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +0000871 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i64, 26 },
872 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000873
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000874 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000875 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i1, 3 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000876 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000877 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
878 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 2 },
879 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i8, 2 },
880 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i8, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000881 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i16, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000882 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
883 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 2 },
884 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i16, 2 },
885 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i16, 2 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000886 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i32, 2 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +0000887 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000888 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
889 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
890 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
891 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i32, 1 },
892 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i32, 1 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +0000893 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 5 },
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +0000894 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 5 },
895 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 12 },
896 { ISD::UINT_TO_FP, MVT::v8f64, MVT::v8i64, 26 },
897
898 { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f32, 1 },
899 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f32, 1 },
900 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 1 },
901 { ISD::FP_TO_UINT, MVT::v16i32, MVT::v16f32, 1 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000902 };
903
Craig Topper4b275762015-10-28 04:02:12 +0000904 static const TypeConversionCostTblEntry AVX2ConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +0000905 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
906 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000907 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
908 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 3 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000909 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
910 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000911 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
912 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
913 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
914 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000915 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
916 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000917 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
918 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 1 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000919 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
920 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 1 },
921
922 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 2 },
923 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 2 },
924 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 2 },
925 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 2 },
926 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 2 },
927 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 4 },
Elena Demikhovsky27012472014-09-16 07:57:37 +0000928
929 { ISD::FP_EXTEND, MVT::v8f64, MVT::v8f32, 3 },
930 { ISD::FP_ROUND, MVT::v8f32, MVT::v8f64, 3 },
Quentin Colombet360460b2014-11-11 02:23:47 +0000931
932 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 8 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000933 };
934
Craig Topper4b275762015-10-28 04:02:12 +0000935 static const TypeConversionCostTblEntry AVXConversionTbl[] = {
Tim Northoverf0e21612014-02-06 18:18:36 +0000936 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i1, 6 },
937 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i1, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000938 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i1, 7 },
939 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i1, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000940 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 6 },
941 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000942 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 7 },
943 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 4 },
944 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
945 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000946 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 6 },
947 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000948 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
949 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000950 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
951 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 4 },
952
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000953 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 4 },
954 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
955 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000956 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i64, 4 },
957 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i64, 4 },
958 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 4 },
Tim Northoverf0e21612014-02-06 18:18:36 +0000959 { ISD::TRUNCATE, MVT::v8i32, MVT::v8i64, 9 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000960
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000961 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i1, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000962 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i1, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000963 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i1, 8 },
964 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i8, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000965 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i8, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000966 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i8, 8 },
967 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i16, 3 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000968 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i16, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000969 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
970 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000971 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000972 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i32, 1 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000973
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000974 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i1, 7 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000975 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i1, 7 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000976 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i1, 6 },
977 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i8, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000978 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000979 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 5 },
980 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000981 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i16, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000982 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 5 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +0000983 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000984 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 6 },
Benjamin Kramer52ceb442013-04-01 10:23:49 +0000985 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i32, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000986 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i32, 9 },
Quentin Colombet85b904d2014-03-27 22:27:41 +0000987 // The generic code to compute the scalar overhead is currently broken.
988 // Workaround this limitation by estimating the scalarization overhead
989 // here. We have roughly 10 instructions per scalar element.
990 // Multiply that by the vector width.
991 // FIXME: remove that when PR19268 is fixed.
Michael Kupersteinf0c59332016-07-11 21:39:44 +0000992 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 10 },
993 { ISD::UINT_TO_FP, MVT::v4f64, MVT::v4i64, 20 },
994 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
995 { ISD::SINT_TO_FP, MVT::v4f64, MVT::v4i64, 13 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +0000996
Renato Goline1fb0592013-01-20 20:57:20 +0000997 { ISD::FP_TO_SINT, MVT::v4i8, MVT::v4f32, 1 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +0000998 { ISD::FP_TO_SINT, MVT::v8i8, MVT::v8f32, 7 },
Adam Nemet6dafe972014-03-30 18:07:13 +0000999 // This node is expanded into scalarized operations but BasicTTI is overly
1000 // optimistic estimating its cost. It computes 3 per element (one
1001 // vector-extract, one scalar conversion and one vector-insert). The
1002 // problem is that the inserts form a read-modify-write chain so latency
1003 // should be factored in too. Inflating the cost per element by 1.
1004 { ISD::FP_TO_UINT, MVT::v8i32, MVT::v8f32, 8*4 },
Adam Nemet10c4ce22014-03-31 21:54:48 +00001005 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f64, 4*4 },
Michael Kupersteinf0c59332016-07-11 21:39:44 +00001006
1007 { ISD::FP_EXTEND, MVT::v4f64, MVT::v4f32, 1 },
1008 { ISD::FP_ROUND, MVT::v4f32, MVT::v4f64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001009 };
1010
Cong Hou59898d82015-12-11 00:31:39 +00001011 static const TypeConversionCostTblEntry SSE41ConversionTbl[] = {
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001012 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
1013 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001014 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1015 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 2 },
1016 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
1017 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 2 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001018
Cong Hou59898d82015-12-11 00:31:39 +00001019 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1020 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 2 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001021 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1022 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 1 },
1023 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1024 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1025 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1026 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 2 },
1027 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1028 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 2 },
1029 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1030 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 4 },
1031 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1032 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1033 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1034 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 2 },
1035 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
1036 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 4 },
Cong Hou59898d82015-12-11 00:31:39 +00001037
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001038 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 2 },
1039 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 1 },
1040 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001041 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 1 },
Cong Hou59898d82015-12-11 00:31:39 +00001042 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 3 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001043 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 3 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001044 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 6 },
1045
Cong Hou59898d82015-12-11 00:31:39 +00001046 };
1047
1048 static const TypeConversionCostTblEntry SSE2ConversionTbl[] = {
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001049 // These are somewhat magic numbers justified by looking at the output of
1050 // Intel's IACA, running some kernels and making sure when we take
1051 // legalization into account the throughput will be overestimated.
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001052 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001053 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1054 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1055 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
Sanjay Patel04b34962016-07-06 19:15:54 +00001056 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 5 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001057 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1058 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
1059 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
Cong Hou59898d82015-12-11 00:31:39 +00001060
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001061 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v16i8, 16*10 },
1062 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v16i8, 8 },
1063 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v8i16, 15 },
1064 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v8i16, 8*10 },
1065 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v4i32, 4*10 },
1066 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 8 },
1067 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 2*10 },
1068 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v2i64, 15 },
Michael Kuperstein9a0542a2016-06-10 17:01:05 +00001069
Simon Pilgrim4ddc92b2016-10-18 07:42:15 +00001070 { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f64, 3 },
1071
Cong Hou59898d82015-12-11 00:31:39 +00001072 { ISD::ZERO_EXTEND, MVT::v4i16, MVT::v4i8, 1 },
1073 { ISD::SIGN_EXTEND, MVT::v4i16, MVT::v4i8, 6 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001074 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i8, 2 },
1075 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i8, 3 },
1076 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i8, 4 },
1077 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i8, 8 },
1078 { ISD::ZERO_EXTEND, MVT::v8i16, MVT::v8i8, 1 },
1079 { ISD::SIGN_EXTEND, MVT::v8i16, MVT::v8i8, 2 },
1080 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1081 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 6 },
1082 { ISD::ZERO_EXTEND, MVT::v16i16, MVT::v16i8, 3 },
1083 { ISD::SIGN_EXTEND, MVT::v16i16, MVT::v16i8, 4 },
1084 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 9 },
1085 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 12 },
1086 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 1 },
1087 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 2 },
1088 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
1089 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 10 },
1090 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i16, 3 },
1091 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i16, 4 },
1092 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i16, 6 },
Simon Pilgrim285d9e42016-07-17 19:02:27 +00001093 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i16, 8 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001094 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i32, 3 },
1095 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i32, 5 },
Cong Hou59898d82015-12-11 00:31:39 +00001096
Cong Hou59898d82015-12-11 00:31:39 +00001097 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i16, 4 },
Michael Kuperstein1b62e0e2016-07-06 18:26:48 +00001098 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i16, 2 },
1099 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i16, 3 },
1100 { ISD::TRUNCATE, MVT::v4i8, MVT::v4i32, 3 },
1101 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 3 },
1102 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 4 },
1103 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 7 },
1104 { ISD::TRUNCATE, MVT::v8i16, MVT::v8i32, 5 },
1105 { ISD::TRUNCATE, MVT::v16i16, MVT::v16i32, 10 },
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001106 };
1107
Chandler Carruth93205eb2015-08-05 18:08:10 +00001108 std::pair<int, MVT> LTSrc = TLI->getTypeLegalizationCost(DL, Src);
1109 std::pair<int, MVT> LTDest = TLI->getTypeLegalizationCost(DL, Dst);
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001110
1111 if (ST->hasSSE2() && !ST->hasAVX()) {
Cong Hou59898d82015-12-11 00:31:39 +00001112 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
Craig Topperee0c8592015-10-27 04:14:24 +00001113 LTDest.second, LTSrc.second))
1114 return LTSrc.first * Entry->Cost;
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001115 }
1116
Simon Pilgrime2c244f2015-07-19 15:36:12 +00001117 EVT SrcTy = TLI->getValueType(DL, Src);
1118 EVT DstTy = TLI->getValueType(DL, Dst);
1119
1120 // The function getSimpleVT only handles simple value types.
1121 if (!SrcTy.isSimple() || !DstTy.isSimple())
1122 return BaseT::getCastInstrCost(Opcode, Dst, Src);
1123
Elena Demikhovskya1a40cc2015-12-02 08:59:47 +00001124 if (ST->hasDQI())
1125 if (const auto *Entry = ConvertCostTableLookup(AVX512DQConversionTbl, ISD,
1126 DstTy.getSimpleVT(),
1127 SrcTy.getSimpleVT()))
1128 return Entry->Cost;
1129
1130 if (ST->hasAVX512())
1131 if (const auto *Entry = ConvertCostTableLookup(AVX512FConversionTbl, ISD,
1132 DstTy.getSimpleVT(),
1133 SrcTy.getSimpleVT()))
1134 return Entry->Cost;
1135
Tim Northoverf0e21612014-02-06 18:18:36 +00001136 if (ST->hasAVX2()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001137 if (const auto *Entry = ConvertCostTableLookup(AVX2ConversionTbl, ISD,
1138 DstTy.getSimpleVT(),
1139 SrcTy.getSimpleVT()))
1140 return Entry->Cost;
Tim Northoverf0e21612014-02-06 18:18:36 +00001141 }
1142
Chandler Carruth664e3542013-01-07 01:37:14 +00001143 if (ST->hasAVX()) {
Craig Topperee0c8592015-10-27 04:14:24 +00001144 if (const auto *Entry = ConvertCostTableLookup(AVXConversionTbl, ISD,
1145 DstTy.getSimpleVT(),
1146 SrcTy.getSimpleVT()))
1147 return Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001148 }
1149
Cong Hou59898d82015-12-11 00:31:39 +00001150 if (ST->hasSSE41()) {
1151 if (const auto *Entry = ConvertCostTableLookup(SSE41ConversionTbl, ISD,
1152 DstTy.getSimpleVT(),
1153 SrcTy.getSimpleVT()))
1154 return Entry->Cost;
1155 }
1156
1157 if (ST->hasSSE2()) {
1158 if (const auto *Entry = ConvertCostTableLookup(SSE2ConversionTbl, ISD,
1159 DstTy.getSimpleVT(),
1160 SrcTy.getSimpleVT()))
1161 return Entry->Cost;
1162 }
1163
Chandler Carruth705b1852015-01-31 03:43:40 +00001164 return BaseT::getCastInstrCost(Opcode, Dst, Src);
Chandler Carruth664e3542013-01-07 01:37:14 +00001165}
1166
Chandler Carruth93205eb2015-08-05 18:08:10 +00001167int X86TTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy, Type *CondTy) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001168 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001169 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Chandler Carruth664e3542013-01-07 01:37:14 +00001170
1171 MVT MTy = LT.second;
1172
1173 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1174 assert(ISD && "Invalid opcode");
1175
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001176 static const CostTblEntry SSE2CostTbl[] = {
1177 { ISD::SETCC, MVT::v2i64, 8 },
1178 { ISD::SETCC, MVT::v4i32, 1 },
1179 { ISD::SETCC, MVT::v8i16, 1 },
1180 { ISD::SETCC, MVT::v16i8, 1 },
1181 };
1182
Craig Topper4b275762015-10-28 04:02:12 +00001183 static const CostTblEntry SSE42CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001184 { ISD::SETCC, MVT::v2f64, 1 },
1185 { ISD::SETCC, MVT::v4f32, 1 },
1186 { ISD::SETCC, MVT::v2i64, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001187 };
1188
Craig Topper4b275762015-10-28 04:02:12 +00001189 static const CostTblEntry AVX1CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001190 { ISD::SETCC, MVT::v4f64, 1 },
1191 { ISD::SETCC, MVT::v8f32, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001192 // AVX1 does not support 8-wide integer compare.
Renato Goline1fb0592013-01-20 20:57:20 +00001193 { ISD::SETCC, MVT::v4i64, 4 },
1194 { ISD::SETCC, MVT::v8i32, 4 },
1195 { ISD::SETCC, MVT::v16i16, 4 },
1196 { ISD::SETCC, MVT::v32i8, 4 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001197 };
1198
Craig Topper4b275762015-10-28 04:02:12 +00001199 static const CostTblEntry AVX2CostTbl[] = {
Renato Goline1fb0592013-01-20 20:57:20 +00001200 { ISD::SETCC, MVT::v4i64, 1 },
1201 { ISD::SETCC, MVT::v8i32, 1 },
1202 { ISD::SETCC, MVT::v16i16, 1 },
1203 { ISD::SETCC, MVT::v32i8, 1 },
Chandler Carruth664e3542013-01-07 01:37:14 +00001204 };
1205
Craig Topper4b275762015-10-28 04:02:12 +00001206 static const CostTblEntry AVX512CostTbl[] = {
Elena Demikhovsky27012472014-09-16 07:57:37 +00001207 { ISD::SETCC, MVT::v8i64, 1 },
1208 { ISD::SETCC, MVT::v16i32, 1 },
1209 { ISD::SETCC, MVT::v8f64, 1 },
1210 { ISD::SETCC, MVT::v16f32, 1 },
1211 };
1212
Craig Topperee0c8592015-10-27 04:14:24 +00001213 if (ST->hasAVX512())
1214 if (const auto *Entry = CostTableLookup(AVX512CostTbl, ISD, MTy))
1215 return LT.first * Entry->Cost;
Elena Demikhovsky27012472014-09-16 07:57:37 +00001216
Craig Topperee0c8592015-10-27 04:14:24 +00001217 if (ST->hasAVX2())
1218 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1219 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001220
Craig Topperee0c8592015-10-27 04:14:24 +00001221 if (ST->hasAVX())
1222 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1223 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001224
Craig Topperee0c8592015-10-27 04:14:24 +00001225 if (ST->hasSSE42())
1226 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1227 return LT.first * Entry->Cost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001228
Simon Pilgrimeec3a952016-05-09 21:14:38 +00001229 if (ST->hasSSE2())
1230 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1231 return LT.first * Entry->Cost;
1232
Chandler Carruth705b1852015-01-31 03:43:40 +00001233 return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy);
Chandler Carruth664e3542013-01-07 01:37:14 +00001234}
1235
Simon Pilgrim14000b32016-05-24 08:17:50 +00001236int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
1237 ArrayRef<Type *> Tys, FastMathFlags FMF) {
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001238 // Costs should match the codegen from:
1239 // BITREVERSE: llvm\test\CodeGen\X86\vector-bitreverse.ll
1240 // BSWAP: llvm\test\CodeGen\X86\bswap-vector.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001241 // CTLZ: llvm\test\CodeGen\X86\vector-lzcnt-*.ll
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001242 // CTPOP: llvm\test\CodeGen\X86\vector-popcnt-*.ll
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001243 // CTTZ: llvm\test\CodeGen\X86\vector-tzcnt-*.ll
Simon Pilgrim14000b32016-05-24 08:17:50 +00001244 static const CostTblEntry XOPCostTbl[] = {
1245 { ISD::BITREVERSE, MVT::v4i64, 4 },
1246 { ISD::BITREVERSE, MVT::v8i32, 4 },
1247 { ISD::BITREVERSE, MVT::v16i16, 4 },
1248 { ISD::BITREVERSE, MVT::v32i8, 4 },
1249 { ISD::BITREVERSE, MVT::v2i64, 1 },
1250 { ISD::BITREVERSE, MVT::v4i32, 1 },
1251 { ISD::BITREVERSE, MVT::v8i16, 1 },
1252 { ISD::BITREVERSE, MVT::v16i8, 1 },
1253 { ISD::BITREVERSE, MVT::i64, 3 },
1254 { ISD::BITREVERSE, MVT::i32, 3 },
1255 { ISD::BITREVERSE, MVT::i16, 3 },
1256 { ISD::BITREVERSE, MVT::i8, 3 }
1257 };
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001258 static const CostTblEntry AVX2CostTbl[] = {
1259 { ISD::BITREVERSE, MVT::v4i64, 5 },
1260 { ISD::BITREVERSE, MVT::v8i32, 5 },
1261 { ISD::BITREVERSE, MVT::v16i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001262 { ISD::BITREVERSE, MVT::v32i8, 5 },
1263 { ISD::BSWAP, MVT::v4i64, 1 },
1264 { ISD::BSWAP, MVT::v8i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001265 { ISD::BSWAP, MVT::v16i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001266 { ISD::CTLZ, MVT::v4i64, 23 },
1267 { ISD::CTLZ, MVT::v8i32, 18 },
1268 { ISD::CTLZ, MVT::v16i16, 14 },
1269 { ISD::CTLZ, MVT::v32i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001270 { ISD::CTPOP, MVT::v4i64, 7 },
1271 { ISD::CTPOP, MVT::v8i32, 11 },
1272 { ISD::CTPOP, MVT::v16i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001273 { ISD::CTPOP, MVT::v32i8, 6 },
1274 { ISD::CTTZ, MVT::v4i64, 10 },
1275 { ISD::CTTZ, MVT::v8i32, 14 },
1276 { ISD::CTTZ, MVT::v16i16, 12 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001277 { ISD::CTTZ, MVT::v32i8, 9 },
1278 { ISD::FSQRT, MVT::f32, 7 }, // Haswell from http://www.agner.org/
1279 { ISD::FSQRT, MVT::v4f32, 7 }, // Haswell from http://www.agner.org/
1280 { ISD::FSQRT, MVT::v8f32, 14 }, // Haswell from http://www.agner.org/
1281 { ISD::FSQRT, MVT::f64, 14 }, // Haswell from http://www.agner.org/
1282 { ISD::FSQRT, MVT::v2f64, 14 }, // Haswell from http://www.agner.org/
1283 { ISD::FSQRT, MVT::v4f64, 28 }, // Haswell from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001284 };
1285 static const CostTblEntry AVX1CostTbl[] = {
1286 { ISD::BITREVERSE, MVT::v4i64, 10 },
1287 { ISD::BITREVERSE, MVT::v8i32, 10 },
1288 { ISD::BITREVERSE, MVT::v16i16, 10 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001289 { ISD::BITREVERSE, MVT::v32i8, 10 },
1290 { ISD::BSWAP, MVT::v4i64, 4 },
1291 { ISD::BSWAP, MVT::v8i32, 4 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001292 { ISD::BSWAP, MVT::v16i16, 4 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001293 { ISD::CTLZ, MVT::v4i64, 46 },
1294 { ISD::CTLZ, MVT::v8i32, 36 },
1295 { ISD::CTLZ, MVT::v16i16, 28 },
1296 { ISD::CTLZ, MVT::v32i8, 18 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001297 { ISD::CTPOP, MVT::v4i64, 14 },
1298 { ISD::CTPOP, MVT::v8i32, 22 },
1299 { ISD::CTPOP, MVT::v16i16, 18 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001300 { ISD::CTPOP, MVT::v32i8, 12 },
1301 { ISD::CTTZ, MVT::v4i64, 20 },
1302 { ISD::CTTZ, MVT::v8i32, 28 },
1303 { ISD::CTTZ, MVT::v16i16, 24 },
1304 { ISD::CTTZ, MVT::v32i8, 18 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001305 { ISD::FSQRT, MVT::f32, 14 }, // SNB from http://www.agner.org/
1306 { ISD::FSQRT, MVT::v4f32, 14 }, // SNB from http://www.agner.org/
1307 { ISD::FSQRT, MVT::v8f32, 28 }, // SNB from http://www.agner.org/
1308 { ISD::FSQRT, MVT::f64, 21 }, // SNB from http://www.agner.org/
1309 { ISD::FSQRT, MVT::v2f64, 21 }, // SNB from http://www.agner.org/
1310 { ISD::FSQRT, MVT::v4f64, 43 }, // SNB from http://www.agner.org/
1311 };
1312 static const CostTblEntry SSE42CostTbl[] = {
1313 { ISD::FSQRT, MVT::f32, 18 }, // Nehalem from http://www.agner.org/
1314 { ISD::FSQRT, MVT::v4f32, 18 }, // Nehalem from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001315 };
1316 static const CostTblEntry SSSE3CostTbl[] = {
1317 { ISD::BITREVERSE, MVT::v2i64, 5 },
1318 { ISD::BITREVERSE, MVT::v4i32, 5 },
1319 { ISD::BITREVERSE, MVT::v8i16, 5 },
Simon Pilgrim356e8232016-06-20 23:08:21 +00001320 { ISD::BITREVERSE, MVT::v16i8, 5 },
1321 { ISD::BSWAP, MVT::v2i64, 1 },
1322 { ISD::BSWAP, MVT::v4i32, 1 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001323 { ISD::BSWAP, MVT::v8i16, 1 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001324 { ISD::CTLZ, MVT::v2i64, 23 },
1325 { ISD::CTLZ, MVT::v4i32, 18 },
1326 { ISD::CTLZ, MVT::v8i16, 14 },
1327 { ISD::CTLZ, MVT::v16i8, 9 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001328 { ISD::CTPOP, MVT::v2i64, 7 },
1329 { ISD::CTPOP, MVT::v4i32, 11 },
1330 { ISD::CTPOP, MVT::v8i16, 9 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001331 { ISD::CTPOP, MVT::v16i8, 6 },
1332 { ISD::CTTZ, MVT::v2i64, 10 },
1333 { ISD::CTTZ, MVT::v4i32, 14 },
1334 { ISD::CTTZ, MVT::v8i16, 12 },
1335 { ISD::CTTZ, MVT::v16i8, 9 }
Simon Pilgrim356e8232016-06-20 23:08:21 +00001336 };
1337 static const CostTblEntry SSE2CostTbl[] = {
1338 { ISD::BSWAP, MVT::v2i64, 7 },
1339 { ISD::BSWAP, MVT::v4i32, 7 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001340 { ISD::BSWAP, MVT::v8i16, 7 },
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001341 { ISD::CTLZ, MVT::v2i64, 25 },
1342 { ISD::CTLZ, MVT::v4i32, 26 },
1343 { ISD::CTLZ, MVT::v8i16, 20 },
1344 { ISD::CTLZ, MVT::v16i8, 17 },
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001345 { ISD::CTPOP, MVT::v2i64, 12 },
1346 { ISD::CTPOP, MVT::v4i32, 15 },
1347 { ISD::CTPOP, MVT::v8i16, 13 },
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001348 { ISD::CTPOP, MVT::v16i8, 10 },
1349 { ISD::CTTZ, MVT::v2i64, 14 },
1350 { ISD::CTTZ, MVT::v4i32, 18 },
1351 { ISD::CTTZ, MVT::v8i16, 16 },
Alexey Bataevd07c7312016-10-31 12:10:53 +00001352 { ISD::CTTZ, MVT::v16i8, 13 },
1353 { ISD::FSQRT, MVT::f64, 32 }, // Nehalem from http://www.agner.org/
1354 { ISD::FSQRT, MVT::v2f64, 32 }, // Nehalem from http://www.agner.org/
1355 };
1356 static const CostTblEntry SSE1CostTbl[] = {
1357 { ISD::FSQRT, MVT::f32, 28 }, // Pentium III from http://www.agner.org/
1358 { ISD::FSQRT, MVT::v4f32, 56 }, // Pentium III from http://www.agner.org/
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001359 };
Simon Pilgrim14000b32016-05-24 08:17:50 +00001360
1361 unsigned ISD = ISD::DELETED_NODE;
1362 switch (IID) {
1363 default:
1364 break;
1365 case Intrinsic::bitreverse:
1366 ISD = ISD::BITREVERSE;
1367 break;
Simon Pilgrim356e8232016-06-20 23:08:21 +00001368 case Intrinsic::bswap:
1369 ISD = ISD::BSWAP;
1370 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001371 case Intrinsic::ctlz:
1372 ISD = ISD::CTLZ;
1373 break;
Simon Pilgrim1b4f5112016-07-20 10:41:28 +00001374 case Intrinsic::ctpop:
1375 ISD = ISD::CTPOP;
1376 break;
Simon Pilgrim5d5ca9c2016-08-04 10:51:41 +00001377 case Intrinsic::cttz:
1378 ISD = ISD::CTTZ;
1379 break;
Alexey Bataevd07c7312016-10-31 12:10:53 +00001380 case Intrinsic::sqrt:
1381 ISD = ISD::FSQRT;
1382 break;
Simon Pilgrim14000b32016-05-24 08:17:50 +00001383 }
1384
1385 // Legalize the type.
1386 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, RetTy);
1387 MVT MTy = LT.second;
1388
1389 // Attempt to lookup cost.
1390 if (ST->hasXOP())
1391 if (const auto *Entry = CostTableLookup(XOPCostTbl, ISD, MTy))
1392 return LT.first * Entry->Cost;
1393
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001394 if (ST->hasAVX2())
1395 if (const auto *Entry = CostTableLookup(AVX2CostTbl, ISD, MTy))
1396 return LT.first * Entry->Cost;
1397
1398 if (ST->hasAVX())
1399 if (const auto *Entry = CostTableLookup(AVX1CostTbl, ISD, MTy))
1400 return LT.first * Entry->Cost;
1401
Alexey Bataevd07c7312016-10-31 12:10:53 +00001402 if (ST->hasSSE42())
1403 if (const auto *Entry = CostTableLookup(SSE42CostTbl, ISD, MTy))
1404 return LT.first * Entry->Cost;
1405
Simon Pilgrim3fc09f72016-06-11 19:23:02 +00001406 if (ST->hasSSSE3())
1407 if (const auto *Entry = CostTableLookup(SSSE3CostTbl, ISD, MTy))
1408 return LT.first * Entry->Cost;
1409
Simon Pilgrim356e8232016-06-20 23:08:21 +00001410 if (ST->hasSSE2())
1411 if (const auto *Entry = CostTableLookup(SSE2CostTbl, ISD, MTy))
1412 return LT.first * Entry->Cost;
1413
Alexey Bataevd07c7312016-10-31 12:10:53 +00001414 if (ST->hasSSE1())
1415 if (const auto *Entry = CostTableLookup(SSE1CostTbl, ISD, MTy))
1416 return LT.first * Entry->Cost;
1417
Simon Pilgrim14000b32016-05-24 08:17:50 +00001418 return BaseT::getIntrinsicInstrCost(IID, RetTy, Tys, FMF);
1419}
1420
1421int X86TTIImpl::getIntrinsicInstrCost(Intrinsic::ID IID, Type *RetTy,
1422 ArrayRef<Value *> Args, FastMathFlags FMF) {
1423 return BaseT::getIntrinsicInstrCost(IID, RetTy, Args, FMF);
1424}
1425
Chandler Carruth93205eb2015-08-05 18:08:10 +00001426int X86TTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val, unsigned Index) {
Chandler Carruth664e3542013-01-07 01:37:14 +00001427 assert(Val->isVectorTy() && "This must be a vector type");
1428
Sanjay Patelaedc3472016-05-25 17:27:54 +00001429 Type *ScalarType = Val->getScalarType();
1430
Chandler Carruth664e3542013-01-07 01:37:14 +00001431 if (Index != -1U) {
1432 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001433 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Val);
Chandler Carruth664e3542013-01-07 01:37:14 +00001434
1435 // This type is legalized to a scalar type.
1436 if (!LT.second.isVector())
1437 return 0;
1438
1439 // The type may be split. Normalize the index to the new type.
1440 unsigned Width = LT.second.getVectorNumElements();
1441 Index = Index % Width;
1442
1443 // Floating point scalars are already located in index #0.
Sanjay Patelaedc3472016-05-25 17:27:54 +00001444 if (ScalarType->isFloatingPointTy() && Index == 0)
Chandler Carruth664e3542013-01-07 01:37:14 +00001445 return 0;
1446 }
1447
Sanjay Patelaedc3472016-05-25 17:27:54 +00001448 // Add to the base cost if we know that the extracted element of a vector is
1449 // destined to be moved to and used in the integer register file.
1450 int RegisterFileMoveCost = 0;
1451 if (Opcode == Instruction::ExtractElement && ScalarType->isPointerTy())
1452 RegisterFileMoveCost = 1;
1453
1454 return BaseT::getVectorInstrCost(Opcode, Val, Index) + RegisterFileMoveCost;
Chandler Carruth664e3542013-01-07 01:37:14 +00001455}
1456
Chandler Carruth93205eb2015-08-05 18:08:10 +00001457int X86TTIImpl::getScalarizationOverhead(Type *Ty, bool Insert, bool Extract) {
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001458 assert (Ty->isVectorTy() && "Can only scalarize vectors");
Chandler Carruth93205eb2015-08-05 18:08:10 +00001459 int Cost = 0;
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001460
1461 for (int i = 0, e = Ty->getVectorNumElements(); i < e; ++i) {
1462 if (Insert)
Chandler Carruth705b1852015-01-31 03:43:40 +00001463 Cost += getVectorInstrCost(Instruction::InsertElement, Ty, i);
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001464 if (Extract)
Chandler Carruth705b1852015-01-31 03:43:40 +00001465 Cost += getVectorInstrCost(Instruction::ExtractElement, Ty, i);
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001466 }
1467
1468 return Cost;
1469}
1470
Chandler Carruth93205eb2015-08-05 18:08:10 +00001471int X86TTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src, unsigned Alignment,
1472 unsigned AddressSpace) {
Alp Tokerf907b892013-12-05 05:44:44 +00001473 // Handle non-power-of-two vectors such as <3 x float>
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001474 if (VectorType *VTy = dyn_cast<VectorType>(Src)) {
1475 unsigned NumElem = VTy->getVectorNumElements();
1476
1477 // Handle a few common cases:
1478 // <3 x float>
1479 if (NumElem == 3 && VTy->getScalarSizeInBits() == 32)
1480 // Cost = 64 bit store + extract + 32 bit store.
1481 return 3;
1482
1483 // <3 x double>
1484 if (NumElem == 3 && VTy->getScalarSizeInBits() == 64)
1485 // Cost = 128 bit store + unpack + 64 bit store.
1486 return 3;
1487
Alp Tokerf907b892013-12-05 05:44:44 +00001488 // Assume that all other non-power-of-two numbers are scalarized.
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001489 if (!isPowerOf2_32(NumElem)) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00001490 int Cost = BaseT::getMemoryOpCost(Opcode, VTy->getScalarType(), Alignment,
1491 AddressSpace);
1492 int SplitCost = getScalarizationOverhead(Src, Opcode == Instruction::Load,
1493 Opcode == Instruction::Store);
Nadav Rotemf9ecbcb2013-06-27 17:52:04 +00001494 return NumElem * Cost + SplitCost;
1495 }
1496 }
1497
Chandler Carruth664e3542013-01-07 01:37:14 +00001498 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001499 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src);
Chandler Carruth664e3542013-01-07 01:37:14 +00001500 assert((Opcode == Instruction::Load || Opcode == Instruction::Store) &&
1501 "Invalid Opcode");
1502
1503 // Each load/store unit costs 1.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001504 int Cost = LT.first * 1;
Chandler Carruth664e3542013-01-07 01:37:14 +00001505
Sanjay Patel9f6c4d52016-03-09 22:23:33 +00001506 // This isn't exactly right. We're using slow unaligned 32-byte accesses as a
1507 // proxy for a double-pumped AVX memory interface such as on Sandybridge.
1508 if (LT.second.getStoreSize() == 32 && ST->isUnalignedMem32Slow())
1509 Cost *= 2;
Chandler Carruth664e3542013-01-07 01:37:14 +00001510
1511 return Cost;
1512}
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001513
Chandler Carruth93205eb2015-08-05 18:08:10 +00001514int X86TTIImpl::getMaskedMemoryOpCost(unsigned Opcode, Type *SrcTy,
1515 unsigned Alignment,
1516 unsigned AddressSpace) {
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001517 VectorType *SrcVTy = dyn_cast<VectorType>(SrcTy);
1518 if (!SrcVTy)
1519 // To calculate scalar take the regular cost, without mask
1520 return getMemoryOpCost(Opcode, SrcTy, Alignment, AddressSpace);
1521
1522 unsigned NumElem = SrcVTy->getVectorNumElements();
1523 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00001524 VectorType::get(Type::getInt8Ty(SrcVTy->getContext()), NumElem);
Elena Demikhovsky20662e32015-10-19 07:43:38 +00001525 if ((Opcode == Instruction::Load && !isLegalMaskedLoad(SrcVTy)) ||
1526 (Opcode == Instruction::Store && !isLegalMaskedStore(SrcVTy)) ||
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001527 !isPowerOf2_32(NumElem)) {
1528 // Scalarization
Chandler Carruth93205eb2015-08-05 18:08:10 +00001529 int MaskSplitCost = getScalarizationOverhead(MaskTy, false, true);
1530 int ScalarCompareCost = getCmpSelInstrCost(
Mehdi Amini867e9142016-04-14 04:36:40 +00001531 Instruction::ICmp, Type::getInt8Ty(SrcVTy->getContext()), nullptr);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001532 int BranchCost = getCFInstrCost(Instruction::Br);
1533 int MaskCmpCost = NumElem * (BranchCost + ScalarCompareCost);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001534
Chandler Carruth93205eb2015-08-05 18:08:10 +00001535 int ValueSplitCost = getScalarizationOverhead(
1536 SrcVTy, Opcode == Instruction::Load, Opcode == Instruction::Store);
1537 int MemopCost =
Chandler Carruth705b1852015-01-31 03:43:40 +00001538 NumElem * BaseT::getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
1539 Alignment, AddressSpace);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001540 return MemopCost + ValueSplitCost + MaskSplitCost + MaskCmpCost;
1541 }
1542
1543 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001544 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, SrcVTy);
Cong Houda4e8ae2015-10-28 18:15:46 +00001545 auto VT = TLI->getValueType(DL, SrcVTy);
Chandler Carruth93205eb2015-08-05 18:08:10 +00001546 int Cost = 0;
Cong Houda4e8ae2015-10-28 18:15:46 +00001547 if (VT.isSimple() && LT.second != VT.getSimpleVT() &&
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001548 LT.second.getVectorNumElements() == NumElem)
1549 // Promotion requires expand/truncate for data and a shuffle for mask.
Hans Wennborg083ca9b2015-10-06 23:24:35 +00001550 Cost += getShuffleCost(TTI::SK_Alternate, SrcVTy, 0, nullptr) +
1551 getShuffleCost(TTI::SK_Alternate, MaskTy, 0, nullptr);
Chandler Carruth705b1852015-01-31 03:43:40 +00001552
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001553 else if (LT.second.getVectorNumElements() > NumElem) {
1554 VectorType *NewMaskTy = VectorType::get(MaskTy->getVectorElementType(),
1555 LT.second.getVectorNumElements());
1556 // Expanding requires fill mask with zeroes
Chandler Carruth705b1852015-01-31 03:43:40 +00001557 Cost += getShuffleCost(TTI::SK_InsertSubvector, NewMaskTy, 0, MaskTy);
Elena Demikhovskya3232f72015-01-25 08:44:46 +00001558 }
1559 if (!ST->hasAVX512())
1560 return Cost + LT.first*4; // Each maskmov costs 4
1561
1562 // AVX-512 masked load/store is cheapper
1563 return Cost+LT.first;
1564}
1565
Chandler Carruth93205eb2015-08-05 18:08:10 +00001566int X86TTIImpl::getAddressComputationCost(Type *Ty, bool IsComplex) {
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001567 // Address computations in vectorized code with non-consecutive addresses will
1568 // likely result in more instructions compared to scalar code where the
1569 // computation can more often be merged into the index mode. The resulting
1570 // extra micro-ops can significantly decrease throughput.
1571 unsigned NumVectorInstToHideOverhead = 10;
1572
1573 if (Ty->isVectorTy() && IsComplex)
1574 return NumVectorInstToHideOverhead;
1575
Chandler Carruth705b1852015-01-31 03:43:40 +00001576 return BaseT::getAddressComputationCost(Ty, IsComplex);
Arnold Schwaighofer6042a262013-07-12 19:16:07 +00001577}
Yi Jiang5c343de2013-09-19 17:48:48 +00001578
Chandler Carruth93205eb2015-08-05 18:08:10 +00001579int X86TTIImpl::getReductionCost(unsigned Opcode, Type *ValTy,
1580 bool IsPairwise) {
Michael Liao5bf95782014-12-04 05:20:33 +00001581
Chandler Carruth93205eb2015-08-05 18:08:10 +00001582 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, ValTy);
Michael Liao5bf95782014-12-04 05:20:33 +00001583
Yi Jiang5c343de2013-09-19 17:48:48 +00001584 MVT MTy = LT.second;
Michael Liao5bf95782014-12-04 05:20:33 +00001585
Yi Jiang5c343de2013-09-19 17:48:48 +00001586 int ISD = TLI->InstructionOpcodeToISD(Opcode);
1587 assert(ISD && "Invalid opcode");
Michael Liao5bf95782014-12-04 05:20:33 +00001588
1589 // We use the Intel Architecture Code Analyzer(IACA) to measure the throughput
1590 // and make it as the cost.
1591
Craig Topper4b275762015-10-28 04:02:12 +00001592 static const CostTblEntry SSE42CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00001593 { ISD::FADD, MVT::v2f64, 2 },
1594 { ISD::FADD, MVT::v4f32, 4 },
1595 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
1596 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
1597 { ISD::ADD, MVT::v8i16, 5 },
1598 };
Michael Liao5bf95782014-12-04 05:20:33 +00001599
Craig Topper4b275762015-10-28 04:02:12 +00001600 static const CostTblEntry AVX1CostTblPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00001601 { ISD::FADD, MVT::v4f32, 4 },
1602 { ISD::FADD, MVT::v4f64, 5 },
1603 { ISD::FADD, MVT::v8f32, 7 },
1604 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
1605 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.5".
1606 { ISD::ADD, MVT::v4i64, 5 }, // The data reported by the IACA tool is "4.8".
1607 { ISD::ADD, MVT::v8i16, 5 },
1608 { ISD::ADD, MVT::v8i32, 5 },
1609 };
1610
Craig Topper4b275762015-10-28 04:02:12 +00001611 static const CostTblEntry SSE42CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00001612 { ISD::FADD, MVT::v2f64, 2 },
1613 { ISD::FADD, MVT::v4f32, 4 },
1614 { ISD::ADD, MVT::v2i64, 2 }, // The data reported by the IACA tool is "1.6".
1615 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "3.3".
1616 { ISD::ADD, MVT::v8i16, 4 }, // The data reported by the IACA tool is "4.3".
1617 };
Michael Liao5bf95782014-12-04 05:20:33 +00001618
Craig Topper4b275762015-10-28 04:02:12 +00001619 static const CostTblEntry AVX1CostTblNoPairWise[] = {
Yi Jiang5c343de2013-09-19 17:48:48 +00001620 { ISD::FADD, MVT::v4f32, 3 },
1621 { ISD::FADD, MVT::v4f64, 3 },
1622 { ISD::FADD, MVT::v8f32, 4 },
1623 { ISD::ADD, MVT::v2i64, 1 }, // The data reported by the IACA tool is "1.5".
1624 { ISD::ADD, MVT::v4i32, 3 }, // The data reported by the IACA tool is "2.8".
1625 { ISD::ADD, MVT::v4i64, 3 },
1626 { ISD::ADD, MVT::v8i16, 4 },
1627 { ISD::ADD, MVT::v8i32, 5 },
1628 };
Michael Liao5bf95782014-12-04 05:20:33 +00001629
Yi Jiang5c343de2013-09-19 17:48:48 +00001630 if (IsPairwise) {
Craig Topperee0c8592015-10-27 04:14:24 +00001631 if (ST->hasAVX())
1632 if (const auto *Entry = CostTableLookup(AVX1CostTblPairWise, ISD, MTy))
1633 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00001634
Craig Topperee0c8592015-10-27 04:14:24 +00001635 if (ST->hasSSE42())
1636 if (const auto *Entry = CostTableLookup(SSE42CostTblPairWise, ISD, MTy))
1637 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00001638 } else {
Craig Topperee0c8592015-10-27 04:14:24 +00001639 if (ST->hasAVX())
1640 if (const auto *Entry = CostTableLookup(AVX1CostTblNoPairWise, ISD, MTy))
1641 return LT.first * Entry->Cost;
Michael Liao5bf95782014-12-04 05:20:33 +00001642
Craig Topperee0c8592015-10-27 04:14:24 +00001643 if (ST->hasSSE42())
1644 if (const auto *Entry = CostTableLookup(SSE42CostTblNoPairWise, ISD, MTy))
1645 return LT.first * Entry->Cost;
Yi Jiang5c343de2013-09-19 17:48:48 +00001646 }
1647
Chandler Carruth705b1852015-01-31 03:43:40 +00001648 return BaseT::getReductionCost(Opcode, ValTy, IsPairwise);
Yi Jiang5c343de2013-09-19 17:48:48 +00001649}
1650
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001651/// \brief Calculate the cost of materializing a 64-bit value. This helper
1652/// method might only calculate a fraction of a larger immediate. Therefore it
1653/// is valid to return a cost of ZERO.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001654int X86TTIImpl::getIntImmCost(int64_t Val) {
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001655 if (Val == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00001656 return TTI::TCC_Free;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001657
1658 if (isInt<32>(Val))
Chandler Carruth705b1852015-01-31 03:43:40 +00001659 return TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001660
Chandler Carruth705b1852015-01-31 03:43:40 +00001661 return 2 * TTI::TCC_Basic;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001662}
1663
Chandler Carruth93205eb2015-08-05 18:08:10 +00001664int X86TTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001665 assert(Ty->isIntegerTy());
1666
1667 unsigned BitSize = Ty->getPrimitiveSizeInBits();
1668 if (BitSize == 0)
1669 return ~0U;
1670
Juergen Ributzka43176172014-05-19 21:00:53 +00001671 // Never hoist constants larger than 128bit, because this might lead to
1672 // incorrect code generation or assertions in codegen.
1673 // Fixme: Create a cost model for types larger than i128 once the codegen
1674 // issues have been fixed.
1675 if (BitSize > 128)
Chandler Carruth705b1852015-01-31 03:43:40 +00001676 return TTI::TCC_Free;
Juergen Ributzka43176172014-05-19 21:00:53 +00001677
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001678 if (Imm == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00001679 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001680
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001681 // Sign-extend all constants to a multiple of 64-bit.
1682 APInt ImmVal = Imm;
1683 if (BitSize & 0x3f)
1684 ImmVal = Imm.sext((BitSize + 63) & ~0x3fU);
1685
1686 // Split the constant into 64-bit chunks and calculate the cost for each
1687 // chunk.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001688 int Cost = 0;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001689 for (unsigned ShiftVal = 0; ShiftVal < BitSize; ShiftVal += 64) {
1690 APInt Tmp = ImmVal.ashr(ShiftVal).sextOrTrunc(64);
1691 int64_t Val = Tmp.getSExtValue();
1692 Cost += getIntImmCost(Val);
1693 }
Sanjay Patel4c7d0942016-04-05 19:27:39 +00001694 // We need at least one instruction to materialize the constant.
Chandler Carruth93205eb2015-08-05 18:08:10 +00001695 return std::max(1, Cost);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001696}
1697
Chandler Carruth93205eb2015-08-05 18:08:10 +00001698int X86TTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx, const APInt &Imm,
1699 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001700 assert(Ty->isIntegerTy());
1701
1702 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00001703 // There is no cost model for constants with a bit size of 0. Return TCC_Free
1704 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001705 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00001706 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001707
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001708 unsigned ImmIdx = ~0U;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001709 switch (Opcode) {
Chandler Carruth705b1852015-01-31 03:43:40 +00001710 default:
1711 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001712 case Instruction::GetElementPtr:
Juergen Ributzka27435b32014-04-02 21:45:36 +00001713 // Always hoist the base address of a GetElementPtr. This prevents the
1714 // creation of new constants for every base constant that gets constant
1715 // folded with the offset.
Juergen Ributzka631c4912014-03-25 18:01:25 +00001716 if (Idx == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00001717 return 2 * TTI::TCC_Basic;
1718 return TTI::TCC_Free;
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001719 case Instruction::Store:
1720 ImmIdx = 0;
1721 break;
Craig Topper074e8452015-12-20 18:41:54 +00001722 case Instruction::ICmp:
1723 // This is an imperfect hack to prevent constant hoisting of
1724 // compares that might be trying to check if a 64-bit value fits in
1725 // 32-bits. The backend can optimize these cases using a right shift by 32.
1726 // Ideally we would check the compare predicate here. There also other
1727 // similar immediates the backend can use shifts for.
1728 if (Idx == 1 && Imm.getBitWidth() == 64) {
1729 uint64_t ImmVal = Imm.getZExtValue();
1730 if (ImmVal == 0x100000000ULL || ImmVal == 0xffffffff)
1731 return TTI::TCC_Free;
1732 }
1733 ImmIdx = 1;
1734 break;
Craig Topper79dd1bf2015-10-06 02:50:24 +00001735 case Instruction::And:
1736 // We support 64-bit ANDs with immediates with 32-bits of leading zeroes
1737 // by using a 32-bit operation with implicit zero extension. Detect such
1738 // immediates here as the normal path expects bit 31 to be sign extended.
1739 if (Idx == 1 && Imm.getBitWidth() == 64 && isUInt<32>(Imm.getZExtValue()))
1740 return TTI::TCC_Free;
Justin Bognerb03fd122016-08-17 05:10:15 +00001741 LLVM_FALLTHROUGH;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001742 case Instruction::Add:
1743 case Instruction::Sub:
1744 case Instruction::Mul:
1745 case Instruction::UDiv:
1746 case Instruction::SDiv:
1747 case Instruction::URem:
1748 case Instruction::SRem:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001749 case Instruction::Or:
1750 case Instruction::Xor:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001751 ImmIdx = 1;
1752 break;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00001753 // Always return TCC_Free for the shift value of a shift instruction.
1754 case Instruction::Shl:
1755 case Instruction::LShr:
1756 case Instruction::AShr:
1757 if (Idx == 1)
Chandler Carruth705b1852015-01-31 03:43:40 +00001758 return TTI::TCC_Free;
Michael Zolotukhin1f4a9602014-04-30 19:17:32 +00001759 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001760 case Instruction::Trunc:
1761 case Instruction::ZExt:
1762 case Instruction::SExt:
1763 case Instruction::IntToPtr:
1764 case Instruction::PtrToInt:
1765 case Instruction::BitCast:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001766 case Instruction::PHI:
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001767 case Instruction::Call:
1768 case Instruction::Select:
1769 case Instruction::Ret:
1770 case Instruction::Load:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001771 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001772 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001773
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001774 if (Idx == ImmIdx) {
Chandler Carruth93205eb2015-08-05 18:08:10 +00001775 int NumConstants = (BitSize + 63) / 64;
1776 int Cost = X86TTIImpl::getIntImmCost(Imm, Ty);
Chandler Carruth705b1852015-01-31 03:43:40 +00001777 return (Cost <= NumConstants * TTI::TCC_Basic)
Chandler Carruth93205eb2015-08-05 18:08:10 +00001778 ? static_cast<int>(TTI::TCC_Free)
Chandler Carruth705b1852015-01-31 03:43:40 +00001779 : Cost;
Juergen Ributzkab2e4edb2014-06-10 00:32:29 +00001780 }
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001781
Chandler Carruth705b1852015-01-31 03:43:40 +00001782 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001783}
1784
Chandler Carruth93205eb2015-08-05 18:08:10 +00001785int X86TTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx, const APInt &Imm,
1786 Type *Ty) {
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001787 assert(Ty->isIntegerTy());
1788
1789 unsigned BitSize = Ty->getPrimitiveSizeInBits();
Juergen Ributzka43176172014-05-19 21:00:53 +00001790 // There is no cost model for constants with a bit size of 0. Return TCC_Free
1791 // here, so that constant hoisting will ignore this constant.
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001792 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +00001793 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001794
1795 switch (IID) {
Chandler Carruth705b1852015-01-31 03:43:40 +00001796 default:
1797 return TTI::TCC_Free;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001798 case Intrinsic::sadd_with_overflow:
1799 case Intrinsic::uadd_with_overflow:
1800 case Intrinsic::ssub_with_overflow:
1801 case Intrinsic::usub_with_overflow:
1802 case Intrinsic::smul_with_overflow:
1803 case Intrinsic::umul_with_overflow:
Juergen Ributzkaf0dff492014-03-21 06:04:45 +00001804 if ((Idx == 1) && Imm.getBitWidth() <= 64 && isInt<32>(Imm.getSExtValue()))
Chandler Carruth705b1852015-01-31 03:43:40 +00001805 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00001806 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001807 case Intrinsic::experimental_stackmap:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00001808 if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00001809 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00001810 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001811 case Intrinsic::experimental_patchpoint_void:
1812 case Intrinsic::experimental_patchpoint_i64:
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00001813 if ((Idx < 4) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +00001814 return TTI::TCC_Free;
Juergen Ributzka5eef98c2014-03-25 18:01:23 +00001815 break;
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001816 }
Chandler Carruth705b1852015-01-31 03:43:40 +00001817 return X86TTIImpl::getIntImmCost(Imm, Ty);
Juergen Ributzkaf26beda2014-01-25 02:02:55 +00001818}
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00001819
Elena Demikhovsky54946982015-12-28 20:10:59 +00001820// Return an average cost of Gather / Scatter instruction, maybe improved later
1821int X86TTIImpl::getGSVectorCost(unsigned Opcode, Type *SrcVTy, Value *Ptr,
1822 unsigned Alignment, unsigned AddressSpace) {
1823
1824 assert(isa<VectorType>(SrcVTy) && "Unexpected type in getGSVectorCost");
1825 unsigned VF = SrcVTy->getVectorNumElements();
1826
1827 // Try to reduce index size from 64 bit (default for GEP)
1828 // to 32. It is essential for VF 16. If the index can't be reduced to 32, the
1829 // operation will use 16 x 64 indices which do not fit in a zmm and needs
1830 // to split. Also check that the base pointer is the same for all lanes,
1831 // and that there's at most one variable index.
1832 auto getIndexSizeInBits = [](Value *Ptr, const DataLayout& DL) {
1833 unsigned IndexSize = DL.getPointerSizeInBits();
1834 GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(Ptr);
1835 if (IndexSize < 64 || !GEP)
1836 return IndexSize;
Simon Pilgrim14000b32016-05-24 08:17:50 +00001837
Elena Demikhovsky54946982015-12-28 20:10:59 +00001838 unsigned NumOfVarIndices = 0;
1839 Value *Ptrs = GEP->getPointerOperand();
1840 if (Ptrs->getType()->isVectorTy() && !getSplatValue(Ptrs))
1841 return IndexSize;
1842 for (unsigned i = 1; i < GEP->getNumOperands(); ++i) {
1843 if (isa<Constant>(GEP->getOperand(i)))
1844 continue;
1845 Type *IndxTy = GEP->getOperand(i)->getType();
1846 if (IndxTy->isVectorTy())
1847 IndxTy = IndxTy->getVectorElementType();
1848 if ((IndxTy->getPrimitiveSizeInBits() == 64 &&
1849 !isa<SExtInst>(GEP->getOperand(i))) ||
1850 ++NumOfVarIndices > 1)
1851 return IndexSize; // 64
1852 }
1853 return (unsigned)32;
1854 };
1855
1856
1857 // Trying to reduce IndexSize to 32 bits for vector 16.
1858 // By default the IndexSize is equal to pointer size.
1859 unsigned IndexSize = (VF >= 16) ? getIndexSizeInBits(Ptr, DL) :
1860 DL.getPointerSizeInBits();
1861
Mehdi Amini867e9142016-04-14 04:36:40 +00001862 Type *IndexVTy = VectorType::get(IntegerType::get(SrcVTy->getContext(),
Elena Demikhovsky54946982015-12-28 20:10:59 +00001863 IndexSize), VF);
1864 std::pair<int, MVT> IdxsLT = TLI->getTypeLegalizationCost(DL, IndexVTy);
1865 std::pair<int, MVT> SrcLT = TLI->getTypeLegalizationCost(DL, SrcVTy);
1866 int SplitFactor = std::max(IdxsLT.first, SrcLT.first);
1867 if (SplitFactor > 1) {
1868 // Handle splitting of vector of pointers
1869 Type *SplitSrcTy = VectorType::get(SrcVTy->getScalarType(), VF / SplitFactor);
1870 return SplitFactor * getGSVectorCost(Opcode, SplitSrcTy, Ptr, Alignment,
1871 AddressSpace);
1872 }
1873
1874 // The gather / scatter cost is given by Intel architects. It is a rough
1875 // number since we are looking at one instruction in a time.
1876 const int GSOverhead = 2;
1877 return GSOverhead + VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
1878 Alignment, AddressSpace);
1879}
1880
1881/// Return the cost of full scalarization of gather / scatter operation.
1882///
1883/// Opcode - Load or Store instruction.
1884/// SrcVTy - The type of the data vector that should be gathered or scattered.
1885/// VariableMask - The mask is non-constant at compile time.
1886/// Alignment - Alignment for one element.
1887/// AddressSpace - pointer[s] address space.
1888///
1889int X86TTIImpl::getGSScalarCost(unsigned Opcode, Type *SrcVTy,
1890 bool VariableMask, unsigned Alignment,
1891 unsigned AddressSpace) {
1892 unsigned VF = SrcVTy->getVectorNumElements();
1893
1894 int MaskUnpackCost = 0;
1895 if (VariableMask) {
1896 VectorType *MaskTy =
Mehdi Amini867e9142016-04-14 04:36:40 +00001897 VectorType::get(Type::getInt1Ty(SrcVTy->getContext()), VF);
Elena Demikhovsky54946982015-12-28 20:10:59 +00001898 MaskUnpackCost = getScalarizationOverhead(MaskTy, false, true);
1899 int ScalarCompareCost =
Mehdi Amini867e9142016-04-14 04:36:40 +00001900 getCmpSelInstrCost(Instruction::ICmp, Type::getInt1Ty(SrcVTy->getContext()),
Elena Demikhovsky54946982015-12-28 20:10:59 +00001901 nullptr);
1902 int BranchCost = getCFInstrCost(Instruction::Br);
1903 MaskUnpackCost += VF * (BranchCost + ScalarCompareCost);
1904 }
1905
1906 // The cost of the scalar loads/stores.
1907 int MemoryOpCost = VF * getMemoryOpCost(Opcode, SrcVTy->getScalarType(),
1908 Alignment, AddressSpace);
1909
1910 int InsertExtractCost = 0;
1911 if (Opcode == Instruction::Load)
1912 for (unsigned i = 0; i < VF; ++i)
1913 // Add the cost of inserting each scalar load into the vector
1914 InsertExtractCost +=
1915 getVectorInstrCost(Instruction::InsertElement, SrcVTy, i);
1916 else
1917 for (unsigned i = 0; i < VF; ++i)
1918 // Add the cost of extracting each element out of the data vector
1919 InsertExtractCost +=
1920 getVectorInstrCost(Instruction::ExtractElement, SrcVTy, i);
1921
1922 return MemoryOpCost + MaskUnpackCost + InsertExtractCost;
1923}
1924
1925/// Calculate the cost of Gather / Scatter operation
1926int X86TTIImpl::getGatherScatterOpCost(unsigned Opcode, Type *SrcVTy,
1927 Value *Ptr, bool VariableMask,
1928 unsigned Alignment) {
1929 assert(SrcVTy->isVectorTy() && "Unexpected data type for Gather/Scatter");
1930 unsigned VF = SrcVTy->getVectorNumElements();
1931 PointerType *PtrTy = dyn_cast<PointerType>(Ptr->getType());
1932 if (!PtrTy && Ptr->getType()->isVectorTy())
1933 PtrTy = dyn_cast<PointerType>(Ptr->getType()->getVectorElementType());
1934 assert(PtrTy && "Unexpected type for Ptr argument");
1935 unsigned AddressSpace = PtrTy->getAddressSpace();
1936
1937 bool Scalarize = false;
1938 if ((Opcode == Instruction::Load && !isLegalMaskedGather(SrcVTy)) ||
1939 (Opcode == Instruction::Store && !isLegalMaskedScatter(SrcVTy)))
1940 Scalarize = true;
1941 // Gather / Scatter for vector 2 is not profitable on KNL / SKX
1942 // Vector-4 of gather/scatter instruction does not exist on KNL.
1943 // We can extend it to 8 elements, but zeroing upper bits of
1944 // the mask vector will add more instructions. Right now we give the scalar
1945 // cost of vector-4 for KNL. TODO: Check, maybe the gather/scatter instruction is
1946 // better in the VariableMask case.
1947 if (VF == 2 || (VF == 4 && !ST->hasVLX()))
1948 Scalarize = true;
1949
1950 if (Scalarize)
1951 return getGSScalarCost(Opcode, SrcVTy, VariableMask, Alignment, AddressSpace);
1952
1953 return getGSVectorCost(Opcode, SrcVTy, Ptr, Alignment, AddressSpace);
1954}
1955
Elena Demikhovsky20662e32015-10-19 07:43:38 +00001956bool X86TTIImpl::isLegalMaskedLoad(Type *DataTy) {
1957 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00001958 int DataWidth = isa<PointerType>(ScalarTy) ?
1959 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00001960
Igor Bregerf44b79d2016-08-02 09:15:28 +00001961 return ((DataWidth == 32 || DataWidth == 64) && ST->hasAVX()) ||
1962 ((DataWidth == 8 || DataWidth == 16) && ST->hasBWI());
NAKAMURA Takumi0b305db2015-07-14 04:03:49 +00001963}
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00001964
Elena Demikhovsky20662e32015-10-19 07:43:38 +00001965bool X86TTIImpl::isLegalMaskedStore(Type *DataType) {
1966 return isLegalMaskedLoad(DataType);
Elena Demikhovskyf1de34b2014-12-04 09:40:44 +00001967}
1968
Elena Demikhovsky09285852015-10-25 15:37:55 +00001969bool X86TTIImpl::isLegalMaskedGather(Type *DataTy) {
1970 // This function is called now in two cases: from the Loop Vectorizer
1971 // and from the Scalarizer.
1972 // When the Loop Vectorizer asks about legality of the feature,
1973 // the vectorization factor is not calculated yet. The Loop Vectorizer
1974 // sends a scalar type and the decision is based on the width of the
1975 // scalar element.
1976 // Later on, the cost model will estimate usage this intrinsic based on
1977 // the vector type.
1978 // The Scalarizer asks again about legality. It sends a vector type.
1979 // In this case we can reject non-power-of-2 vectors.
1980 if (isa<VectorType>(DataTy) && !isPowerOf2_32(DataTy->getVectorNumElements()))
1981 return false;
1982 Type *ScalarTy = DataTy->getScalarType();
Elena Demikhovsky1ca72e12015-11-19 07:17:16 +00001983 int DataWidth = isa<PointerType>(ScalarTy) ?
1984 DL.getPointerSizeInBits() : ScalarTy->getPrimitiveSizeInBits();
Elena Demikhovsky09285852015-10-25 15:37:55 +00001985
1986 // AVX-512 allows gather and scatter
Igor Bregerf44b79d2016-08-02 09:15:28 +00001987 return (DataWidth == 32 || DataWidth == 64) && ST->hasAVX512();
Elena Demikhovsky09285852015-10-25 15:37:55 +00001988}
1989
1990bool X86TTIImpl::isLegalMaskedScatter(Type *DataType) {
1991 return isLegalMaskedGather(DataType);
1992}
1993
Eric Christopherd566fb12015-07-29 22:09:48 +00001994bool X86TTIImpl::areInlineCompatible(const Function *Caller,
1995 const Function *Callee) const {
Eric Christophere1002262015-07-02 01:11:50 +00001996 const TargetMachine &TM = getTLI()->getTargetMachine();
1997
1998 // Work this as a subsetting of subtarget features.
1999 const FeatureBitset &CallerBits =
2000 TM.getSubtargetImpl(*Caller)->getFeatureBits();
2001 const FeatureBitset &CalleeBits =
2002 TM.getSubtargetImpl(*Callee)->getFeatureBits();
2003
2004 // FIXME: This is likely too limiting as it will include subtarget features
2005 // that we might not care about for inlining, but it is conservatively
2006 // correct.
2007 return (CallerBits & CalleeBits) == CalleeBits;
2008}
Michael Kupersteinb2443ed2016-10-20 21:04:31 +00002009
2010bool X86TTIImpl::enableInterleavedAccessVectorization() {
2011 // TODO: We expect this to be beneficial regardless of arch,
2012 // but there are currently some unexplained performance artifacts on Atom.
2013 // As a temporary solution, disable on Atom.
2014 return !(ST->isAtom() || ST->isSLM());
2015}