[PowerPC] Include the permutation cost for unaligned vector loads
[oota-llvm.git] / lib / Target / PowerPC / PPCTargetTransformInfo.cpp
1 //===-- PPCTargetTransformInfo.cpp - PPC specific TTI ---------------------===//
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
10 #include "PPCTargetTransformInfo.h"
11 #include "llvm/Analysis/TargetTransformInfo.h"
12 #include "llvm/CodeGen/BasicTTIImpl.h"
13 #include "llvm/Support/CommandLine.h"
14 #include "llvm/Support/Debug.h"
15 #include "llvm/Target/CostTable.h"
16 #include "llvm/Target/TargetLowering.h"
17 using namespace llvm;
18
19 #define DEBUG_TYPE "ppctti"
20
21 static cl::opt<bool> DisablePPCConstHoist("disable-ppc-constant-hoisting",
22 cl::desc("disable constant hoisting on PPC"), cl::init(false), cl::Hidden);
23
24 //===----------------------------------------------------------------------===//
25 //
26 // PPC cost model.
27 //
28 //===----------------------------------------------------------------------===//
29
30 TargetTransformInfo::PopcntSupportKind
31 PPCTTIImpl::getPopcntSupport(unsigned TyWidth) {
32   assert(isPowerOf2_32(TyWidth) && "Ty width must be power of 2");
33   if (ST->hasPOPCNTD() && TyWidth <= 64)
34     return TTI::PSK_FastHardware;
35   return TTI::PSK_Software;
36 }
37
38 int PPCTTIImpl::getIntImmCost(const APInt &Imm, Type *Ty) {
39   if (DisablePPCConstHoist)
40     return BaseT::getIntImmCost(Imm, Ty);
41
42   assert(Ty->isIntegerTy());
43
44   unsigned BitSize = Ty->getPrimitiveSizeInBits();
45   if (BitSize == 0)
46     return ~0U;
47
48   if (Imm == 0)
49     return TTI::TCC_Free;
50
51   if (Imm.getBitWidth() <= 64) {
52     if (isInt<16>(Imm.getSExtValue()))
53       return TTI::TCC_Basic;
54
55     if (isInt<32>(Imm.getSExtValue())) {
56       // A constant that can be materialized using lis.
57       if ((Imm.getZExtValue() & 0xFFFF) == 0)
58         return TTI::TCC_Basic;
59
60       return 2 * TTI::TCC_Basic;
61     }
62   }
63
64   return 4 * TTI::TCC_Basic;
65 }
66
67 int PPCTTIImpl::getIntImmCost(Intrinsic::ID IID, unsigned Idx, const APInt &Imm,
68                               Type *Ty) {
69   if (DisablePPCConstHoist)
70     return BaseT::getIntImmCost(IID, Idx, Imm, Ty);
71
72   assert(Ty->isIntegerTy());
73
74   unsigned BitSize = Ty->getPrimitiveSizeInBits();
75   if (BitSize == 0)
76     return ~0U;
77
78   switch (IID) {
79   default:
80     return TTI::TCC_Free;
81   case Intrinsic::sadd_with_overflow:
82   case Intrinsic::uadd_with_overflow:
83   case Intrinsic::ssub_with_overflow:
84   case Intrinsic::usub_with_overflow:
85     if ((Idx == 1) && Imm.getBitWidth() <= 64 && isInt<16>(Imm.getSExtValue()))
86       return TTI::TCC_Free;
87     break;
88   case Intrinsic::experimental_stackmap:
89     if ((Idx < 2) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
90       return TTI::TCC_Free;
91     break;
92   case Intrinsic::experimental_patchpoint_void:
93   case Intrinsic::experimental_patchpoint_i64:
94     if ((Idx < 4) || (Imm.getBitWidth() <= 64 && isInt<64>(Imm.getSExtValue())))
95       return TTI::TCC_Free;
96     break;
97   }
98   return PPCTTIImpl::getIntImmCost(Imm, Ty);
99 }
100
101 int PPCTTIImpl::getIntImmCost(unsigned Opcode, unsigned Idx, const APInt &Imm,
102                               Type *Ty) {
103   if (DisablePPCConstHoist)
104     return BaseT::getIntImmCost(Opcode, Idx, Imm, Ty);
105
106   assert(Ty->isIntegerTy());
107
108   unsigned BitSize = Ty->getPrimitiveSizeInBits();
109   if (BitSize == 0)
110     return ~0U;
111
112   unsigned ImmIdx = ~0U;
113   bool ShiftedFree = false, RunFree = false, UnsignedFree = false,
114        ZeroFree = false;
115   switch (Opcode) {
116   default:
117     return TTI::TCC_Free;
118   case Instruction::GetElementPtr:
119     // Always hoist the base address of a GetElementPtr. This prevents the
120     // creation of new constants for every base constant that gets constant
121     // folded with the offset.
122     if (Idx == 0)
123       return 2 * TTI::TCC_Basic;
124     return TTI::TCC_Free;
125   case Instruction::And:
126     RunFree = true; // (for the rotate-and-mask instructions)
127     // Fallthrough...
128   case Instruction::Add:
129   case Instruction::Or:
130   case Instruction::Xor:
131     ShiftedFree = true;
132     // Fallthrough...
133   case Instruction::Sub:
134   case Instruction::Mul:
135   case Instruction::Shl:
136   case Instruction::LShr:
137   case Instruction::AShr:
138     ImmIdx = 1;
139     break;
140   case Instruction::ICmp:
141     UnsignedFree = true;
142     ImmIdx = 1;
143     // Fallthrough... (zero comparisons can use record-form instructions)
144   case Instruction::Select:
145     ZeroFree = true;
146     break;
147   case Instruction::PHI:
148   case Instruction::Call:
149   case Instruction::Ret:
150   case Instruction::Load:
151   case Instruction::Store:
152     break;
153   }
154
155   if (ZeroFree && Imm == 0)
156     return TTI::TCC_Free;
157
158   if (Idx == ImmIdx && Imm.getBitWidth() <= 64) {
159     if (isInt<16>(Imm.getSExtValue()))
160       return TTI::TCC_Free;
161
162     if (RunFree) {
163       if (Imm.getBitWidth() <= 32 &&
164           (isShiftedMask_32(Imm.getZExtValue()) ||
165            isShiftedMask_32(~Imm.getZExtValue())))
166         return TTI::TCC_Free;
167
168       if (ST->isPPC64() &&
169           (isShiftedMask_64(Imm.getZExtValue()) ||
170            isShiftedMask_64(~Imm.getZExtValue())))
171         return TTI::TCC_Free;
172     }
173
174     if (UnsignedFree && isUInt<16>(Imm.getZExtValue()))
175       return TTI::TCC_Free;
176
177     if (ShiftedFree && (Imm.getZExtValue() & 0xFFFF) == 0)
178       return TTI::TCC_Free;
179   }
180
181   return PPCTTIImpl::getIntImmCost(Imm, Ty);
182 }
183
184 void PPCTTIImpl::getUnrollingPreferences(Loop *L,
185                                          TTI::UnrollingPreferences &UP) {
186   if (ST->getDarwinDirective() == PPC::DIR_A2) {
187     // The A2 is in-order with a deep pipeline, and concatenation unrolling
188     // helps expose latency-hiding opportunities to the instruction scheduler.
189     UP.Partial = UP.Runtime = true;
190
191     // We unroll a lot on the A2 (hundreds of instructions), and the benefits
192     // often outweigh the cost of a division to compute the trip count.
193     UP.AllowExpensiveTripCount = true;
194   }
195
196   BaseT::getUnrollingPreferences(L, UP);
197 }
198
199 bool PPCTTIImpl::enableAggressiveInterleaving(bool LoopHasReductions) {
200   return LoopHasReductions;
201 }
202
203 unsigned PPCTTIImpl::getNumberOfRegisters(bool Vector) {
204   if (Vector && !ST->hasAltivec() && !ST->hasQPX())
205     return 0;
206   return ST->hasVSX() ? 64 : 32;
207 }
208
209 unsigned PPCTTIImpl::getRegisterBitWidth(bool Vector) {
210   if (Vector) {
211     if (ST->hasQPX()) return 256;
212     if (ST->hasAltivec()) return 128;
213     return 0;
214   }
215
216   if (ST->isPPC64())
217     return 64;
218   return 32;
219
220 }
221
222 unsigned PPCTTIImpl::getMaxInterleaveFactor(unsigned VF) {
223   unsigned Directive = ST->getDarwinDirective();
224   // The 440 has no SIMD support, but floating-point instructions
225   // have a 5-cycle latency, so unroll by 5x for latency hiding.
226   if (Directive == PPC::DIR_440)
227     return 5;
228
229   // The A2 has no SIMD support, but floating-point instructions
230   // have a 6-cycle latency, so unroll by 6x for latency hiding.
231   if (Directive == PPC::DIR_A2)
232     return 6;
233
234   // FIXME: For lack of any better information, do no harm...
235   if (Directive == PPC::DIR_E500mc || Directive == PPC::DIR_E5500)
236     return 1;
237
238   // For P7 and P8, floating-point instructions have a 6-cycle latency and
239   // there are two execution units, so unroll by 12x for latency hiding.
240   if (Directive == PPC::DIR_PWR7 ||
241       Directive == PPC::DIR_PWR8)
242     return 12;
243
244   // For most things, modern systems have two execution units (and
245   // out-of-order execution).
246   return 2;
247 }
248
249 int PPCTTIImpl::getArithmeticInstrCost(
250     unsigned Opcode, Type *Ty, TTI::OperandValueKind Op1Info,
251     TTI::OperandValueKind Op2Info, TTI::OperandValueProperties Opd1PropInfo,
252     TTI::OperandValueProperties Opd2PropInfo) {
253   assert(TLI->InstructionOpcodeToISD(Opcode) && "Invalid opcode");
254
255   // Fallback to the default implementation.
256   return BaseT::getArithmeticInstrCost(Opcode, Ty, Op1Info, Op2Info,
257                                        Opd1PropInfo, Opd2PropInfo);
258 }
259
260 int PPCTTIImpl::getShuffleCost(TTI::ShuffleKind Kind, Type *Tp, int Index,
261                                Type *SubTp) {
262   return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
263 }
264
265 int PPCTTIImpl::getCastInstrCost(unsigned Opcode, Type *Dst, Type *Src) {
266   assert(TLI->InstructionOpcodeToISD(Opcode) && "Invalid opcode");
267
268   return BaseT::getCastInstrCost(Opcode, Dst, Src);
269 }
270
271 int PPCTTIImpl::getCmpSelInstrCost(unsigned Opcode, Type *ValTy, Type *CondTy) {
272   return BaseT::getCmpSelInstrCost(Opcode, ValTy, CondTy);
273 }
274
275 int PPCTTIImpl::getVectorInstrCost(unsigned Opcode, Type *Val, unsigned Index) {
276   assert(Val->isVectorTy() && "This must be a vector type");
277
278   int ISD = TLI->InstructionOpcodeToISD(Opcode);
279   assert(ISD && "Invalid opcode");
280
281   if (ST->hasVSX() && Val->getScalarType()->isDoubleTy()) {
282     // Double-precision scalars are already located in index #0.
283     if (Index == 0)
284       return 0;
285
286     return BaseT::getVectorInstrCost(Opcode, Val, Index);
287   } else if (ST->hasQPX() && Val->getScalarType()->isFloatingPointTy()) {
288     // Floating point scalars are already located in index #0.
289     if (Index == 0)
290       return 0;
291
292     return BaseT::getVectorInstrCost(Opcode, Val, Index);
293   }
294
295   // Estimated cost of a load-hit-store delay.  This was obtained
296   // experimentally as a minimum needed to prevent unprofitable
297   // vectorization for the paq8p benchmark.  It may need to be
298   // raised further if other unprofitable cases remain.
299   unsigned LHSPenalty = 2;
300   if (ISD == ISD::INSERT_VECTOR_ELT)
301     LHSPenalty += 7;
302
303   // Vector element insert/extract with Altivec is very expensive,
304   // because they require store and reload with the attendant
305   // processor stall for load-hit-store.  Until VSX is available,
306   // these need to be estimated as very costly.
307   if (ISD == ISD::EXTRACT_VECTOR_ELT ||
308       ISD == ISD::INSERT_VECTOR_ELT)
309     return LHSPenalty + BaseT::getVectorInstrCost(Opcode, Val, Index);
310
311   return BaseT::getVectorInstrCost(Opcode, Val, Index);
312 }
313
314 int PPCTTIImpl::getMemoryOpCost(unsigned Opcode, Type *Src, unsigned Alignment,
315                                 unsigned AddressSpace) {
316   // Legalize the type.
317   std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Src);
318   assert((Opcode == Instruction::Load || Opcode == Instruction::Store) &&
319          "Invalid Opcode");
320
321   int Cost = BaseT::getMemoryOpCost(Opcode, Src, Alignment, AddressSpace);
322
323   // Aligned loads and stores are easy.
324   unsigned SrcBytes = LT.second.getStoreSize();
325   if (!SrcBytes || !Alignment || Alignment >= SrcBytes)
326     return Cost;
327
328   bool IsAltivecType = ST->hasAltivec() &&
329                        (LT.second == MVT::v16i8 || LT.second == MVT::v8i16 ||
330                         LT.second == MVT::v4i32 || LT.second == MVT::v4f32);
331   bool IsVSXType = ST->hasVSX() &&
332                    (LT.second == MVT::v2f64 || LT.second == MVT::v2i64);
333   bool IsQPXType = ST->hasQPX() &&
334                    (LT.second == MVT::v4f64 || LT.second == MVT::v4f32);
335
336   // If we can use the permutation-based load sequence, then this is also
337   // relatively cheap (not counting loop-invariant instructions): one load plus
338   // one permute (the last load in a series has extra cost, but we're
339   // neglecting that here). Note that on the P7, we should do unaligned loads
340   // for Altivec types using the VSX instructions, but that's more expensive
341   // than using the permutation-based load sequence. On the P8, that's no
342   // longer true.
343   if (Opcode == Instruction::Load &&
344       ((!ST->hasP8Vector() && IsAltivecType) || IsQPXType) &&
345       Alignment >= LT.second.getScalarType().getStoreSize())
346     return Cost + LT.first; // Add the cost of the permutations.
347
348   // For VSX, we can do unaligned loads and stores on Altivec/VSX types. On the
349   // P7, unaligned vector loads are more expensive than the permutation-based
350   // load sequence, so that might be used instead, but regardless, the net cost
351   // is about the same (not counting loop-invariant instructions).
352   if (IsVSXType || (ST->hasVSX() && IsAltivecType))
353     return Cost;
354
355   // PPC in general does not support unaligned loads and stores. They'll need
356   // to be decomposed based on the alignment factor.
357
358   // Add the cost of each scalar load or store.
359   Cost += LT.first*(SrcBytes/Alignment-1);
360
361   // For a vector type, there is also scalarization overhead (only for
362   // stores, loads are expanded using the vector-load + permutation sequence,
363   // which is much less expensive).
364   if (Src->isVectorTy() && Opcode == Instruction::Store)
365     for (int i = 0, e = Src->getVectorNumElements(); i < e; ++i)
366       Cost += getVectorInstrCost(Instruction::ExtractElement, Src, i);
367
368   return Cost;
369 }
370