blob: 79f657fa26b09a716c3bad156f1efbb383d21616 [file] [log] [blame]
Chandler Carruth93dcdc42015-01-31 11:17:59 +00001//===-- AArch64TargetTransformInfo.cpp - AArch64 specific TTI -------------===//
Tim Northover3b0846e2014-05-24 12:50:23 +00002//
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//===----------------------------------------------------------------------===//
Tim Northover3b0846e2014-05-24 12:50:23 +00009
Chandler Carruth93dcdc42015-01-31 11:17:59 +000010#include "AArch64TargetTransformInfo.h"
Tim Northover3b0846e2014-05-24 12:50:23 +000011#include "MCTargetDesc/AArch64AddressingModes.h"
12#include "llvm/Analysis/TargetTransformInfo.h"
Kevin Qinaef68412015-03-09 06:14:28 +000013#include "llvm/Analysis/LoopInfo.h"
Chandler Carruth705b1852015-01-31 03:43:40 +000014#include "llvm/CodeGen/BasicTTIImpl.h"
Tim Northover3b0846e2014-05-24 12:50:23 +000015#include "llvm/Support/Debug.h"
16#include "llvm/Target/CostTable.h"
17#include "llvm/Target/TargetLowering.h"
18#include <algorithm>
19using namespace llvm;
20
21#define DEBUG_TYPE "aarch64tti"
22
Tim Northover3b0846e2014-05-24 12:50:23 +000023/// \brief Calculate the cost of materializing a 64-bit value. This helper
24/// method might only calculate a fraction of a larger immediate. Therefore it
25/// is valid to return a cost of ZERO.
Chandler Carruth93205eb2015-08-05 18:08:10 +000026int AArch64TTIImpl::getIntImmCost(int64_t Val) {
Tim Northover3b0846e2014-05-24 12:50:23 +000027 // Check if the immediate can be encoded within an instruction.
28 if (Val == 0 || AArch64_AM::isLogicalImmediate(Val, 64))
29 return 0;
30
31 if (Val < 0)
32 Val = ~Val;
33
34 // Calculate how many moves we will need to materialize this constant.
35 unsigned LZ = countLeadingZeros((uint64_t)Val);
36 return (64 - LZ + 15) / 16;
37}
38
39/// \brief Calculate the cost of materializing the given constant.
Chandler Carruth93205eb2015-08-05 18:08:10 +000040int AArch64TTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) {
Tim Northover3b0846e2014-05-24 12:50:23 +000041 assert(Ty->isIntegerTy());
42
43 unsigned BitSize = Ty->getPrimitiveSizeInBits();
44 if (BitSize == 0)
45 return ~0U;
46
47 // Sign-extend all constants to a multiple of 64-bit.
48 APInt ImmVal = Imm;
49 if (BitSize & 0x3f)
50 ImmVal = Imm.sext((BitSize + 63) & ~0x3fU);
51
52 // Split the constant into 64-bit chunks and calculate the cost for each
53 // chunk.
Chandler Carruth93205eb2015-08-05 18:08:10 +000054 int Cost = 0;
Tim Northover3b0846e2014-05-24 12:50:23 +000055 for (unsigned ShiftVal = 0; ShiftVal < BitSize; ShiftVal += 64) {
56 APInt Tmp = ImmVal.ashr(ShiftVal).sextOrTrunc(64);
57 int64_t Val = Tmp.getSExtValue();
58 Cost += getIntImmCost(Val);
59 }
60 // We need at least one instruction to materialze the constant.
Chandler Carruth93205eb2015-08-05 18:08:10 +000061 return std::max(1, Cost);
Tim Northover3b0846e2014-05-24 12:50:23 +000062}
63
Chandler Carruth93205eb2015-08-05 18:08:10 +000064int AArch64TTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx,
65 const APInt &Imm, Type *Ty) {
Tim Northover3b0846e2014-05-24 12:50:23 +000066 assert(Ty->isIntegerTy());
67
68 unsigned BitSize = Ty->getPrimitiveSizeInBits();
69 // There is no cost model for constants with a bit size of 0. Return TCC_Free
70 // here, so that constant hoisting will ignore this constant.
71 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +000072 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +000073
74 unsigned ImmIdx = ~0U;
75 switch (Opcode) {
76 default:
Chandler Carruth705b1852015-01-31 03:43:40 +000077 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +000078 case Instruction::GetElementPtr:
79 // Always hoist the base address of a GetElementPtr.
80 if (Idx == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +000081 return 2 * TTI::TCC_Basic;
82 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +000083 case Instruction::Store:
84 ImmIdx = 0;
85 break;
86 case Instruction::Add:
87 case Instruction::Sub:
88 case Instruction::Mul:
89 case Instruction::UDiv:
90 case Instruction::SDiv:
91 case Instruction::URem:
92 case Instruction::SRem:
93 case Instruction::And:
94 case Instruction::Or:
95 case Instruction::Xor:
96 case Instruction::ICmp:
97 ImmIdx = 1;
98 break;
99 // Always return TCC_Free for the shift value of a shift instruction.
100 case Instruction::Shl:
101 case Instruction::LShr:
102 case Instruction::AShr:
103 if (Idx == 1)
Chandler Carruth705b1852015-01-31 03:43:40 +0000104 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +0000105 break;
106 case Instruction::Trunc:
107 case Instruction::ZExt:
108 case Instruction::SExt:
109 case Instruction::IntToPtr:
110 case Instruction::PtrToInt:
111 case Instruction::BitCast:
112 case Instruction::PHI:
113 case Instruction::Call:
114 case Instruction::Select:
115 case Instruction::Ret:
116 case Instruction::Load:
117 break;
118 }
119
120 if (Idx == ImmIdx) {
Chandler Carruth93205eb2015-08-05 18:08:10 +0000121 int NumConstants = (BitSize + 63) / 64;
122 int Cost = AArch64TTIImpl::getIntImmCost(Imm, Ty);
Chandler Carruth705b1852015-01-31 03:43:40 +0000123 return (Cost <= NumConstants * TTI::TCC_Basic)
Chandler Carruth93205eb2015-08-05 18:08:10 +0000124 ? static_cast<int>(TTI::TCC_Free)
Chandler Carruth705b1852015-01-31 03:43:40 +0000125 : Cost;
Tim Northover3b0846e2014-05-24 12:50:23 +0000126 }
Chandler Carruth705b1852015-01-31 03:43:40 +0000127 return AArch64TTIImpl::getIntImmCost(Imm, Ty);
Tim Northover3b0846e2014-05-24 12:50:23 +0000128}
129
Chandler Carruth93205eb2015-08-05 18:08:10 +0000130int AArch64TTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx,
131 const APInt &Imm, Type *Ty) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000132 assert(Ty->isIntegerTy());
133
134 unsigned BitSize = Ty->getPrimitiveSizeInBits();
135 // There is no cost model for constants with a bit size of 0. Return TCC_Free
136 // here, so that constant hoisting will ignore this constant.
137 if (BitSize == 0)
Chandler Carruth705b1852015-01-31 03:43:40 +0000138 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +0000139
140 switch (IID) {
141 default:
Chandler Carruth705b1852015-01-31 03:43:40 +0000142 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +0000143 case Intrinsic::sadd_with_overflow:
144 case Intrinsic::uadd_with_overflow:
145 case Intrinsic::ssub_with_overflow:
146 case Intrinsic::usub_with_overflow:
147 case Intrinsic::smul_with_overflow:
148 case Intrinsic::umul_with_overflow:
149 if (Idx == 1) {
Chandler Carruth93205eb2015-08-05 18:08:10 +0000150 int NumConstants = (BitSize + 63) / 64;
151 int Cost = AArch64TTIImpl::getIntImmCost(Imm, Ty);
Chandler Carruth705b1852015-01-31 03:43:40 +0000152 return (Cost <= NumConstants * TTI::TCC_Basic)
Chandler Carruth93205eb2015-08-05 18:08:10 +0000153 ? static_cast<int>(TTI::TCC_Free)
Chandler Carruth705b1852015-01-31 03:43:40 +0000154 : Cost;
Tim Northover3b0846e2014-05-24 12:50:23 +0000155 }
156 break;
157 case Intrinsic::experimental_stackmap:
158 if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +0000159 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +0000160 break;
161 case Intrinsic::experimental_patchpoint_void:
162 case Intrinsic::experimental_patchpoint_i64:
163 if ((Idx < 4) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
Chandler Carruth705b1852015-01-31 03:43:40 +0000164 return TTI::TCC_Free;
Tim Northover3b0846e2014-05-24 12:50:23 +0000165 break;
166 }
Chandler Carruth705b1852015-01-31 03:43:40 +0000167 return AArch64TTIImpl::getIntImmCost(Imm, Ty);
Tim Northover3b0846e2014-05-24 12:50:23 +0000168}
169
Chandler Carruth705b1852015-01-31 03:43:40 +0000170TargetTransformInfo::PopcntSupportKind
171AArch64TTIImpl::getPopcntSupport(unsigned TyWidth) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000172 assert(isPowerOf2_32(TyWidth) && "Ty width must be power of 2");
173 if (TyWidth == 32 || TyWidth == 64)
Chandler Carruth705b1852015-01-31 03:43:40 +0000174 return TTI::PSK_FastHardware;
Tim Northover3b0846e2014-05-24 12:50:23 +0000175 // TODO: AArch64TargetLowering::LowerCTPOP() supports 128bit popcount.
Chandler Carruth705b1852015-01-31 03:43:40 +0000176 return TTI::PSK_Software;
Tim Northover3b0846e2014-05-24 12:50:23 +0000177}
178
Chandler Carruth93205eb2015-08-05 18:08:10 +0000179int AArch64TTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000180 int ISD = TLI->InstructionOpcodeToISD(Opcode);
181 assert(ISD && "Invalid opcode");
182
Mehdi Amini44ede332015-07-09 02:09:04 +0000183 EVT SrcTy = TLI->getValueType(DL, Src);
184 EVT DstTy = TLI->getValueType(DL, Dst);
Tim Northover3b0846e2014-05-24 12:50:23 +0000185
186 if (!SrcTy.isSimple() || !DstTy.isSimple())
Chandler Carruth705b1852015-01-31 03:43:40 +0000187 return BaseT::getCastInstrCost(Opcode, Dst, Src);
Tim Northover3b0846e2014-05-24 12:50:23 +0000188
189 static const TypeConversionCostTblEntry<MVT> ConversionTbl[] = {
Silviu Barangab322aa62015-08-17 16:05:09 +0000190 { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 0 },
191 { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 0 },
192 { ISD::SIGN_EXTEND, MVT::v2i64, MVT::v2i32, 1 },
193 { ISD::ZERO_EXTEND, MVT::v2i64, MVT::v2i32, 1 },
194 { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 0 },
195 { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 1 },
196
197 // The number of shll instructions for the extension.
198 { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
199 { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 },
200 { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
201 { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 3 },
202 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i8, 7 },
203 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i8, 7 },
204 { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i16, 6 },
205 { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i16, 6 },
206 { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 6 },
207 { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 6 },
208
209 { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 6 },
210 { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 3 },
211
Tim Northover3b0846e2014-05-24 12:50:23 +0000212 // LowerVectorINT_TO_FP:
213 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i32, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000214 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
Tim Northover3b0846e2014-05-24 12:50:23 +0000215 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
216 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i32, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000217 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 },
Tim Northover3b0846e2014-05-24 12:50:23 +0000218 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000219
220 // Complex: to v2f32
221 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i8, 3 },
222 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i16, 3 },
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000223 { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i64, 2 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000224 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i8, 3 },
225 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i16, 3 },
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000226 { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 2 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000227
228 // Complex: to v4f32
229 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i8, 4 },
230 { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 },
231 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i8, 3 },
232 { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 },
233
Silviu Barangab322aa62015-08-17 16:05:09 +0000234 // Complex: to v8f32
235 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i8, 10 },
236 { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i16, 4 },
237 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 10 },
238 { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 4 },
239
240 // Complex: to v16f32
241 { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i8, 21 },
242 { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i8, 21 },
243
Tim Northoveref0d7602014-06-15 09:27:06 +0000244 // Complex: to v2f64
245 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i8, 4 },
246 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i16, 4 },
247 { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i32, 2 },
248 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i8, 4 },
249 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i16, 4 },
250 { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 2 },
251
252
Tim Northover3b0846e2014-05-24 12:50:23 +0000253 // LowerVectorFP_TO_INT
Tim Northoveref0d7602014-06-15 09:27:06 +0000254 { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f32, 1 },
Tim Northover3b0846e2014-05-24 12:50:23 +0000255 { ISD::FP_TO_SINT, MVT::v4i32, MVT::v4f32, 1 },
256 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f64, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000257 { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f32, 1 },
Tim Northover3b0846e2014-05-24 12:50:23 +0000258 { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f32, 1 },
259 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f64, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000260
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000261 // Complex, from v2f32: legal type is v2i32 (no cost) or v2i64 (1 ext).
Tim Northoveref0d7602014-06-15 09:27:06 +0000262 { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f32, 2 },
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000263 { ISD::FP_TO_SINT, MVT::v2i16, MVT::v2f32, 1 },
264 { ISD::FP_TO_SINT, MVT::v2i8, MVT::v2f32, 1 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000265 { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f32, 2 },
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000266 { ISD::FP_TO_UINT, MVT::v2i16, MVT::v2f32, 1 },
267 { ISD::FP_TO_UINT, MVT::v2i8, MVT::v2f32, 1 },
268
269 // Complex, from v4f32: legal type is v4i16, 1 narrowing => ~2
270 { ISD::FP_TO_SINT, MVT::v4i16, MVT::v4f32, 2 },
271 { ISD::FP_TO_SINT, MVT::v4i8, MVT::v4f32, 2 },
Tim Northoveref0d7602014-06-15 09:27:06 +0000272 { ISD::FP_TO_UINT, MVT::v4i16, MVT::v4f32, 2 },
Tim Northoverdbecc3b2014-06-15 09:27:15 +0000273 { ISD::FP_TO_UINT, MVT::v4i8, MVT::v4f32, 2 },
274
275 // Complex, from v2f64: legal type is v2i32, 1 narrowing => ~2.
276 { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f64, 2 },
277 { ISD::FP_TO_SINT, MVT::v2i16, MVT::v2f64, 2 },
278 { ISD::FP_TO_SINT, MVT::v2i8, MVT::v2f64, 2 },
279 { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f64, 2 },
280 { ISD::FP_TO_UINT, MVT::v2i16, MVT::v2f64, 2 },
281 { ISD::FP_TO_UINT, MVT::v2i8, MVT::v2f64, 2 },
Tim Northover3b0846e2014-05-24 12:50:23 +0000282 };
283
284 int Idx = ConvertCostTableLookup<MVT>(
285 ConversionTbl, array_lengthof(ConversionTbl), ISD, DstTy.getSimpleVT(),
286 SrcTy.getSimpleVT());
287 if (Idx != -1)
288 return ConversionTbl[Idx].Cost;
289
Chandler Carruth705b1852015-01-31 03:43:40 +0000290 return BaseT::getCastInstrCost(Opcode, Dst, Src);
Tim Northover3b0846e2014-05-24 12:50:23 +0000291}
292
Chandler Carruth93205eb2015-08-05 18:08:10 +0000293int AArch64TTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val,
294 unsigned Index) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000295 assert(Val->isVectorTy() && "This must be a vector type");
296
297 if (Index != -1U) {
298 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000299 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Val);
Tim Northover3b0846e2014-05-24 12:50:23 +0000300
301 // This type is legalized to a scalar type.
302 if (!LT.second.isVector())
303 return 0;
304
305 // The type may be split. Normalize the index to the new type.
306 unsigned Width = LT.second.getVectorNumElements();
307 Index = Index % Width;
308
309 // The element at index zero is already inside the vector.
310 if (Index == 0)
311 return 0;
312 }
313
314 // All other insert/extracts cost this much.
Silviu Barangab322aa62015-08-17 16:05:09 +0000315 return 3;
Tim Northover3b0846e2014-05-24 12:50:23 +0000316}
317
Chandler Carruth93205eb2015-08-05 18:08:10 +0000318int AArch64TTIImpl::getArithmeticInstrCost(
Chandler Carruth705b1852015-01-31 03:43:40 +0000319 unsigned Opcode, Type *Ty, TTI::OperandValueKind Opd1Info,
320 TTI::OperandValueKind Opd2Info, TTI::OperandValueProperties Opd1PropInfo,
321 TTI::OperandValueProperties Opd2PropInfo) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000322 // Legalize the type.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000323 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Ty);
Tim Northover3b0846e2014-05-24 12:50:23 +0000324
325 int ISD = TLI->InstructionOpcodeToISD(Opcode);
326
Chad Rosier70d54ac2014-09-29 13:59:31 +0000327 if (ISD == ISD::SDIV &&
328 Opd2Info == TargetTransformInfo::OK_UniformConstantValue &&
329 Opd2PropInfo == TargetTransformInfo::OP_PowerOf2) {
330 // On AArch64, scalar signed division by constants power-of-two are
331 // normally expanded to the sequence ADD + CMP + SELECT + SRA.
332 // The OperandValue properties many not be same as that of previous
333 // operation; conservatively assume OP_None.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000334 int Cost = getArithmeticInstrCost(Instruction::Add, Ty, Opd1Info, Opd2Info,
335 TargetTransformInfo::OP_None,
336 TargetTransformInfo::OP_None);
Chad Rosier70d54ac2014-09-29 13:59:31 +0000337 Cost += getArithmeticInstrCost(Instruction::Sub, Ty, Opd1Info, Opd2Info,
338 TargetTransformInfo::OP_None,
339 TargetTransformInfo::OP_None);
340 Cost += getArithmeticInstrCost(Instruction::Select, Ty, Opd1Info, Opd2Info,
341 TargetTransformInfo::OP_None,
342 TargetTransformInfo::OP_None);
343 Cost += getArithmeticInstrCost(Instruction::AShr, Ty, Opd1Info, Opd2Info,
344 TargetTransformInfo::OP_None,
345 TargetTransformInfo::OP_None);
346 return Cost;
347 }
348
Tim Northover3b0846e2014-05-24 12:50:23 +0000349 switch (ISD) {
350 default:
Chandler Carruth705b1852015-01-31 03:43:40 +0000351 return BaseT::getArithmeticInstrCost(Opcode, Ty, Opd1Info, Opd2Info,
352 Opd1PropInfo, Opd2PropInfo);
Tim Northover3b0846e2014-05-24 12:50:23 +0000353 case ISD::ADD:
354 case ISD::MUL:
355 case ISD::XOR:
356 case ISD::OR:
357 case ISD::AND:
358 // These nodes are marked as 'custom' for combining purposes only.
359 // We know that they are legal. See LowerAdd in ISelLowering.
360 return 1 * LT.first;
361 }
362}
363
Chandler Carruth93205eb2015-08-05 18:08:10 +0000364int AArch64TTIImpl::getAddressComputationCost(Type *Ty, bool IsComplex) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000365 // Address computations in vectorized code with non-consecutive addresses will
366 // likely result in more instructions compared to scalar code where the
367 // computation can more often be merged into the index mode. The resulting
368 // extra micro-ops can significantly decrease throughput.
369 unsigned NumVectorInstToHideOverhead = 10;
370
371 if (Ty->isVectorTy() && IsComplex)
372 return NumVectorInstToHideOverhead;
373
374 // In many cases the address computation is not merged into the instruction
375 // addressing mode.
376 return 1;
377}
378
Chandler Carruth93205eb2015-08-05 18:08:10 +0000379int AArch64TTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy,
380 Type *CondTy) {
Tim Northover3b0846e2014-05-24 12:50:23 +0000381
382 int ISD = TLI->InstructionOpcodeToISD(Opcode);
383 // We don't lower vector selects well that are wider than the register width.
384 if (ValTy->isVectorTy() && ISD == ISD::SELECT) {
385 // We would need this many instructions to hide the scalarization happening.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000386 const int AmortizationCost = 20;
Tim Northover3b0846e2014-05-24 12:50:23 +0000387 static const TypeConversionCostTblEntry<MVT::SimpleValueType>
388 VectorSelectTbl[] = {
389 { ISD::SELECT, MVT::v16i1, MVT::v16i16, 16 * AmortizationCost },
390 { ISD::SELECT, MVT::v8i1, MVT::v8i32, 8 * AmortizationCost },
391 { ISD::SELECT, MVT::v16i1, MVT::v16i32, 16 * AmortizationCost },
392 { ISD::SELECT, MVT::v4i1, MVT::v4i64, 4 * AmortizationCost },
393 { ISD::SELECT, MVT::v8i1, MVT::v8i64, 8 * AmortizationCost },
394 { ISD::SELECT, MVT::v16i1, MVT::v16i64, 16 * AmortizationCost }
395 };
396
Mehdi Amini44ede332015-07-09 02:09:04 +0000397 EVT SelCondTy = TLI->getValueType(DL, CondTy);
398 EVT SelValTy = TLI->getValueType(DL, ValTy);
Tim Northover3b0846e2014-05-24 12:50:23 +0000399 if (SelCondTy.isSimple() && SelValTy.isSimple()) {
400 int Idx =
401 ConvertCostTableLookup(VectorSelectTbl, ISD, SelCondTy.getSimpleVT(),
402 SelValTy.getSimpleVT());
403 if (Idx != -1)
404 return VectorSelectTbl[Idx].Cost;
405 }
406 }
Chandler Carruth705b1852015-01-31 03:43:40 +0000407 return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy);
Tim Northover3b0846e2014-05-24 12:50:23 +0000408}
409
Chandler Carruth93205eb2015-08-05 18:08:10 +0000410int AArch64TTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src,
411 unsigned Alignment, unsigned AddressSpace) {
412 std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src);
Tim Northover3b0846e2014-05-24 12:50:23 +0000413
414 if (Opcode == Instruction::Store && Src->isVectorTy() && Alignment != 16 &&
415 Src->getVectorElementType()->isIntegerTy(64)) {
416 // Unaligned stores are extremely inefficient. We don't split
417 // unaligned v2i64 stores because the negative impact that has shown in
418 // practice on inlined memcpy code.
419 // We make v2i64 stores expensive so that we will only vectorize if there
420 // are 6 other instructions getting vectorized.
Chandler Carruth93205eb2015-08-05 18:08:10 +0000421 int AmortizationCost = 6;
Tim Northover3b0846e2014-05-24 12:50:23 +0000422
423 return LT.first * 2 * AmortizationCost;
424 }
425
426 if (Src->isVectorTy() && Src->getVectorElementType()->isIntegerTy(8) &&
427 Src->getVectorNumElements() < 8) {
428 // We scalarize the loads/stores because there is not v.4b register and we
429 // have to promote the elements to v.4h.
430 unsigned NumVecElts = Src->getVectorNumElements();
431 unsigned NumVectorizableInstsToAmortize = NumVecElts * 2;
432 // We generate 2 instructions per vector element.
433 return NumVectorizableInstsToAmortize * NumVecElts * 2;
434 }
435
436 return LT.first;
437}
James Molloy2b8933c2014-08-05 12:30:34 +0000438
Chandler Carruth93205eb2015-08-05 18:08:10 +0000439int AArch64TTIImpl::getInterleavedMemoryOpCost(unsigned Opcode, Type *VecTy,
440 unsigned Factor,
441 ArrayRef<unsigned> Indices,
442 unsigned Alignment,
443 unsigned AddressSpace) {
Hao Liu7ec8ee32015-06-26 02:32:07 +0000444 assert(Factor >= 2 && "Invalid interleave factor");
445 assert(isa<VectorType>(VecTy) && "Expect a vector type");
446
447 if (Factor <= TLI->getMaxSupportedInterleaveFactor()) {
448 unsigned NumElts = VecTy->getVectorNumElements();
449 Type *SubVecTy = VectorType::get(VecTy->getScalarType(), NumElts / Factor);
Silviu Baranga7581d222015-07-27 14:39:34 +0000450 unsigned SubVecSize = DL.getTypeAllocSizeInBits(SubVecTy);
Hao Liu7ec8ee32015-06-26 02:32:07 +0000451
452 // ldN/stN only support legal vector types of size 64 or 128 in bits.
453 if (NumElts % Factor == 0 && (SubVecSize == 64 || SubVecSize == 128))
454 return Factor;
455 }
456
457 return BaseT::getInterleavedMemoryOpCost(Opcode, VecTy, Factor, Indices,
458 Alignment, AddressSpace);
459}
460
Chandler Carruth93205eb2015-08-05 18:08:10 +0000461int AArch64TTIImpl::getCostOfKeepingLiveOverCall(ArrayRef<Type *> Tys) {
462 int Cost = 0;
James Molloy2b8933c2014-08-05 12:30:34 +0000463 for (auto *I : Tys) {
464 if (!I->isVectorTy())
465 continue;
466 if (I->getScalarSizeInBits() * I->getVectorNumElements() == 128)
467 Cost += getMemoryOpCost(Instruction::Store, I, 128, 0) +
468 getMemoryOpCost(Instruction::Load, I, 128, 0);
469 }
470 return Cost;
471}
James Molloya88896b2014-08-21 00:02:51 +0000472
Wei Mi062c7442015-05-06 17:12:25 +0000473unsigned AArch64TTIImpl::getMaxInterleaveFactor(unsigned VF) {
Gerolf Hoflehner7b0abb82014-09-10 20:31:57 +0000474 if (ST->isCortexA57())
James Molloya88896b2014-08-21 00:02:51 +0000475 return 4;
476 return 2;
477}
Kevin Qin72a799a2014-10-09 10:13:27 +0000478
Chandler Carruthab5cb362015-02-01 14:31:23 +0000479void AArch64TTIImpl::getUnrollingPreferences(Loop *L,
Chandler Carruth705b1852015-01-31 03:43:40 +0000480 TTI::UnrollingPreferences &UP) {
Kevin Qinaef68412015-03-09 06:14:28 +0000481 // Enable partial unrolling and runtime unrolling.
482 BaseT::getUnrollingPreferences(L, UP);
483
484 // For inner loop, it is more likely to be a hot one, and the runtime check
485 // can be promoted out from LICM pass, so the overhead is less, let's try
486 // a larger threshold to unroll more loops.
487 if (L->getLoopDepth() > 1)
488 UP.PartialThreshold *= 2;
489
Kevin Qin72a799a2014-10-09 10:13:27 +0000490 // Disable partial & runtime unrolling on -Os.
491 UP.PartialOptSizeThreshold = 0;
492}
Chad Rosierf9327d62015-01-26 22:51:15 +0000493
Chandler Carruth705b1852015-01-31 03:43:40 +0000494Value *AArch64TTIImpl::getOrCreateResultFromMemIntrinsic(IntrinsicInst *Inst,
495 Type *ExpectedType) {
Chad Rosierf9327d62015-01-26 22:51:15 +0000496 switch (Inst->getIntrinsicID()) {
497 default:
498 return nullptr;
499 case Intrinsic::aarch64_neon_st2:
500 case Intrinsic::aarch64_neon_st3:
501 case Intrinsic::aarch64_neon_st4: {
502 // Create a struct type
503 StructType *ST = dyn_cast<StructType>(ExpectedType);
504 if (!ST)
505 return nullptr;
506 unsigned NumElts = Inst->getNumArgOperands() - 1;
507 if (ST->getNumElements() != NumElts)
508 return nullptr;
509 for (unsigned i = 0, e = NumElts; i != e; ++i) {
510 if (Inst->getArgOperand(i)->getType() != ST->getElementType(i))
511 return nullptr;
512 }
513 Value *Res = UndefValue::get(ExpectedType);
514 IRBuilder<> Builder(Inst);
515 for (unsigned i = 0, e = NumElts; i != e; ++i) {
516 Value *L = Inst->getArgOperand(i);
517 Res = Builder.CreateInsertValue(Res, L, i);
518 }
519 return Res;
520 }
521 case Intrinsic::aarch64_neon_ld2:
522 case Intrinsic::aarch64_neon_ld3:
523 case Intrinsic::aarch64_neon_ld4:
524 if (Inst->getType() == ExpectedType)
525 return Inst;
526 return nullptr;
527 }
528}
529
Chandler Carruth705b1852015-01-31 03:43:40 +0000530bool AArch64TTIImpl::getTgtMemIntrinsic(IntrinsicInst *Inst,
531 MemIntrinsicInfo &Info) {
Chad Rosierf9327d62015-01-26 22:51:15 +0000532 switch (Inst->getIntrinsicID()) {
533 default:
534 break;
535 case Intrinsic::aarch64_neon_ld2:
536 case Intrinsic::aarch64_neon_ld3:
537 case Intrinsic::aarch64_neon_ld4:
538 Info.ReadMem = true;
539 Info.WriteMem = false;
540 Info.Vol = false;
541 Info.NumMemRefs = 1;
542 Info.PtrVal = Inst->getArgOperand(0);
543 break;
544 case Intrinsic::aarch64_neon_st2:
545 case Intrinsic::aarch64_neon_st3:
546 case Intrinsic::aarch64_neon_st4:
547 Info.ReadMem = false;
548 Info.WriteMem = true;
549 Info.Vol = false;
550 Info.NumMemRefs = 1;
551 Info.PtrVal = Inst->getArgOperand(Inst->getNumArgOperands() - 1);
552 break;
553 }
554
555 switch (Inst->getIntrinsicID()) {
556 default:
557 return false;
558 case Intrinsic::aarch64_neon_ld2:
559 case Intrinsic::aarch64_neon_st2:
560 Info.MatchingId = VECTOR_LDST_TWO_ELEMENTS;
561 break;
562 case Intrinsic::aarch64_neon_ld3:
563 case Intrinsic::aarch64_neon_st3:
564 Info.MatchingId = VECTOR_LDST_THREE_ELEMENTS;
565 break;
566 case Intrinsic::aarch64_neon_ld4:
567 case Intrinsic::aarch64_neon_st4:
568 Info.MatchingId = VECTOR_LDST_FOUR_ELEMENTS;
569 break;
570 }
571 return true;
572}