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