1 //===- MachineFunction.cpp ------------------------------------------------===// 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 // 9 // Collect native machine code information for a function. This allows 10 // target-specific information about the generated code to be stored with each 11 // function. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "llvm/CodeGen/MachineFunction.h" 16 #include "llvm/ADT/BitVector.h" 17 #include "llvm/ADT/DenseMap.h" 18 #include "llvm/ADT/DenseSet.h" 19 #include "llvm/ADT/STLExtras.h" 20 #include "llvm/ADT/SmallString.h" 21 #include "llvm/ADT/SmallVector.h" 22 #include "llvm/ADT/StringRef.h" 23 #include "llvm/ADT/Twine.h" 24 #include "llvm/Analysis/ConstantFolding.h" 25 #include "llvm/Analysis/EHPersonalities.h" 26 #include "llvm/CodeGen/MachineBasicBlock.h" 27 #include "llvm/CodeGen/MachineConstantPool.h" 28 #include "llvm/CodeGen/MachineFrameInfo.h" 29 #include "llvm/CodeGen/MachineInstr.h" 30 #include "llvm/CodeGen/MachineJumpTableInfo.h" 31 #include "llvm/CodeGen/MachineMemOperand.h" 32 #include "llvm/CodeGen/MachineModuleInfo.h" 33 #include "llvm/CodeGen/MachineRegisterInfo.h" 34 #include "llvm/CodeGen/PseudoSourceValue.h" 35 #include "llvm/CodeGen/TargetFrameLowering.h" 36 #include "llvm/CodeGen/TargetInstrInfo.h" 37 #include "llvm/CodeGen/TargetLowering.h" 38 #include "llvm/CodeGen/TargetRegisterInfo.h" 39 #include "llvm/CodeGen/TargetSubtargetInfo.h" 40 #include "llvm/CodeGen/WasmEHFuncInfo.h" 41 #include "llvm/CodeGen/WinEHFuncInfo.h" 42 #include "llvm/Config/llvm-config.h" 43 #include "llvm/IR/Attributes.h" 44 #include "llvm/IR/BasicBlock.h" 45 #include "llvm/IR/Constant.h" 46 #include "llvm/IR/DataLayout.h" 47 #include "llvm/IR/DerivedTypes.h" 48 #include "llvm/IR/Function.h" 49 #include "llvm/IR/GlobalValue.h" 50 #include "llvm/IR/Instruction.h" 51 #include "llvm/IR/Instructions.h" 52 #include "llvm/IR/Metadata.h" 53 #include "llvm/IR/Module.h" 54 #include "llvm/IR/ModuleSlotTracker.h" 55 #include "llvm/IR/Value.h" 56 #include "llvm/MC/MCContext.h" 57 #include "llvm/MC/MCSymbol.h" 58 #include "llvm/MC/SectionKind.h" 59 #include "llvm/Support/Casting.h" 60 #include "llvm/Support/CommandLine.h" 61 #include "llvm/Support/Compiler.h" 62 #include "llvm/Support/DOTGraphTraits.h" 63 #include "llvm/Support/ErrorHandling.h" 64 #include "llvm/Support/GraphWriter.h" 65 #include "llvm/Support/raw_ostream.h" 66 #include "llvm/Target/TargetMachine.h" 67 #include <algorithm> 68 #include <cassert> 69 #include <cstddef> 70 #include <cstdint> 71 #include <iterator> 72 #include <string> 73 #include <type_traits> 74 #include <utility> 75 #include <vector> 76 77 #include "LiveDebugValues/LiveDebugValues.h" 78 79 using namespace llvm; 80 81 #define DEBUG_TYPE "codegen" 82 83 static cl::opt<unsigned> AlignAllFunctions( 84 "align-all-functions", 85 cl::desc("Force the alignment of all functions in log2 format (e.g. 4 " 86 "means align on 16B boundaries)."), 87 cl::init(0), cl::Hidden); 88 89 static const char *getPropertyName(MachineFunctionProperties::Property Prop) { 90 using P = MachineFunctionProperties::Property; 91 92 // clang-format off 93 switch(Prop) { 94 case P::FailedISel: return "FailedISel"; 95 case P::IsSSA: return "IsSSA"; 96 case P::Legalized: return "Legalized"; 97 case P::NoPHIs: return "NoPHIs"; 98 case P::NoVRegs: return "NoVRegs"; 99 case P::RegBankSelected: return "RegBankSelected"; 100 case P::Selected: return "Selected"; 101 case P::TracksLiveness: return "TracksLiveness"; 102 case P::TiedOpsRewritten: return "TiedOpsRewritten"; 103 case P::FailsVerification: return "FailsVerification"; 104 case P::TracksDebugUserValues: return "TracksDebugUserValues"; 105 } 106 // clang-format on 107 llvm_unreachable("Invalid machine function property"); 108 } 109 110 void setUnsafeStackSize(const Function &F, MachineFrameInfo &FrameInfo) { 111 if (!F.hasFnAttribute(Attribute::SafeStack)) 112 return; 113 114 auto *Existing = 115 dyn_cast_or_null<MDTuple>(F.getMetadata(LLVMContext::MD_annotation)); 116 117 if (!Existing || Existing->getNumOperands() != 2) 118 return; 119 120 auto *MetadataName = "unsafe-stack-size"; 121 if (auto &N = Existing->getOperand(0)) { 122 if (cast<MDString>(N.get())->getString() == MetadataName) { 123 if (auto &Op = Existing->getOperand(1)) { 124 auto Val = mdconst::extract<ConstantInt>(Op)->getZExtValue(); 125 FrameInfo.setUnsafeStackSize(Val); 126 } 127 } 128 } 129 } 130 131 // Pin the vtable to this file. 132 void MachineFunction::Delegate::anchor() {} 133 134 void MachineFunctionProperties::print(raw_ostream &OS) const { 135 const char *Separator = ""; 136 for (BitVector::size_type I = 0; I < Properties.size(); ++I) { 137 if (!Properties[I]) 138 continue; 139 OS << Separator << getPropertyName(static_cast<Property>(I)); 140 Separator = ", "; 141 } 142 } 143 144 //===----------------------------------------------------------------------===// 145 // MachineFunction implementation 146 //===----------------------------------------------------------------------===// 147 148 // Out-of-line virtual method. 149 MachineFunctionInfo::~MachineFunctionInfo() = default; 150 151 void ilist_alloc_traits<MachineBasicBlock>::deleteNode(MachineBasicBlock *MBB) { 152 MBB->getParent()->deleteMachineBasicBlock(MBB); 153 } 154 155 static inline Align getFnStackAlignment(const TargetSubtargetInfo *STI, 156 const Function &F) { 157 if (auto MA = F.getFnStackAlign()) 158 return *MA; 159 return STI->getFrameLowering()->getStackAlign(); 160 } 161 162 MachineFunction::MachineFunction(Function &F, const LLVMTargetMachine &Target, 163 const TargetSubtargetInfo &STI, 164 unsigned FunctionNum, MachineModuleInfo &mmi) 165 : F(F), Target(Target), STI(&STI), Ctx(mmi.getContext()), MMI(mmi) { 166 FunctionNumber = FunctionNum; 167 init(); 168 } 169 170 void MachineFunction::handleInsertion(MachineInstr &MI) { 171 if (TheDelegate) 172 TheDelegate->MF_HandleInsertion(MI); 173 } 174 175 void MachineFunction::handleRemoval(MachineInstr &MI) { 176 if (TheDelegate) 177 TheDelegate->MF_HandleRemoval(MI); 178 } 179 180 void MachineFunction::init() { 181 // Assume the function starts in SSA form with correct liveness. 182 Properties.set(MachineFunctionProperties::Property::IsSSA); 183 Properties.set(MachineFunctionProperties::Property::TracksLiveness); 184 if (STI->getRegisterInfo()) 185 RegInfo = new (Allocator) MachineRegisterInfo(this); 186 else 187 RegInfo = nullptr; 188 189 MFInfo = nullptr; 190 // We can realign the stack if the target supports it and the user hasn't 191 // explicitly asked us not to. 192 bool CanRealignSP = STI->getFrameLowering()->isStackRealignable() && 193 !F.hasFnAttribute("no-realign-stack"); 194 FrameInfo = new (Allocator) MachineFrameInfo( 195 getFnStackAlignment(STI, F), /*StackRealignable=*/CanRealignSP, 196 /*ForcedRealign=*/CanRealignSP && 197 F.hasFnAttribute(Attribute::StackAlignment)); 198 199 setUnsafeStackSize(F, *FrameInfo); 200 201 if (F.hasFnAttribute(Attribute::StackAlignment)) 202 FrameInfo->ensureMaxAlignment(*F.getFnStackAlign()); 203 204 ConstantPool = new (Allocator) MachineConstantPool(getDataLayout()); 205 Alignment = STI->getTargetLowering()->getMinFunctionAlignment(); 206 207 // FIXME: Shouldn't use pref alignment if explicit alignment is set on F. 208 // FIXME: Use Function::hasOptSize(). 209 if (!F.hasFnAttribute(Attribute::OptimizeForSize)) 210 Alignment = std::max(Alignment, 211 STI->getTargetLowering()->getPrefFunctionAlignment()); 212 213 if (AlignAllFunctions) 214 Alignment = Align(1ULL << AlignAllFunctions); 215 216 JumpTableInfo = nullptr; 217 218 if (isFuncletEHPersonality(classifyEHPersonality( 219 F.hasPersonalityFn() ? F.getPersonalityFn() : nullptr))) { 220 WinEHInfo = new (Allocator) WinEHFuncInfo(); 221 } 222 223 if (isScopedEHPersonality(classifyEHPersonality( 224 F.hasPersonalityFn() ? F.getPersonalityFn() : nullptr))) { 225 WasmEHInfo = new (Allocator) WasmEHFuncInfo(); 226 } 227 228 assert(Target.isCompatibleDataLayout(getDataLayout()) && 229 "Can't create a MachineFunction using a Module with a " 230 "Target-incompatible DataLayout attached\n"); 231 232 PSVManager = std::make_unique<PseudoSourceValueManager>(getTarget()); 233 } 234 235 MachineFunction::~MachineFunction() { 236 clear(); 237 } 238 239 void MachineFunction::clear() { 240 Properties.reset(); 241 // Don't call destructors on MachineInstr and MachineOperand. All of their 242 // memory comes from the BumpPtrAllocator which is about to be purged. 243 // 244 // Do call MachineBasicBlock destructors, it contains std::vectors. 245 for (iterator I = begin(), E = end(); I != E; I = BasicBlocks.erase(I)) 246 I->Insts.clearAndLeakNodesUnsafely(); 247 MBBNumbering.clear(); 248 249 InstructionRecycler.clear(Allocator); 250 OperandRecycler.clear(Allocator); 251 BasicBlockRecycler.clear(Allocator); 252 CodeViewAnnotations.clear(); 253 VariableDbgInfos.clear(); 254 if (RegInfo) { 255 RegInfo->~MachineRegisterInfo(); 256 Allocator.Deallocate(RegInfo); 257 } 258 if (MFInfo) { 259 MFInfo->~MachineFunctionInfo(); 260 Allocator.Deallocate(MFInfo); 261 } 262 263 FrameInfo->~MachineFrameInfo(); 264 Allocator.Deallocate(FrameInfo); 265 266 ConstantPool->~MachineConstantPool(); 267 Allocator.Deallocate(ConstantPool); 268 269 if (JumpTableInfo) { 270 JumpTableInfo->~MachineJumpTableInfo(); 271 Allocator.Deallocate(JumpTableInfo); 272 } 273 274 if (WinEHInfo) { 275 WinEHInfo->~WinEHFuncInfo(); 276 Allocator.Deallocate(WinEHInfo); 277 } 278 279 if (WasmEHInfo) { 280 WasmEHInfo->~WasmEHFuncInfo(); 281 Allocator.Deallocate(WasmEHInfo); 282 } 283 } 284 285 const DataLayout &MachineFunction::getDataLayout() const { 286 return F.getParent()->getDataLayout(); 287 } 288 289 /// Get the JumpTableInfo for this function. 290 /// If it does not already exist, allocate one. 291 MachineJumpTableInfo *MachineFunction:: 292 getOrCreateJumpTableInfo(unsigned EntryKind) { 293 if (JumpTableInfo) return JumpTableInfo; 294 295 JumpTableInfo = new (Allocator) 296 MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind); 297 return JumpTableInfo; 298 } 299 300 DenormalMode MachineFunction::getDenormalMode(const fltSemantics &FPType) const { 301 return F.getDenormalMode(FPType); 302 } 303 304 /// Should we be emitting segmented stack stuff for the function 305 bool MachineFunction::shouldSplitStack() const { 306 return getFunction().hasFnAttribute("split-stack"); 307 } 308 309 [[nodiscard]] unsigned 310 MachineFunction::addFrameInst(const MCCFIInstruction &Inst) { 311 FrameInstructions.push_back(Inst); 312 return FrameInstructions.size() - 1; 313 } 314 315 /// This discards all of the MachineBasicBlock numbers and recomputes them. 316 /// This guarantees that the MBB numbers are sequential, dense, and match the 317 /// ordering of the blocks within the function. If a specific MachineBasicBlock 318 /// is specified, only that block and those after it are renumbered. 319 void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) { 320 if (empty()) { MBBNumbering.clear(); return; } 321 MachineFunction::iterator MBBI, E = end(); 322 if (MBB == nullptr) 323 MBBI = begin(); 324 else 325 MBBI = MBB->getIterator(); 326 327 // Figure out the block number this should have. 328 unsigned BlockNo = 0; 329 if (MBBI != begin()) 330 BlockNo = std::prev(MBBI)->getNumber() + 1; 331 332 for (; MBBI != E; ++MBBI, ++BlockNo) { 333 if (MBBI->getNumber() != (int)BlockNo) { 334 // Remove use of the old number. 335 if (MBBI->getNumber() != -1) { 336 assert(MBBNumbering[MBBI->getNumber()] == &*MBBI && 337 "MBB number mismatch!"); 338 MBBNumbering[MBBI->getNumber()] = nullptr; 339 } 340 341 // If BlockNo is already taken, set that block's number to -1. 342 if (MBBNumbering[BlockNo]) 343 MBBNumbering[BlockNo]->setNumber(-1); 344 345 MBBNumbering[BlockNo] = &*MBBI; 346 MBBI->setNumber(BlockNo); 347 } 348 } 349 350 // Okay, all the blocks are renumbered. If we have compactified the block 351 // numbering, shrink MBBNumbering now. 352 assert(BlockNo <= MBBNumbering.size() && "Mismatch!"); 353 MBBNumbering.resize(BlockNo); 354 } 355 356 /// This method iterates over the basic blocks and assigns their IsBeginSection 357 /// and IsEndSection fields. This must be called after MBB layout is finalized 358 /// and the SectionID's are assigned to MBBs. 359 void MachineFunction::assignBeginEndSections() { 360 front().setIsBeginSection(); 361 auto CurrentSectionID = front().getSectionID(); 362 for (auto MBBI = std::next(begin()), E = end(); MBBI != E; ++MBBI) { 363 if (MBBI->getSectionID() == CurrentSectionID) 364 continue; 365 MBBI->setIsBeginSection(); 366 std::prev(MBBI)->setIsEndSection(); 367 CurrentSectionID = MBBI->getSectionID(); 368 } 369 back().setIsEndSection(); 370 } 371 372 /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'. 373 MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID, 374 DebugLoc DL, 375 bool NoImplicit) { 376 return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 377 MachineInstr(*this, MCID, std::move(DL), NoImplicit); 378 } 379 380 /// Create a new MachineInstr which is a copy of the 'Orig' instruction, 381 /// identical in all ways except the instruction has no parent, prev, or next. 382 MachineInstr * 383 MachineFunction::CloneMachineInstr(const MachineInstr *Orig) { 384 return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 385 MachineInstr(*this, *Orig); 386 } 387 388 MachineInstr &MachineFunction::cloneMachineInstrBundle( 389 MachineBasicBlock &MBB, MachineBasicBlock::iterator InsertBefore, 390 const MachineInstr &Orig) { 391 MachineInstr *FirstClone = nullptr; 392 MachineBasicBlock::const_instr_iterator I = Orig.getIterator(); 393 while (true) { 394 MachineInstr *Cloned = CloneMachineInstr(&*I); 395 MBB.insert(InsertBefore, Cloned); 396 if (FirstClone == nullptr) { 397 FirstClone = Cloned; 398 } else { 399 Cloned->bundleWithPred(); 400 } 401 402 if (!I->isBundledWithSucc()) 403 break; 404 ++I; 405 } 406 // Copy over call site info to the cloned instruction if needed. If Orig is in 407 // a bundle, copyCallSiteInfo takes care of finding the call instruction in 408 // the bundle. 409 if (Orig.shouldUpdateCallSiteInfo()) 410 copyCallSiteInfo(&Orig, FirstClone); 411 return *FirstClone; 412 } 413 414 /// Delete the given MachineInstr. 415 /// 416 /// This function also serves as the MachineInstr destructor - the real 417 /// ~MachineInstr() destructor must be empty. 418 void MachineFunction::deleteMachineInstr(MachineInstr *MI) { 419 // Verify that a call site info is at valid state. This assertion should 420 // be triggered during the implementation of support for the 421 // call site info of a new architecture. If the assertion is triggered, 422 // back trace will tell where to insert a call to updateCallSiteInfo(). 423 assert((!MI->isCandidateForCallSiteEntry() || 424 CallSitesInfo.find(MI) == CallSitesInfo.end()) && 425 "Call site info was not updated!"); 426 // Strip it for parts. The operand array and the MI object itself are 427 // independently recyclable. 428 if (MI->Operands) 429 deallocateOperandArray(MI->CapOperands, MI->Operands); 430 // Don't call ~MachineInstr() which must be trivial anyway because 431 // ~MachineFunction drops whole lists of MachineInstrs wihout calling their 432 // destructors. 433 InstructionRecycler.Deallocate(Allocator, MI); 434 } 435 436 /// Allocate a new MachineBasicBlock. Use this instead of 437 /// `new MachineBasicBlock'. 438 MachineBasicBlock * 439 MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) { 440 return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator)) 441 MachineBasicBlock(*this, bb); 442 } 443 444 /// Delete the given MachineBasicBlock. 445 void MachineFunction::deleteMachineBasicBlock(MachineBasicBlock *MBB) { 446 assert(MBB->getParent() == this && "MBB parent mismatch!"); 447 // Clean up any references to MBB in jump tables before deleting it. 448 if (JumpTableInfo) 449 JumpTableInfo->RemoveMBBFromJumpTables(MBB); 450 MBB->~MachineBasicBlock(); 451 BasicBlockRecycler.Deallocate(Allocator, MBB); 452 } 453 454 MachineMemOperand *MachineFunction::getMachineMemOperand( 455 MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s, 456 Align base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges, 457 SyncScope::ID SSID, AtomicOrdering Ordering, 458 AtomicOrdering FailureOrdering) { 459 return new (Allocator) 460 MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges, 461 SSID, Ordering, FailureOrdering); 462 } 463 464 MachineMemOperand *MachineFunction::getMachineMemOperand( 465 MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, LLT MemTy, 466 Align base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges, 467 SyncScope::ID SSID, AtomicOrdering Ordering, 468 AtomicOrdering FailureOrdering) { 469 return new (Allocator) 470 MachineMemOperand(PtrInfo, f, MemTy, base_alignment, AAInfo, Ranges, SSID, 471 Ordering, FailureOrdering); 472 } 473 474 MachineMemOperand *MachineFunction::getMachineMemOperand( 475 const MachineMemOperand *MMO, const MachinePointerInfo &PtrInfo, uint64_t Size) { 476 return new (Allocator) 477 MachineMemOperand(PtrInfo, MMO->getFlags(), Size, MMO->getBaseAlign(), 478 AAMDNodes(), nullptr, MMO->getSyncScopeID(), 479 MMO->getSuccessOrdering(), MMO->getFailureOrdering()); 480 } 481 482 MachineMemOperand *MachineFunction::getMachineMemOperand( 483 const MachineMemOperand *MMO, const MachinePointerInfo &PtrInfo, LLT Ty) { 484 return new (Allocator) 485 MachineMemOperand(PtrInfo, MMO->getFlags(), Ty, MMO->getBaseAlign(), 486 AAMDNodes(), nullptr, MMO->getSyncScopeID(), 487 MMO->getSuccessOrdering(), MMO->getFailureOrdering()); 488 } 489 490 MachineMemOperand * 491 MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO, 492 int64_t Offset, LLT Ty) { 493 const MachinePointerInfo &PtrInfo = MMO->getPointerInfo(); 494 495 // If there is no pointer value, the offset isn't tracked so we need to adjust 496 // the base alignment. 497 Align Alignment = PtrInfo.V.isNull() 498 ? commonAlignment(MMO->getBaseAlign(), Offset) 499 : MMO->getBaseAlign(); 500 501 // Do not preserve ranges, since we don't necessarily know what the high bits 502 // are anymore. 503 return new (Allocator) MachineMemOperand( 504 PtrInfo.getWithOffset(Offset), MMO->getFlags(), Ty, Alignment, 505 MMO->getAAInfo(), nullptr, MMO->getSyncScopeID(), 506 MMO->getSuccessOrdering(), MMO->getFailureOrdering()); 507 } 508 509 MachineMemOperand * 510 MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO, 511 const AAMDNodes &AAInfo) { 512 MachinePointerInfo MPI = MMO->getValue() ? 513 MachinePointerInfo(MMO->getValue(), MMO->getOffset()) : 514 MachinePointerInfo(MMO->getPseudoValue(), MMO->getOffset()); 515 516 return new (Allocator) MachineMemOperand( 517 MPI, MMO->getFlags(), MMO->getSize(), MMO->getBaseAlign(), AAInfo, 518 MMO->getRanges(), MMO->getSyncScopeID(), MMO->getSuccessOrdering(), 519 MMO->getFailureOrdering()); 520 } 521 522 MachineMemOperand * 523 MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO, 524 MachineMemOperand::Flags Flags) { 525 return new (Allocator) MachineMemOperand( 526 MMO->getPointerInfo(), Flags, MMO->getSize(), MMO->getBaseAlign(), 527 MMO->getAAInfo(), MMO->getRanges(), MMO->getSyncScopeID(), 528 MMO->getSuccessOrdering(), MMO->getFailureOrdering()); 529 } 530 531 MachineInstr::ExtraInfo *MachineFunction::createMIExtraInfo( 532 ArrayRef<MachineMemOperand *> MMOs, MCSymbol *PreInstrSymbol, 533 MCSymbol *PostInstrSymbol, MDNode *HeapAllocMarker, MDNode *PCSections, 534 uint32_t CFIType) { 535 return MachineInstr::ExtraInfo::create(Allocator, MMOs, PreInstrSymbol, 536 PostInstrSymbol, HeapAllocMarker, 537 PCSections, CFIType); 538 } 539 540 const char *MachineFunction::createExternalSymbolName(StringRef Name) { 541 char *Dest = Allocator.Allocate<char>(Name.size() + 1); 542 llvm::copy(Name, Dest); 543 Dest[Name.size()] = 0; 544 return Dest; 545 } 546 547 uint32_t *MachineFunction::allocateRegMask() { 548 unsigned NumRegs = getSubtarget().getRegisterInfo()->getNumRegs(); 549 unsigned Size = MachineOperand::getRegMaskSize(NumRegs); 550 uint32_t *Mask = Allocator.Allocate<uint32_t>(Size); 551 memset(Mask, 0, Size * sizeof(Mask[0])); 552 return Mask; 553 } 554 555 ArrayRef<int> MachineFunction::allocateShuffleMask(ArrayRef<int> Mask) { 556 int* AllocMask = Allocator.Allocate<int>(Mask.size()); 557 copy(Mask, AllocMask); 558 return {AllocMask, Mask.size()}; 559 } 560 561 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 562 LLVM_DUMP_METHOD void MachineFunction::dump() const { 563 print(dbgs()); 564 } 565 #endif 566 567 StringRef MachineFunction::getName() const { 568 return getFunction().getName(); 569 } 570 571 void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const { 572 OS << "# Machine code for function " << getName() << ": "; 573 getProperties().print(OS); 574 OS << '\n'; 575 576 // Print Frame Information 577 FrameInfo->print(*this, OS); 578 579 // Print JumpTable Information 580 if (JumpTableInfo) 581 JumpTableInfo->print(OS); 582 583 // Print Constant Pool 584 ConstantPool->print(OS); 585 586 const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo(); 587 588 if (RegInfo && !RegInfo->livein_empty()) { 589 OS << "Function Live Ins: "; 590 for (MachineRegisterInfo::livein_iterator 591 I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) { 592 OS << printReg(I->first, TRI); 593 if (I->second) 594 OS << " in " << printReg(I->second, TRI); 595 if (std::next(I) != E) 596 OS << ", "; 597 } 598 OS << '\n'; 599 } 600 601 ModuleSlotTracker MST(getFunction().getParent()); 602 MST.incorporateFunction(getFunction()); 603 for (const auto &BB : *this) { 604 OS << '\n'; 605 // If we print the whole function, print it at its most verbose level. 606 BB.print(OS, MST, Indexes, /*IsStandalone=*/true); 607 } 608 609 OS << "\n# End machine code for function " << getName() << ".\n\n"; 610 } 611 612 /// True if this function needs frame moves for debug or exceptions. 613 bool MachineFunction::needsFrameMoves() const { 614 return getMMI().hasDebugInfo() || 615 getTarget().Options.ForceDwarfFrameSection || 616 F.needsUnwindTableEntry(); 617 } 618 619 namespace llvm { 620 621 template<> 622 struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits { 623 DOTGraphTraits(bool isSimple = false) : DefaultDOTGraphTraits(isSimple) {} 624 625 static std::string getGraphName(const MachineFunction *F) { 626 return ("CFG for '" + F->getName() + "' function").str(); 627 } 628 629 std::string getNodeLabel(const MachineBasicBlock *Node, 630 const MachineFunction *Graph) { 631 std::string OutStr; 632 { 633 raw_string_ostream OSS(OutStr); 634 635 if (isSimple()) { 636 OSS << printMBBReference(*Node); 637 if (const BasicBlock *BB = Node->getBasicBlock()) 638 OSS << ": " << BB->getName(); 639 } else 640 Node->print(OSS); 641 } 642 643 if (OutStr[0] == '\n') OutStr.erase(OutStr.begin()); 644 645 // Process string output to make it nicer... 646 for (unsigned i = 0; i != OutStr.length(); ++i) 647 if (OutStr[i] == '\n') { // Left justify 648 OutStr[i] = '\\'; 649 OutStr.insert(OutStr.begin()+i+1, 'l'); 650 } 651 return OutStr; 652 } 653 }; 654 655 } // end namespace llvm 656 657 void MachineFunction::viewCFG() const 658 { 659 #ifndef NDEBUG 660 ViewGraph(this, "mf" + getName()); 661 #else 662 errs() << "MachineFunction::viewCFG is only available in debug builds on " 663 << "systems with Graphviz or gv!\n"; 664 #endif // NDEBUG 665 } 666 667 void MachineFunction::viewCFGOnly() const 668 { 669 #ifndef NDEBUG 670 ViewGraph(this, "mf" + getName(), true); 671 #else 672 errs() << "MachineFunction::viewCFGOnly is only available in debug builds on " 673 << "systems with Graphviz or gv!\n"; 674 #endif // NDEBUG 675 } 676 677 /// Add the specified physical register as a live-in value and 678 /// create a corresponding virtual register for it. 679 Register MachineFunction::addLiveIn(MCRegister PReg, 680 const TargetRegisterClass *RC) { 681 MachineRegisterInfo &MRI = getRegInfo(); 682 Register VReg = MRI.getLiveInVirtReg(PReg); 683 if (VReg) { 684 const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg); 685 (void)VRegRC; 686 // A physical register can be added several times. 687 // Between two calls, the register class of the related virtual register 688 // may have been constrained to match some operation constraints. 689 // In that case, check that the current register class includes the 690 // physical register and is a sub class of the specified RC. 691 assert((VRegRC == RC || (VRegRC->contains(PReg) && 692 RC->hasSubClassEq(VRegRC))) && 693 "Register class mismatch!"); 694 return VReg; 695 } 696 VReg = MRI.createVirtualRegister(RC); 697 MRI.addLiveIn(PReg, VReg); 698 return VReg; 699 } 700 701 /// Return the MCSymbol for the specified non-empty jump table. 702 /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a 703 /// normal 'L' label is returned. 704 MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx, 705 bool isLinkerPrivate) const { 706 const DataLayout &DL = getDataLayout(); 707 assert(JumpTableInfo && "No jump tables"); 708 assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!"); 709 710 StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix() 711 : DL.getPrivateGlobalPrefix(); 712 SmallString<60> Name; 713 raw_svector_ostream(Name) 714 << Prefix << "JTI" << getFunctionNumber() << '_' << JTI; 715 return Ctx.getOrCreateSymbol(Name); 716 } 717 718 /// Return a function-local symbol to represent the PIC base. 719 MCSymbol *MachineFunction::getPICBaseSymbol() const { 720 const DataLayout &DL = getDataLayout(); 721 return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) + 722 Twine(getFunctionNumber()) + "$pb"); 723 } 724 725 /// \name Exception Handling 726 /// \{ 727 728 LandingPadInfo & 729 MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) { 730 unsigned N = LandingPads.size(); 731 for (unsigned i = 0; i < N; ++i) { 732 LandingPadInfo &LP = LandingPads[i]; 733 if (LP.LandingPadBlock == LandingPad) 734 return LP; 735 } 736 737 LandingPads.push_back(LandingPadInfo(LandingPad)); 738 return LandingPads[N]; 739 } 740 741 void MachineFunction::addInvoke(MachineBasicBlock *LandingPad, 742 MCSymbol *BeginLabel, MCSymbol *EndLabel) { 743 LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 744 LP.BeginLabels.push_back(BeginLabel); 745 LP.EndLabels.push_back(EndLabel); 746 } 747 748 MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) { 749 MCSymbol *LandingPadLabel = Ctx.createTempSymbol(); 750 LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 751 LP.LandingPadLabel = LandingPadLabel; 752 753 const Instruction *FirstI = LandingPad->getBasicBlock()->getFirstNonPHI(); 754 if (const auto *LPI = dyn_cast<LandingPadInst>(FirstI)) { 755 if (const auto *PF = 756 dyn_cast<Function>(F.getPersonalityFn()->stripPointerCasts())) 757 getMMI().addPersonality(PF); 758 759 if (LPI->isCleanup()) 760 addCleanup(LandingPad); 761 762 // FIXME: New EH - Add the clauses in reverse order. This isn't 100% 763 // correct, but we need to do it this way because of how the DWARF EH 764 // emitter processes the clauses. 765 for (unsigned I = LPI->getNumClauses(); I != 0; --I) { 766 Value *Val = LPI->getClause(I - 1); 767 if (LPI->isCatch(I - 1)) { 768 addCatchTypeInfo(LandingPad, 769 dyn_cast<GlobalValue>(Val->stripPointerCasts())); 770 } else { 771 // Add filters in a list. 772 auto *CVal = cast<Constant>(Val); 773 SmallVector<const GlobalValue *, 4> FilterList; 774 for (const Use &U : CVal->operands()) 775 FilterList.push_back(cast<GlobalValue>(U->stripPointerCasts())); 776 777 addFilterTypeInfo(LandingPad, FilterList); 778 } 779 } 780 781 } else if (const auto *CPI = dyn_cast<CatchPadInst>(FirstI)) { 782 for (unsigned I = CPI->getNumArgOperands(); I != 0; --I) { 783 Value *TypeInfo = CPI->getArgOperand(I - 1)->stripPointerCasts(); 784 addCatchTypeInfo(LandingPad, dyn_cast<GlobalValue>(TypeInfo)); 785 } 786 787 } else { 788 assert(isa<CleanupPadInst>(FirstI) && "Invalid landingpad!"); 789 } 790 791 return LandingPadLabel; 792 } 793 794 void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad, 795 ArrayRef<const GlobalValue *> TyInfo) { 796 LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 797 for (const GlobalValue *GV : llvm::reverse(TyInfo)) 798 LP.TypeIds.push_back(getTypeIDFor(GV)); 799 } 800 801 void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad, 802 ArrayRef<const GlobalValue *> TyInfo) { 803 LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 804 std::vector<unsigned> IdsInFilter(TyInfo.size()); 805 for (unsigned I = 0, E = TyInfo.size(); I != E; ++I) 806 IdsInFilter[I] = getTypeIDFor(TyInfo[I]); 807 LP.TypeIds.push_back(getFilterIDFor(IdsInFilter)); 808 } 809 810 void MachineFunction::tidyLandingPads(DenseMap<MCSymbol *, uintptr_t> *LPMap, 811 bool TidyIfNoBeginLabels) { 812 for (unsigned i = 0; i != LandingPads.size(); ) { 813 LandingPadInfo &LandingPad = LandingPads[i]; 814 if (LandingPad.LandingPadLabel && 815 !LandingPad.LandingPadLabel->isDefined() && 816 (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0)) 817 LandingPad.LandingPadLabel = nullptr; 818 819 // Special case: we *should* emit LPs with null LP MBB. This indicates 820 // "nounwind" case. 821 if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) { 822 LandingPads.erase(LandingPads.begin() + i); 823 continue; 824 } 825 826 if (TidyIfNoBeginLabels) { 827 for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) { 828 MCSymbol *BeginLabel = LandingPad.BeginLabels[j]; 829 MCSymbol *EndLabel = LandingPad.EndLabels[j]; 830 if ((BeginLabel->isDefined() || (LPMap && (*LPMap)[BeginLabel] != 0)) && 831 (EndLabel->isDefined() || (LPMap && (*LPMap)[EndLabel] != 0))) 832 continue; 833 834 LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j); 835 LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j); 836 --j; 837 --e; 838 } 839 840 // Remove landing pads with no try-ranges. 841 if (LandingPads[i].BeginLabels.empty()) { 842 LandingPads.erase(LandingPads.begin() + i); 843 continue; 844 } 845 } 846 847 // If there is no landing pad, ensure that the list of typeids is empty. 848 // If the only typeid is a cleanup, this is the same as having no typeids. 849 if (!LandingPad.LandingPadBlock || 850 (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0])) 851 LandingPad.TypeIds.clear(); 852 ++i; 853 } 854 } 855 856 void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) { 857 LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 858 LP.TypeIds.push_back(0); 859 } 860 861 void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym, 862 ArrayRef<unsigned> Sites) { 863 LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end()); 864 } 865 866 unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) { 867 for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i) 868 if (TypeInfos[i] == TI) return i + 1; 869 870 TypeInfos.push_back(TI); 871 return TypeInfos.size(); 872 } 873 874 int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) { 875 // If the new filter coincides with the tail of an existing filter, then 876 // re-use the existing filter. Folding filters more than this requires 877 // re-ordering filters and/or their elements - probably not worth it. 878 for (unsigned i : FilterEnds) { 879 unsigned j = TyIds.size(); 880 881 while (i && j) 882 if (FilterIds[--i] != TyIds[--j]) 883 goto try_next; 884 885 if (!j) 886 // The new filter coincides with range [i, end) of the existing filter. 887 return -(1 + i); 888 889 try_next:; 890 } 891 892 // Add the new filter. 893 int FilterID = -(1 + FilterIds.size()); 894 FilterIds.reserve(FilterIds.size() + TyIds.size() + 1); 895 llvm::append_range(FilterIds, TyIds); 896 FilterEnds.push_back(FilterIds.size()); 897 FilterIds.push_back(0); // terminator 898 return FilterID; 899 } 900 901 MachineFunction::CallSiteInfoMap::iterator 902 MachineFunction::getCallSiteInfo(const MachineInstr *MI) { 903 assert(MI->isCandidateForCallSiteEntry() && 904 "Call site info refers only to call (MI) candidates"); 905 906 if (!Target.Options.EmitCallSiteInfo) 907 return CallSitesInfo.end(); 908 return CallSitesInfo.find(MI); 909 } 910 911 /// Return the call machine instruction or find a call within bundle. 912 static const MachineInstr *getCallInstr(const MachineInstr *MI) { 913 if (!MI->isBundle()) 914 return MI; 915 916 for (const auto &BMI : make_range(getBundleStart(MI->getIterator()), 917 getBundleEnd(MI->getIterator()))) 918 if (BMI.isCandidateForCallSiteEntry()) 919 return &BMI; 920 921 llvm_unreachable("Unexpected bundle without a call site candidate"); 922 } 923 924 void MachineFunction::eraseCallSiteInfo(const MachineInstr *MI) { 925 assert(MI->shouldUpdateCallSiteInfo() && 926 "Call site info refers only to call (MI) candidates or " 927 "candidates inside bundles"); 928 929 const MachineInstr *CallMI = getCallInstr(MI); 930 CallSiteInfoMap::iterator CSIt = getCallSiteInfo(CallMI); 931 if (CSIt == CallSitesInfo.end()) 932 return; 933 CallSitesInfo.erase(CSIt); 934 } 935 936 void MachineFunction::copyCallSiteInfo(const MachineInstr *Old, 937 const MachineInstr *New) { 938 assert(Old->shouldUpdateCallSiteInfo() && 939 "Call site info refers only to call (MI) candidates or " 940 "candidates inside bundles"); 941 942 if (!New->isCandidateForCallSiteEntry()) 943 return eraseCallSiteInfo(Old); 944 945 const MachineInstr *OldCallMI = getCallInstr(Old); 946 CallSiteInfoMap::iterator CSIt = getCallSiteInfo(OldCallMI); 947 if (CSIt == CallSitesInfo.end()) 948 return; 949 950 CallSiteInfo CSInfo = CSIt->second; 951 CallSitesInfo[New] = CSInfo; 952 } 953 954 void MachineFunction::moveCallSiteInfo(const MachineInstr *Old, 955 const MachineInstr *New) { 956 assert(Old->shouldUpdateCallSiteInfo() && 957 "Call site info refers only to call (MI) candidates or " 958 "candidates inside bundles"); 959 960 if (!New->isCandidateForCallSiteEntry()) 961 return eraseCallSiteInfo(Old); 962 963 const MachineInstr *OldCallMI = getCallInstr(Old); 964 CallSiteInfoMap::iterator CSIt = getCallSiteInfo(OldCallMI); 965 if (CSIt == CallSitesInfo.end()) 966 return; 967 968 CallSiteInfo CSInfo = std::move(CSIt->second); 969 CallSitesInfo.erase(CSIt); 970 CallSitesInfo[New] = CSInfo; 971 } 972 973 void MachineFunction::setDebugInstrNumberingCount(unsigned Num) { 974 DebugInstrNumberingCount = Num; 975 } 976 977 void MachineFunction::makeDebugValueSubstitution(DebugInstrOperandPair A, 978 DebugInstrOperandPair B, 979 unsigned Subreg) { 980 // Catch any accidental self-loops. 981 assert(A.first != B.first); 982 // Don't allow any substitutions _from_ the memory operand number. 983 assert(A.second != DebugOperandMemNumber); 984 985 DebugValueSubstitutions.push_back({A, B, Subreg}); 986 } 987 988 void MachineFunction::substituteDebugValuesForInst(const MachineInstr &Old, 989 MachineInstr &New, 990 unsigned MaxOperand) { 991 // If the Old instruction wasn't tracked at all, there is no work to do. 992 unsigned OldInstrNum = Old.peekDebugInstrNum(); 993 if (!OldInstrNum) 994 return; 995 996 // Iterate over all operands looking for defs to create substitutions for. 997 // Avoid creating new instr numbers unless we create a new substitution. 998 // While this has no functional effect, it risks confusing someone reading 999 // MIR output. 1000 // Examine all the operands, or the first N specified by the caller. 1001 MaxOperand = std::min(MaxOperand, Old.getNumOperands()); 1002 for (unsigned int I = 0; I < MaxOperand; ++I) { 1003 const auto &OldMO = Old.getOperand(I); 1004 auto &NewMO = New.getOperand(I); 1005 (void)NewMO; 1006 1007 if (!OldMO.isReg() || !OldMO.isDef()) 1008 continue; 1009 assert(NewMO.isDef()); 1010 1011 unsigned NewInstrNum = New.getDebugInstrNum(); 1012 makeDebugValueSubstitution(std::make_pair(OldInstrNum, I), 1013 std::make_pair(NewInstrNum, I)); 1014 } 1015 } 1016 1017 auto MachineFunction::salvageCopySSA( 1018 MachineInstr &MI, DenseMap<Register, DebugInstrOperandPair> &DbgPHICache) 1019 -> DebugInstrOperandPair { 1020 const TargetInstrInfo &TII = *getSubtarget().getInstrInfo(); 1021 1022 // Check whether this copy-like instruction has already been salvaged into 1023 // an operand pair. 1024 Register Dest; 1025 if (auto CopyDstSrc = TII.isCopyInstr(MI)) { 1026 Dest = CopyDstSrc->Destination->getReg(); 1027 } else { 1028 assert(MI.isSubregToReg()); 1029 Dest = MI.getOperand(0).getReg(); 1030 } 1031 1032 auto CacheIt = DbgPHICache.find(Dest); 1033 if (CacheIt != DbgPHICache.end()) 1034 return CacheIt->second; 1035 1036 // Calculate the instruction number to use, or install a DBG_PHI. 1037 auto OperandPair = salvageCopySSAImpl(MI); 1038 DbgPHICache.insert({Dest, OperandPair}); 1039 return OperandPair; 1040 } 1041 1042 auto MachineFunction::salvageCopySSAImpl(MachineInstr &MI) 1043 -> DebugInstrOperandPair { 1044 MachineRegisterInfo &MRI = getRegInfo(); 1045 const TargetRegisterInfo &TRI = *MRI.getTargetRegisterInfo(); 1046 const TargetInstrInfo &TII = *getSubtarget().getInstrInfo(); 1047 1048 // Chase the value read by a copy-like instruction back to the instruction 1049 // that ultimately _defines_ that value. This may pass: 1050 // * Through multiple intermediate copies, including subregister moves / 1051 // copies, 1052 // * Copies from physical registers that must then be traced back to the 1053 // defining instruction, 1054 // * Or, physical registers may be live-in to (only) the entry block, which 1055 // requires a DBG_PHI to be created. 1056 // We can pursue this problem in that order: trace back through copies, 1057 // optionally through a physical register, to a defining instruction. We 1058 // should never move from physreg to vreg. As we're still in SSA form, no need 1059 // to worry about partial definitions of registers. 1060 1061 // Helper lambda to interpret a copy-like instruction. Takes instruction, 1062 // returns the register read and any subregister identifying which part is 1063 // read. 1064 auto GetRegAndSubreg = 1065 [&](const MachineInstr &Cpy) -> std::pair<Register, unsigned> { 1066 Register NewReg, OldReg; 1067 unsigned SubReg; 1068 if (Cpy.isCopy()) { 1069 OldReg = Cpy.getOperand(0).getReg(); 1070 NewReg = Cpy.getOperand(1).getReg(); 1071 SubReg = Cpy.getOperand(1).getSubReg(); 1072 } else if (Cpy.isSubregToReg()) { 1073 OldReg = Cpy.getOperand(0).getReg(); 1074 NewReg = Cpy.getOperand(2).getReg(); 1075 SubReg = Cpy.getOperand(3).getImm(); 1076 } else { 1077 auto CopyDetails = *TII.isCopyInstr(Cpy); 1078 const MachineOperand &Src = *CopyDetails.Source; 1079 const MachineOperand &Dest = *CopyDetails.Destination; 1080 OldReg = Dest.getReg(); 1081 NewReg = Src.getReg(); 1082 SubReg = Src.getSubReg(); 1083 } 1084 1085 return {NewReg, SubReg}; 1086 }; 1087 1088 // First seek either the defining instruction, or a copy from a physreg. 1089 // During search, the current state is the current copy instruction, and which 1090 // register we've read. Accumulate qualifying subregisters into SubregsSeen; 1091 // deal with those later. 1092 auto State = GetRegAndSubreg(MI); 1093 auto CurInst = MI.getIterator(); 1094 SmallVector<unsigned, 4> SubregsSeen; 1095 while (true) { 1096 // If we've found a copy from a physreg, first portion of search is over. 1097 if (!State.first.isVirtual()) 1098 break; 1099 1100 // Record any subregister qualifier. 1101 if (State.second) 1102 SubregsSeen.push_back(State.second); 1103 1104 assert(MRI.hasOneDef(State.first)); 1105 MachineInstr &Inst = *MRI.def_begin(State.first)->getParent(); 1106 CurInst = Inst.getIterator(); 1107 1108 // Any non-copy instruction is the defining instruction we're seeking. 1109 if (!Inst.isCopyLike() && !TII.isCopyInstr(Inst)) 1110 break; 1111 State = GetRegAndSubreg(Inst); 1112 }; 1113 1114 // Helper lambda to apply additional subregister substitutions to a known 1115 // instruction/operand pair. Adds new (fake) substitutions so that we can 1116 // record the subregister. FIXME: this isn't very space efficient if multiple 1117 // values are tracked back through the same copies; cache something later. 1118 auto ApplySubregisters = 1119 [&](DebugInstrOperandPair P) -> DebugInstrOperandPair { 1120 for (unsigned Subreg : reverse(SubregsSeen)) { 1121 // Fetch a new instruction number, not attached to an actual instruction. 1122 unsigned NewInstrNumber = getNewDebugInstrNum(); 1123 // Add a substitution from the "new" number to the known one, with a 1124 // qualifying subreg. 1125 makeDebugValueSubstitution({NewInstrNumber, 0}, P, Subreg); 1126 // Return the new number; to find the underlying value, consumers need to 1127 // deal with the qualifying subreg. 1128 P = {NewInstrNumber, 0}; 1129 } 1130 return P; 1131 }; 1132 1133 // If we managed to find the defining instruction after COPYs, return an 1134 // instruction / operand pair after adding subregister qualifiers. 1135 if (State.first.isVirtual()) { 1136 // Virtual register def -- we can just look up where this happens. 1137 MachineInstr *Inst = MRI.def_begin(State.first)->getParent(); 1138 for (auto &MO : Inst->operands()) { 1139 if (!MO.isReg() || !MO.isDef() || MO.getReg() != State.first) 1140 continue; 1141 return ApplySubregisters( 1142 {Inst->getDebugInstrNum(), Inst->getOperandNo(&MO)}); 1143 } 1144 1145 llvm_unreachable("Vreg def with no corresponding operand?"); 1146 } 1147 1148 // Our search ended in a copy from a physreg: walk back up the function 1149 // looking for whatever defines the physreg. 1150 assert(CurInst->isCopyLike() || TII.isCopyInstr(*CurInst)); 1151 State = GetRegAndSubreg(*CurInst); 1152 Register RegToSeek = State.first; 1153 1154 auto RMII = CurInst->getReverseIterator(); 1155 auto PrevInstrs = make_range(RMII, CurInst->getParent()->instr_rend()); 1156 for (auto &ToExamine : PrevInstrs) { 1157 for (auto &MO : ToExamine.operands()) { 1158 // Test for operand that defines something aliasing RegToSeek. 1159 if (!MO.isReg() || !MO.isDef() || 1160 !TRI.regsOverlap(RegToSeek, MO.getReg())) 1161 continue; 1162 1163 return ApplySubregisters( 1164 {ToExamine.getDebugInstrNum(), ToExamine.getOperandNo(&MO)}); 1165 } 1166 } 1167 1168 MachineBasicBlock &InsertBB = *CurInst->getParent(); 1169 1170 // We reached the start of the block before finding a defining instruction. 1171 // There are numerous scenarios where this can happen: 1172 // * Constant physical registers, 1173 // * Several intrinsics that allow LLVM-IR to read arbitary registers, 1174 // * Arguments in the entry block, 1175 // * Exception handling landing pads. 1176 // Validating all of them is too difficult, so just insert a DBG_PHI reading 1177 // the variable value at this position, rather than checking it makes sense. 1178 1179 // Create DBG_PHI for specified physreg. 1180 auto Builder = BuildMI(InsertBB, InsertBB.getFirstNonPHI(), DebugLoc(), 1181 TII.get(TargetOpcode::DBG_PHI)); 1182 Builder.addReg(State.first); 1183 unsigned NewNum = getNewDebugInstrNum(); 1184 Builder.addImm(NewNum); 1185 return ApplySubregisters({NewNum, 0u}); 1186 } 1187 1188 void MachineFunction::finalizeDebugInstrRefs() { 1189 auto *TII = getSubtarget().getInstrInfo(); 1190 1191 auto MakeUndefDbgValue = [&](MachineInstr &MI) { 1192 const MCInstrDesc &RefII = TII->get(TargetOpcode::DBG_VALUE); 1193 MI.setDesc(RefII); 1194 MI.getOperand(0).setReg(0); 1195 MI.getOperand(1).ChangeToRegister(0, false); 1196 }; 1197 1198 DenseMap<Register, DebugInstrOperandPair> ArgDbgPHIs; 1199 for (auto &MBB : *this) { 1200 for (auto &MI : MBB) { 1201 if (!MI.isDebugRef() || !MI.getOperand(0).isReg()) 1202 continue; 1203 1204 Register Reg = MI.getOperand(0).getReg(); 1205 1206 // Some vregs can be deleted as redundant in the meantime. Mark those 1207 // as DBG_VALUE $noreg. Additionally, some normal instructions are 1208 // quickly deleted, leaving dangling references to vregs with no def. 1209 if (Reg == 0 || !RegInfo->hasOneDef(Reg)) { 1210 MakeUndefDbgValue(MI); 1211 continue; 1212 } 1213 1214 assert(Reg.isVirtual()); 1215 MachineInstr &DefMI = *RegInfo->def_instr_begin(Reg); 1216 1217 // If we've found a copy-like instruction, follow it back to the 1218 // instruction that defines the source value, see salvageCopySSA docs 1219 // for why this is important. 1220 if (DefMI.isCopyLike() || TII->isCopyInstr(DefMI)) { 1221 auto Result = salvageCopySSA(DefMI, ArgDbgPHIs); 1222 MI.getOperand(0).ChangeToImmediate(Result.first); 1223 MI.getOperand(1).setImm(Result.second); 1224 } else { 1225 // Otherwise, identify the operand number that the VReg refers to. 1226 unsigned OperandIdx = 0; 1227 for (const auto &MO : DefMI.operands()) { 1228 if (MO.isReg() && MO.isDef() && MO.getReg() == Reg) 1229 break; 1230 ++OperandIdx; 1231 } 1232 assert(OperandIdx < DefMI.getNumOperands()); 1233 1234 // Morph this instr ref to point at the given instruction and operand. 1235 unsigned ID = DefMI.getDebugInstrNum(); 1236 MI.getOperand(0).ChangeToImmediate(ID); 1237 MI.getOperand(1).setImm(OperandIdx); 1238 } 1239 } 1240 } 1241 } 1242 1243 bool MachineFunction::useDebugInstrRef() const { 1244 // Disable instr-ref at -O0: it's very slow (in compile time). We can still 1245 // have optimized code inlined into this unoptimized code, however with 1246 // fewer and less aggressive optimizations happening, coverage and accuracy 1247 // should not suffer. 1248 if (getTarget().getOptLevel() == CodeGenOpt::None) 1249 return false; 1250 1251 // Don't use instr-ref if this function is marked optnone. 1252 if (F.hasFnAttribute(Attribute::OptimizeNone)) 1253 return false; 1254 1255 if (llvm::debuginfoShouldUseDebugInstrRef(getTarget().getTargetTriple())) 1256 return true; 1257 1258 return false; 1259 } 1260 1261 // Use one million as a high / reserved number. 1262 const unsigned MachineFunction::DebugOperandMemNumber = 1000000; 1263 1264 /// \} 1265 1266 //===----------------------------------------------------------------------===// 1267 // MachineJumpTableInfo implementation 1268 //===----------------------------------------------------------------------===// 1269 1270 /// Return the size of each entry in the jump table. 1271 unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const { 1272 // The size of a jump table entry is 4 bytes unless the entry is just the 1273 // address of a block, in which case it is the pointer size. 1274 switch (getEntryKind()) { 1275 case MachineJumpTableInfo::EK_BlockAddress: 1276 return TD.getPointerSize(); 1277 case MachineJumpTableInfo::EK_GPRel64BlockAddress: 1278 return 8; 1279 case MachineJumpTableInfo::EK_GPRel32BlockAddress: 1280 case MachineJumpTableInfo::EK_LabelDifference32: 1281 case MachineJumpTableInfo::EK_Custom32: 1282 return 4; 1283 case MachineJumpTableInfo::EK_Inline: 1284 return 0; 1285 } 1286 llvm_unreachable("Unknown jump table encoding!"); 1287 } 1288 1289 /// Return the alignment of each entry in the jump table. 1290 unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const { 1291 // The alignment of a jump table entry is the alignment of int32 unless the 1292 // entry is just the address of a block, in which case it is the pointer 1293 // alignment. 1294 switch (getEntryKind()) { 1295 case MachineJumpTableInfo::EK_BlockAddress: 1296 return TD.getPointerABIAlignment(0).value(); 1297 case MachineJumpTableInfo::EK_GPRel64BlockAddress: 1298 return TD.getABIIntegerTypeAlignment(64).value(); 1299 case MachineJumpTableInfo::EK_GPRel32BlockAddress: 1300 case MachineJumpTableInfo::EK_LabelDifference32: 1301 case MachineJumpTableInfo::EK_Custom32: 1302 return TD.getABIIntegerTypeAlignment(32).value(); 1303 case MachineJumpTableInfo::EK_Inline: 1304 return 1; 1305 } 1306 llvm_unreachable("Unknown jump table encoding!"); 1307 } 1308 1309 /// Create a new jump table entry in the jump table info. 1310 unsigned MachineJumpTableInfo::createJumpTableIndex( 1311 const std::vector<MachineBasicBlock*> &DestBBs) { 1312 assert(!DestBBs.empty() && "Cannot create an empty jump table!"); 1313 JumpTables.push_back(MachineJumpTableEntry(DestBBs)); 1314 return JumpTables.size()-1; 1315 } 1316 1317 /// If Old is the target of any jump tables, update the jump tables to branch 1318 /// to New instead. 1319 bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old, 1320 MachineBasicBlock *New) { 1321 assert(Old != New && "Not making a change?"); 1322 bool MadeChange = false; 1323 for (size_t i = 0, e = JumpTables.size(); i != e; ++i) 1324 ReplaceMBBInJumpTable(i, Old, New); 1325 return MadeChange; 1326 } 1327 1328 /// If MBB is present in any jump tables, remove it. 1329 bool MachineJumpTableInfo::RemoveMBBFromJumpTables(MachineBasicBlock *MBB) { 1330 bool MadeChange = false; 1331 for (MachineJumpTableEntry &JTE : JumpTables) { 1332 auto removeBeginItr = std::remove(JTE.MBBs.begin(), JTE.MBBs.end(), MBB); 1333 MadeChange |= (removeBeginItr != JTE.MBBs.end()); 1334 JTE.MBBs.erase(removeBeginItr, JTE.MBBs.end()); 1335 } 1336 return MadeChange; 1337 } 1338 1339 /// If Old is a target of the jump tables, update the jump table to branch to 1340 /// New instead. 1341 bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx, 1342 MachineBasicBlock *Old, 1343 MachineBasicBlock *New) { 1344 assert(Old != New && "Not making a change?"); 1345 bool MadeChange = false; 1346 MachineJumpTableEntry &JTE = JumpTables[Idx]; 1347 for (MachineBasicBlock *&MBB : JTE.MBBs) 1348 if (MBB == Old) { 1349 MBB = New; 1350 MadeChange = true; 1351 } 1352 return MadeChange; 1353 } 1354 1355 void MachineJumpTableInfo::print(raw_ostream &OS) const { 1356 if (JumpTables.empty()) return; 1357 1358 OS << "Jump Tables:\n"; 1359 1360 for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) { 1361 OS << printJumpTableEntryReference(i) << ':'; 1362 for (const MachineBasicBlock *MBB : JumpTables[i].MBBs) 1363 OS << ' ' << printMBBReference(*MBB); 1364 if (i != e) 1365 OS << '\n'; 1366 } 1367 1368 OS << '\n'; 1369 } 1370 1371 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 1372 LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); } 1373 #endif 1374 1375 Printable llvm::printJumpTableEntryReference(unsigned Idx) { 1376 return Printable([Idx](raw_ostream &OS) { OS << "%jump-table." << Idx; }); 1377 } 1378 1379 //===----------------------------------------------------------------------===// 1380 // MachineConstantPool implementation 1381 //===----------------------------------------------------------------------===// 1382 1383 void MachineConstantPoolValue::anchor() {} 1384 1385 unsigned MachineConstantPoolValue::getSizeInBytes(const DataLayout &DL) const { 1386 return DL.getTypeAllocSize(Ty); 1387 } 1388 1389 unsigned MachineConstantPoolEntry::getSizeInBytes(const DataLayout &DL) const { 1390 if (isMachineConstantPoolEntry()) 1391 return Val.MachineCPVal->getSizeInBytes(DL); 1392 return DL.getTypeAllocSize(Val.ConstVal->getType()); 1393 } 1394 1395 bool MachineConstantPoolEntry::needsRelocation() const { 1396 if (isMachineConstantPoolEntry()) 1397 return true; 1398 return Val.ConstVal->needsDynamicRelocation(); 1399 } 1400 1401 SectionKind 1402 MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const { 1403 if (needsRelocation()) 1404 return SectionKind::getReadOnlyWithRel(); 1405 switch (getSizeInBytes(*DL)) { 1406 case 4: 1407 return SectionKind::getMergeableConst4(); 1408 case 8: 1409 return SectionKind::getMergeableConst8(); 1410 case 16: 1411 return SectionKind::getMergeableConst16(); 1412 case 32: 1413 return SectionKind::getMergeableConst32(); 1414 default: 1415 return SectionKind::getReadOnly(); 1416 } 1417 } 1418 1419 MachineConstantPool::~MachineConstantPool() { 1420 // A constant may be a member of both Constants and MachineCPVsSharingEntries, 1421 // so keep track of which we've deleted to avoid double deletions. 1422 DenseSet<MachineConstantPoolValue*> Deleted; 1423 for (const MachineConstantPoolEntry &C : Constants) 1424 if (C.isMachineConstantPoolEntry()) { 1425 Deleted.insert(C.Val.MachineCPVal); 1426 delete C.Val.MachineCPVal; 1427 } 1428 for (MachineConstantPoolValue *CPV : MachineCPVsSharingEntries) { 1429 if (Deleted.count(CPV) == 0) 1430 delete CPV; 1431 } 1432 } 1433 1434 /// Test whether the given two constants can be allocated the same constant pool 1435 /// entry. 1436 static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B, 1437 const DataLayout &DL) { 1438 // Handle the trivial case quickly. 1439 if (A == B) return true; 1440 1441 // If they have the same type but weren't the same constant, quickly 1442 // reject them. 1443 if (A->getType() == B->getType()) return false; 1444 1445 // We can't handle structs or arrays. 1446 if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) || 1447 isa<StructType>(B->getType()) || isa<ArrayType>(B->getType())) 1448 return false; 1449 1450 // For now, only support constants with the same size. 1451 uint64_t StoreSize = DL.getTypeStoreSize(A->getType()); 1452 if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128) 1453 return false; 1454 1455 Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8); 1456 1457 // Try constant folding a bitcast of both instructions to an integer. If we 1458 // get two identical ConstantInt's, then we are good to share them. We use 1459 // the constant folding APIs to do this so that we get the benefit of 1460 // DataLayout. 1461 if (isa<PointerType>(A->getType())) 1462 A = ConstantFoldCastOperand(Instruction::PtrToInt, 1463 const_cast<Constant *>(A), IntTy, DL); 1464 else if (A->getType() != IntTy) 1465 A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A), 1466 IntTy, DL); 1467 if (isa<PointerType>(B->getType())) 1468 B = ConstantFoldCastOperand(Instruction::PtrToInt, 1469 const_cast<Constant *>(B), IntTy, DL); 1470 else if (B->getType() != IntTy) 1471 B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B), 1472 IntTy, DL); 1473 1474 return A == B; 1475 } 1476 1477 /// Create a new entry in the constant pool or return an existing one. 1478 /// User must specify the log2 of the minimum required alignment for the object. 1479 unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C, 1480 Align Alignment) { 1481 if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1482 1483 // Check to see if we already have this constant. 1484 // 1485 // FIXME, this could be made much more efficient for large constant pools. 1486 for (unsigned i = 0, e = Constants.size(); i != e; ++i) 1487 if (!Constants[i].isMachineConstantPoolEntry() && 1488 CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) { 1489 if (Constants[i].getAlign() < Alignment) 1490 Constants[i].Alignment = Alignment; 1491 return i; 1492 } 1493 1494 Constants.push_back(MachineConstantPoolEntry(C, Alignment)); 1495 return Constants.size()-1; 1496 } 1497 1498 unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V, 1499 Align Alignment) { 1500 if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1501 1502 // Check to see if we already have this constant. 1503 // 1504 // FIXME, this could be made much more efficient for large constant pools. 1505 int Idx = V->getExistingMachineCPValue(this, Alignment); 1506 if (Idx != -1) { 1507 MachineCPVsSharingEntries.insert(V); 1508 return (unsigned)Idx; 1509 } 1510 1511 Constants.push_back(MachineConstantPoolEntry(V, Alignment)); 1512 return Constants.size()-1; 1513 } 1514 1515 void MachineConstantPool::print(raw_ostream &OS) const { 1516 if (Constants.empty()) return; 1517 1518 OS << "Constant Pool:\n"; 1519 for (unsigned i = 0, e = Constants.size(); i != e; ++i) { 1520 OS << " cp#" << i << ": "; 1521 if (Constants[i].isMachineConstantPoolEntry()) 1522 Constants[i].Val.MachineCPVal->print(OS); 1523 else 1524 Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false); 1525 OS << ", align=" << Constants[i].getAlign().value(); 1526 OS << "\n"; 1527 } 1528 } 1529 1530 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 1531 LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); } 1532 #endif 1533