1 //===- llvm/CodeGen/GlobalISel/IRTranslator.cpp - IRTranslator ---*- C++ -*-==// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 /// \file 9 /// This file implements the IRTranslator class. 10 //===----------------------------------------------------------------------===// 11 12 #include "llvm/CodeGen/GlobalISel/IRTranslator.h" 13 #include "llvm/ADT/PostOrderIterator.h" 14 #include "llvm/ADT/STLExtras.h" 15 #include "llvm/ADT/ScopeExit.h" 16 #include "llvm/ADT/SmallSet.h" 17 #include "llvm/ADT/SmallVector.h" 18 #include "llvm/Analysis/BranchProbabilityInfo.h" 19 #include "llvm/Analysis/OptimizationRemarkEmitter.h" 20 #include "llvm/Analysis/ValueTracking.h" 21 #include "llvm/CodeGen/Analysis.h" 22 #include "llvm/CodeGen/FunctionLoweringInfo.h" 23 #include "llvm/CodeGen/GlobalISel/CallLowering.h" 24 #include "llvm/CodeGen/GlobalISel/GISelChangeObserver.h" 25 #include "llvm/CodeGen/LowLevelType.h" 26 #include "llvm/CodeGen/MachineBasicBlock.h" 27 #include "llvm/CodeGen/MachineFrameInfo.h" 28 #include "llvm/CodeGen/MachineFunction.h" 29 #include "llvm/CodeGen/MachineInstrBuilder.h" 30 #include "llvm/CodeGen/MachineMemOperand.h" 31 #include "llvm/CodeGen/MachineOperand.h" 32 #include "llvm/CodeGen/MachineRegisterInfo.h" 33 #include "llvm/CodeGen/StackProtector.h" 34 #include "llvm/CodeGen/TargetFrameLowering.h" 35 #include "llvm/CodeGen/TargetLowering.h" 36 #include "llvm/CodeGen/TargetPassConfig.h" 37 #include "llvm/CodeGen/TargetRegisterInfo.h" 38 #include "llvm/CodeGen/TargetSubtargetInfo.h" 39 #include "llvm/IR/BasicBlock.h" 40 #include "llvm/IR/CFG.h" 41 #include "llvm/IR/Constant.h" 42 #include "llvm/IR/Constants.h" 43 #include "llvm/IR/DataLayout.h" 44 #include "llvm/IR/DebugInfo.h" 45 #include "llvm/IR/DerivedTypes.h" 46 #include "llvm/IR/Function.h" 47 #include "llvm/IR/GetElementPtrTypeIterator.h" 48 #include "llvm/IR/InlineAsm.h" 49 #include "llvm/IR/InstrTypes.h" 50 #include "llvm/IR/Instructions.h" 51 #include "llvm/IR/IntrinsicInst.h" 52 #include "llvm/IR/Intrinsics.h" 53 #include "llvm/IR/LLVMContext.h" 54 #include "llvm/IR/Metadata.h" 55 #include "llvm/IR/Type.h" 56 #include "llvm/IR/User.h" 57 #include "llvm/IR/Value.h" 58 #include "llvm/MC/MCContext.h" 59 #include "llvm/Pass.h" 60 #include "llvm/Support/Casting.h" 61 #include "llvm/Support/CodeGen.h" 62 #include "llvm/Support/Debug.h" 63 #include "llvm/Support/ErrorHandling.h" 64 #include "llvm/Support/LowLevelTypeImpl.h" 65 #include "llvm/Support/MathExtras.h" 66 #include "llvm/Support/raw_ostream.h" 67 #include "llvm/Target/TargetIntrinsicInfo.h" 68 #include "llvm/Target/TargetMachine.h" 69 #include <algorithm> 70 #include <cassert> 71 #include <cstdint> 72 #include <iterator> 73 #include <string> 74 #include <utility> 75 #include <vector> 76 77 #define DEBUG_TYPE "irtranslator" 78 79 using namespace llvm; 80 81 static cl::opt<bool> 82 EnableCSEInIRTranslator("enable-cse-in-irtranslator", 83 cl::desc("Should enable CSE in irtranslator"), 84 cl::Optional, cl::init(false)); 85 char IRTranslator::ID = 0; 86 87 INITIALIZE_PASS_BEGIN(IRTranslator, DEBUG_TYPE, "IRTranslator LLVM IR -> MI", 88 false, false) 89 INITIALIZE_PASS_DEPENDENCY(TargetPassConfig) 90 INITIALIZE_PASS_DEPENDENCY(GISelCSEAnalysisWrapperPass) 91 INITIALIZE_PASS_END(IRTranslator, DEBUG_TYPE, "IRTranslator LLVM IR -> MI", 92 false, false) 93 94 static void reportTranslationError(MachineFunction &MF, 95 const TargetPassConfig &TPC, 96 OptimizationRemarkEmitter &ORE, 97 OptimizationRemarkMissed &R) { 98 MF.getProperties().set(MachineFunctionProperties::Property::FailedISel); 99 100 // Print the function name explicitly if we don't have a debug location (which 101 // makes the diagnostic less useful) or if we're going to emit a raw error. 102 if (!R.getLocation().isValid() || TPC.isGlobalISelAbortEnabled()) 103 R << (" (in function: " + MF.getName() + ")").str(); 104 105 if (TPC.isGlobalISelAbortEnabled()) 106 report_fatal_error(R.getMsg()); 107 else 108 ORE.emit(R); 109 } 110 111 IRTranslator::IRTranslator() : MachineFunctionPass(ID) { } 112 113 #ifndef NDEBUG 114 namespace { 115 /// Verify that every instruction created has the same DILocation as the 116 /// instruction being translated. 117 class DILocationVerifier : public GISelChangeObserver { 118 const Instruction *CurrInst = nullptr; 119 120 public: 121 DILocationVerifier() = default; 122 ~DILocationVerifier() = default; 123 124 const Instruction *getCurrentInst() const { return CurrInst; } 125 void setCurrentInst(const Instruction *Inst) { CurrInst = Inst; } 126 127 void erasingInstr(MachineInstr &MI) override {} 128 void changingInstr(MachineInstr &MI) override {} 129 void changedInstr(MachineInstr &MI) override {} 130 131 void createdInstr(MachineInstr &MI) override { 132 assert(getCurrentInst() && "Inserted instruction without a current MI"); 133 134 // Only print the check message if we're actually checking it. 135 #ifndef NDEBUG 136 LLVM_DEBUG(dbgs() << "Checking DILocation from " << *CurrInst 137 << " was copied to " << MI); 138 #endif 139 // We allow insts in the entry block to have a debug loc line of 0 because 140 // they could have originated from constants, and we don't want a jumpy 141 // debug experience. 142 assert((CurrInst->getDebugLoc() == MI.getDebugLoc() || 143 MI.getDebugLoc().getLine() == 0) && 144 "Line info was not transferred to all instructions"); 145 } 146 }; 147 } // namespace 148 #endif // ifndef NDEBUG 149 150 151 void IRTranslator::getAnalysisUsage(AnalysisUsage &AU) const { 152 AU.addRequired<StackProtector>(); 153 AU.addRequired<TargetPassConfig>(); 154 AU.addRequired<GISelCSEAnalysisWrapperPass>(); 155 getSelectionDAGFallbackAnalysisUsage(AU); 156 MachineFunctionPass::getAnalysisUsage(AU); 157 } 158 159 IRTranslator::ValueToVRegInfo::VRegListT & 160 IRTranslator::allocateVRegs(const Value &Val) { 161 assert(!VMap.contains(Val) && "Value already allocated in VMap"); 162 auto *Regs = VMap.getVRegs(Val); 163 auto *Offsets = VMap.getOffsets(Val); 164 SmallVector<LLT, 4> SplitTys; 165 computeValueLLTs(*DL, *Val.getType(), SplitTys, 166 Offsets->empty() ? Offsets : nullptr); 167 for (unsigned i = 0; i < SplitTys.size(); ++i) 168 Regs->push_back(0); 169 return *Regs; 170 } 171 172 ArrayRef<Register> IRTranslator::getOrCreateVRegs(const Value &Val) { 173 auto VRegsIt = VMap.findVRegs(Val); 174 if (VRegsIt != VMap.vregs_end()) 175 return *VRegsIt->second; 176 177 if (Val.getType()->isVoidTy()) 178 return *VMap.getVRegs(Val); 179 180 // Create entry for this type. 181 auto *VRegs = VMap.getVRegs(Val); 182 auto *Offsets = VMap.getOffsets(Val); 183 184 assert(Val.getType()->isSized() && 185 "Don't know how to create an empty vreg"); 186 187 SmallVector<LLT, 4> SplitTys; 188 computeValueLLTs(*DL, *Val.getType(), SplitTys, 189 Offsets->empty() ? Offsets : nullptr); 190 191 if (!isa<Constant>(Val)) { 192 for (auto Ty : SplitTys) 193 VRegs->push_back(MRI->createGenericVirtualRegister(Ty)); 194 return *VRegs; 195 } 196 197 if (Val.getType()->isAggregateType()) { 198 // UndefValue, ConstantAggregateZero 199 auto &C = cast<Constant>(Val); 200 unsigned Idx = 0; 201 while (auto Elt = C.getAggregateElement(Idx++)) { 202 auto EltRegs = getOrCreateVRegs(*Elt); 203 llvm::copy(EltRegs, std::back_inserter(*VRegs)); 204 } 205 } else { 206 assert(SplitTys.size() == 1 && "unexpectedly split LLT"); 207 VRegs->push_back(MRI->createGenericVirtualRegister(SplitTys[0])); 208 bool Success = translate(cast<Constant>(Val), VRegs->front()); 209 if (!Success) { 210 OptimizationRemarkMissed R("gisel-irtranslator", "GISelFailure", 211 MF->getFunction().getSubprogram(), 212 &MF->getFunction().getEntryBlock()); 213 R << "unable to translate constant: " << ore::NV("Type", Val.getType()); 214 reportTranslationError(*MF, *TPC, *ORE, R); 215 return *VRegs; 216 } 217 } 218 219 return *VRegs; 220 } 221 222 int IRTranslator::getOrCreateFrameIndex(const AllocaInst &AI) { 223 if (FrameIndices.find(&AI) != FrameIndices.end()) 224 return FrameIndices[&AI]; 225 226 unsigned ElementSize = DL->getTypeAllocSize(AI.getAllocatedType()); 227 unsigned Size = 228 ElementSize * cast<ConstantInt>(AI.getArraySize())->getZExtValue(); 229 230 // Always allocate at least one byte. 231 Size = std::max(Size, 1u); 232 233 unsigned Alignment = AI.getAlignment(); 234 if (!Alignment) 235 Alignment = DL->getABITypeAlignment(AI.getAllocatedType()); 236 237 int &FI = FrameIndices[&AI]; 238 FI = MF->getFrameInfo().CreateStackObject(Size, Alignment, false, &AI); 239 return FI; 240 } 241 242 unsigned IRTranslator::getMemOpAlignment(const Instruction &I) { 243 unsigned Alignment = 0; 244 Type *ValTy = nullptr; 245 if (const StoreInst *SI = dyn_cast<StoreInst>(&I)) { 246 Alignment = SI->getAlignment(); 247 ValTy = SI->getValueOperand()->getType(); 248 } else if (const LoadInst *LI = dyn_cast<LoadInst>(&I)) { 249 Alignment = LI->getAlignment(); 250 ValTy = LI->getType(); 251 } else if (const AtomicCmpXchgInst *AI = dyn_cast<AtomicCmpXchgInst>(&I)) { 252 // TODO(PR27168): This instruction has no alignment attribute, but unlike 253 // the default alignment for load/store, the default here is to assume 254 // it has NATURAL alignment, not DataLayout-specified alignment. 255 const DataLayout &DL = AI->getModule()->getDataLayout(); 256 Alignment = DL.getTypeStoreSize(AI->getCompareOperand()->getType()); 257 ValTy = AI->getCompareOperand()->getType(); 258 } else if (const AtomicRMWInst *AI = dyn_cast<AtomicRMWInst>(&I)) { 259 // TODO(PR27168): This instruction has no alignment attribute, but unlike 260 // the default alignment for load/store, the default here is to assume 261 // it has NATURAL alignment, not DataLayout-specified alignment. 262 const DataLayout &DL = AI->getModule()->getDataLayout(); 263 Alignment = DL.getTypeStoreSize(AI->getValOperand()->getType()); 264 ValTy = AI->getType(); 265 } else { 266 OptimizationRemarkMissed R("gisel-irtranslator", "", &I); 267 R << "unable to translate memop: " << ore::NV("Opcode", &I); 268 reportTranslationError(*MF, *TPC, *ORE, R); 269 return 1; 270 } 271 272 return Alignment ? Alignment : DL->getABITypeAlignment(ValTy); 273 } 274 275 MachineBasicBlock &IRTranslator::getMBB(const BasicBlock &BB) { 276 MachineBasicBlock *&MBB = BBToMBB[&BB]; 277 assert(MBB && "BasicBlock was not encountered before"); 278 return *MBB; 279 } 280 281 void IRTranslator::addMachineCFGPred(CFGEdge Edge, MachineBasicBlock *NewPred) { 282 assert(NewPred && "new predecessor must be a real MachineBasicBlock"); 283 MachinePreds[Edge].push_back(NewPred); 284 } 285 286 bool IRTranslator::translateBinaryOp(unsigned Opcode, const User &U, 287 MachineIRBuilder &MIRBuilder) { 288 // Get or create a virtual register for each value. 289 // Unless the value is a Constant => loadimm cst? 290 // or inline constant each time? 291 // Creation of a virtual register needs to have a size. 292 Register Op0 = getOrCreateVReg(*U.getOperand(0)); 293 Register Op1 = getOrCreateVReg(*U.getOperand(1)); 294 Register Res = getOrCreateVReg(U); 295 uint16_t Flags = 0; 296 if (isa<Instruction>(U)) { 297 const Instruction &I = cast<Instruction>(U); 298 Flags = MachineInstr::copyFlagsFromInstruction(I); 299 } 300 301 MIRBuilder.buildInstr(Opcode, {Res}, {Op0, Op1}, Flags); 302 return true; 303 } 304 305 bool IRTranslator::translateFSub(const User &U, MachineIRBuilder &MIRBuilder) { 306 // -0.0 - X --> G_FNEG 307 if (isa<Constant>(U.getOperand(0)) && 308 U.getOperand(0) == ConstantFP::getZeroValueForNegation(U.getType())) { 309 Register Op1 = getOrCreateVReg(*U.getOperand(1)); 310 Register Res = getOrCreateVReg(U); 311 uint16_t Flags = 0; 312 if (isa<Instruction>(U)) { 313 const Instruction &I = cast<Instruction>(U); 314 Flags = MachineInstr::copyFlagsFromInstruction(I); 315 } 316 // Negate the last operand of the FSUB 317 MIRBuilder.buildInstr(TargetOpcode::G_FNEG, {Res}, {Op1}, Flags); 318 return true; 319 } 320 return translateBinaryOp(TargetOpcode::G_FSUB, U, MIRBuilder); 321 } 322 323 bool IRTranslator::translateFNeg(const User &U, MachineIRBuilder &MIRBuilder) { 324 Register Op0 = getOrCreateVReg(*U.getOperand(0)); 325 Register Res = getOrCreateVReg(U); 326 uint16_t Flags = 0; 327 if (isa<Instruction>(U)) { 328 const Instruction &I = cast<Instruction>(U); 329 Flags = MachineInstr::copyFlagsFromInstruction(I); 330 } 331 MIRBuilder.buildInstr(TargetOpcode::G_FNEG, {Res}, {Op0}, Flags); 332 return true; 333 } 334 335 bool IRTranslator::translateCompare(const User &U, 336 MachineIRBuilder &MIRBuilder) { 337 const CmpInst *CI = dyn_cast<CmpInst>(&U); 338 Register Op0 = getOrCreateVReg(*U.getOperand(0)); 339 Register Op1 = getOrCreateVReg(*U.getOperand(1)); 340 Register Res = getOrCreateVReg(U); 341 CmpInst::Predicate Pred = 342 CI ? CI->getPredicate() : static_cast<CmpInst::Predicate>( 343 cast<ConstantExpr>(U).getPredicate()); 344 if (CmpInst::isIntPredicate(Pred)) 345 MIRBuilder.buildICmp(Pred, Res, Op0, Op1); 346 else if (Pred == CmpInst::FCMP_FALSE) 347 MIRBuilder.buildCopy( 348 Res, getOrCreateVReg(*Constant::getNullValue(CI->getType()))); 349 else if (Pred == CmpInst::FCMP_TRUE) 350 MIRBuilder.buildCopy( 351 Res, getOrCreateVReg(*Constant::getAllOnesValue(CI->getType()))); 352 else { 353 MIRBuilder.buildInstr(TargetOpcode::G_FCMP, {Res}, {Pred, Op0, Op1}, 354 MachineInstr::copyFlagsFromInstruction(*CI)); 355 } 356 357 return true; 358 } 359 360 bool IRTranslator::translateRet(const User &U, MachineIRBuilder &MIRBuilder) { 361 const ReturnInst &RI = cast<ReturnInst>(U); 362 const Value *Ret = RI.getReturnValue(); 363 if (Ret && DL->getTypeStoreSize(Ret->getType()) == 0) 364 Ret = nullptr; 365 366 ArrayRef<Register> VRegs; 367 if (Ret) 368 VRegs = getOrCreateVRegs(*Ret); 369 370 Register SwiftErrorVReg = 0; 371 if (CLI->supportSwiftError() && SwiftError.getFunctionArg()) { 372 SwiftErrorVReg = SwiftError.getOrCreateVRegUseAt( 373 &RI, &MIRBuilder.getMBB(), SwiftError.getFunctionArg()); 374 } 375 376 // The target may mess up with the insertion point, but 377 // this is not important as a return is the last instruction 378 // of the block anyway. 379 return CLI->lowerReturn(MIRBuilder, Ret, VRegs, SwiftErrorVReg); 380 } 381 382 bool IRTranslator::translateBr(const User &U, MachineIRBuilder &MIRBuilder) { 383 const BranchInst &BrInst = cast<BranchInst>(U); 384 unsigned Succ = 0; 385 if (!BrInst.isUnconditional()) { 386 // We want a G_BRCOND to the true BB followed by an unconditional branch. 387 Register Tst = getOrCreateVReg(*BrInst.getCondition()); 388 const BasicBlock &TrueTgt = *cast<BasicBlock>(BrInst.getSuccessor(Succ++)); 389 MachineBasicBlock &TrueBB = getMBB(TrueTgt); 390 MIRBuilder.buildBrCond(Tst, TrueBB); 391 } 392 393 const BasicBlock &BrTgt = *cast<BasicBlock>(BrInst.getSuccessor(Succ)); 394 MachineBasicBlock &TgtBB = getMBB(BrTgt); 395 MachineBasicBlock &CurBB = MIRBuilder.getMBB(); 396 397 // If the unconditional target is the layout successor, fallthrough. 398 if (!CurBB.isLayoutSuccessor(&TgtBB)) 399 MIRBuilder.buildBr(TgtBB); 400 401 // Link successors. 402 for (const BasicBlock *Succ : successors(&BrInst)) 403 CurBB.addSuccessor(&getMBB(*Succ)); 404 return true; 405 } 406 407 void IRTranslator::addSuccessorWithProb(MachineBasicBlock *Src, 408 MachineBasicBlock *Dst, 409 BranchProbability Prob) { 410 if (!FuncInfo.BPI) { 411 Src->addSuccessorWithoutProb(Dst); 412 return; 413 } 414 if (Prob.isUnknown()) 415 Prob = getEdgeProbability(Src, Dst); 416 Src->addSuccessor(Dst, Prob); 417 } 418 419 BranchProbability 420 IRTranslator::getEdgeProbability(const MachineBasicBlock *Src, 421 const MachineBasicBlock *Dst) const { 422 const BasicBlock *SrcBB = Src->getBasicBlock(); 423 const BasicBlock *DstBB = Dst->getBasicBlock(); 424 if (!FuncInfo.BPI) { 425 // If BPI is not available, set the default probability as 1 / N, where N is 426 // the number of successors. 427 auto SuccSize = std::max<uint32_t>(succ_size(SrcBB), 1); 428 return BranchProbability(1, SuccSize); 429 } 430 return FuncInfo.BPI->getEdgeProbability(SrcBB, DstBB); 431 } 432 433 bool IRTranslator::translateSwitch(const User &U, MachineIRBuilder &MIB) { 434 using namespace SwitchCG; 435 // Extract cases from the switch. 436 const SwitchInst &SI = cast<SwitchInst>(U); 437 BranchProbabilityInfo *BPI = FuncInfo.BPI; 438 CaseClusterVector Clusters; 439 Clusters.reserve(SI.getNumCases()); 440 for (auto &I : SI.cases()) { 441 MachineBasicBlock *Succ = &getMBB(*I.getCaseSuccessor()); 442 assert(Succ && "Could not find successor mbb in mapping"); 443 const ConstantInt *CaseVal = I.getCaseValue(); 444 BranchProbability Prob = 445 BPI ? BPI->getEdgeProbability(SI.getParent(), I.getSuccessorIndex()) 446 : BranchProbability(1, SI.getNumCases() + 1); 447 Clusters.push_back(CaseCluster::range(CaseVal, CaseVal, Succ, Prob)); 448 } 449 450 MachineBasicBlock *DefaultMBB = &getMBB(*SI.getDefaultDest()); 451 452 // Cluster adjacent cases with the same destination. We do this at all 453 // optimization levels because it's cheap to do and will make codegen faster 454 // if there are many clusters. 455 sortAndRangeify(Clusters); 456 457 MachineBasicBlock *SwitchMBB = &getMBB(*SI.getParent()); 458 459 // If there is only the default destination, jump there directly. 460 if (Clusters.empty()) { 461 SwitchMBB->addSuccessor(DefaultMBB); 462 if (DefaultMBB != SwitchMBB->getNextNode()) 463 MIB.buildBr(*DefaultMBB); 464 return true; 465 } 466 467 SL->findJumpTables(Clusters, &SI, DefaultMBB); 468 469 LLVM_DEBUG({ 470 dbgs() << "Case clusters: "; 471 for (const CaseCluster &C : Clusters) { 472 if (C.Kind == CC_JumpTable) 473 dbgs() << "JT:"; 474 if (C.Kind == CC_BitTests) 475 dbgs() << "BT:"; 476 477 C.Low->getValue().print(dbgs(), true); 478 if (C.Low != C.High) { 479 dbgs() << '-'; 480 C.High->getValue().print(dbgs(), true); 481 } 482 dbgs() << ' '; 483 } 484 dbgs() << '\n'; 485 }); 486 487 assert(!Clusters.empty()); 488 SwitchWorkList WorkList; 489 CaseClusterIt First = Clusters.begin(); 490 CaseClusterIt Last = Clusters.end() - 1; 491 auto DefaultProb = getEdgeProbability(SwitchMBB, DefaultMBB); 492 WorkList.push_back({SwitchMBB, First, Last, nullptr, nullptr, DefaultProb}); 493 494 // FIXME: At the moment we don't do any splitting optimizations here like 495 // SelectionDAG does, so this worklist only has one entry. 496 while (!WorkList.empty()) { 497 SwitchWorkListItem W = WorkList.back(); 498 WorkList.pop_back(); 499 if (!lowerSwitchWorkItem(W, SI.getCondition(), SwitchMBB, DefaultMBB, MIB)) 500 return false; 501 } 502 return true; 503 } 504 505 void IRTranslator::emitJumpTable(SwitchCG::JumpTable &JT, 506 MachineBasicBlock *MBB) { 507 // Emit the code for the jump table 508 assert(JT.Reg != -1U && "Should lower JT Header first!"); 509 MachineIRBuilder MIB(*MBB->getParent()); 510 MIB.setMBB(*MBB); 511 MIB.setDebugLoc(CurBuilder->getDebugLoc()); 512 513 Type *PtrIRTy = Type::getInt8PtrTy(MF->getFunction().getContext()); 514 const LLT PtrTy = getLLTForType(*PtrIRTy, *DL); 515 516 auto Table = MIB.buildJumpTable(PtrTy, JT.JTI); 517 MIB.buildBrJT(Table.getReg(0), JT.JTI, JT.Reg); 518 } 519 520 bool IRTranslator::emitJumpTableHeader(SwitchCG::JumpTable &JT, 521 SwitchCG::JumpTableHeader &JTH, 522 MachineBasicBlock *HeaderBB) { 523 MachineIRBuilder MIB(*HeaderBB->getParent()); 524 MIB.setMBB(*HeaderBB); 525 MIB.setDebugLoc(CurBuilder->getDebugLoc()); 526 527 const Value &SValue = *JTH.SValue; 528 // Subtract the lowest switch case value from the value being switched on. 529 const LLT SwitchTy = getLLTForType(*SValue.getType(), *DL); 530 Register SwitchOpReg = getOrCreateVReg(SValue); 531 auto FirstCst = MIB.buildConstant(SwitchTy, JTH.First); 532 auto Sub = MIB.buildSub({SwitchTy}, SwitchOpReg, FirstCst); 533 534 // This value may be smaller or larger than the target's pointer type, and 535 // therefore require extension or truncating. 536 Type *PtrIRTy = SValue.getType()->getPointerTo(); 537 const LLT PtrScalarTy = LLT::scalar(DL->getTypeSizeInBits(PtrIRTy)); 538 Sub = MIB.buildZExtOrTrunc(PtrScalarTy, Sub); 539 540 JT.Reg = Sub.getReg(0); 541 542 if (JTH.OmitRangeCheck) { 543 if (JT.MBB != HeaderBB->getNextNode()) 544 MIB.buildBr(*JT.MBB); 545 return true; 546 } 547 548 // Emit the range check for the jump table, and branch to the default block 549 // for the switch statement if the value being switched on exceeds the 550 // largest case in the switch. 551 auto Cst = getOrCreateVReg( 552 *ConstantInt::get(SValue.getType(), JTH.Last - JTH.First)); 553 Cst = MIB.buildZExtOrTrunc(PtrScalarTy, Cst).getReg(0); 554 auto Cmp = MIB.buildICmp(CmpInst::ICMP_UGT, LLT::scalar(1), Sub, Cst); 555 556 auto BrCond = MIB.buildBrCond(Cmp.getReg(0), *JT.Default); 557 558 // Avoid emitting unnecessary branches to the next block. 559 if (JT.MBB != HeaderBB->getNextNode()) 560 BrCond = MIB.buildBr(*JT.MBB); 561 return true; 562 } 563 564 void IRTranslator::emitSwitchCase(SwitchCG::CaseBlock &CB, 565 MachineBasicBlock *SwitchBB, 566 MachineIRBuilder &MIB) { 567 Register CondLHS = getOrCreateVReg(*CB.CmpLHS); 568 Register Cond; 569 DebugLoc OldDbgLoc = MIB.getDebugLoc(); 570 MIB.setDebugLoc(CB.DbgLoc); 571 MIB.setMBB(*CB.ThisBB); 572 573 if (CB.PredInfo.NoCmp) { 574 // Branch or fall through to TrueBB. 575 addSuccessorWithProb(CB.ThisBB, CB.TrueBB, CB.TrueProb); 576 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.TrueBB->getBasicBlock()}, 577 CB.ThisBB); 578 CB.ThisBB->normalizeSuccProbs(); 579 if (CB.TrueBB != CB.ThisBB->getNextNode()) 580 MIB.buildBr(*CB.TrueBB); 581 MIB.setDebugLoc(OldDbgLoc); 582 return; 583 } 584 585 const LLT i1Ty = LLT::scalar(1); 586 // Build the compare. 587 if (!CB.CmpMHS) { 588 Register CondRHS = getOrCreateVReg(*CB.CmpRHS); 589 Cond = MIB.buildICmp(CB.PredInfo.Pred, i1Ty, CondLHS, CondRHS).getReg(0); 590 } else { 591 assert(CB.PredInfo.Pred == CmpInst::ICMP_ULE && 592 "Can only handle ULE ranges"); 593 594 const APInt& Low = cast<ConstantInt>(CB.CmpLHS)->getValue(); 595 const APInt& High = cast<ConstantInt>(CB.CmpRHS)->getValue(); 596 597 Register CmpOpReg = getOrCreateVReg(*CB.CmpMHS); 598 if (cast<ConstantInt>(CB.CmpLHS)->isMinValue(true)) { 599 Register CondRHS = getOrCreateVReg(*CB.CmpRHS); 600 Cond = 601 MIB.buildICmp(CmpInst::ICMP_ULE, i1Ty, CmpOpReg, CondRHS).getReg(0); 602 } else { 603 const LLT &CmpTy = MRI->getType(CmpOpReg); 604 auto Sub = MIB.buildSub({CmpTy}, CmpOpReg, CondLHS); 605 auto Diff = MIB.buildConstant(CmpTy, High - Low); 606 Cond = MIB.buildICmp(CmpInst::ICMP_ULE, i1Ty, Sub, Diff).getReg(0); 607 } 608 } 609 610 // Update successor info 611 addSuccessorWithProb(CB.ThisBB, CB.TrueBB, CB.TrueProb); 612 613 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.TrueBB->getBasicBlock()}, 614 CB.ThisBB); 615 616 // TrueBB and FalseBB are always different unless the incoming IR is 617 // degenerate. This only happens when running llc on weird IR. 618 if (CB.TrueBB != CB.FalseBB) 619 addSuccessorWithProb(CB.ThisBB, CB.FalseBB, CB.FalseProb); 620 CB.ThisBB->normalizeSuccProbs(); 621 622 // if (SwitchBB->getBasicBlock() != CB.FalseBB->getBasicBlock()) 623 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.FalseBB->getBasicBlock()}, 624 CB.ThisBB); 625 626 // If the lhs block is the next block, invert the condition so that we can 627 // fall through to the lhs instead of the rhs block. 628 if (CB.TrueBB == CB.ThisBB->getNextNode()) { 629 std::swap(CB.TrueBB, CB.FalseBB); 630 auto True = MIB.buildConstant(i1Ty, 1); 631 Cond = MIB.buildInstr(TargetOpcode::G_XOR, {i1Ty}, {Cond, True}, None) 632 .getReg(0); 633 } 634 635 MIB.buildBrCond(Cond, *CB.TrueBB); 636 MIB.buildBr(*CB.FalseBB); 637 MIB.setDebugLoc(OldDbgLoc); 638 } 639 640 bool IRTranslator::lowerJumpTableWorkItem(SwitchCG::SwitchWorkListItem W, 641 MachineBasicBlock *SwitchMBB, 642 MachineBasicBlock *CurMBB, 643 MachineBasicBlock *DefaultMBB, 644 MachineIRBuilder &MIB, 645 MachineFunction::iterator BBI, 646 BranchProbability UnhandledProbs, 647 SwitchCG::CaseClusterIt I, 648 MachineBasicBlock *Fallthrough, 649 bool FallthroughUnreachable) { 650 using namespace SwitchCG; 651 MachineFunction *CurMF = SwitchMBB->getParent(); 652 // FIXME: Optimize away range check based on pivot comparisons. 653 JumpTableHeader *JTH = &SL->JTCases[I->JTCasesIndex].first; 654 SwitchCG::JumpTable *JT = &SL->JTCases[I->JTCasesIndex].second; 655 BranchProbability DefaultProb = W.DefaultProb; 656 657 // The jump block hasn't been inserted yet; insert it here. 658 MachineBasicBlock *JumpMBB = JT->MBB; 659 CurMF->insert(BBI, JumpMBB); 660 661 // Since the jump table block is separate from the switch block, we need 662 // to keep track of it as a machine predecessor to the default block, 663 // otherwise we lose the phi edges. 664 addMachineCFGPred({SwitchMBB->getBasicBlock(), DefaultMBB->getBasicBlock()}, 665 CurMBB); 666 addMachineCFGPred({SwitchMBB->getBasicBlock(), DefaultMBB->getBasicBlock()}, 667 JumpMBB); 668 669 auto JumpProb = I->Prob; 670 auto FallthroughProb = UnhandledProbs; 671 672 // If the default statement is a target of the jump table, we evenly 673 // distribute the default probability to successors of CurMBB. Also 674 // update the probability on the edge from JumpMBB to Fallthrough. 675 for (MachineBasicBlock::succ_iterator SI = JumpMBB->succ_begin(), 676 SE = JumpMBB->succ_end(); 677 SI != SE; ++SI) { 678 if (*SI == DefaultMBB) { 679 JumpProb += DefaultProb / 2; 680 FallthroughProb -= DefaultProb / 2; 681 JumpMBB->setSuccProbability(SI, DefaultProb / 2); 682 JumpMBB->normalizeSuccProbs(); 683 } else { 684 // Also record edges from the jump table block to it's successors. 685 addMachineCFGPred({SwitchMBB->getBasicBlock(), (*SI)->getBasicBlock()}, 686 JumpMBB); 687 } 688 } 689 690 // Skip the range check if the fallthrough block is unreachable. 691 if (FallthroughUnreachable) 692 JTH->OmitRangeCheck = true; 693 694 if (!JTH->OmitRangeCheck) 695 addSuccessorWithProb(CurMBB, Fallthrough, FallthroughProb); 696 addSuccessorWithProb(CurMBB, JumpMBB, JumpProb); 697 CurMBB->normalizeSuccProbs(); 698 699 // The jump table header will be inserted in our current block, do the 700 // range check, and fall through to our fallthrough block. 701 JTH->HeaderBB = CurMBB; 702 JT->Default = Fallthrough; // FIXME: Move Default to JumpTableHeader. 703 704 // If we're in the right place, emit the jump table header right now. 705 if (CurMBB == SwitchMBB) { 706 if (!emitJumpTableHeader(*JT, *JTH, CurMBB)) 707 return false; 708 JTH->Emitted = true; 709 } 710 return true; 711 } 712 bool IRTranslator::lowerSwitchRangeWorkItem(SwitchCG::CaseClusterIt I, 713 Value *Cond, 714 MachineBasicBlock *Fallthrough, 715 bool FallthroughUnreachable, 716 BranchProbability UnhandledProbs, 717 MachineBasicBlock *CurMBB, 718 MachineIRBuilder &MIB, 719 MachineBasicBlock *SwitchMBB) { 720 using namespace SwitchCG; 721 const Value *RHS, *LHS, *MHS; 722 CmpInst::Predicate Pred; 723 if (I->Low == I->High) { 724 // Check Cond == I->Low. 725 Pred = CmpInst::ICMP_EQ; 726 LHS = Cond; 727 RHS = I->Low; 728 MHS = nullptr; 729 } else { 730 // Check I->Low <= Cond <= I->High. 731 Pred = CmpInst::ICMP_ULE; 732 LHS = I->Low; 733 MHS = Cond; 734 RHS = I->High; 735 } 736 737 // If Fallthrough is unreachable, fold away the comparison. 738 // The false probability is the sum of all unhandled cases. 739 CaseBlock CB(Pred, FallthroughUnreachable, LHS, RHS, MHS, I->MBB, Fallthrough, 740 CurMBB, MIB.getDebugLoc(), I->Prob, UnhandledProbs); 741 742 emitSwitchCase(CB, SwitchMBB, MIB); 743 return true; 744 } 745 746 bool IRTranslator::lowerSwitchWorkItem(SwitchCG::SwitchWorkListItem W, 747 Value *Cond, 748 MachineBasicBlock *SwitchMBB, 749 MachineBasicBlock *DefaultMBB, 750 MachineIRBuilder &MIB) { 751 using namespace SwitchCG; 752 MachineFunction *CurMF = FuncInfo.MF; 753 MachineBasicBlock *NextMBB = nullptr; 754 MachineFunction::iterator BBI(W.MBB); 755 if (++BBI != FuncInfo.MF->end()) 756 NextMBB = &*BBI; 757 758 if (EnableOpts) { 759 // Here, we order cases by probability so the most likely case will be 760 // checked first. However, two clusters can have the same probability in 761 // which case their relative ordering is non-deterministic. So we use Low 762 // as a tie-breaker as clusters are guaranteed to never overlap. 763 llvm::sort(W.FirstCluster, W.LastCluster + 1, 764 [](const CaseCluster &a, const CaseCluster &b) { 765 return a.Prob != b.Prob 766 ? a.Prob > b.Prob 767 : a.Low->getValue().slt(b.Low->getValue()); 768 }); 769 770 // Rearrange the case blocks so that the last one falls through if possible 771 // without changing the order of probabilities. 772 for (CaseClusterIt I = W.LastCluster; I > W.FirstCluster;) { 773 --I; 774 if (I->Prob > W.LastCluster->Prob) 775 break; 776 if (I->Kind == CC_Range && I->MBB == NextMBB) { 777 std::swap(*I, *W.LastCluster); 778 break; 779 } 780 } 781 } 782 783 // Compute total probability. 784 BranchProbability DefaultProb = W.DefaultProb; 785 BranchProbability UnhandledProbs = DefaultProb; 786 for (CaseClusterIt I = W.FirstCluster; I <= W.LastCluster; ++I) 787 UnhandledProbs += I->Prob; 788 789 MachineBasicBlock *CurMBB = W.MBB; 790 for (CaseClusterIt I = W.FirstCluster, E = W.LastCluster; I <= E; ++I) { 791 bool FallthroughUnreachable = false; 792 MachineBasicBlock *Fallthrough; 793 if (I == W.LastCluster) { 794 // For the last cluster, fall through to the default destination. 795 Fallthrough = DefaultMBB; 796 FallthroughUnreachable = isa<UnreachableInst>( 797 DefaultMBB->getBasicBlock()->getFirstNonPHIOrDbg()); 798 } else { 799 Fallthrough = CurMF->CreateMachineBasicBlock(CurMBB->getBasicBlock()); 800 CurMF->insert(BBI, Fallthrough); 801 } 802 UnhandledProbs -= I->Prob; 803 804 switch (I->Kind) { 805 case CC_BitTests: { 806 LLVM_DEBUG(dbgs() << "Switch to bit test optimization unimplemented"); 807 return false; // Bit tests currently unimplemented. 808 } 809 case CC_JumpTable: { 810 if (!lowerJumpTableWorkItem(W, SwitchMBB, CurMBB, DefaultMBB, MIB, BBI, 811 UnhandledProbs, I, Fallthrough, 812 FallthroughUnreachable)) { 813 LLVM_DEBUG(dbgs() << "Failed to lower jump table"); 814 return false; 815 } 816 break; 817 } 818 case CC_Range: { 819 if (!lowerSwitchRangeWorkItem(I, Cond, Fallthrough, 820 FallthroughUnreachable, UnhandledProbs, 821 CurMBB, MIB, SwitchMBB)) { 822 LLVM_DEBUG(dbgs() << "Failed to lower switch range"); 823 return false; 824 } 825 break; 826 } 827 } 828 CurMBB = Fallthrough; 829 } 830 831 return true; 832 } 833 834 bool IRTranslator::translateIndirectBr(const User &U, 835 MachineIRBuilder &MIRBuilder) { 836 const IndirectBrInst &BrInst = cast<IndirectBrInst>(U); 837 838 const Register Tgt = getOrCreateVReg(*BrInst.getAddress()); 839 MIRBuilder.buildBrIndirect(Tgt); 840 841 // Link successors. 842 MachineBasicBlock &CurBB = MIRBuilder.getMBB(); 843 for (const BasicBlock *Succ : successors(&BrInst)) 844 CurBB.addSuccessor(&getMBB(*Succ)); 845 846 return true; 847 } 848 849 static bool isSwiftError(const Value *V) { 850 if (auto Arg = dyn_cast<Argument>(V)) 851 return Arg->hasSwiftErrorAttr(); 852 if (auto AI = dyn_cast<AllocaInst>(V)) 853 return AI->isSwiftError(); 854 return false; 855 } 856 857 bool IRTranslator::translateLoad(const User &U, MachineIRBuilder &MIRBuilder) { 858 const LoadInst &LI = cast<LoadInst>(U); 859 860 auto Flags = LI.isVolatile() ? MachineMemOperand::MOVolatile 861 : MachineMemOperand::MONone; 862 Flags |= MachineMemOperand::MOLoad; 863 864 if (DL->getTypeStoreSize(LI.getType()) == 0) 865 return true; 866 867 ArrayRef<Register> Regs = getOrCreateVRegs(LI); 868 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(LI); 869 Register Base = getOrCreateVReg(*LI.getPointerOperand()); 870 871 Type *OffsetIRTy = DL->getIntPtrType(LI.getPointerOperandType()); 872 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL); 873 874 if (CLI->supportSwiftError() && isSwiftError(LI.getPointerOperand())) { 875 assert(Regs.size() == 1 && "swifterror should be single pointer"); 876 Register VReg = SwiftError.getOrCreateVRegUseAt(&LI, &MIRBuilder.getMBB(), 877 LI.getPointerOperand()); 878 MIRBuilder.buildCopy(Regs[0], VReg); 879 return true; 880 } 881 882 const MDNode *Ranges = 883 Regs.size() == 1 ? LI.getMetadata(LLVMContext::MD_range) : nullptr; 884 for (unsigned i = 0; i < Regs.size(); ++i) { 885 Register Addr; 886 MIRBuilder.materializeGEP(Addr, Base, OffsetTy, Offsets[i] / 8); 887 888 MachinePointerInfo Ptr(LI.getPointerOperand(), Offsets[i] / 8); 889 unsigned BaseAlign = getMemOpAlignment(LI); 890 auto MMO = MF->getMachineMemOperand( 891 Ptr, Flags, (MRI->getType(Regs[i]).getSizeInBits() + 7) / 8, 892 MinAlign(BaseAlign, Offsets[i] / 8), AAMDNodes(), Ranges, 893 LI.getSyncScopeID(), LI.getOrdering()); 894 MIRBuilder.buildLoad(Regs[i], Addr, *MMO); 895 } 896 897 return true; 898 } 899 900 bool IRTranslator::translateStore(const User &U, MachineIRBuilder &MIRBuilder) { 901 const StoreInst &SI = cast<StoreInst>(U); 902 auto Flags = SI.isVolatile() ? MachineMemOperand::MOVolatile 903 : MachineMemOperand::MONone; 904 Flags |= MachineMemOperand::MOStore; 905 906 if (DL->getTypeStoreSize(SI.getValueOperand()->getType()) == 0) 907 return true; 908 909 ArrayRef<Register> Vals = getOrCreateVRegs(*SI.getValueOperand()); 910 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(*SI.getValueOperand()); 911 Register Base = getOrCreateVReg(*SI.getPointerOperand()); 912 913 Type *OffsetIRTy = DL->getIntPtrType(SI.getPointerOperandType()); 914 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL); 915 916 if (CLI->supportSwiftError() && isSwiftError(SI.getPointerOperand())) { 917 assert(Vals.size() == 1 && "swifterror should be single pointer"); 918 919 Register VReg = SwiftError.getOrCreateVRegDefAt(&SI, &MIRBuilder.getMBB(), 920 SI.getPointerOperand()); 921 MIRBuilder.buildCopy(VReg, Vals[0]); 922 return true; 923 } 924 925 for (unsigned i = 0; i < Vals.size(); ++i) { 926 Register Addr; 927 MIRBuilder.materializeGEP(Addr, Base, OffsetTy, Offsets[i] / 8); 928 929 MachinePointerInfo Ptr(SI.getPointerOperand(), Offsets[i] / 8); 930 unsigned BaseAlign = getMemOpAlignment(SI); 931 auto MMO = MF->getMachineMemOperand( 932 Ptr, Flags, (MRI->getType(Vals[i]).getSizeInBits() + 7) / 8, 933 MinAlign(BaseAlign, Offsets[i] / 8), AAMDNodes(), nullptr, 934 SI.getSyncScopeID(), SI.getOrdering()); 935 MIRBuilder.buildStore(Vals[i], Addr, *MMO); 936 } 937 return true; 938 } 939 940 static uint64_t getOffsetFromIndices(const User &U, const DataLayout &DL) { 941 const Value *Src = U.getOperand(0); 942 Type *Int32Ty = Type::getInt32Ty(U.getContext()); 943 944 // getIndexedOffsetInType is designed for GEPs, so the first index is the 945 // usual array element rather than looking into the actual aggregate. 946 SmallVector<Value *, 1> Indices; 947 Indices.push_back(ConstantInt::get(Int32Ty, 0)); 948 949 if (const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(&U)) { 950 for (auto Idx : EVI->indices()) 951 Indices.push_back(ConstantInt::get(Int32Ty, Idx)); 952 } else if (const InsertValueInst *IVI = dyn_cast<InsertValueInst>(&U)) { 953 for (auto Idx : IVI->indices()) 954 Indices.push_back(ConstantInt::get(Int32Ty, Idx)); 955 } else { 956 for (unsigned i = 1; i < U.getNumOperands(); ++i) 957 Indices.push_back(U.getOperand(i)); 958 } 959 960 return 8 * static_cast<uint64_t>( 961 DL.getIndexedOffsetInType(Src->getType(), Indices)); 962 } 963 964 bool IRTranslator::translateExtractValue(const User &U, 965 MachineIRBuilder &MIRBuilder) { 966 const Value *Src = U.getOperand(0); 967 uint64_t Offset = getOffsetFromIndices(U, *DL); 968 ArrayRef<Register> SrcRegs = getOrCreateVRegs(*Src); 969 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(*Src); 970 unsigned Idx = llvm::lower_bound(Offsets, Offset) - Offsets.begin(); 971 auto &DstRegs = allocateVRegs(U); 972 973 for (unsigned i = 0; i < DstRegs.size(); ++i) 974 DstRegs[i] = SrcRegs[Idx++]; 975 976 return true; 977 } 978 979 bool IRTranslator::translateInsertValue(const User &U, 980 MachineIRBuilder &MIRBuilder) { 981 const Value *Src = U.getOperand(0); 982 uint64_t Offset = getOffsetFromIndices(U, *DL); 983 auto &DstRegs = allocateVRegs(U); 984 ArrayRef<uint64_t> DstOffsets = *VMap.getOffsets(U); 985 ArrayRef<Register> SrcRegs = getOrCreateVRegs(*Src); 986 ArrayRef<Register> InsertedRegs = getOrCreateVRegs(*U.getOperand(1)); 987 auto InsertedIt = InsertedRegs.begin(); 988 989 for (unsigned i = 0; i < DstRegs.size(); ++i) { 990 if (DstOffsets[i] >= Offset && InsertedIt != InsertedRegs.end()) 991 DstRegs[i] = *InsertedIt++; 992 else 993 DstRegs[i] = SrcRegs[i]; 994 } 995 996 return true; 997 } 998 999 bool IRTranslator::translateSelect(const User &U, 1000 MachineIRBuilder &MIRBuilder) { 1001 Register Tst = getOrCreateVReg(*U.getOperand(0)); 1002 ArrayRef<Register> ResRegs = getOrCreateVRegs(U); 1003 ArrayRef<Register> Op0Regs = getOrCreateVRegs(*U.getOperand(1)); 1004 ArrayRef<Register> Op1Regs = getOrCreateVRegs(*U.getOperand(2)); 1005 1006 const SelectInst &SI = cast<SelectInst>(U); 1007 uint16_t Flags = 0; 1008 if (const CmpInst *Cmp = dyn_cast<CmpInst>(SI.getCondition())) 1009 Flags = MachineInstr::copyFlagsFromInstruction(*Cmp); 1010 1011 for (unsigned i = 0; i < ResRegs.size(); ++i) { 1012 MIRBuilder.buildInstr(TargetOpcode::G_SELECT, {ResRegs[i]}, 1013 {Tst, Op0Regs[i], Op1Regs[i]}, Flags); 1014 } 1015 1016 return true; 1017 } 1018 1019 bool IRTranslator::translateBitCast(const User &U, 1020 MachineIRBuilder &MIRBuilder) { 1021 // If we're bitcasting to the source type, we can reuse the source vreg. 1022 if (getLLTForType(*U.getOperand(0)->getType(), *DL) == 1023 getLLTForType(*U.getType(), *DL)) { 1024 Register SrcReg = getOrCreateVReg(*U.getOperand(0)); 1025 auto &Regs = *VMap.getVRegs(U); 1026 // If we already assigned a vreg for this bitcast, we can't change that. 1027 // Emit a copy to satisfy the users we already emitted. 1028 if (!Regs.empty()) 1029 MIRBuilder.buildCopy(Regs[0], SrcReg); 1030 else { 1031 Regs.push_back(SrcReg); 1032 VMap.getOffsets(U)->push_back(0); 1033 } 1034 return true; 1035 } 1036 return translateCast(TargetOpcode::G_BITCAST, U, MIRBuilder); 1037 } 1038 1039 bool IRTranslator::translateCast(unsigned Opcode, const User &U, 1040 MachineIRBuilder &MIRBuilder) { 1041 Register Op = getOrCreateVReg(*U.getOperand(0)); 1042 Register Res = getOrCreateVReg(U); 1043 MIRBuilder.buildInstr(Opcode, {Res}, {Op}); 1044 return true; 1045 } 1046 1047 bool IRTranslator::translateGetElementPtr(const User &U, 1048 MachineIRBuilder &MIRBuilder) { 1049 // FIXME: support vector GEPs. 1050 if (U.getType()->isVectorTy()) 1051 return false; 1052 1053 Value &Op0 = *U.getOperand(0); 1054 Register BaseReg = getOrCreateVReg(Op0); 1055 Type *PtrIRTy = Op0.getType(); 1056 LLT PtrTy = getLLTForType(*PtrIRTy, *DL); 1057 Type *OffsetIRTy = DL->getIntPtrType(PtrIRTy); 1058 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL); 1059 1060 int64_t Offset = 0; 1061 for (gep_type_iterator GTI = gep_type_begin(&U), E = gep_type_end(&U); 1062 GTI != E; ++GTI) { 1063 const Value *Idx = GTI.getOperand(); 1064 if (StructType *StTy = GTI.getStructTypeOrNull()) { 1065 unsigned Field = cast<Constant>(Idx)->getUniqueInteger().getZExtValue(); 1066 Offset += DL->getStructLayout(StTy)->getElementOffset(Field); 1067 continue; 1068 } else { 1069 uint64_t ElementSize = DL->getTypeAllocSize(GTI.getIndexedType()); 1070 1071 // If this is a scalar constant or a splat vector of constants, 1072 // handle it quickly. 1073 if (const auto *CI = dyn_cast<ConstantInt>(Idx)) { 1074 Offset += ElementSize * CI->getSExtValue(); 1075 continue; 1076 } 1077 1078 if (Offset != 0) { 1079 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL); 1080 auto OffsetMIB = MIRBuilder.buildConstant({OffsetTy}, Offset); 1081 BaseReg = 1082 MIRBuilder.buildGEP(PtrTy, BaseReg, OffsetMIB.getReg(0)).getReg(0); 1083 Offset = 0; 1084 } 1085 1086 Register IdxReg = getOrCreateVReg(*Idx); 1087 if (MRI->getType(IdxReg) != OffsetTy) 1088 IdxReg = MIRBuilder.buildSExtOrTrunc(OffsetTy, IdxReg).getReg(0); 1089 1090 // N = N + Idx * ElementSize; 1091 // Avoid doing it for ElementSize of 1. 1092 Register GepOffsetReg; 1093 if (ElementSize != 1) { 1094 auto ElementSizeMIB = MIRBuilder.buildConstant( 1095 getLLTForType(*OffsetIRTy, *DL), ElementSize); 1096 GepOffsetReg = 1097 MIRBuilder.buildMul(OffsetTy, ElementSizeMIB, IdxReg).getReg(0); 1098 } else 1099 GepOffsetReg = IdxReg; 1100 1101 BaseReg = MIRBuilder.buildGEP(PtrTy, BaseReg, GepOffsetReg).getReg(0); 1102 } 1103 } 1104 1105 if (Offset != 0) { 1106 auto OffsetMIB = 1107 MIRBuilder.buildConstant(getLLTForType(*OffsetIRTy, *DL), Offset); 1108 MIRBuilder.buildGEP(getOrCreateVReg(U), BaseReg, OffsetMIB.getReg(0)); 1109 return true; 1110 } 1111 1112 MIRBuilder.buildCopy(getOrCreateVReg(U), BaseReg); 1113 return true; 1114 } 1115 1116 bool IRTranslator::translateMemFunc(const CallInst &CI, 1117 MachineIRBuilder &MIRBuilder, 1118 Intrinsic::ID ID) { 1119 1120 // If the source is undef, then just emit a nop. 1121 if (isa<UndefValue>(CI.getArgOperand(1))) 1122 return true; 1123 1124 ArrayRef<Register> Res; 1125 auto ICall = MIRBuilder.buildIntrinsic(ID, Res, true); 1126 for (auto AI = CI.arg_begin(), AE = CI.arg_end(); std::next(AI) != AE; ++AI) 1127 ICall.addUse(getOrCreateVReg(**AI)); 1128 1129 unsigned DstAlign = 0, SrcAlign = 0; 1130 unsigned IsVol = 1131 cast<ConstantInt>(CI.getArgOperand(CI.getNumArgOperands() - 1)) 1132 ->getZExtValue(); 1133 1134 if (auto *MCI = dyn_cast<MemCpyInst>(&CI)) { 1135 DstAlign = std::max<unsigned>(MCI->getDestAlignment(), 1); 1136 SrcAlign = std::max<unsigned>(MCI->getSourceAlignment(), 1); 1137 } else if (auto *MMI = dyn_cast<MemMoveInst>(&CI)) { 1138 DstAlign = std::max<unsigned>(MMI->getDestAlignment(), 1); 1139 SrcAlign = std::max<unsigned>(MMI->getSourceAlignment(), 1); 1140 } else { 1141 auto *MSI = cast<MemSetInst>(&CI); 1142 DstAlign = std::max<unsigned>(MSI->getDestAlignment(), 1); 1143 } 1144 1145 // Create mem operands to store the alignment and volatile info. 1146 auto VolFlag = IsVol ? MachineMemOperand::MOVolatile : MachineMemOperand::MONone; 1147 ICall.addMemOperand(MF->getMachineMemOperand( 1148 MachinePointerInfo(CI.getArgOperand(0)), 1149 MachineMemOperand::MOStore | VolFlag, 1, DstAlign)); 1150 if (ID != Intrinsic::memset) 1151 ICall.addMemOperand(MF->getMachineMemOperand( 1152 MachinePointerInfo(CI.getArgOperand(1)), 1153 MachineMemOperand::MOLoad | VolFlag, 1, SrcAlign)); 1154 1155 return true; 1156 } 1157 1158 void IRTranslator::getStackGuard(Register DstReg, 1159 MachineIRBuilder &MIRBuilder) { 1160 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo(); 1161 MRI->setRegClass(DstReg, TRI->getPointerRegClass(*MF)); 1162 auto MIB = MIRBuilder.buildInstr(TargetOpcode::LOAD_STACK_GUARD); 1163 MIB.addDef(DstReg); 1164 1165 auto &TLI = *MF->getSubtarget().getTargetLowering(); 1166 Value *Global = TLI.getSDagStackGuard(*MF->getFunction().getParent()); 1167 if (!Global) 1168 return; 1169 1170 MachinePointerInfo MPInfo(Global); 1171 auto Flags = MachineMemOperand::MOLoad | MachineMemOperand::MOInvariant | 1172 MachineMemOperand::MODereferenceable; 1173 MachineMemOperand *MemRef = 1174 MF->getMachineMemOperand(MPInfo, Flags, DL->getPointerSizeInBits() / 8, 1175 DL->getPointerABIAlignment(0)); 1176 MIB.setMemRefs({MemRef}); 1177 } 1178 1179 bool IRTranslator::translateOverflowIntrinsic(const CallInst &CI, unsigned Op, 1180 MachineIRBuilder &MIRBuilder) { 1181 ArrayRef<Register> ResRegs = getOrCreateVRegs(CI); 1182 MIRBuilder.buildInstr(Op) 1183 .addDef(ResRegs[0]) 1184 .addDef(ResRegs[1]) 1185 .addUse(getOrCreateVReg(*CI.getOperand(0))) 1186 .addUse(getOrCreateVReg(*CI.getOperand(1))); 1187 1188 return true; 1189 } 1190 1191 unsigned IRTranslator::getSimpleIntrinsicOpcode(Intrinsic::ID ID) { 1192 switch (ID) { 1193 default: 1194 break; 1195 case Intrinsic::bswap: 1196 return TargetOpcode::G_BSWAP; 1197 case Intrinsic::bitreverse: 1198 return TargetOpcode::G_BITREVERSE; 1199 case Intrinsic::ceil: 1200 return TargetOpcode::G_FCEIL; 1201 case Intrinsic::cos: 1202 return TargetOpcode::G_FCOS; 1203 case Intrinsic::ctpop: 1204 return TargetOpcode::G_CTPOP; 1205 case Intrinsic::exp: 1206 return TargetOpcode::G_FEXP; 1207 case Intrinsic::exp2: 1208 return TargetOpcode::G_FEXP2; 1209 case Intrinsic::fabs: 1210 return TargetOpcode::G_FABS; 1211 case Intrinsic::copysign: 1212 return TargetOpcode::G_FCOPYSIGN; 1213 case Intrinsic::minnum: 1214 return TargetOpcode::G_FMINNUM; 1215 case Intrinsic::maxnum: 1216 return TargetOpcode::G_FMAXNUM; 1217 case Intrinsic::minimum: 1218 return TargetOpcode::G_FMINIMUM; 1219 case Intrinsic::maximum: 1220 return TargetOpcode::G_FMAXIMUM; 1221 case Intrinsic::canonicalize: 1222 return TargetOpcode::G_FCANONICALIZE; 1223 case Intrinsic::floor: 1224 return TargetOpcode::G_FFLOOR; 1225 case Intrinsic::fma: 1226 return TargetOpcode::G_FMA; 1227 case Intrinsic::log: 1228 return TargetOpcode::G_FLOG; 1229 case Intrinsic::log2: 1230 return TargetOpcode::G_FLOG2; 1231 case Intrinsic::log10: 1232 return TargetOpcode::G_FLOG10; 1233 case Intrinsic::nearbyint: 1234 return TargetOpcode::G_FNEARBYINT; 1235 case Intrinsic::pow: 1236 return TargetOpcode::G_FPOW; 1237 case Intrinsic::rint: 1238 return TargetOpcode::G_FRINT; 1239 case Intrinsic::round: 1240 return TargetOpcode::G_INTRINSIC_ROUND; 1241 case Intrinsic::sin: 1242 return TargetOpcode::G_FSIN; 1243 case Intrinsic::sqrt: 1244 return TargetOpcode::G_FSQRT; 1245 case Intrinsic::trunc: 1246 return TargetOpcode::G_INTRINSIC_TRUNC; 1247 } 1248 return Intrinsic::not_intrinsic; 1249 } 1250 1251 bool IRTranslator::translateSimpleIntrinsic(const CallInst &CI, 1252 Intrinsic::ID ID, 1253 MachineIRBuilder &MIRBuilder) { 1254 1255 unsigned Op = getSimpleIntrinsicOpcode(ID); 1256 1257 // Is this a simple intrinsic? 1258 if (Op == Intrinsic::not_intrinsic) 1259 return false; 1260 1261 // Yes. Let's translate it. 1262 SmallVector<llvm::SrcOp, 4> VRegs; 1263 for (auto &Arg : CI.arg_operands()) 1264 VRegs.push_back(getOrCreateVReg(*Arg)); 1265 1266 MIRBuilder.buildInstr(Op, {getOrCreateVReg(CI)}, VRegs, 1267 MachineInstr::copyFlagsFromInstruction(CI)); 1268 return true; 1269 } 1270 1271 bool IRTranslator::translateKnownIntrinsic(const CallInst &CI, Intrinsic::ID ID, 1272 MachineIRBuilder &MIRBuilder) { 1273 1274 // If this is a simple intrinsic (that is, we just need to add a def of 1275 // a vreg, and uses for each arg operand, then translate it. 1276 if (translateSimpleIntrinsic(CI, ID, MIRBuilder)) 1277 return true; 1278 1279 switch (ID) { 1280 default: 1281 break; 1282 case Intrinsic::lifetime_start: 1283 case Intrinsic::lifetime_end: { 1284 // No stack colouring in O0, discard region information. 1285 if (MF->getTarget().getOptLevel() == CodeGenOpt::None) 1286 return true; 1287 1288 unsigned Op = ID == Intrinsic::lifetime_start ? TargetOpcode::LIFETIME_START 1289 : TargetOpcode::LIFETIME_END; 1290 1291 // Get the underlying objects for the location passed on the lifetime 1292 // marker. 1293 SmallVector<const Value *, 4> Allocas; 1294 GetUnderlyingObjects(CI.getArgOperand(1), Allocas, *DL); 1295 1296 // Iterate over each underlying object, creating lifetime markers for each 1297 // static alloca. Quit if we find a non-static alloca. 1298 for (const Value *V : Allocas) { 1299 const AllocaInst *AI = dyn_cast<AllocaInst>(V); 1300 if (!AI) 1301 continue; 1302 1303 if (!AI->isStaticAlloca()) 1304 return true; 1305 1306 MIRBuilder.buildInstr(Op).addFrameIndex(getOrCreateFrameIndex(*AI)); 1307 } 1308 return true; 1309 } 1310 case Intrinsic::dbg_declare: { 1311 const DbgDeclareInst &DI = cast<DbgDeclareInst>(CI); 1312 assert(DI.getVariable() && "Missing variable"); 1313 1314 const Value *Address = DI.getAddress(); 1315 if (!Address || isa<UndefValue>(Address)) { 1316 LLVM_DEBUG(dbgs() << "Dropping debug info for " << DI << "\n"); 1317 return true; 1318 } 1319 1320 assert(DI.getVariable()->isValidLocationForIntrinsic( 1321 MIRBuilder.getDebugLoc()) && 1322 "Expected inlined-at fields to agree"); 1323 auto AI = dyn_cast<AllocaInst>(Address); 1324 if (AI && AI->isStaticAlloca()) { 1325 // Static allocas are tracked at the MF level, no need for DBG_VALUE 1326 // instructions (in fact, they get ignored if they *do* exist). 1327 MF->setVariableDbgInfo(DI.getVariable(), DI.getExpression(), 1328 getOrCreateFrameIndex(*AI), DI.getDebugLoc()); 1329 } else { 1330 // A dbg.declare describes the address of a source variable, so lower it 1331 // into an indirect DBG_VALUE. 1332 MIRBuilder.buildIndirectDbgValue(getOrCreateVReg(*Address), 1333 DI.getVariable(), DI.getExpression()); 1334 } 1335 return true; 1336 } 1337 case Intrinsic::dbg_label: { 1338 const DbgLabelInst &DI = cast<DbgLabelInst>(CI); 1339 assert(DI.getLabel() && "Missing label"); 1340 1341 assert(DI.getLabel()->isValidLocationForIntrinsic( 1342 MIRBuilder.getDebugLoc()) && 1343 "Expected inlined-at fields to agree"); 1344 1345 MIRBuilder.buildDbgLabel(DI.getLabel()); 1346 return true; 1347 } 1348 case Intrinsic::vaend: 1349 // No target I know of cares about va_end. Certainly no in-tree target 1350 // does. Simplest intrinsic ever! 1351 return true; 1352 case Intrinsic::vastart: { 1353 auto &TLI = *MF->getSubtarget().getTargetLowering(); 1354 Value *Ptr = CI.getArgOperand(0); 1355 unsigned ListSize = TLI.getVaListSizeInBits(*DL) / 8; 1356 1357 // FIXME: Get alignment 1358 MIRBuilder.buildInstr(TargetOpcode::G_VASTART) 1359 .addUse(getOrCreateVReg(*Ptr)) 1360 .addMemOperand(MF->getMachineMemOperand( 1361 MachinePointerInfo(Ptr), MachineMemOperand::MOStore, ListSize, 1)); 1362 return true; 1363 } 1364 case Intrinsic::dbg_value: { 1365 // This form of DBG_VALUE is target-independent. 1366 const DbgValueInst &DI = cast<DbgValueInst>(CI); 1367 const Value *V = DI.getValue(); 1368 assert(DI.getVariable()->isValidLocationForIntrinsic( 1369 MIRBuilder.getDebugLoc()) && 1370 "Expected inlined-at fields to agree"); 1371 if (!V) { 1372 // Currently the optimizer can produce this; insert an undef to 1373 // help debugging. Probably the optimizer should not do this. 1374 MIRBuilder.buildIndirectDbgValue(0, DI.getVariable(), DI.getExpression()); 1375 } else if (const auto *CI = dyn_cast<Constant>(V)) { 1376 MIRBuilder.buildConstDbgValue(*CI, DI.getVariable(), DI.getExpression()); 1377 } else { 1378 for (Register Reg : getOrCreateVRegs(*V)) { 1379 // FIXME: This does not handle register-indirect values at offset 0. The 1380 // direct/indirect thing shouldn't really be handled by something as 1381 // implicit as reg+noreg vs reg+imm in the first place, but it seems 1382 // pretty baked in right now. 1383 MIRBuilder.buildDirectDbgValue(Reg, DI.getVariable(), DI.getExpression()); 1384 } 1385 } 1386 return true; 1387 } 1388 case Intrinsic::uadd_with_overflow: 1389 return translateOverflowIntrinsic(CI, TargetOpcode::G_UADDO, MIRBuilder); 1390 case Intrinsic::sadd_with_overflow: 1391 return translateOverflowIntrinsic(CI, TargetOpcode::G_SADDO, MIRBuilder); 1392 case Intrinsic::usub_with_overflow: 1393 return translateOverflowIntrinsic(CI, TargetOpcode::G_USUBO, MIRBuilder); 1394 case Intrinsic::ssub_with_overflow: 1395 return translateOverflowIntrinsic(CI, TargetOpcode::G_SSUBO, MIRBuilder); 1396 case Intrinsic::umul_with_overflow: 1397 return translateOverflowIntrinsic(CI, TargetOpcode::G_UMULO, MIRBuilder); 1398 case Intrinsic::smul_with_overflow: 1399 return translateOverflowIntrinsic(CI, TargetOpcode::G_SMULO, MIRBuilder); 1400 case Intrinsic::fmuladd: { 1401 const TargetMachine &TM = MF->getTarget(); 1402 const TargetLowering &TLI = *MF->getSubtarget().getTargetLowering(); 1403 Register Dst = getOrCreateVReg(CI); 1404 Register Op0 = getOrCreateVReg(*CI.getArgOperand(0)); 1405 Register Op1 = getOrCreateVReg(*CI.getArgOperand(1)); 1406 Register Op2 = getOrCreateVReg(*CI.getArgOperand(2)); 1407 if (TM.Options.AllowFPOpFusion != FPOpFusion::Strict && 1408 TLI.isFMAFasterThanFMulAndFAdd(TLI.getValueType(*DL, CI.getType()))) { 1409 // TODO: Revisit this to see if we should move this part of the 1410 // lowering to the combiner. 1411 MIRBuilder.buildInstr(TargetOpcode::G_FMA, {Dst}, {Op0, Op1, Op2}, 1412 MachineInstr::copyFlagsFromInstruction(CI)); 1413 } else { 1414 LLT Ty = getLLTForType(*CI.getType(), *DL); 1415 auto FMul = MIRBuilder.buildInstr(TargetOpcode::G_FMUL, {Ty}, {Op0, Op1}, 1416 MachineInstr::copyFlagsFromInstruction(CI)); 1417 MIRBuilder.buildInstr(TargetOpcode::G_FADD, {Dst}, {FMul, Op2}, 1418 MachineInstr::copyFlagsFromInstruction(CI)); 1419 } 1420 return true; 1421 } 1422 case Intrinsic::memcpy: 1423 case Intrinsic::memmove: 1424 case Intrinsic::memset: 1425 return translateMemFunc(CI, MIRBuilder, ID); 1426 case Intrinsic::eh_typeid_for: { 1427 GlobalValue *GV = ExtractTypeInfo(CI.getArgOperand(0)); 1428 Register Reg = getOrCreateVReg(CI); 1429 unsigned TypeID = MF->getTypeIDFor(GV); 1430 MIRBuilder.buildConstant(Reg, TypeID); 1431 return true; 1432 } 1433 case Intrinsic::objectsize: { 1434 // If we don't know by now, we're never going to know. 1435 const ConstantInt *Min = cast<ConstantInt>(CI.getArgOperand(1)); 1436 1437 MIRBuilder.buildConstant(getOrCreateVReg(CI), Min->isZero() ? -1ULL : 0); 1438 return true; 1439 } 1440 case Intrinsic::is_constant: 1441 // If this wasn't constant-folded away by now, then it's not a 1442 // constant. 1443 MIRBuilder.buildConstant(getOrCreateVReg(CI), 0); 1444 return true; 1445 case Intrinsic::stackguard: 1446 getStackGuard(getOrCreateVReg(CI), MIRBuilder); 1447 return true; 1448 case Intrinsic::stackprotector: { 1449 LLT PtrTy = getLLTForType(*CI.getArgOperand(0)->getType(), *DL); 1450 Register GuardVal = MRI->createGenericVirtualRegister(PtrTy); 1451 getStackGuard(GuardVal, MIRBuilder); 1452 1453 AllocaInst *Slot = cast<AllocaInst>(CI.getArgOperand(1)); 1454 int FI = getOrCreateFrameIndex(*Slot); 1455 MF->getFrameInfo().setStackProtectorIndex(FI); 1456 1457 MIRBuilder.buildStore( 1458 GuardVal, getOrCreateVReg(*Slot), 1459 *MF->getMachineMemOperand(MachinePointerInfo::getFixedStack(*MF, FI), 1460 MachineMemOperand::MOStore | 1461 MachineMemOperand::MOVolatile, 1462 PtrTy.getSizeInBits() / 8, 8)); 1463 return true; 1464 } 1465 case Intrinsic::stacksave: { 1466 // Save the stack pointer to the location provided by the intrinsic. 1467 Register Reg = getOrCreateVReg(CI); 1468 Register StackPtr = MF->getSubtarget() 1469 .getTargetLowering() 1470 ->getStackPointerRegisterToSaveRestore(); 1471 1472 // If the target doesn't specify a stack pointer, then fall back. 1473 if (!StackPtr) 1474 return false; 1475 1476 MIRBuilder.buildCopy(Reg, StackPtr); 1477 return true; 1478 } 1479 case Intrinsic::stackrestore: { 1480 // Restore the stack pointer from the location provided by the intrinsic. 1481 Register Reg = getOrCreateVReg(*CI.getArgOperand(0)); 1482 Register StackPtr = MF->getSubtarget() 1483 .getTargetLowering() 1484 ->getStackPointerRegisterToSaveRestore(); 1485 1486 // If the target doesn't specify a stack pointer, then fall back. 1487 if (!StackPtr) 1488 return false; 1489 1490 MIRBuilder.buildCopy(StackPtr, Reg); 1491 return true; 1492 } 1493 case Intrinsic::cttz: 1494 case Intrinsic::ctlz: { 1495 ConstantInt *Cst = cast<ConstantInt>(CI.getArgOperand(1)); 1496 bool isTrailing = ID == Intrinsic::cttz; 1497 unsigned Opcode = isTrailing 1498 ? Cst->isZero() ? TargetOpcode::G_CTTZ 1499 : TargetOpcode::G_CTTZ_ZERO_UNDEF 1500 : Cst->isZero() ? TargetOpcode::G_CTLZ 1501 : TargetOpcode::G_CTLZ_ZERO_UNDEF; 1502 MIRBuilder.buildInstr(Opcode) 1503 .addDef(getOrCreateVReg(CI)) 1504 .addUse(getOrCreateVReg(*CI.getArgOperand(0))); 1505 return true; 1506 } 1507 case Intrinsic::invariant_start: { 1508 LLT PtrTy = getLLTForType(*CI.getArgOperand(0)->getType(), *DL); 1509 Register Undef = MRI->createGenericVirtualRegister(PtrTy); 1510 MIRBuilder.buildUndef(Undef); 1511 return true; 1512 } 1513 case Intrinsic::invariant_end: 1514 return true; 1515 case Intrinsic::assume: 1516 case Intrinsic::var_annotation: 1517 case Intrinsic::sideeffect: 1518 // Discard annotate attributes, assumptions, and artificial side-effects. 1519 return true; 1520 } 1521 return false; 1522 } 1523 1524 bool IRTranslator::translateInlineAsm(const CallInst &CI, 1525 MachineIRBuilder &MIRBuilder) { 1526 const InlineAsm &IA = cast<InlineAsm>(*CI.getCalledValue()); 1527 if (!IA.getConstraintString().empty()) 1528 return false; 1529 1530 unsigned ExtraInfo = 0; 1531 if (IA.hasSideEffects()) 1532 ExtraInfo |= InlineAsm::Extra_HasSideEffects; 1533 if (IA.getDialect() == InlineAsm::AD_Intel) 1534 ExtraInfo |= InlineAsm::Extra_AsmDialect; 1535 1536 MIRBuilder.buildInstr(TargetOpcode::INLINEASM) 1537 .addExternalSymbol(IA.getAsmString().c_str()) 1538 .addImm(ExtraInfo); 1539 1540 return true; 1541 } 1542 1543 bool IRTranslator::translateCallSite(const ImmutableCallSite &CS, 1544 MachineIRBuilder &MIRBuilder) { 1545 const Instruction &I = *CS.getInstruction(); 1546 ArrayRef<Register> Res = getOrCreateVRegs(I); 1547 1548 SmallVector<ArrayRef<Register>, 8> Args; 1549 Register SwiftInVReg = 0; 1550 Register SwiftErrorVReg = 0; 1551 for (auto &Arg : CS.args()) { 1552 if (CLI->supportSwiftError() && isSwiftError(Arg)) { 1553 assert(SwiftInVReg == 0 && "Expected only one swift error argument"); 1554 LLT Ty = getLLTForType(*Arg->getType(), *DL); 1555 SwiftInVReg = MRI->createGenericVirtualRegister(Ty); 1556 MIRBuilder.buildCopy(SwiftInVReg, SwiftError.getOrCreateVRegUseAt( 1557 &I, &MIRBuilder.getMBB(), Arg)); 1558 Args.emplace_back(makeArrayRef(SwiftInVReg)); 1559 SwiftErrorVReg = 1560 SwiftError.getOrCreateVRegDefAt(&I, &MIRBuilder.getMBB(), Arg); 1561 continue; 1562 } 1563 Args.push_back(getOrCreateVRegs(*Arg)); 1564 } 1565 1566 MF->getFrameInfo().setHasCalls(true); 1567 bool Success = 1568 CLI->lowerCall(MIRBuilder, CS, Res, Args, SwiftErrorVReg, 1569 [&]() { return getOrCreateVReg(*CS.getCalledValue()); }); 1570 1571 return Success; 1572 } 1573 1574 bool IRTranslator::translateCall(const User &U, MachineIRBuilder &MIRBuilder) { 1575 const CallInst &CI = cast<CallInst>(U); 1576 auto TII = MF->getTarget().getIntrinsicInfo(); 1577 const Function *F = CI.getCalledFunction(); 1578 1579 // FIXME: support Windows dllimport function calls. 1580 if (F && F->hasDLLImportStorageClass()) 1581 return false; 1582 1583 if (CI.isInlineAsm()) 1584 return translateInlineAsm(CI, MIRBuilder); 1585 1586 Intrinsic::ID ID = Intrinsic::not_intrinsic; 1587 if (F && F->isIntrinsic()) { 1588 ID = F->getIntrinsicID(); 1589 if (TII && ID == Intrinsic::not_intrinsic) 1590 ID = static_cast<Intrinsic::ID>(TII->getIntrinsicID(F)); 1591 } 1592 1593 if (!F || !F->isIntrinsic() || ID == Intrinsic::not_intrinsic) 1594 return translateCallSite(&CI, MIRBuilder); 1595 1596 assert(ID != Intrinsic::not_intrinsic && "unknown intrinsic"); 1597 1598 if (translateKnownIntrinsic(CI, ID, MIRBuilder)) 1599 return true; 1600 1601 ArrayRef<Register> ResultRegs; 1602 if (!CI.getType()->isVoidTy()) 1603 ResultRegs = getOrCreateVRegs(CI); 1604 1605 // Ignore the callsite attributes. Backend code is most likely not expecting 1606 // an intrinsic to sometimes have side effects and sometimes not. 1607 MachineInstrBuilder MIB = 1608 MIRBuilder.buildIntrinsic(ID, ResultRegs, !F->doesNotAccessMemory()); 1609 if (isa<FPMathOperator>(CI)) 1610 MIB->copyIRFlags(CI); 1611 1612 for (auto &Arg : CI.arg_operands()) { 1613 // Some intrinsics take metadata parameters. Reject them. 1614 if (isa<MetadataAsValue>(Arg)) 1615 return false; 1616 ArrayRef<Register> VRegs = getOrCreateVRegs(*Arg); 1617 if (VRegs.size() > 1) 1618 return false; 1619 MIB.addUse(VRegs[0]); 1620 } 1621 1622 // Add a MachineMemOperand if it is a target mem intrinsic. 1623 const TargetLowering &TLI = *MF->getSubtarget().getTargetLowering(); 1624 TargetLowering::IntrinsicInfo Info; 1625 // TODO: Add a GlobalISel version of getTgtMemIntrinsic. 1626 if (TLI.getTgtMemIntrinsic(Info, CI, *MF, ID)) { 1627 MaybeAlign Align = Info.align; 1628 if (!Align) 1629 Align = MaybeAlign( 1630 DL->getABITypeAlignment(Info.memVT.getTypeForEVT(F->getContext()))); 1631 1632 uint64_t Size = Info.memVT.getStoreSize(); 1633 MIB.addMemOperand(MF->getMachineMemOperand( 1634 MachinePointerInfo(Info.ptrVal), Info.flags, Size, Align->value())); 1635 } 1636 1637 return true; 1638 } 1639 1640 bool IRTranslator::translateInvoke(const User &U, 1641 MachineIRBuilder &MIRBuilder) { 1642 const InvokeInst &I = cast<InvokeInst>(U); 1643 MCContext &Context = MF->getContext(); 1644 1645 const BasicBlock *ReturnBB = I.getSuccessor(0); 1646 const BasicBlock *EHPadBB = I.getSuccessor(1); 1647 1648 const Value *Callee = I.getCalledValue(); 1649 const Function *Fn = dyn_cast<Function>(Callee); 1650 if (isa<InlineAsm>(Callee)) 1651 return false; 1652 1653 // FIXME: support invoking patchpoint and statepoint intrinsics. 1654 if (Fn && Fn->isIntrinsic()) 1655 return false; 1656 1657 // FIXME: support whatever these are. 1658 if (I.countOperandBundlesOfType(LLVMContext::OB_deopt)) 1659 return false; 1660 1661 // FIXME: support Windows exception handling. 1662 if (!isa<LandingPadInst>(EHPadBB->front())) 1663 return false; 1664 1665 // Emit the actual call, bracketed by EH_LABELs so that the MF knows about 1666 // the region covered by the try. 1667 MCSymbol *BeginSymbol = Context.createTempSymbol(); 1668 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL).addSym(BeginSymbol); 1669 1670 if (!translateCallSite(&I, MIRBuilder)) 1671 return false; 1672 1673 MCSymbol *EndSymbol = Context.createTempSymbol(); 1674 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL).addSym(EndSymbol); 1675 1676 // FIXME: track probabilities. 1677 MachineBasicBlock &EHPadMBB = getMBB(*EHPadBB), 1678 &ReturnMBB = getMBB(*ReturnBB); 1679 MF->addInvoke(&EHPadMBB, BeginSymbol, EndSymbol); 1680 MIRBuilder.getMBB().addSuccessor(&ReturnMBB); 1681 MIRBuilder.getMBB().addSuccessor(&EHPadMBB); 1682 MIRBuilder.buildBr(ReturnMBB); 1683 1684 return true; 1685 } 1686 1687 bool IRTranslator::translateCallBr(const User &U, 1688 MachineIRBuilder &MIRBuilder) { 1689 // FIXME: Implement this. 1690 return false; 1691 } 1692 1693 bool IRTranslator::translateLandingPad(const User &U, 1694 MachineIRBuilder &MIRBuilder) { 1695 const LandingPadInst &LP = cast<LandingPadInst>(U); 1696 1697 MachineBasicBlock &MBB = MIRBuilder.getMBB(); 1698 1699 MBB.setIsEHPad(); 1700 1701 // If there aren't registers to copy the values into (e.g., during SjLj 1702 // exceptions), then don't bother. 1703 auto &TLI = *MF->getSubtarget().getTargetLowering(); 1704 const Constant *PersonalityFn = MF->getFunction().getPersonalityFn(); 1705 if (TLI.getExceptionPointerRegister(PersonalityFn) == 0 && 1706 TLI.getExceptionSelectorRegister(PersonalityFn) == 0) 1707 return true; 1708 1709 // If landingpad's return type is token type, we don't create DAG nodes 1710 // for its exception pointer and selector value. The extraction of exception 1711 // pointer or selector value from token type landingpads is not currently 1712 // supported. 1713 if (LP.getType()->isTokenTy()) 1714 return true; 1715 1716 // Add a label to mark the beginning of the landing pad. Deletion of the 1717 // landing pad can thus be detected via the MachineModuleInfo. 1718 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL) 1719 .addSym(MF->addLandingPad(&MBB)); 1720 1721 LLT Ty = getLLTForType(*LP.getType(), *DL); 1722 Register Undef = MRI->createGenericVirtualRegister(Ty); 1723 MIRBuilder.buildUndef(Undef); 1724 1725 SmallVector<LLT, 2> Tys; 1726 for (Type *Ty : cast<StructType>(LP.getType())->elements()) 1727 Tys.push_back(getLLTForType(*Ty, *DL)); 1728 assert(Tys.size() == 2 && "Only two-valued landingpads are supported"); 1729 1730 // Mark exception register as live in. 1731 Register ExceptionReg = TLI.getExceptionPointerRegister(PersonalityFn); 1732 if (!ExceptionReg) 1733 return false; 1734 1735 MBB.addLiveIn(ExceptionReg); 1736 ArrayRef<Register> ResRegs = getOrCreateVRegs(LP); 1737 MIRBuilder.buildCopy(ResRegs[0], ExceptionReg); 1738 1739 Register SelectorReg = TLI.getExceptionSelectorRegister(PersonalityFn); 1740 if (!SelectorReg) 1741 return false; 1742 1743 MBB.addLiveIn(SelectorReg); 1744 Register PtrVReg = MRI->createGenericVirtualRegister(Tys[0]); 1745 MIRBuilder.buildCopy(PtrVReg, SelectorReg); 1746 MIRBuilder.buildCast(ResRegs[1], PtrVReg); 1747 1748 return true; 1749 } 1750 1751 bool IRTranslator::translateAlloca(const User &U, 1752 MachineIRBuilder &MIRBuilder) { 1753 auto &AI = cast<AllocaInst>(U); 1754 1755 if (AI.isSwiftError()) 1756 return true; 1757 1758 if (AI.isStaticAlloca()) { 1759 Register Res = getOrCreateVReg(AI); 1760 int FI = getOrCreateFrameIndex(AI); 1761 MIRBuilder.buildFrameIndex(Res, FI); 1762 return true; 1763 } 1764 1765 // FIXME: support stack probing for Windows. 1766 if (MF->getTarget().getTargetTriple().isOSWindows()) 1767 return false; 1768 1769 // Now we're in the harder dynamic case. 1770 Type *Ty = AI.getAllocatedType(); 1771 unsigned Align = 1772 std::max((unsigned)DL->getPrefTypeAlignment(Ty), AI.getAlignment()); 1773 1774 Register NumElts = getOrCreateVReg(*AI.getArraySize()); 1775 1776 Type *IntPtrIRTy = DL->getIntPtrType(AI.getType()); 1777 LLT IntPtrTy = getLLTForType(*IntPtrIRTy, *DL); 1778 if (MRI->getType(NumElts) != IntPtrTy) { 1779 Register ExtElts = MRI->createGenericVirtualRegister(IntPtrTy); 1780 MIRBuilder.buildZExtOrTrunc(ExtElts, NumElts); 1781 NumElts = ExtElts; 1782 } 1783 1784 Register AllocSize = MRI->createGenericVirtualRegister(IntPtrTy); 1785 Register TySize = 1786 getOrCreateVReg(*ConstantInt::get(IntPtrIRTy, DL->getTypeAllocSize(Ty))); 1787 MIRBuilder.buildMul(AllocSize, NumElts, TySize); 1788 1789 unsigned StackAlign = 1790 MF->getSubtarget().getFrameLowering()->getStackAlignment(); 1791 if (Align <= StackAlign) 1792 Align = 0; 1793 1794 // Round the size of the allocation up to the stack alignment size 1795 // by add SA-1 to the size. This doesn't overflow because we're computing 1796 // an address inside an alloca. 1797 auto SAMinusOne = MIRBuilder.buildConstant(IntPtrTy, StackAlign - 1); 1798 auto AllocAdd = MIRBuilder.buildAdd(IntPtrTy, AllocSize, SAMinusOne, 1799 MachineInstr::NoUWrap); 1800 auto AlignCst = 1801 MIRBuilder.buildConstant(IntPtrTy, ~(uint64_t)(StackAlign - 1)); 1802 auto AlignedAlloc = MIRBuilder.buildAnd(IntPtrTy, AllocAdd, AlignCst); 1803 1804 MIRBuilder.buildDynStackAlloc(getOrCreateVReg(AI), AlignedAlloc, Align); 1805 1806 MF->getFrameInfo().CreateVariableSizedObject(Align ? Align : 1, &AI); 1807 assert(MF->getFrameInfo().hasVarSizedObjects()); 1808 return true; 1809 } 1810 1811 bool IRTranslator::translateVAArg(const User &U, MachineIRBuilder &MIRBuilder) { 1812 // FIXME: We may need more info about the type. Because of how LLT works, 1813 // we're completely discarding the i64/double distinction here (amongst 1814 // others). Fortunately the ABIs I know of where that matters don't use va_arg 1815 // anyway but that's not guaranteed. 1816 MIRBuilder.buildInstr(TargetOpcode::G_VAARG) 1817 .addDef(getOrCreateVReg(U)) 1818 .addUse(getOrCreateVReg(*U.getOperand(0))) 1819 .addImm(DL->getABITypeAlignment(U.getType())); 1820 return true; 1821 } 1822 1823 bool IRTranslator::translateInsertElement(const User &U, 1824 MachineIRBuilder &MIRBuilder) { 1825 // If it is a <1 x Ty> vector, use the scalar as it is 1826 // not a legal vector type in LLT. 1827 if (U.getType()->getVectorNumElements() == 1) { 1828 Register Elt = getOrCreateVReg(*U.getOperand(1)); 1829 auto &Regs = *VMap.getVRegs(U); 1830 if (Regs.empty()) { 1831 Regs.push_back(Elt); 1832 VMap.getOffsets(U)->push_back(0); 1833 } else { 1834 MIRBuilder.buildCopy(Regs[0], Elt); 1835 } 1836 return true; 1837 } 1838 1839 Register Res = getOrCreateVReg(U); 1840 Register Val = getOrCreateVReg(*U.getOperand(0)); 1841 Register Elt = getOrCreateVReg(*U.getOperand(1)); 1842 Register Idx = getOrCreateVReg(*U.getOperand(2)); 1843 MIRBuilder.buildInsertVectorElement(Res, Val, Elt, Idx); 1844 return true; 1845 } 1846 1847 bool IRTranslator::translateExtractElement(const User &U, 1848 MachineIRBuilder &MIRBuilder) { 1849 // If it is a <1 x Ty> vector, use the scalar as it is 1850 // not a legal vector type in LLT. 1851 if (U.getOperand(0)->getType()->getVectorNumElements() == 1) { 1852 Register Elt = getOrCreateVReg(*U.getOperand(0)); 1853 auto &Regs = *VMap.getVRegs(U); 1854 if (Regs.empty()) { 1855 Regs.push_back(Elt); 1856 VMap.getOffsets(U)->push_back(0); 1857 } else { 1858 MIRBuilder.buildCopy(Regs[0], Elt); 1859 } 1860 return true; 1861 } 1862 Register Res = getOrCreateVReg(U); 1863 Register Val = getOrCreateVReg(*U.getOperand(0)); 1864 const auto &TLI = *MF->getSubtarget().getTargetLowering(); 1865 unsigned PreferredVecIdxWidth = TLI.getVectorIdxTy(*DL).getSizeInBits(); 1866 Register Idx; 1867 if (auto *CI = dyn_cast<ConstantInt>(U.getOperand(1))) { 1868 if (CI->getBitWidth() != PreferredVecIdxWidth) { 1869 APInt NewIdx = CI->getValue().sextOrTrunc(PreferredVecIdxWidth); 1870 auto *NewIdxCI = ConstantInt::get(CI->getContext(), NewIdx); 1871 Idx = getOrCreateVReg(*NewIdxCI); 1872 } 1873 } 1874 if (!Idx) 1875 Idx = getOrCreateVReg(*U.getOperand(1)); 1876 if (MRI->getType(Idx).getSizeInBits() != PreferredVecIdxWidth) { 1877 const LLT &VecIdxTy = LLT::scalar(PreferredVecIdxWidth); 1878 Idx = MIRBuilder.buildSExtOrTrunc(VecIdxTy, Idx)->getOperand(0).getReg(); 1879 } 1880 MIRBuilder.buildExtractVectorElement(Res, Val, Idx); 1881 return true; 1882 } 1883 1884 bool IRTranslator::translateShuffleVector(const User &U, 1885 MachineIRBuilder &MIRBuilder) { 1886 MIRBuilder.buildInstr(TargetOpcode::G_SHUFFLE_VECTOR) 1887 .addDef(getOrCreateVReg(U)) 1888 .addUse(getOrCreateVReg(*U.getOperand(0))) 1889 .addUse(getOrCreateVReg(*U.getOperand(1))) 1890 .addShuffleMask(cast<Constant>(U.getOperand(2))); 1891 return true; 1892 } 1893 1894 bool IRTranslator::translatePHI(const User &U, MachineIRBuilder &MIRBuilder) { 1895 const PHINode &PI = cast<PHINode>(U); 1896 1897 SmallVector<MachineInstr *, 4> Insts; 1898 for (auto Reg : getOrCreateVRegs(PI)) { 1899 auto MIB = MIRBuilder.buildInstr(TargetOpcode::G_PHI, {Reg}, {}); 1900 Insts.push_back(MIB.getInstr()); 1901 } 1902 1903 PendingPHIs.emplace_back(&PI, std::move(Insts)); 1904 return true; 1905 } 1906 1907 bool IRTranslator::translateAtomicCmpXchg(const User &U, 1908 MachineIRBuilder &MIRBuilder) { 1909 const AtomicCmpXchgInst &I = cast<AtomicCmpXchgInst>(U); 1910 1911 if (I.isWeak()) 1912 return false; 1913 1914 auto Flags = I.isVolatile() ? MachineMemOperand::MOVolatile 1915 : MachineMemOperand::MONone; 1916 Flags |= MachineMemOperand::MOLoad | MachineMemOperand::MOStore; 1917 1918 Type *ResType = I.getType(); 1919 Type *ValType = ResType->Type::getStructElementType(0); 1920 1921 auto Res = getOrCreateVRegs(I); 1922 Register OldValRes = Res[0]; 1923 Register SuccessRes = Res[1]; 1924 Register Addr = getOrCreateVReg(*I.getPointerOperand()); 1925 Register Cmp = getOrCreateVReg(*I.getCompareOperand()); 1926 Register NewVal = getOrCreateVReg(*I.getNewValOperand()); 1927 1928 MIRBuilder.buildAtomicCmpXchgWithSuccess( 1929 OldValRes, SuccessRes, Addr, Cmp, NewVal, 1930 *MF->getMachineMemOperand(MachinePointerInfo(I.getPointerOperand()), 1931 Flags, DL->getTypeStoreSize(ValType), 1932 getMemOpAlignment(I), AAMDNodes(), nullptr, 1933 I.getSyncScopeID(), I.getSuccessOrdering(), 1934 I.getFailureOrdering())); 1935 return true; 1936 } 1937 1938 bool IRTranslator::translateAtomicRMW(const User &U, 1939 MachineIRBuilder &MIRBuilder) { 1940 const AtomicRMWInst &I = cast<AtomicRMWInst>(U); 1941 1942 auto Flags = I.isVolatile() ? MachineMemOperand::MOVolatile 1943 : MachineMemOperand::MONone; 1944 Flags |= MachineMemOperand::MOLoad | MachineMemOperand::MOStore; 1945 1946 Type *ResType = I.getType(); 1947 1948 Register Res = getOrCreateVReg(I); 1949 Register Addr = getOrCreateVReg(*I.getPointerOperand()); 1950 Register Val = getOrCreateVReg(*I.getValOperand()); 1951 1952 unsigned Opcode = 0; 1953 switch (I.getOperation()) { 1954 default: 1955 return false; 1956 case AtomicRMWInst::Xchg: 1957 Opcode = TargetOpcode::G_ATOMICRMW_XCHG; 1958 break; 1959 case AtomicRMWInst::Add: 1960 Opcode = TargetOpcode::G_ATOMICRMW_ADD; 1961 break; 1962 case AtomicRMWInst::Sub: 1963 Opcode = TargetOpcode::G_ATOMICRMW_SUB; 1964 break; 1965 case AtomicRMWInst::And: 1966 Opcode = TargetOpcode::G_ATOMICRMW_AND; 1967 break; 1968 case AtomicRMWInst::Nand: 1969 Opcode = TargetOpcode::G_ATOMICRMW_NAND; 1970 break; 1971 case AtomicRMWInst::Or: 1972 Opcode = TargetOpcode::G_ATOMICRMW_OR; 1973 break; 1974 case AtomicRMWInst::Xor: 1975 Opcode = TargetOpcode::G_ATOMICRMW_XOR; 1976 break; 1977 case AtomicRMWInst::Max: 1978 Opcode = TargetOpcode::G_ATOMICRMW_MAX; 1979 break; 1980 case AtomicRMWInst::Min: 1981 Opcode = TargetOpcode::G_ATOMICRMW_MIN; 1982 break; 1983 case AtomicRMWInst::UMax: 1984 Opcode = TargetOpcode::G_ATOMICRMW_UMAX; 1985 break; 1986 case AtomicRMWInst::UMin: 1987 Opcode = TargetOpcode::G_ATOMICRMW_UMIN; 1988 break; 1989 case AtomicRMWInst::FAdd: 1990 Opcode = TargetOpcode::G_ATOMICRMW_FADD; 1991 break; 1992 case AtomicRMWInst::FSub: 1993 Opcode = TargetOpcode::G_ATOMICRMW_FSUB; 1994 break; 1995 } 1996 1997 MIRBuilder.buildAtomicRMW( 1998 Opcode, Res, Addr, Val, 1999 *MF->getMachineMemOperand(MachinePointerInfo(I.getPointerOperand()), 2000 Flags, DL->getTypeStoreSize(ResType), 2001 getMemOpAlignment(I), AAMDNodes(), nullptr, 2002 I.getSyncScopeID(), I.getOrdering())); 2003 return true; 2004 } 2005 2006 bool IRTranslator::translateFence(const User &U, 2007 MachineIRBuilder &MIRBuilder) { 2008 const FenceInst &Fence = cast<FenceInst>(U); 2009 MIRBuilder.buildFence(static_cast<unsigned>(Fence.getOrdering()), 2010 Fence.getSyncScopeID()); 2011 return true; 2012 } 2013 2014 void IRTranslator::finishPendingPhis() { 2015 #ifndef NDEBUG 2016 DILocationVerifier Verifier; 2017 GISelObserverWrapper WrapperObserver(&Verifier); 2018 RAIIDelegateInstaller DelInstall(*MF, &WrapperObserver); 2019 #endif // ifndef NDEBUG 2020 for (auto &Phi : PendingPHIs) { 2021 const PHINode *PI = Phi.first; 2022 ArrayRef<MachineInstr *> ComponentPHIs = Phi.second; 2023 MachineBasicBlock *PhiMBB = ComponentPHIs[0]->getParent(); 2024 EntryBuilder->setDebugLoc(PI->getDebugLoc()); 2025 #ifndef NDEBUG 2026 Verifier.setCurrentInst(PI); 2027 #endif // ifndef NDEBUG 2028 2029 SmallSet<const MachineBasicBlock *, 16> SeenPreds; 2030 for (unsigned i = 0; i < PI->getNumIncomingValues(); ++i) { 2031 auto IRPred = PI->getIncomingBlock(i); 2032 ArrayRef<Register> ValRegs = getOrCreateVRegs(*PI->getIncomingValue(i)); 2033 for (auto Pred : getMachinePredBBs({IRPred, PI->getParent()})) { 2034 if (SeenPreds.count(Pred) || !PhiMBB->isPredecessor(Pred)) 2035 continue; 2036 SeenPreds.insert(Pred); 2037 for (unsigned j = 0; j < ValRegs.size(); ++j) { 2038 MachineInstrBuilder MIB(*MF, ComponentPHIs[j]); 2039 MIB.addUse(ValRegs[j]); 2040 MIB.addMBB(Pred); 2041 } 2042 } 2043 } 2044 } 2045 } 2046 2047 bool IRTranslator::valueIsSplit(const Value &V, 2048 SmallVectorImpl<uint64_t> *Offsets) { 2049 SmallVector<LLT, 4> SplitTys; 2050 if (Offsets && !Offsets->empty()) 2051 Offsets->clear(); 2052 computeValueLLTs(*DL, *V.getType(), SplitTys, Offsets); 2053 return SplitTys.size() > 1; 2054 } 2055 2056 bool IRTranslator::translate(const Instruction &Inst) { 2057 CurBuilder->setDebugLoc(Inst.getDebugLoc()); 2058 // We only emit constants into the entry block from here. To prevent jumpy 2059 // debug behaviour set the line to 0. 2060 if (const DebugLoc &DL = Inst.getDebugLoc()) 2061 EntryBuilder->setDebugLoc( 2062 DebugLoc::get(0, 0, DL.getScope(), DL.getInlinedAt())); 2063 else 2064 EntryBuilder->setDebugLoc(DebugLoc()); 2065 2066 switch (Inst.getOpcode()) { 2067 #define HANDLE_INST(NUM, OPCODE, CLASS) \ 2068 case Instruction::OPCODE: \ 2069 return translate##OPCODE(Inst, *CurBuilder.get()); 2070 #include "llvm/IR/Instruction.def" 2071 default: 2072 return false; 2073 } 2074 } 2075 2076 bool IRTranslator::translate(const Constant &C, Register Reg) { 2077 if (auto CI = dyn_cast<ConstantInt>(&C)) 2078 EntryBuilder->buildConstant(Reg, *CI); 2079 else if (auto CF = dyn_cast<ConstantFP>(&C)) 2080 EntryBuilder->buildFConstant(Reg, *CF); 2081 else if (isa<UndefValue>(C)) 2082 EntryBuilder->buildUndef(Reg); 2083 else if (isa<ConstantPointerNull>(C)) { 2084 // As we are trying to build a constant val of 0 into a pointer, 2085 // insert a cast to make them correct with respect to types. 2086 unsigned NullSize = DL->getTypeSizeInBits(C.getType()); 2087 auto *ZeroTy = Type::getIntNTy(C.getContext(), NullSize); 2088 auto *ZeroVal = ConstantInt::get(ZeroTy, 0); 2089 Register ZeroReg = getOrCreateVReg(*ZeroVal); 2090 EntryBuilder->buildCast(Reg, ZeroReg); 2091 } else if (auto GV = dyn_cast<GlobalValue>(&C)) 2092 EntryBuilder->buildGlobalValue(Reg, GV); 2093 else if (auto CAZ = dyn_cast<ConstantAggregateZero>(&C)) { 2094 if (!CAZ->getType()->isVectorTy()) 2095 return false; 2096 // Return the scalar if it is a <1 x Ty> vector. 2097 if (CAZ->getNumElements() == 1) 2098 return translate(*CAZ->getElementValue(0u), Reg); 2099 SmallVector<Register, 4> Ops; 2100 for (unsigned i = 0; i < CAZ->getNumElements(); ++i) { 2101 Constant &Elt = *CAZ->getElementValue(i); 2102 Ops.push_back(getOrCreateVReg(Elt)); 2103 } 2104 EntryBuilder->buildBuildVector(Reg, Ops); 2105 } else if (auto CV = dyn_cast<ConstantDataVector>(&C)) { 2106 // Return the scalar if it is a <1 x Ty> vector. 2107 if (CV->getNumElements() == 1) 2108 return translate(*CV->getElementAsConstant(0), Reg); 2109 SmallVector<Register, 4> Ops; 2110 for (unsigned i = 0; i < CV->getNumElements(); ++i) { 2111 Constant &Elt = *CV->getElementAsConstant(i); 2112 Ops.push_back(getOrCreateVReg(Elt)); 2113 } 2114 EntryBuilder->buildBuildVector(Reg, Ops); 2115 } else if (auto CE = dyn_cast<ConstantExpr>(&C)) { 2116 switch(CE->getOpcode()) { 2117 #define HANDLE_INST(NUM, OPCODE, CLASS) \ 2118 case Instruction::OPCODE: \ 2119 return translate##OPCODE(*CE, *EntryBuilder.get()); 2120 #include "llvm/IR/Instruction.def" 2121 default: 2122 return false; 2123 } 2124 } else if (auto CV = dyn_cast<ConstantVector>(&C)) { 2125 if (CV->getNumOperands() == 1) 2126 return translate(*CV->getOperand(0), Reg); 2127 SmallVector<Register, 4> Ops; 2128 for (unsigned i = 0; i < CV->getNumOperands(); ++i) { 2129 Ops.push_back(getOrCreateVReg(*CV->getOperand(i))); 2130 } 2131 EntryBuilder->buildBuildVector(Reg, Ops); 2132 } else if (auto *BA = dyn_cast<BlockAddress>(&C)) { 2133 EntryBuilder->buildBlockAddress(Reg, BA); 2134 } else 2135 return false; 2136 2137 return true; 2138 } 2139 2140 void IRTranslator::finalizeBasicBlock() { 2141 for (auto &JTCase : SL->JTCases) { 2142 // Emit header first, if it wasn't already emitted. 2143 if (!JTCase.first.Emitted) 2144 emitJumpTableHeader(JTCase.second, JTCase.first, JTCase.first.HeaderBB); 2145 2146 emitJumpTable(JTCase.second, JTCase.second.MBB); 2147 } 2148 SL->JTCases.clear(); 2149 } 2150 2151 void IRTranslator::finalizeFunction() { 2152 // Release the memory used by the different maps we 2153 // needed during the translation. 2154 PendingPHIs.clear(); 2155 VMap.reset(); 2156 FrameIndices.clear(); 2157 MachinePreds.clear(); 2158 // MachineIRBuilder::DebugLoc can outlive the DILocation it holds. Clear it 2159 // to avoid accessing free’d memory (in runOnMachineFunction) and to avoid 2160 // destroying it twice (in ~IRTranslator() and ~LLVMContext()) 2161 EntryBuilder.reset(); 2162 CurBuilder.reset(); 2163 FuncInfo.clear(); 2164 } 2165 2166 bool IRTranslator::runOnMachineFunction(MachineFunction &CurMF) { 2167 MF = &CurMF; 2168 const Function &F = MF->getFunction(); 2169 if (F.empty()) 2170 return false; 2171 GISelCSEAnalysisWrapper &Wrapper = 2172 getAnalysis<GISelCSEAnalysisWrapperPass>().getCSEWrapper(); 2173 // Set the CSEConfig and run the analysis. 2174 GISelCSEInfo *CSEInfo = nullptr; 2175 TPC = &getAnalysis<TargetPassConfig>(); 2176 bool EnableCSE = EnableCSEInIRTranslator.getNumOccurrences() 2177 ? EnableCSEInIRTranslator 2178 : TPC->isGISelCSEEnabled(); 2179 2180 if (EnableCSE) { 2181 EntryBuilder = std::make_unique<CSEMIRBuilder>(CurMF); 2182 CSEInfo = &Wrapper.get(TPC->getCSEConfig()); 2183 EntryBuilder->setCSEInfo(CSEInfo); 2184 CurBuilder = std::make_unique<CSEMIRBuilder>(CurMF); 2185 CurBuilder->setCSEInfo(CSEInfo); 2186 } else { 2187 EntryBuilder = std::make_unique<MachineIRBuilder>(); 2188 CurBuilder = std::make_unique<MachineIRBuilder>(); 2189 } 2190 CLI = MF->getSubtarget().getCallLowering(); 2191 CurBuilder->setMF(*MF); 2192 EntryBuilder->setMF(*MF); 2193 MRI = &MF->getRegInfo(); 2194 DL = &F.getParent()->getDataLayout(); 2195 ORE = std::make_unique<OptimizationRemarkEmitter>(&F); 2196 FuncInfo.MF = MF; 2197 FuncInfo.BPI = nullptr; 2198 const auto &TLI = *MF->getSubtarget().getTargetLowering(); 2199 const TargetMachine &TM = MF->getTarget(); 2200 SL = std::make_unique<GISelSwitchLowering>(this, FuncInfo); 2201 SL->init(TLI, TM, *DL); 2202 2203 EnableOpts = TM.getOptLevel() != CodeGenOpt::None && !skipFunction(F); 2204 2205 assert(PendingPHIs.empty() && "stale PHIs"); 2206 2207 if (!DL->isLittleEndian()) { 2208 // Currently we don't properly handle big endian code. 2209 OptimizationRemarkMissed R("gisel-irtranslator", "GISelFailure", 2210 F.getSubprogram(), &F.getEntryBlock()); 2211 R << "unable to translate in big endian mode"; 2212 reportTranslationError(*MF, *TPC, *ORE, R); 2213 } 2214 2215 // Release the per-function state when we return, whether we succeeded or not. 2216 auto FinalizeOnReturn = make_scope_exit([this]() { finalizeFunction(); }); 2217 2218 // Setup a separate basic-block for the arguments and constants 2219 MachineBasicBlock *EntryBB = MF->CreateMachineBasicBlock(); 2220 MF->push_back(EntryBB); 2221 EntryBuilder->setMBB(*EntryBB); 2222 2223 DebugLoc DbgLoc = F.getEntryBlock().getFirstNonPHI()->getDebugLoc(); 2224 SwiftError.setFunction(CurMF); 2225 SwiftError.createEntriesInEntryBlock(DbgLoc); 2226 2227 // Create all blocks, in IR order, to preserve the layout. 2228 for (const BasicBlock &BB: F) { 2229 auto *&MBB = BBToMBB[&BB]; 2230 2231 MBB = MF->CreateMachineBasicBlock(&BB); 2232 MF->push_back(MBB); 2233 2234 if (BB.hasAddressTaken()) 2235 MBB->setHasAddressTaken(); 2236 } 2237 2238 // Make our arguments/constants entry block fallthrough to the IR entry block. 2239 EntryBB->addSuccessor(&getMBB(F.front())); 2240 2241 // Lower the actual args into this basic block. 2242 SmallVector<ArrayRef<Register>, 8> VRegArgs; 2243 for (const Argument &Arg: F.args()) { 2244 if (DL->getTypeStoreSize(Arg.getType()) == 0) 2245 continue; // Don't handle zero sized types. 2246 ArrayRef<Register> VRegs = getOrCreateVRegs(Arg); 2247 VRegArgs.push_back(VRegs); 2248 2249 if (Arg.hasSwiftErrorAttr()) { 2250 assert(VRegs.size() == 1 && "Too many vregs for Swift error"); 2251 SwiftError.setCurrentVReg(EntryBB, SwiftError.getFunctionArg(), VRegs[0]); 2252 } 2253 } 2254 2255 if (!CLI->lowerFormalArguments(*EntryBuilder.get(), F, VRegArgs)) { 2256 OptimizationRemarkMissed R("gisel-irtranslator", "GISelFailure", 2257 F.getSubprogram(), &F.getEntryBlock()); 2258 R << "unable to lower arguments: " << ore::NV("Prototype", F.getType()); 2259 reportTranslationError(*MF, *TPC, *ORE, R); 2260 return false; 2261 } 2262 2263 // Need to visit defs before uses when translating instructions. 2264 GISelObserverWrapper WrapperObserver; 2265 if (EnableCSE && CSEInfo) 2266 WrapperObserver.addObserver(CSEInfo); 2267 { 2268 ReversePostOrderTraversal<const Function *> RPOT(&F); 2269 #ifndef NDEBUG 2270 DILocationVerifier Verifier; 2271 WrapperObserver.addObserver(&Verifier); 2272 #endif // ifndef NDEBUG 2273 RAIIDelegateInstaller DelInstall(*MF, &WrapperObserver); 2274 for (const BasicBlock *BB : RPOT) { 2275 MachineBasicBlock &MBB = getMBB(*BB); 2276 // Set the insertion point of all the following translations to 2277 // the end of this basic block. 2278 CurBuilder->setMBB(MBB); 2279 2280 for (const Instruction &Inst : *BB) { 2281 #ifndef NDEBUG 2282 Verifier.setCurrentInst(&Inst); 2283 #endif // ifndef NDEBUG 2284 if (translate(Inst)) 2285 continue; 2286 2287 OptimizationRemarkMissed R("gisel-irtranslator", "GISelFailure", 2288 Inst.getDebugLoc(), BB); 2289 R << "unable to translate instruction: " << ore::NV("Opcode", &Inst); 2290 2291 if (ORE->allowExtraAnalysis("gisel-irtranslator")) { 2292 std::string InstStrStorage; 2293 raw_string_ostream InstStr(InstStrStorage); 2294 InstStr << Inst; 2295 2296 R << ": '" << InstStr.str() << "'"; 2297 } 2298 2299 reportTranslationError(*MF, *TPC, *ORE, R); 2300 return false; 2301 } 2302 2303 finalizeBasicBlock(); 2304 } 2305 #ifndef NDEBUG 2306 WrapperObserver.removeObserver(&Verifier); 2307 #endif 2308 } 2309 2310 finishPendingPhis(); 2311 2312 SwiftError.propagateVRegs(); 2313 2314 // Merge the argument lowering and constants block with its single 2315 // successor, the LLVM-IR entry block. We want the basic block to 2316 // be maximal. 2317 assert(EntryBB->succ_size() == 1 && 2318 "Custom BB used for lowering should have only one successor"); 2319 // Get the successor of the current entry block. 2320 MachineBasicBlock &NewEntryBB = **EntryBB->succ_begin(); 2321 assert(NewEntryBB.pred_size() == 1 && 2322 "LLVM-IR entry block has a predecessor!?"); 2323 // Move all the instruction from the current entry block to the 2324 // new entry block. 2325 NewEntryBB.splice(NewEntryBB.begin(), EntryBB, EntryBB->begin(), 2326 EntryBB->end()); 2327 2328 // Update the live-in information for the new entry block. 2329 for (const MachineBasicBlock::RegisterMaskPair &LiveIn : EntryBB->liveins()) 2330 NewEntryBB.addLiveIn(LiveIn); 2331 NewEntryBB.sortUniqueLiveIns(); 2332 2333 // Get rid of the now empty basic block. 2334 EntryBB->removeSuccessor(&NewEntryBB); 2335 MF->remove(EntryBB); 2336 MF->DeleteMachineBasicBlock(EntryBB); 2337 2338 assert(&MF->front() == &NewEntryBB && 2339 "New entry wasn't next in the list of basic block!"); 2340 2341 // Initialize stack protector information. 2342 StackProtector &SP = getAnalysis<StackProtector>(); 2343 SP.copyToMachineFrameInfo(MF->getFrameInfo()); 2344 2345 return false; 2346 } 2347