1 //===- AMDGPUTargetTransformInfo.cpp - AMDGPU 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 //
10 // \file
11 // This file implements a TargetTransformInfo analysis pass specific to the
12 // AMDGPU target machine. It uses the target's detailed information to provide
13 // more precise answers to certain TTI queries, while letting the target
14 // independent and default TTI implementations handle the rest.
15 //
16 //===----------------------------------------------------------------------===//
17 
18 #include "AMDGPUTargetTransformInfo.h"
19 #include "AMDGPUSubtarget.h"
20 #include "llvm/ADT/STLExtras.h"
21 #include "llvm/Analysis/LoopInfo.h"
22 #include "llvm/Analysis/TargetTransformInfo.h"
23 #include "llvm/Analysis/ValueTracking.h"
24 #include "llvm/CodeGen/ISDOpcodes.h"
25 #include "llvm/CodeGen/MachineValueType.h"
26 #include "llvm/CodeGen/ValueTypes.h"
27 #include "llvm/IR/Argument.h"
28 #include "llvm/IR/Attributes.h"
29 #include "llvm/IR/BasicBlock.h"
30 #include "llvm/IR/CallingConv.h"
31 #include "llvm/IR/DataLayout.h"
32 #include "llvm/IR/DerivedTypes.h"
33 #include "llvm/IR/Function.h"
34 #include "llvm/IR/Instruction.h"
35 #include "llvm/IR/Instructions.h"
36 #include "llvm/IR/IntrinsicInst.h"
37 #include "llvm/IR/Module.h"
38 #include "llvm/IR/PatternMatch.h"
39 #include "llvm/IR/Type.h"
40 #include "llvm/IR/Value.h"
41 #include "llvm/MC/SubtargetFeature.h"
42 #include "llvm/Support/Casting.h"
43 #include "llvm/Support/CommandLine.h"
44 #include "llvm/Support/Debug.h"
45 #include "llvm/Support/ErrorHandling.h"
46 #include "llvm/Support/raw_ostream.h"
47 #include "llvm/Target/TargetMachine.h"
48 #include <algorithm>
49 #include <cassert>
50 #include <limits>
51 #include <utility>
52 
53 using namespace llvm;
54 
55 #define DEBUG_TYPE "AMDGPUtti"
56 
57 static cl::opt<unsigned> UnrollThresholdPrivate(
58   "amdgpu-unroll-threshold-private",
59   cl::desc("Unroll threshold for AMDGPU if private memory used in a loop"),
60   cl::init(2500), cl::Hidden);
61 
62 static cl::opt<unsigned> UnrollThresholdLocal(
63   "amdgpu-unroll-threshold-local",
64   cl::desc("Unroll threshold for AMDGPU if local memory used in a loop"),
65   cl::init(1000), cl::Hidden);
66 
67 static cl::opt<unsigned> UnrollThresholdIf(
68   "amdgpu-unroll-threshold-if",
69   cl::desc("Unroll threshold increment for AMDGPU for each if statement inside loop"),
70   cl::init(150), cl::Hidden);
71 
72 static bool dependsOnLocalPhi(const Loop *L, const Value *Cond,
73                               unsigned Depth = 0) {
74   const Instruction *I = dyn_cast<Instruction>(Cond);
75   if (!I)
76     return false;
77 
78   for (const Value *V : I->operand_values()) {
79     if (!L->contains(I))
80       continue;
81     if (const PHINode *PHI = dyn_cast<PHINode>(V)) {
82       if (llvm::none_of(L->getSubLoops(), [PHI](const Loop* SubLoop) {
83                   return SubLoop->contains(PHI); }))
84         return true;
85     } else if (Depth < 10 && dependsOnLocalPhi(L, V, Depth+1))
86       return true;
87   }
88   return false;
89 }
90 
91 void AMDGPUTTIImpl::getUnrollingPreferences(Loop *L, ScalarEvolution &SE,
92                                             TTI::UnrollingPreferences &UP) {
93   UP.Threshold = 300; // Twice the default.
94   UP.MaxCount = std::numeric_limits<unsigned>::max();
95   UP.Partial = true;
96 
97   // TODO: Do we want runtime unrolling?
98 
99   // Maximum alloca size than can fit registers. Reserve 16 registers.
100   const unsigned MaxAlloca = (256 - 16) * 4;
101   unsigned ThresholdPrivate = UnrollThresholdPrivate;
102   unsigned ThresholdLocal = UnrollThresholdLocal;
103   unsigned MaxBoost = std::max(ThresholdPrivate, ThresholdLocal);
104   AMDGPUAS ASST = ST->getAMDGPUAS();
105   for (const BasicBlock *BB : L->getBlocks()) {
106     const DataLayout &DL = BB->getModule()->getDataLayout();
107     unsigned LocalGEPsSeen = 0;
108 
109     if (llvm::any_of(L->getSubLoops(), [BB](const Loop* SubLoop) {
110                return SubLoop->contains(BB); }))
111         continue; // Block belongs to an inner loop.
112 
113     for (const Instruction &I : *BB) {
114       // Unroll a loop which contains an "if" statement whose condition
115       // defined by a PHI belonging to the loop. This may help to eliminate
116       // if region and potentially even PHI itself, saving on both divergence
117       // and registers used for the PHI.
118       // Add a small bonus for each of such "if" statements.
119       if (const BranchInst *Br = dyn_cast<BranchInst>(&I)) {
120         if (UP.Threshold < MaxBoost && Br->isConditional()) {
121           if (L->isLoopExiting(Br->getSuccessor(0)) ||
122               L->isLoopExiting(Br->getSuccessor(1)))
123             continue;
124           if (dependsOnLocalPhi(L, Br->getCondition())) {
125             UP.Threshold += UnrollThresholdIf;
126             DEBUG(dbgs() << "Set unroll threshold " << UP.Threshold
127                          << " for loop:\n" << *L << " due to " << *Br << '\n');
128             if (UP.Threshold >= MaxBoost)
129               return;
130           }
131         }
132         continue;
133       }
134 
135       const GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(&I);
136       if (!GEP)
137         continue;
138 
139       unsigned AS = GEP->getAddressSpace();
140       unsigned Threshold = 0;
141       if (AS == ASST.PRIVATE_ADDRESS)
142         Threshold = ThresholdPrivate;
143       else if (AS == ASST.LOCAL_ADDRESS)
144         Threshold = ThresholdLocal;
145       else
146         continue;
147 
148       if (UP.Threshold >= Threshold)
149         continue;
150 
151       if (AS == ASST.PRIVATE_ADDRESS) {
152         const Value *Ptr = GEP->getPointerOperand();
153         const AllocaInst *Alloca =
154             dyn_cast<AllocaInst>(GetUnderlyingObject(Ptr, DL));
155         if (!Alloca || !Alloca->isStaticAlloca())
156           continue;
157         Type *Ty = Alloca->getAllocatedType();
158         unsigned AllocaSize = Ty->isSized() ? DL.getTypeAllocSize(Ty) : 0;
159         if (AllocaSize > MaxAlloca)
160           continue;
161       } else if (AS == ASST.LOCAL_ADDRESS) {
162         LocalGEPsSeen++;
163         // Inhibit unroll for local memory if we have seen addressing not to
164         // a variable, most likely we will be unable to combine it.
165         // Do not unroll too deep inner loops for local memory to give a chance
166         // to unroll an outer loop for a more important reason.
167         if (LocalGEPsSeen > 1 || L->getLoopDepth() > 2 ||
168             (!isa<GlobalVariable>(GEP->getPointerOperand()) &&
169              !isa<Argument>(GEP->getPointerOperand())))
170           continue;
171       }
172 
173       // Check if GEP depends on a value defined by this loop itself.
174       bool HasLoopDef = false;
175       for (const Value *Op : GEP->operands()) {
176         const Instruction *Inst = dyn_cast<Instruction>(Op);
177         if (!Inst || L->isLoopInvariant(Op))
178           continue;
179 
180         if (llvm::any_of(L->getSubLoops(), [Inst](const Loop* SubLoop) {
181              return SubLoop->contains(Inst); }))
182           continue;
183         HasLoopDef = true;
184         break;
185       }
186       if (!HasLoopDef)
187         continue;
188 
189       // We want to do whatever we can to limit the number of alloca
190       // instructions that make it through to the code generator.  allocas
191       // require us to use indirect addressing, which is slow and prone to
192       // compiler bugs.  If this loop does an address calculation on an
193       // alloca ptr, then we want to use a higher than normal loop unroll
194       // threshold. This will give SROA a better chance to eliminate these
195       // allocas.
196       //
197       // We also want to have more unrolling for local memory to let ds
198       // instructions with different offsets combine.
199       //
200       // Don't use the maximum allowed value here as it will make some
201       // programs way too big.
202       UP.Threshold = Threshold;
203       DEBUG(dbgs() << "Set unroll threshold " << Threshold << " for loop:\n"
204                    << *L << " due to " << *GEP << '\n');
205       if (UP.Threshold >= MaxBoost)
206         return;
207     }
208   }
209 }
210 
211 unsigned AMDGPUTTIImpl::getHardwareNumberOfRegisters(bool Vec) const {
212   // The concept of vector registers doesn't really exist. Some packed vector
213   // operations operate on the normal 32-bit registers.
214 
215   // Number of VGPRs on SI.
216   if (ST->getGeneration() >= AMDGPUSubtarget::SOUTHERN_ISLANDS)
217     return 256;
218 
219   return 4 * 128; // XXX - 4 channels. Should these count as vector instead?
220 }
221 
222 unsigned AMDGPUTTIImpl::getNumberOfRegisters(bool Vec) const {
223   // This is really the number of registers to fill when vectorizing /
224   // interleaving loops, so we lie to avoid trying to use all registers.
225   return getHardwareNumberOfRegisters(Vec) >> 3;
226 }
227 
228 unsigned AMDGPUTTIImpl::getRegisterBitWidth(bool Vector) const {
229   return 32;
230 }
231 
232 unsigned AMDGPUTTIImpl::getMinVectorRegisterBitWidth() const {
233   return 32;
234 }
235 
236 unsigned AMDGPUTTIImpl::getLoadStoreVecRegBitWidth(unsigned AddrSpace) const {
237   AMDGPUAS AS = ST->getAMDGPUAS();
238   if (AddrSpace == AS.GLOBAL_ADDRESS ||
239       AddrSpace == AS.CONSTANT_ADDRESS ||
240       AddrSpace == AS.CONSTANT_ADDRESS_32BIT ||
241       AddrSpace == AS.FLAT_ADDRESS)
242     return 128;
243   if (AddrSpace == AS.LOCAL_ADDRESS ||
244       AddrSpace == AS.REGION_ADDRESS)
245     return 64;
246   if (AddrSpace == AS.PRIVATE_ADDRESS)
247     return 8 * ST->getMaxPrivateElementSize();
248 
249   if (ST->getGeneration() <= AMDGPUSubtarget::NORTHERN_ISLANDS &&
250       (AddrSpace == AS.PARAM_D_ADDRESS ||
251       AddrSpace == AS.PARAM_I_ADDRESS ||
252       (AddrSpace >= AS.CONSTANT_BUFFER_0 &&
253       AddrSpace <= AS.CONSTANT_BUFFER_15)))
254     return 128;
255   llvm_unreachable("unhandled address space");
256 }
257 
258 bool AMDGPUTTIImpl::isLegalToVectorizeMemChain(unsigned ChainSizeInBytes,
259                                                unsigned Alignment,
260                                                unsigned AddrSpace) const {
261   // We allow vectorization of flat stores, even though we may need to decompose
262   // them later if they may access private memory. We don't have enough context
263   // here, and legalization can handle it.
264   if (AddrSpace == ST->getAMDGPUAS().PRIVATE_ADDRESS) {
265     return (Alignment >= 4 || ST->hasUnalignedScratchAccess()) &&
266       ChainSizeInBytes <= ST->getMaxPrivateElementSize();
267   }
268   return true;
269 }
270 
271 bool AMDGPUTTIImpl::isLegalToVectorizeLoadChain(unsigned ChainSizeInBytes,
272                                                 unsigned Alignment,
273                                                 unsigned AddrSpace) const {
274   return isLegalToVectorizeMemChain(ChainSizeInBytes, Alignment, AddrSpace);
275 }
276 
277 bool AMDGPUTTIImpl::isLegalToVectorizeStoreChain(unsigned ChainSizeInBytes,
278                                                  unsigned Alignment,
279                                                  unsigned AddrSpace) const {
280   return isLegalToVectorizeMemChain(ChainSizeInBytes, Alignment, AddrSpace);
281 }
282 
283 unsigned AMDGPUTTIImpl::getMaxInterleaveFactor(unsigned VF) {
284   // Disable unrolling if the loop is not vectorized.
285   // TODO: Enable this again.
286   if (VF == 1)
287     return 1;
288 
289   return 8;
290 }
291 
292 bool AMDGPUTTIImpl::getTgtMemIntrinsic(IntrinsicInst *Inst,
293                                        MemIntrinsicInfo &Info) const {
294   switch (Inst->getIntrinsicID()) {
295   case Intrinsic::amdgcn_atomic_inc:
296   case Intrinsic::amdgcn_atomic_dec:
297   case Intrinsic::amdgcn_ds_fadd:
298   case Intrinsic::amdgcn_ds_fmin:
299   case Intrinsic::amdgcn_ds_fmax: {
300     auto *Ordering = dyn_cast<ConstantInt>(Inst->getArgOperand(2));
301     auto *Volatile = dyn_cast<ConstantInt>(Inst->getArgOperand(4));
302     if (!Ordering || !Volatile)
303       return false; // Invalid.
304 
305     unsigned OrderingVal = Ordering->getZExtValue();
306     if (OrderingVal > static_cast<unsigned>(AtomicOrdering::SequentiallyConsistent))
307       return false;
308 
309     Info.PtrVal = Inst->getArgOperand(0);
310     Info.Ordering = static_cast<AtomicOrdering>(OrderingVal);
311     Info.ReadMem = true;
312     Info.WriteMem = true;
313     Info.IsVolatile = !Volatile->isNullValue();
314     return true;
315   }
316   default:
317     return false;
318   }
319 }
320 
321 int AMDGPUTTIImpl::getArithmeticInstrCost(
322     unsigned Opcode, Type *Ty, TTI::OperandValueKind Opd1Info,
323     TTI::OperandValueKind Opd2Info, TTI::OperandValueProperties Opd1PropInfo,
324     TTI::OperandValueProperties Opd2PropInfo, ArrayRef<const Value *> Args ) {
325   EVT OrigTy = TLI->getValueType(DL, Ty);
326   if (!OrigTy.isSimple()) {
327     return BaseT::getArithmeticInstrCost(Opcode, Ty, Opd1Info, Opd2Info,
328                                          Opd1PropInfo, Opd2PropInfo);
329   }
330 
331   // Legalize the type.
332   std::pair<int, MVT> LT = TLI->getTypeLegalizationCost(DL, Ty);
333   int ISD = TLI->InstructionOpcodeToISD(Opcode);
334 
335   // Because we don't have any legal vector operations, but the legal types, we
336   // need to account for split vectors.
337   unsigned NElts = LT.second.isVector() ?
338     LT.second.getVectorNumElements() : 1;
339 
340   MVT::SimpleValueType SLT = LT.second.getScalarType().SimpleTy;
341 
342   switch (ISD) {
343   case ISD::SHL:
344   case ISD::SRL:
345   case ISD::SRA:
346     if (SLT == MVT::i64)
347       return get64BitInstrCost() * LT.first * NElts;
348 
349     // i32
350     return getFullRateInstrCost() * LT.first * NElts;
351   case ISD::ADD:
352   case ISD::SUB:
353   case ISD::AND:
354   case ISD::OR:
355   case ISD::XOR:
356     if (SLT == MVT::i64){
357       // and, or and xor are typically split into 2 VALU instructions.
358       return 2 * getFullRateInstrCost() * LT.first * NElts;
359     }
360 
361     return LT.first * NElts * getFullRateInstrCost();
362   case ISD::MUL: {
363     const int QuarterRateCost = getQuarterRateInstrCost();
364     if (SLT == MVT::i64) {
365       const int FullRateCost = getFullRateInstrCost();
366       return (4 * QuarterRateCost + (2 * 2) * FullRateCost) * LT.first * NElts;
367     }
368 
369     // i32
370     return QuarterRateCost * NElts * LT.first;
371   }
372   case ISD::FADD:
373   case ISD::FSUB:
374   case ISD::FMUL:
375     if (SLT == MVT::f64)
376       return LT.first * NElts * get64BitInstrCost();
377 
378     if (SLT == MVT::f32 || SLT == MVT::f16)
379       return LT.first * NElts * getFullRateInstrCost();
380     break;
381   case ISD::FDIV:
382   case ISD::FREM:
383     // FIXME: frem should be handled separately. The fdiv in it is most of it,
384     // but the current lowering is also not entirely correct.
385     if (SLT == MVT::f64) {
386       int Cost = 4 * get64BitInstrCost() + 7 * getQuarterRateInstrCost();
387       // Add cost of workaround.
388       if (ST->getGeneration() == AMDGPUSubtarget::SOUTHERN_ISLANDS)
389         Cost += 3 * getFullRateInstrCost();
390 
391       return LT.first * Cost * NElts;
392     }
393 
394     if (!Args.empty() && match(Args[0], PatternMatch::m_FPOne())) {
395       // TODO: This is more complicated, unsafe flags etc.
396       if ((SLT == MVT::f32 && !ST->hasFP32Denormals()) ||
397           (SLT == MVT::f16 && ST->has16BitInsts())) {
398         return LT.first * getQuarterRateInstrCost() * NElts;
399       }
400     }
401 
402     if (SLT == MVT::f16 && ST->has16BitInsts()) {
403       // 2 x v_cvt_f32_f16
404       // f32 rcp
405       // f32 fmul
406       // v_cvt_f16_f32
407       // f16 div_fixup
408       int Cost = 4 * getFullRateInstrCost() + 2 * getQuarterRateInstrCost();
409       return LT.first * Cost * NElts;
410     }
411 
412     if (SLT == MVT::f32 || SLT == MVT::f16) {
413       int Cost = 7 * getFullRateInstrCost() + 1 * getQuarterRateInstrCost();
414 
415       if (!ST->hasFP32Denormals()) {
416         // FP mode switches.
417         Cost += 2 * getFullRateInstrCost();
418       }
419 
420       return LT.first * NElts * Cost;
421     }
422     break;
423   default:
424     break;
425   }
426 
427   return BaseT::getArithmeticInstrCost(Opcode, Ty, Opd1Info, Opd2Info,
428                                        Opd1PropInfo, Opd2PropInfo);
429 }
430 
431 unsigned AMDGPUTTIImpl::getCFInstrCost(unsigned Opcode) {
432   // XXX - For some reason this isn't called for switch.
433   switch (Opcode) {
434   case Instruction::Br:
435   case Instruction::Ret:
436     return 10;
437   default:
438     return BaseT::getCFInstrCost(Opcode);
439   }
440 }
441 
442 int AMDGPUTTIImpl::getVectorInstrCost(unsigned Opcode, Type *ValTy,
443                                       unsigned Index) {
444   switch (Opcode) {
445   case Instruction::ExtractElement:
446   case Instruction::InsertElement: {
447     unsigned EltSize
448       = DL.getTypeSizeInBits(cast<VectorType>(ValTy)->getElementType());
449     if (EltSize < 32) {
450       if (EltSize == 16 && Index == 0 && ST->has16BitInsts())
451         return 0;
452       return BaseT::getVectorInstrCost(Opcode, ValTy, Index);
453     }
454 
455     // Extracts are just reads of a subregister, so are free. Inserts are
456     // considered free because we don't want to have any cost for scalarizing
457     // operations, and we don't have to copy into a different register class.
458 
459     // Dynamic indexing isn't free and is best avoided.
460     return Index == ~0u ? 2 : 0;
461   }
462   default:
463     return BaseT::getVectorInstrCost(Opcode, ValTy, Index);
464   }
465 }
466 
467 static bool isIntrinsicSourceOfDivergence(const IntrinsicInst *I) {
468   switch (I->getIntrinsicID()) {
469   case Intrinsic::amdgcn_workitem_id_x:
470   case Intrinsic::amdgcn_workitem_id_y:
471   case Intrinsic::amdgcn_workitem_id_z:
472   case Intrinsic::amdgcn_interp_mov:
473   case Intrinsic::amdgcn_interp_p1:
474   case Intrinsic::amdgcn_interp_p2:
475   case Intrinsic::amdgcn_mbcnt_hi:
476   case Intrinsic::amdgcn_mbcnt_lo:
477   case Intrinsic::r600_read_tidig_x:
478   case Intrinsic::r600_read_tidig_y:
479   case Intrinsic::r600_read_tidig_z:
480   case Intrinsic::amdgcn_atomic_inc:
481   case Intrinsic::amdgcn_atomic_dec:
482   case Intrinsic::amdgcn_ds_fadd:
483   case Intrinsic::amdgcn_ds_fmin:
484   case Intrinsic::amdgcn_ds_fmax:
485   case Intrinsic::amdgcn_image_atomic_swap:
486   case Intrinsic::amdgcn_image_atomic_add:
487   case Intrinsic::amdgcn_image_atomic_sub:
488   case Intrinsic::amdgcn_image_atomic_smin:
489   case Intrinsic::amdgcn_image_atomic_umin:
490   case Intrinsic::amdgcn_image_atomic_smax:
491   case Intrinsic::amdgcn_image_atomic_umax:
492   case Intrinsic::amdgcn_image_atomic_and:
493   case Intrinsic::amdgcn_image_atomic_or:
494   case Intrinsic::amdgcn_image_atomic_xor:
495   case Intrinsic::amdgcn_image_atomic_inc:
496   case Intrinsic::amdgcn_image_atomic_dec:
497   case Intrinsic::amdgcn_image_atomic_cmpswap:
498   case Intrinsic::amdgcn_buffer_atomic_swap:
499   case Intrinsic::amdgcn_buffer_atomic_add:
500   case Intrinsic::amdgcn_buffer_atomic_sub:
501   case Intrinsic::amdgcn_buffer_atomic_smin:
502   case Intrinsic::amdgcn_buffer_atomic_umin:
503   case Intrinsic::amdgcn_buffer_atomic_smax:
504   case Intrinsic::amdgcn_buffer_atomic_umax:
505   case Intrinsic::amdgcn_buffer_atomic_and:
506   case Intrinsic::amdgcn_buffer_atomic_or:
507   case Intrinsic::amdgcn_buffer_atomic_xor:
508   case Intrinsic::amdgcn_buffer_atomic_cmpswap:
509   case Intrinsic::amdgcn_ps_live:
510   case Intrinsic::amdgcn_ds_swizzle:
511     return true;
512   default:
513     return false;
514   }
515 }
516 
517 static bool isArgPassedInSGPR(const Argument *A) {
518   const Function *F = A->getParent();
519 
520   // Arguments to compute shaders are never a source of divergence.
521   CallingConv::ID CC = F->getCallingConv();
522   switch (CC) {
523   case CallingConv::AMDGPU_KERNEL:
524   case CallingConv::SPIR_KERNEL:
525     return true;
526   case CallingConv::AMDGPU_VS:
527   case CallingConv::AMDGPU_LS:
528   case CallingConv::AMDGPU_HS:
529   case CallingConv::AMDGPU_ES:
530   case CallingConv::AMDGPU_GS:
531   case CallingConv::AMDGPU_PS:
532   case CallingConv::AMDGPU_CS:
533     // For non-compute shaders, SGPR inputs are marked with either inreg or byval.
534     // Everything else is in VGPRs.
535     return F->getAttributes().hasParamAttribute(A->getArgNo(), Attribute::InReg) ||
536            F->getAttributes().hasParamAttribute(A->getArgNo(), Attribute::ByVal);
537   default:
538     // TODO: Should calls support inreg for SGPR inputs?
539     return false;
540   }
541 }
542 
543 /// \returns true if the result of the value could potentially be
544 /// different across workitems in a wavefront.
545 bool AMDGPUTTIImpl::isSourceOfDivergence(const Value *V) const {
546   if (const Argument *A = dyn_cast<Argument>(V))
547     return !isArgPassedInSGPR(A);
548 
549   // Loads from the private address space are divergent, because threads
550   // can execute the load instruction with the same inputs and get different
551   // results.
552   //
553   // All other loads are not divergent, because if threads issue loads with the
554   // same arguments, they will always get the same result.
555   if (const LoadInst *Load = dyn_cast<LoadInst>(V))
556     return Load->getPointerAddressSpace() == ST->getAMDGPUAS().PRIVATE_ADDRESS;
557 
558   // Atomics are divergent because they are executed sequentially: when an
559   // atomic operation refers to the same address in each thread, then each
560   // thread after the first sees the value written by the previous thread as
561   // original value.
562   if (isa<AtomicRMWInst>(V) || isa<AtomicCmpXchgInst>(V))
563     return true;
564 
565   if (const IntrinsicInst *Intrinsic = dyn_cast<IntrinsicInst>(V))
566     return isIntrinsicSourceOfDivergence(Intrinsic);
567 
568   // Assume all function calls are a source of divergence.
569   if (isa<CallInst>(V) || isa<InvokeInst>(V))
570     return true;
571 
572   return false;
573 }
574 
575 bool AMDGPUTTIImpl::isAlwaysUniform(const Value *V) const {
576   if (const IntrinsicInst *Intrinsic = dyn_cast<IntrinsicInst>(V)) {
577     switch (Intrinsic->getIntrinsicID()) {
578     default:
579       return false;
580     case Intrinsic::amdgcn_readfirstlane:
581     case Intrinsic::amdgcn_readlane:
582       return true;
583     }
584   }
585   return false;
586 }
587 
588 unsigned AMDGPUTTIImpl::getShuffleCost(TTI::ShuffleKind Kind, Type *Tp, int Index,
589                                        Type *SubTp) {
590   if (ST->hasVOP3PInsts()) {
591     VectorType *VT = cast<VectorType>(Tp);
592     if (VT->getNumElements() == 2 &&
593         DL.getTypeSizeInBits(VT->getElementType()) == 16) {
594       // With op_sel VOP3P instructions freely can access the low half or high
595       // half of a register, so any swizzle is free.
596 
597       switch (Kind) {
598       case TTI::SK_Broadcast:
599       case TTI::SK_Reverse:
600       case TTI::SK_PermuteSingleSrc:
601         return 0;
602       default:
603         break;
604       }
605     }
606   }
607 
608   return BaseT::getShuffleCost(Kind, Tp, Index, SubTp);
609 }
610 
611 bool AMDGPUTTIImpl::areInlineCompatible(const Function *Caller,
612                                         const Function *Callee) const {
613   const TargetMachine &TM = getTLI()->getTargetMachine();
614   const FeatureBitset &CallerBits =
615     TM.getSubtargetImpl(*Caller)->getFeatureBits();
616   const FeatureBitset &CalleeBits =
617     TM.getSubtargetImpl(*Callee)->getFeatureBits();
618 
619   FeatureBitset RealCallerBits = CallerBits & ~InlineFeatureIgnoreList;
620   FeatureBitset RealCalleeBits = CalleeBits & ~InlineFeatureIgnoreList;
621   return ((RealCallerBits & RealCalleeBits) == RealCalleeBits);
622 }
623