Chandler Carruth | 93dcdc4 | 2015-01-31 11:17:59 +0000 | [diff] [blame] | 1 | //===-- AArch64TargetTransformInfo.cpp - AArch64 specific TTI -------------===// |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 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 | //===----------------------------------------------------------------------===// |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 9 | |
Chandler Carruth | 93dcdc4 | 2015-01-31 11:17:59 +0000 | [diff] [blame] | 10 | #include "AArch64TargetTransformInfo.h" |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 11 | #include "MCTargetDesc/AArch64AddressingModes.h" |
| 12 | #include "llvm/Analysis/TargetTransformInfo.h" |
Kevin Qin | aef6841 | 2015-03-09 06:14:28 +0000 | [diff] [blame] | 13 | #include "llvm/Analysis/LoopInfo.h" |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 14 | #include "llvm/CodeGen/BasicTTIImpl.h" |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 15 | #include "llvm/Support/Debug.h" |
| 16 | #include "llvm/Target/CostTable.h" |
| 17 | #include "llvm/Target/TargetLowering.h" |
| 18 | #include <algorithm> |
| 19 | using namespace llvm; |
| 20 | |
| 21 | #define DEBUG_TYPE "aarch64tti" |
| 22 | |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 23 | /// \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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 26 | int AArch64TTIImpl::getIntImmCost(int64_t Val) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 27 | // 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 40 | int AArch64TTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 41 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 54 | int Cost = 0; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 55 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 61 | return std::max(1, Cost); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 62 | } |
| 63 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 64 | int AArch64TTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx, |
| 65 | const APInt &Imm, Type *Ty) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 66 | 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 Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 72 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 73 | |
| 74 | unsigned ImmIdx = ~0U; |
| 75 | switch (Opcode) { |
| 76 | default: |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 77 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 78 | case Instruction::GetElementPtr: |
| 79 | // Always hoist the base address of a GetElementPtr. |
| 80 | if (Idx == 0) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 81 | return 2 * TTI::TCC_Basic; |
| 82 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 83 | 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 Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 104 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 105 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 121 | int NumConstants = (BitSize + 63) / 64; |
| 122 | int Cost = AArch64TTIImpl::getIntImmCost(Imm, Ty); |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 123 | return (Cost <= NumConstants * TTI::TCC_Basic) |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 124 | ? static_cast<int>(TTI::TCC_Free) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 125 | : Cost; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 126 | } |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 127 | return AArch64TTIImpl::getIntImmCost(Imm, Ty); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 128 | } |
| 129 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 130 | int AArch64TTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx, |
| 131 | const APInt &Imm, Type *Ty) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 132 | 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 Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 138 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 139 | |
| 140 | switch (IID) { |
| 141 | default: |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 142 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 143 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 150 | int NumConstants = (BitSize + 63) / 64; |
| 151 | int Cost = AArch64TTIImpl::getIntImmCost(Imm, Ty); |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 152 | return (Cost <= NumConstants * TTI::TCC_Basic) |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 153 | ? static_cast<int>(TTI::TCC_Free) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 154 | : Cost; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 155 | } |
| 156 | break; |
| 157 | case Intrinsic::experimental_stackmap: |
| 158 | if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue()))) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 159 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 160 | 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 Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 164 | return TTI::TCC_Free; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 165 | break; |
| 166 | } |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 167 | return AArch64TTIImpl::getIntImmCost(Imm, Ty); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 168 | } |
| 169 | |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 170 | TargetTransformInfo::PopcntSupportKind |
| 171 | AArch64TTIImpl::getPopcntSupport(unsigned TyWidth) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 172 | assert(isPowerOf2_32(TyWidth) && "Ty width must be power of 2"); |
| 173 | if (TyWidth == 32 || TyWidth == 64) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 174 | return TTI::PSK_FastHardware; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 175 | // TODO: AArch64TargetLowering::LowerCTPOP() supports 128bit popcount. |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 176 | return TTI::PSK_Software; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 177 | } |
| 178 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 179 | int AArch64TTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 180 | int ISD = TLI->InstructionOpcodeToISD(Opcode); |
| 181 | assert(ISD && "Invalid opcode"); |
| 182 | |
Mehdi Amini | 44ede33 | 2015-07-09 02:09:04 +0000 | [diff] [blame] | 183 | EVT SrcTy = TLI->getValueType(DL, Src); |
| 184 | EVT DstTy = TLI->getValueType(DL, Dst); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 185 | |
| 186 | if (!SrcTy.isSimple() || !DstTy.isSimple()) |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 187 | return BaseT::getCastInstrCost(Opcode, Dst, Src); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 188 | |
Craig Topper | 7bf52c9 | 2015-10-25 00:27:14 +0000 | [diff] [blame] | 189 | static const TypeConversionCostTblEntry<MVT::SimpleValueType> |
| 190 | ConversionTbl[] = { |
Silviu Baranga | b322aa6 | 2015-08-17 16:05:09 +0000 | [diff] [blame] | 191 | { ISD::SIGN_EXTEND, MVT::v4i32, MVT::v4i16, 0 }, |
| 192 | { ISD::ZERO_EXTEND, MVT::v4i32, MVT::v4i16, 0 }, |
| 193 | { ISD::SIGN_EXTEND, MVT::v2i64, MVT::v2i32, 1 }, |
| 194 | { ISD::ZERO_EXTEND, MVT::v2i64, MVT::v2i32, 1 }, |
| 195 | { ISD::TRUNCATE, MVT::v4i32, MVT::v4i64, 0 }, |
| 196 | { ISD::TRUNCATE, MVT::v4i16, MVT::v4i32, 1 }, |
| 197 | |
| 198 | // The number of shll instructions for the extension. |
| 199 | { ISD::SIGN_EXTEND, MVT::v4i64, MVT::v4i16, 3 }, |
| 200 | { ISD::ZERO_EXTEND, MVT::v4i64, MVT::v4i16, 3 }, |
| 201 | { ISD::SIGN_EXTEND, MVT::v8i32, MVT::v8i8, 3 }, |
| 202 | { ISD::ZERO_EXTEND, MVT::v8i32, MVT::v8i8, 3 }, |
| 203 | { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i8, 7 }, |
| 204 | { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i8, 7 }, |
| 205 | { ISD::SIGN_EXTEND, MVT::v8i64, MVT::v8i16, 6 }, |
| 206 | { ISD::ZERO_EXTEND, MVT::v8i64, MVT::v8i16, 6 }, |
| 207 | { ISD::SIGN_EXTEND, MVT::v16i32, MVT::v16i8, 6 }, |
| 208 | { ISD::ZERO_EXTEND, MVT::v16i32, MVT::v16i8, 6 }, |
| 209 | |
| 210 | { ISD::TRUNCATE, MVT::v16i8, MVT::v16i32, 6 }, |
| 211 | { ISD::TRUNCATE, MVT::v8i8, MVT::v8i32, 3 }, |
| 212 | |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 213 | // LowerVectorINT_TO_FP: |
| 214 | { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i32, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 215 | { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 216 | { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 }, |
| 217 | { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i32, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 218 | { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i32, 1 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 219 | { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i64, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 220 | |
| 221 | // Complex: to v2f32 |
| 222 | { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i8, 3 }, |
| 223 | { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i16, 3 }, |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 224 | { ISD::SINT_TO_FP, MVT::v2f32, MVT::v2i64, 2 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 225 | { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i8, 3 }, |
| 226 | { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i16, 3 }, |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 227 | { ISD::UINT_TO_FP, MVT::v2f32, MVT::v2i64, 2 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 228 | |
| 229 | // Complex: to v4f32 |
| 230 | { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i8, 4 }, |
| 231 | { ISD::SINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 }, |
| 232 | { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i8, 3 }, |
| 233 | { ISD::UINT_TO_FP, MVT::v4f32, MVT::v4i16, 2 }, |
| 234 | |
Silviu Baranga | b322aa6 | 2015-08-17 16:05:09 +0000 | [diff] [blame] | 235 | // Complex: to v8f32 |
| 236 | { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i8, 10 }, |
| 237 | { ISD::SINT_TO_FP, MVT::v8f32, MVT::v8i16, 4 }, |
| 238 | { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i8, 10 }, |
| 239 | { ISD::UINT_TO_FP, MVT::v8f32, MVT::v8i16, 4 }, |
| 240 | |
| 241 | // Complex: to v16f32 |
| 242 | { ISD::SINT_TO_FP, MVT::v16f32, MVT::v16i8, 21 }, |
| 243 | { ISD::UINT_TO_FP, MVT::v16f32, MVT::v16i8, 21 }, |
| 244 | |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 245 | // Complex: to v2f64 |
| 246 | { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i8, 4 }, |
| 247 | { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i16, 4 }, |
| 248 | { ISD::SINT_TO_FP, MVT::v2f64, MVT::v2i32, 2 }, |
| 249 | { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i8, 4 }, |
| 250 | { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i16, 4 }, |
| 251 | { ISD::UINT_TO_FP, MVT::v2f64, MVT::v2i32, 2 }, |
| 252 | |
| 253 | |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 254 | // LowerVectorFP_TO_INT |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 255 | { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f32, 1 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 256 | { ISD::FP_TO_SINT, MVT::v4i32, MVT::v4f32, 1 }, |
| 257 | { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f64, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 258 | { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f32, 1 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 259 | { ISD::FP_TO_UINT, MVT::v4i32, MVT::v4f32, 1 }, |
| 260 | { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f64, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 261 | |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 262 | // Complex, from v2f32: legal type is v2i32 (no cost) or v2i64 (1 ext). |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 263 | { ISD::FP_TO_SINT, MVT::v2i64, MVT::v2f32, 2 }, |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 264 | { ISD::FP_TO_SINT, MVT::v2i16, MVT::v2f32, 1 }, |
| 265 | { ISD::FP_TO_SINT, MVT::v2i8, MVT::v2f32, 1 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 266 | { ISD::FP_TO_UINT, MVT::v2i64, MVT::v2f32, 2 }, |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 267 | { ISD::FP_TO_UINT, MVT::v2i16, MVT::v2f32, 1 }, |
| 268 | { ISD::FP_TO_UINT, MVT::v2i8, MVT::v2f32, 1 }, |
| 269 | |
| 270 | // Complex, from v4f32: legal type is v4i16, 1 narrowing => ~2 |
| 271 | { ISD::FP_TO_SINT, MVT::v4i16, MVT::v4f32, 2 }, |
| 272 | { ISD::FP_TO_SINT, MVT::v4i8, MVT::v4f32, 2 }, |
Tim Northover | ef0d760 | 2014-06-15 09:27:06 +0000 | [diff] [blame] | 273 | { ISD::FP_TO_UINT, MVT::v4i16, MVT::v4f32, 2 }, |
Tim Northover | dbecc3b | 2014-06-15 09:27:15 +0000 | [diff] [blame] | 274 | { ISD::FP_TO_UINT, MVT::v4i8, MVT::v4f32, 2 }, |
| 275 | |
| 276 | // Complex, from v2f64: legal type is v2i32, 1 narrowing => ~2. |
| 277 | { ISD::FP_TO_SINT, MVT::v2i32, MVT::v2f64, 2 }, |
| 278 | { ISD::FP_TO_SINT, MVT::v2i16, MVT::v2f64, 2 }, |
| 279 | { ISD::FP_TO_SINT, MVT::v2i8, MVT::v2f64, 2 }, |
| 280 | { ISD::FP_TO_UINT, MVT::v2i32, MVT::v2f64, 2 }, |
| 281 | { ISD::FP_TO_UINT, MVT::v2i16, MVT::v2f64, 2 }, |
| 282 | { ISD::FP_TO_UINT, MVT::v2i8, MVT::v2f64, 2 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 283 | }; |
| 284 | |
Craig Topper | ee0c859 | 2015-10-27 04:14:24 +0000 | [diff] [blame^] | 285 | if (const auto *Entry = ConvertCostTableLookup(ConversionTbl, ISD, |
| 286 | DstTy.getSimpleVT(), |
| 287 | SrcTy.getSimpleVT())) |
| 288 | return Entry->Cost; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 289 | |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 290 | return BaseT::getCastInstrCost(Opcode, Dst, Src); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 291 | } |
| 292 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 293 | int AArch64TTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val, |
| 294 | unsigned Index) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 295 | assert(Val->isVectorTy() && "This must be a vector type"); |
| 296 | |
| 297 | if (Index != -1U) { |
| 298 | // Legalize the type. |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 299 | std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Val); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 300 | |
| 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 Baranga | b322aa6 | 2015-08-17 16:05:09 +0000 | [diff] [blame] | 315 | return 3; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 316 | } |
| 317 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 318 | int AArch64TTIImpl::getArithmeticInstrCost( |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 319 | unsigned Opcode, Type *Ty, TTI::OperandValueKind Opd1Info, |
| 320 | TTI::OperandValueKind Opd2Info, TTI::OperandValueProperties Opd1PropInfo, |
| 321 | TTI::OperandValueProperties Opd2PropInfo) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 322 | // Legalize the type. |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 323 | std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Ty); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 324 | |
| 325 | int ISD = TLI->InstructionOpcodeToISD(Opcode); |
| 326 | |
Chad Rosier | 70d54ac | 2014-09-29 13:59:31 +0000 | [diff] [blame] | 327 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 334 | int Cost = getArithmeticInstrCost(Instruction::Add, Ty, Opd1Info, Opd2Info, |
| 335 | TargetTransformInfo::OP_None, |
| 336 | TargetTransformInfo::OP_None); |
Chad Rosier | 70d54ac | 2014-09-29 13:59:31 +0000 | [diff] [blame] | 337 | 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 Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 349 | switch (ISD) { |
| 350 | default: |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 351 | return BaseT::getArithmeticInstrCost(Opcode, Ty, Opd1Info, Opd2Info, |
| 352 | Opd1PropInfo, Opd2PropInfo); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 353 | 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 364 | int AArch64TTIImpl::getAddressComputationCost(Type *Ty, bool IsComplex) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 365 | // 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 379 | int AArch64TTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy, |
| 380 | Type *CondTy) { |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 381 | |
| 382 | int ISD = TLI->InstructionOpcodeToISD(Opcode); |
Silviu Baranga | a3e27ed | 2015-09-09 15:35:02 +0000 | [diff] [blame] | 383 | // We don't lower some vector selects well that are wider than the register |
| 384 | // width. |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 385 | if (ValTy->isVectorTy() && ISD == ISD::SELECT) { |
| 386 | // We would need this many instructions to hide the scalarization happening. |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 387 | const int AmortizationCost = 20; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 388 | static const TypeConversionCostTblEntry<MVT::SimpleValueType> |
| 389 | VectorSelectTbl[] = { |
Silviu Baranga | a3e27ed | 2015-09-09 15:35:02 +0000 | [diff] [blame] | 390 | { ISD::SELECT, MVT::v16i1, MVT::v16i16, 16 }, |
| 391 | { ISD::SELECT, MVT::v8i1, MVT::v8i32, 8 }, |
| 392 | { ISD::SELECT, MVT::v16i1, MVT::v16i32, 16 }, |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 393 | { ISD::SELECT, MVT::v4i1, MVT::v4i64, 4 * AmortizationCost }, |
| 394 | { ISD::SELECT, MVT::v8i1, MVT::v8i64, 8 * AmortizationCost }, |
| 395 | { ISD::SELECT, MVT::v16i1, MVT::v16i64, 16 * AmortizationCost } |
| 396 | }; |
| 397 | |
Mehdi Amini | 44ede33 | 2015-07-09 02:09:04 +0000 | [diff] [blame] | 398 | EVT SelCondTy = TLI->getValueType(DL, CondTy); |
| 399 | EVT SelValTy = TLI->getValueType(DL, ValTy); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 400 | if (SelCondTy.isSimple() && SelValTy.isSimple()) { |
Craig Topper | ee0c859 | 2015-10-27 04:14:24 +0000 | [diff] [blame^] | 401 | if (const auto *Entry = ConvertCostTableLookup(VectorSelectTbl, ISD, |
| 402 | SelCondTy.getSimpleVT(), |
| 403 | SelValTy.getSimpleVT())) |
| 404 | return Entry->Cost; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 405 | } |
| 406 | } |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 407 | return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 408 | } |
| 409 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 410 | int AArch64TTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src, |
| 411 | unsigned Alignment, unsigned AddressSpace) { |
| 412 | std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src); |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 413 | |
| 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 421 | int AmortizationCost = 6; |
Tim Northover | 3b0846e | 2014-05-24 12:50:23 +0000 | [diff] [blame] | 422 | |
| 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 Molloy | 2b8933c | 2014-08-05 12:30:34 +0000 | [diff] [blame] | 438 | |
Chandler Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 439 | int AArch64TTIImpl::getInterleavedMemoryOpCost(unsigned Opcode, Type *VecTy, |
| 440 | unsigned Factor, |
| 441 | ArrayRef<unsigned> Indices, |
| 442 | unsigned Alignment, |
| 443 | unsigned AddressSpace) { |
Hao Liu | 7ec8ee3 | 2015-06-26 02:32:07 +0000 | [diff] [blame] | 444 | 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 Baranga | 7581d22 | 2015-07-27 14:39:34 +0000 | [diff] [blame] | 450 | unsigned SubVecSize = DL.getTypeAllocSizeInBits(SubVecTy); |
Hao Liu | 7ec8ee3 | 2015-06-26 02:32:07 +0000 | [diff] [blame] | 451 | |
| 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 Carruth | 93205eb | 2015-08-05 18:08:10 +0000 | [diff] [blame] | 461 | int AArch64TTIImpl::getCostOfKeepingLiveOverCall(ArrayRef<Type *> Tys) { |
| 462 | int Cost = 0; |
James Molloy | 2b8933c | 2014-08-05 12:30:34 +0000 | [diff] [blame] | 463 | 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 Molloy | a88896b | 2014-08-21 00:02:51 +0000 | [diff] [blame] | 472 | |
Wei Mi | 062c744 | 2015-05-06 17:12:25 +0000 | [diff] [blame] | 473 | unsigned AArch64TTIImpl::getMaxInterleaveFactor(unsigned VF) { |
Gerolf Hoflehner | 7b0abb8 | 2014-09-10 20:31:57 +0000 | [diff] [blame] | 474 | if (ST->isCortexA57()) |
James Molloy | a88896b | 2014-08-21 00:02:51 +0000 | [diff] [blame] | 475 | return 4; |
| 476 | return 2; |
| 477 | } |
Kevin Qin | 72a799a | 2014-10-09 10:13:27 +0000 | [diff] [blame] | 478 | |
Chandler Carruth | ab5cb36 | 2015-02-01 14:31:23 +0000 | [diff] [blame] | 479 | void AArch64TTIImpl::getUnrollingPreferences(Loop *L, |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 480 | TTI::UnrollingPreferences &UP) { |
Kevin Qin | aef6841 | 2015-03-09 06:14:28 +0000 | [diff] [blame] | 481 | // 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 Qin | 72a799a | 2014-10-09 10:13:27 +0000 | [diff] [blame] | 490 | // Disable partial & runtime unrolling on -Os. |
| 491 | UP.PartialOptSizeThreshold = 0; |
| 492 | } |
Chad Rosier | f9327d6 | 2015-01-26 22:51:15 +0000 | [diff] [blame] | 493 | |
Chandler Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 494 | Value *AArch64TTIImpl::getOrCreateResultFromMemIntrinsic(IntrinsicInst *Inst, |
| 495 | Type *ExpectedType) { |
Chad Rosier | f9327d6 | 2015-01-26 22:51:15 +0000 | [diff] [blame] | 496 | 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 Carruth | 705b185 | 2015-01-31 03:43:40 +0000 | [diff] [blame] | 530 | bool AArch64TTIImpl::getTgtMemIntrinsic(IntrinsicInst *Inst, |
| 531 | MemIntrinsicInfo &Info) { |
Chad Rosier | f9327d6 | 2015-01-26 22:51:15 +0000 | [diff] [blame] | 532 | 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 | } |