1ae6f7882SJeremy Morse //===- InstrRefBasedImpl.cpp - Tracking Debug Value MIs -------------------===// 2ae6f7882SJeremy Morse // 3ae6f7882SJeremy Morse // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4ae6f7882SJeremy Morse // See https://llvm.org/LICENSE.txt for license information. 5ae6f7882SJeremy Morse // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6ae6f7882SJeremy Morse // 7ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 8ae6f7882SJeremy Morse /// \file InstrRefBasedImpl.cpp 9ae6f7882SJeremy Morse /// 10ae6f7882SJeremy Morse /// This is a separate implementation of LiveDebugValues, see 11ae6f7882SJeremy Morse /// LiveDebugValues.cpp and VarLocBasedImpl.cpp for more information. 12ae6f7882SJeremy Morse /// 13ae6f7882SJeremy Morse /// This pass propagates variable locations between basic blocks, resolving 14a3936a6cSJeremy Morse /// control flow conflicts between them. The problem is SSA construction, where 15a3936a6cSJeremy Morse /// each debug instruction assigns the *value* that a variable has, and every 16a3936a6cSJeremy Morse /// instruction where the variable is in scope uses that variable. The resulting 17a3936a6cSJeremy Morse /// map of instruction-to-value is then translated into a register (or spill) 18a3936a6cSJeremy Morse /// location for each variable over each instruction. 19ae6f7882SJeremy Morse /// 20a3936a6cSJeremy Morse /// The primary difference from normal SSA construction is that we cannot 21a3936a6cSJeremy Morse /// _create_ PHI values that contain variable values. CodeGen has already 22a3936a6cSJeremy Morse /// completed, and we can't alter it just to make debug-info complete. Thus: 23a3936a6cSJeremy Morse /// we can identify function positions where we would like a PHI value for a 24a3936a6cSJeremy Morse /// variable, but must search the MachineFunction to see whether such a PHI is 25a3936a6cSJeremy Morse /// available. If no such PHI exists, the variable location must be dropped. 26ae6f7882SJeremy Morse /// 27a3936a6cSJeremy Morse /// To achieve this, we perform two kinds of analysis. First, we identify 28ae6f7882SJeremy Morse /// every value defined by every instruction (ignoring those that only move 29a3936a6cSJeremy Morse /// another value), then re-compute an SSA-form representation of the 30a3936a6cSJeremy Morse /// MachineFunction, using value propagation to eliminate any un-necessary 31a3936a6cSJeremy Morse /// PHI values. This gives us a map of every value computed in the function, 32a3936a6cSJeremy Morse /// and its location within the register file / stack. 33ae6f7882SJeremy Morse /// 34a3936a6cSJeremy Morse /// Secondly, for each variable we perform the same analysis, where each debug 35a3936a6cSJeremy Morse /// instruction is considered a def, and every instruction where the variable 36a3936a6cSJeremy Morse /// is in lexical scope as a use. Value propagation is used again to eliminate 37a3936a6cSJeremy Morse /// any un-necessary PHIs. This gives us a map of each variable to the value 38a3936a6cSJeremy Morse /// it should have in a block. 39ae6f7882SJeremy Morse /// 40a3936a6cSJeremy Morse /// Once both are complete, we have two maps for each block: 41a3936a6cSJeremy Morse /// * Variables to the values they should have, 42a3936a6cSJeremy Morse /// * Values to the register / spill slot they are located in. 43a3936a6cSJeremy Morse /// After which we can marry-up variable values with a location, and emit 44a3936a6cSJeremy Morse /// DBG_VALUE instructions specifying those locations. Variable locations may 45a3936a6cSJeremy Morse /// be dropped in this process due to the desired variable value not being 46a3936a6cSJeremy Morse /// resident in any machine location, or because there is no PHI value in any 47a3936a6cSJeremy Morse /// location that accurately represents the desired value. The building of 48a3936a6cSJeremy Morse /// location lists for each block is left to DbgEntityHistoryCalculator. 49ae6f7882SJeremy Morse /// 50a3936a6cSJeremy Morse /// This pass is kept efficient because the size of the first SSA problem 51a3936a6cSJeremy Morse /// is proportional to the working-set size of the function, which the compiler 52a3936a6cSJeremy Morse /// tries to keep small. (It's also proportional to the number of blocks). 53a3936a6cSJeremy Morse /// Additionally, we repeatedly perform the second SSA problem analysis with 54a3936a6cSJeremy Morse /// only the variables and blocks in a single lexical scope, exploiting their 55a3936a6cSJeremy Morse /// locality. 56ae6f7882SJeremy Morse /// 57ae6f7882SJeremy Morse /// ### Terminology 58ae6f7882SJeremy Morse /// 59ae6f7882SJeremy Morse /// A machine location is a register or spill slot, a value is something that's 60ae6f7882SJeremy Morse /// defined by an instruction or PHI node, while a variable value is the value 61ae6f7882SJeremy Morse /// assigned to a variable. A variable location is a machine location, that must 62ae6f7882SJeremy Morse /// contain the appropriate variable value. A value that is a PHI node is 63ae6f7882SJeremy Morse /// occasionally called an mphi. 64ae6f7882SJeremy Morse /// 65a3936a6cSJeremy Morse /// The first SSA problem is the "machine value location" problem, 66ae6f7882SJeremy Morse /// because we're determining which machine locations contain which values. 67ae6f7882SJeremy Morse /// The "locations" are constant: what's unknown is what value they contain. 68ae6f7882SJeremy Morse /// 69a3936a6cSJeremy Morse /// The second SSA problem (the one for variables) is the "variable value 70ae6f7882SJeremy Morse /// problem", because it's determining what values a variable has, rather than 71a3936a6cSJeremy Morse /// what location those values are placed in. 72ae6f7882SJeremy Morse /// 73ae6f7882SJeremy Morse /// TODO: 74ae6f7882SJeremy Morse /// Overlapping fragments 75ae6f7882SJeremy Morse /// Entry values 76ae6f7882SJeremy Morse /// Add back DEBUG statements for debugging this 77ae6f7882SJeremy Morse /// Collect statistics 78ae6f7882SJeremy Morse /// 79ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 80ae6f7882SJeremy Morse 81ae6f7882SJeremy Morse #include "llvm/ADT/DenseMap.h" 82ae6f7882SJeremy Morse #include "llvm/ADT/PostOrderIterator.h" 83010108bbSJeremy Morse #include "llvm/ADT/STLExtras.h" 84ae6f7882SJeremy Morse #include "llvm/ADT/SmallPtrSet.h" 85ae6f7882SJeremy Morse #include "llvm/ADT/SmallSet.h" 86ae6f7882SJeremy Morse #include "llvm/ADT/SmallVector.h" 87*989f1c72Sserge-sans-paille #include "llvm/BinaryFormat/Dwarf.h" 88ae6f7882SJeremy Morse #include "llvm/CodeGen/LexicalScopes.h" 89ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineBasicBlock.h" 90a3936a6cSJeremy Morse #include "llvm/CodeGen/MachineDominators.h" 91ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineFrameInfo.h" 92ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineFunction.h" 93ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineInstr.h" 94ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineInstrBuilder.h" 951575583fSJeremy Morse #include "llvm/CodeGen/MachineInstrBundle.h" 96ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineMemOperand.h" 97ae6f7882SJeremy Morse #include "llvm/CodeGen/MachineOperand.h" 98ae6f7882SJeremy Morse #include "llvm/CodeGen/PseudoSourceValue.h" 99ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetFrameLowering.h" 100ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetInstrInfo.h" 101ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetLowering.h" 102ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetPassConfig.h" 103ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetRegisterInfo.h" 104ae6f7882SJeremy Morse #include "llvm/CodeGen/TargetSubtargetInfo.h" 105ae6f7882SJeremy Morse #include "llvm/Config/llvm-config.h" 106ae6f7882SJeremy Morse #include "llvm/IR/DebugInfoMetadata.h" 107ae6f7882SJeremy Morse #include "llvm/IR/DebugLoc.h" 108ae6f7882SJeremy Morse #include "llvm/IR/Function.h" 109ae6f7882SJeremy Morse #include "llvm/MC/MCRegisterInfo.h" 110ae6f7882SJeremy Morse #include "llvm/Support/Casting.h" 111ae6f7882SJeremy Morse #include "llvm/Support/Compiler.h" 112ae6f7882SJeremy Morse #include "llvm/Support/Debug.h" 113*989f1c72Sserge-sans-paille #include "llvm/Support/GenericIteratedDominanceFrontier.h" 11484a11209SSander de Smalen #include "llvm/Support/TypeSize.h" 115ae6f7882SJeremy Morse #include "llvm/Support/raw_ostream.h" 1161575583fSJeremy Morse #include "llvm/Target/TargetMachine.h" 117010108bbSJeremy Morse #include "llvm/Transforms/Utils/SSAUpdaterImpl.h" 118ae6f7882SJeremy Morse #include <algorithm> 119ae6f7882SJeremy Morse #include <cassert> 120*989f1c72Sserge-sans-paille #include <climits> 121ae6f7882SJeremy Morse #include <cstdint> 122ae6f7882SJeremy Morse #include <functional> 123ae6f7882SJeremy Morse #include <queue> 124ae6f7882SJeremy Morse #include <tuple> 125ae6f7882SJeremy Morse #include <utility> 126ae6f7882SJeremy Morse #include <vector> 127ae6f7882SJeremy Morse 128838b4a53SJeremy Morse #include "InstrRefBasedImpl.h" 129ae6f7882SJeremy Morse #include "LiveDebugValues.h" 130ae6f7882SJeremy Morse 131ae6f7882SJeremy Morse using namespace llvm; 132838b4a53SJeremy Morse using namespace LiveDebugValues; 133ae6f7882SJeremy Morse 134010108bbSJeremy Morse // SSAUpdaterImple sets DEBUG_TYPE, change it. 135010108bbSJeremy Morse #undef DEBUG_TYPE 136ae6f7882SJeremy Morse #define DEBUG_TYPE "livedebugvalues" 137ae6f7882SJeremy Morse 138ae6f7882SJeremy Morse // Act more like the VarLoc implementation, by propagating some locations too 139ae6f7882SJeremy Morse // far and ignoring some transfers. 140ae6f7882SJeremy Morse static cl::opt<bool> EmulateOldLDV("emulate-old-livedebugvalues", cl::Hidden, 141ae6f7882SJeremy Morse cl::desc("Act like old LiveDebugValues did"), 142ae6f7882SJeremy Morse cl::init(false)); 143ae6f7882SJeremy Morse 14414aaaa12SJeremy Morse // Limit for the maximum number of stack slots we should track, past which we 14514aaaa12SJeremy Morse // will ignore any spills. InstrRefBasedLDV gathers detailed information on all 14614aaaa12SJeremy Morse // stack slots which leads to high memory consumption, and in some scenarios 14714aaaa12SJeremy Morse // (such as asan with very many locals) the working set of the function can be 14814aaaa12SJeremy Morse // very large, causing many spills. In these scenarios, it is very unlikely that 14914aaaa12SJeremy Morse // the developer has hundreds of variables live at the same time that they're 15014aaaa12SJeremy Morse // carefully thinking about -- instead, they probably autogenerated the code. 15114aaaa12SJeremy Morse // When this happens, gracefully stop tracking excess spill slots, rather than 15214aaaa12SJeremy Morse // consuming all the developer's memory. 15314aaaa12SJeremy Morse static cl::opt<unsigned> 15414aaaa12SJeremy Morse StackWorkingSetLimit("livedebugvalues-max-stack-slots", cl::Hidden, 15514aaaa12SJeremy Morse cl::desc("livedebugvalues-stack-ws-limit"), 15614aaaa12SJeremy Morse cl::init(250)); 15714aaaa12SJeremy Morse 158ae6f7882SJeremy Morse /// Tracker for converting machine value locations and variable values into 159ae6f7882SJeremy Morse /// variable locations (the output of LiveDebugValues), recorded as DBG_VALUEs 160ae6f7882SJeremy Morse /// specifying block live-in locations and transfers within blocks. 161ae6f7882SJeremy Morse /// 162ae6f7882SJeremy Morse /// Operating on a per-block basis, this class takes a (pre-loaded) MLocTracker 163ae6f7882SJeremy Morse /// and must be initialized with the set of variable values that are live-in to 164ae6f7882SJeremy Morse /// the block. The caller then repeatedly calls process(). TransferTracker picks 165ae6f7882SJeremy Morse /// out variable locations for the live-in variable values (if there _is_ a 166ae6f7882SJeremy Morse /// location) and creates the corresponding DBG_VALUEs. Then, as the block is 167ae6f7882SJeremy Morse /// stepped through, transfers of values between machine locations are 168ae6f7882SJeremy Morse /// identified and if profitable, a DBG_VALUE created. 169ae6f7882SJeremy Morse /// 170ae6f7882SJeremy Morse /// This is where debug use-before-defs would be resolved: a variable with an 171ae6f7882SJeremy Morse /// unavailable value could materialize in the middle of a block, when the 172ae6f7882SJeremy Morse /// value becomes available. Or, we could detect clobbers and re-specify the 173ae6f7882SJeremy Morse /// variable in a backup location. (XXX these are unimplemented). 174ae6f7882SJeremy Morse class TransferTracker { 175ae6f7882SJeremy Morse public: 176ae6f7882SJeremy Morse const TargetInstrInfo *TII; 1771575583fSJeremy Morse const TargetLowering *TLI; 178ae6f7882SJeremy Morse /// This machine location tracker is assumed to always contain the up-to-date 179ae6f7882SJeremy Morse /// value mapping for all machine locations. TransferTracker only reads 180ae6f7882SJeremy Morse /// information from it. (XXX make it const?) 181ae6f7882SJeremy Morse MLocTracker *MTracker; 182ae6f7882SJeremy Morse MachineFunction &MF; 1831575583fSJeremy Morse bool ShouldEmitDebugEntryValues; 184ae6f7882SJeremy Morse 185ae6f7882SJeremy Morse /// Record of all changes in variable locations at a block position. Awkwardly 186ae6f7882SJeremy Morse /// we allow inserting either before or after the point: MBB != nullptr 187ae6f7882SJeremy Morse /// indicates it's before, otherwise after. 188ae6f7882SJeremy Morse struct Transfer { 1891575583fSJeremy Morse MachineBasicBlock::instr_iterator Pos; /// Position to insert DBG_VALUes 190ae6f7882SJeremy Morse MachineBasicBlock *MBB; /// non-null if we should insert after. 191ae6f7882SJeremy Morse SmallVector<MachineInstr *, 4> Insts; /// Vector of DBG_VALUEs to insert. 192ae6f7882SJeremy Morse }; 193ae6f7882SJeremy Morse 19459d90fe8SFangrui Song struct LocAndProperties { 195ae6f7882SJeremy Morse LocIdx Loc; 196ae6f7882SJeremy Morse DbgValueProperties Properties; 19759d90fe8SFangrui Song }; 198ae6f7882SJeremy Morse 199ae6f7882SJeremy Morse /// Collection of transfers (DBG_VALUEs) to be inserted. 200ae6f7882SJeremy Morse SmallVector<Transfer, 32> Transfers; 201ae6f7882SJeremy Morse 202ae6f7882SJeremy Morse /// Local cache of what-value-is-in-what-LocIdx. Used to identify differences 203ae6f7882SJeremy Morse /// between TransferTrackers view of variable locations and MLocTrackers. For 204ae6f7882SJeremy Morse /// example, MLocTracker observes all clobbers, but TransferTracker lazily 205ae6f7882SJeremy Morse /// does not. 2064136897bSJeremy Morse SmallVector<ValueIDNum, 32> VarLocs; 207ae6f7882SJeremy Morse 208ae6f7882SJeremy Morse /// Map from LocIdxes to which DebugVariables are based that location. 209ae6f7882SJeremy Morse /// Mantained while stepping through the block. Not accurate if 210ae6f7882SJeremy Morse /// VarLocs[Idx] != MTracker->LocIdxToIDNum[Idx]. 2114136897bSJeremy Morse DenseMap<LocIdx, SmallSet<DebugVariable, 4>> ActiveMLocs; 212ae6f7882SJeremy Morse 213ae6f7882SJeremy Morse /// Map from DebugVariable to it's current location and qualifying meta 214ae6f7882SJeremy Morse /// information. To be used in conjunction with ActiveMLocs to construct 215ae6f7882SJeremy Morse /// enough information for the DBG_VALUEs for a particular LocIdx. 216ae6f7882SJeremy Morse DenseMap<DebugVariable, LocAndProperties> ActiveVLocs; 217ae6f7882SJeremy Morse 218ae6f7882SJeremy Morse /// Temporary cache of DBG_VALUEs to be entered into the Transfers collection. 219ae6f7882SJeremy Morse SmallVector<MachineInstr *, 4> PendingDbgValues; 220ae6f7882SJeremy Morse 221b1b2c6abSJeremy Morse /// Record of a use-before-def: created when a value that's live-in to the 222b1b2c6abSJeremy Morse /// current block isn't available in any machine location, but it will be 223b1b2c6abSJeremy Morse /// defined in this block. 224b1b2c6abSJeremy Morse struct UseBeforeDef { 225b1b2c6abSJeremy Morse /// Value of this variable, def'd in block. 226b1b2c6abSJeremy Morse ValueIDNum ID; 227b1b2c6abSJeremy Morse /// Identity of this variable. 228b1b2c6abSJeremy Morse DebugVariable Var; 229b1b2c6abSJeremy Morse /// Additional variable properties. 230b1b2c6abSJeremy Morse DbgValueProperties Properties; 231b1b2c6abSJeremy Morse }; 232b1b2c6abSJeremy Morse 233b1b2c6abSJeremy Morse /// Map from instruction index (within the block) to the set of UseBeforeDefs 234b1b2c6abSJeremy Morse /// that become defined at that instruction. 235b1b2c6abSJeremy Morse DenseMap<unsigned, SmallVector<UseBeforeDef, 1>> UseBeforeDefs; 236b1b2c6abSJeremy Morse 237b1b2c6abSJeremy Morse /// The set of variables that are in UseBeforeDefs and can become a location 238b1b2c6abSJeremy Morse /// once the relevant value is defined. An element being erased from this 239b1b2c6abSJeremy Morse /// collection prevents the use-before-def materializing. 240b1b2c6abSJeremy Morse DenseSet<DebugVariable> UseBeforeDefVariables; 241b1b2c6abSJeremy Morse 242ae6f7882SJeremy Morse const TargetRegisterInfo &TRI; 243ae6f7882SJeremy Morse const BitVector &CalleeSavedRegs; 244ae6f7882SJeremy Morse 245ae6f7882SJeremy Morse TransferTracker(const TargetInstrInfo *TII, MLocTracker *MTracker, 246ae6f7882SJeremy Morse MachineFunction &MF, const TargetRegisterInfo &TRI, 2471575583fSJeremy Morse const BitVector &CalleeSavedRegs, const TargetPassConfig &TPC) 248ae6f7882SJeremy Morse : TII(TII), MTracker(MTracker), MF(MF), TRI(TRI), 2491575583fSJeremy Morse CalleeSavedRegs(CalleeSavedRegs) { 2501575583fSJeremy Morse TLI = MF.getSubtarget().getTargetLowering(); 2511575583fSJeremy Morse auto &TM = TPC.getTM<TargetMachine>(); 2521575583fSJeremy Morse ShouldEmitDebugEntryValues = TM.Options.ShouldEmitDebugEntryValues(); 2531575583fSJeremy Morse } 254ae6f7882SJeremy Morse 255ae6f7882SJeremy Morse /// Load object with live-in variable values. \p mlocs contains the live-in 256ae6f7882SJeremy Morse /// values in each machine location, while \p vlocs the live-in variable 257ae6f7882SJeremy Morse /// values. This method picks variable locations for the live-in variables, 258ae6f7882SJeremy Morse /// creates DBG_VALUEs and puts them in #Transfers, then prepares the other 259ae6f7882SJeremy Morse /// object fields to track variable locations as we step through the block. 260ae6f7882SJeremy Morse /// FIXME: could just examine mloctracker instead of passing in \p mlocs? 2610d51b6abSNikita Popov void 262ab49dce0SJeremy Morse loadInlocs(MachineBasicBlock &MBB, ValueTable &MLocs, 2630d51b6abSNikita Popov const SmallVectorImpl<std::pair<DebugVariable, DbgValue>> &VLocs, 264ae6f7882SJeremy Morse unsigned NumLocs) { 265ae6f7882SJeremy Morse ActiveMLocs.clear(); 266ae6f7882SJeremy Morse ActiveVLocs.clear(); 267ae6f7882SJeremy Morse VarLocs.clear(); 268ae6f7882SJeremy Morse VarLocs.reserve(NumLocs); 269b1b2c6abSJeremy Morse UseBeforeDefs.clear(); 270b1b2c6abSJeremy Morse UseBeforeDefVariables.clear(); 271ae6f7882SJeremy Morse 272ae6f7882SJeremy Morse auto isCalleeSaved = [&](LocIdx L) { 273ae6f7882SJeremy Morse unsigned Reg = MTracker->LocIdxToLocID[L]; 274ae6f7882SJeremy Morse if (Reg >= MTracker->NumRegs) 275ae6f7882SJeremy Morse return false; 276ae6f7882SJeremy Morse for (MCRegAliasIterator RAI(Reg, &TRI, true); RAI.isValid(); ++RAI) 277ae6f7882SJeremy Morse if (CalleeSavedRegs.test(*RAI)) 278ae6f7882SJeremy Morse return true; 279ae6f7882SJeremy Morse return false; 280ae6f7882SJeremy Morse }; 281ae6f7882SJeremy Morse 282ae6f7882SJeremy Morse // Map of the preferred location for each value. 283cbaae614SNikita Popov DenseMap<ValueIDNum, LocIdx> ValueToLoc; 28491fb66cfSJeremy Morse 28591fb66cfSJeremy Morse // Initialized the preferred-location map with illegal locations, to be 28691fb66cfSJeremy Morse // filled in later. 28791fb66cfSJeremy Morse for (auto &VLoc : VLocs) 28891fb66cfSJeremy Morse if (VLoc.second.Kind == DbgValue::Def) 28991fb66cfSJeremy Morse ValueToLoc.insert({VLoc.second.ID, LocIdx::MakeIllegalLoc()}); 29091fb66cfSJeremy Morse 2914136897bSJeremy Morse ActiveMLocs.reserve(VLocs.size()); 2924136897bSJeremy Morse ActiveVLocs.reserve(VLocs.size()); 293ae6f7882SJeremy Morse 294ae6f7882SJeremy Morse // Produce a map of value numbers to the current machine locs they live 295ae6f7882SJeremy Morse // in. When emulating VarLocBasedImpl, there should only be one 296ae6f7882SJeremy Morse // location; when not, we get to pick. 297ae6f7882SJeremy Morse for (auto Location : MTracker->locations()) { 298ae6f7882SJeremy Morse LocIdx Idx = Location.Idx; 299ae6f7882SJeremy Morse ValueIDNum &VNum = MLocs[Idx.asU64()]; 300ae6f7882SJeremy Morse VarLocs.push_back(VNum); 301764e52f0SEugene Zhulenev 30291fb66cfSJeremy Morse // Is there a variable that wants a location for this value? If not, skip. 30391fb66cfSJeremy Morse auto VIt = ValueToLoc.find(VNum); 30491fb66cfSJeremy Morse if (VIt == ValueToLoc.end()) 305764e52f0SEugene Zhulenev continue; 306764e52f0SEugene Zhulenev 30791fb66cfSJeremy Morse LocIdx CurLoc = VIt->second; 308ae6f7882SJeremy Morse // In order of preference, pick: 309ae6f7882SJeremy Morse // * Callee saved registers, 310ae6f7882SJeremy Morse // * Other registers, 311ae6f7882SJeremy Morse // * Spill slots. 31291fb66cfSJeremy Morse if (CurLoc.isIllegal() || MTracker->isSpill(CurLoc) || 31391fb66cfSJeremy Morse (!isCalleeSaved(CurLoc) && isCalleeSaved(Idx.asU64()))) { 314ae6f7882SJeremy Morse // Insert, or overwrite if insertion failed. 31591fb66cfSJeremy Morse VIt->second = Idx; 316ae6f7882SJeremy Morse } 317ae6f7882SJeremy Morse } 318ae6f7882SJeremy Morse 319ae6f7882SJeremy Morse // Now map variables to their picked LocIdxes. 3200d51b6abSNikita Popov for (const auto &Var : VLocs) { 321ae6f7882SJeremy Morse if (Var.second.Kind == DbgValue::Const) { 322ae6f7882SJeremy Morse PendingDbgValues.push_back( 323b5426cedSJeremy Morse emitMOLoc(*Var.second.MO, Var.first, Var.second.Properties)); 324ae6f7882SJeremy Morse continue; 325ae6f7882SJeremy Morse } 326ae6f7882SJeremy Morse 327ae6f7882SJeremy Morse // If the value has no location, we can't make a variable location. 328b1b2c6abSJeremy Morse const ValueIDNum &Num = Var.second.ID; 329b1b2c6abSJeremy Morse auto ValuesPreferredLoc = ValueToLoc.find(Num); 33091fb66cfSJeremy Morse if (ValuesPreferredLoc->second.isIllegal()) { 331b1b2c6abSJeremy Morse // If it's a def that occurs in this block, register it as a 332b1b2c6abSJeremy Morse // use-before-def to be resolved as we step through the block. 333b1b2c6abSJeremy Morse if (Num.getBlock() == (unsigned)MBB.getNumber() && !Num.isPHI()) 334b1b2c6abSJeremy Morse addUseBeforeDef(Var.first, Var.second.Properties, Num); 3351575583fSJeremy Morse else 3361575583fSJeremy Morse recoverAsEntryValue(Var.first, Var.second.Properties, Num); 337ae6f7882SJeremy Morse continue; 338b1b2c6abSJeremy Morse } 339ae6f7882SJeremy Morse 340ae6f7882SJeremy Morse LocIdx M = ValuesPreferredLoc->second; 341ae6f7882SJeremy Morse auto NewValue = LocAndProperties{M, Var.second.Properties}; 342ae6f7882SJeremy Morse auto Result = ActiveVLocs.insert(std::make_pair(Var.first, NewValue)); 343ae6f7882SJeremy Morse if (!Result.second) 344ae6f7882SJeremy Morse Result.first->second = NewValue; 345ae6f7882SJeremy Morse ActiveMLocs[M].insert(Var.first); 346ae6f7882SJeremy Morse PendingDbgValues.push_back( 347ae6f7882SJeremy Morse MTracker->emitLoc(M, Var.first, Var.second.Properties)); 348ae6f7882SJeremy Morse } 349ae6f7882SJeremy Morse flushDbgValues(MBB.begin(), &MBB); 350ae6f7882SJeremy Morse } 351ae6f7882SJeremy Morse 352b1b2c6abSJeremy Morse /// Record that \p Var has value \p ID, a value that becomes available 353b1b2c6abSJeremy Morse /// later in the function. 354b1b2c6abSJeremy Morse void addUseBeforeDef(const DebugVariable &Var, 355b1b2c6abSJeremy Morse const DbgValueProperties &Properties, ValueIDNum ID) { 356b1b2c6abSJeremy Morse UseBeforeDef UBD = {ID, Var, Properties}; 357b1b2c6abSJeremy Morse UseBeforeDefs[ID.getInst()].push_back(UBD); 358b1b2c6abSJeremy Morse UseBeforeDefVariables.insert(Var); 359b1b2c6abSJeremy Morse } 360b1b2c6abSJeremy Morse 361b1b2c6abSJeremy Morse /// After the instruction at index \p Inst and position \p pos has been 362b1b2c6abSJeremy Morse /// processed, check whether it defines a variable value in a use-before-def. 363b1b2c6abSJeremy Morse /// If so, and the variable value hasn't changed since the start of the 364b1b2c6abSJeremy Morse /// block, create a DBG_VALUE. 365b1b2c6abSJeremy Morse void checkInstForNewValues(unsigned Inst, MachineBasicBlock::iterator pos) { 366b1b2c6abSJeremy Morse auto MIt = UseBeforeDefs.find(Inst); 367b1b2c6abSJeremy Morse if (MIt == UseBeforeDefs.end()) 368b1b2c6abSJeremy Morse return; 369b1b2c6abSJeremy Morse 370b1b2c6abSJeremy Morse for (auto &Use : MIt->second) { 371b1b2c6abSJeremy Morse LocIdx L = Use.ID.getLoc(); 372b1b2c6abSJeremy Morse 373b1b2c6abSJeremy Morse // If something goes very wrong, we might end up labelling a COPY 374b1b2c6abSJeremy Morse // instruction or similar with an instruction number, where it doesn't 375b1b2c6abSJeremy Morse // actually define a new value, instead it moves a value. In case this 376b1b2c6abSJeremy Morse // happens, discard. 377d9eebe3cSJeremy Morse if (MTracker->readMLoc(L) != Use.ID) 378b1b2c6abSJeremy Morse continue; 379b1b2c6abSJeremy Morse 380b1b2c6abSJeremy Morse // If a different debug instruction defined the variable value / location 381b1b2c6abSJeremy Morse // since the start of the block, don't materialize this use-before-def. 382b1b2c6abSJeremy Morse if (!UseBeforeDefVariables.count(Use.Var)) 383b1b2c6abSJeremy Morse continue; 384b1b2c6abSJeremy Morse 385b1b2c6abSJeremy Morse PendingDbgValues.push_back(MTracker->emitLoc(L, Use.Var, Use.Properties)); 386b1b2c6abSJeremy Morse } 387b1b2c6abSJeremy Morse flushDbgValues(pos, nullptr); 388b1b2c6abSJeremy Morse } 389b1b2c6abSJeremy Morse 390ae6f7882SJeremy Morse /// Helper to move created DBG_VALUEs into Transfers collection. 391ae6f7882SJeremy Morse void flushDbgValues(MachineBasicBlock::iterator Pos, MachineBasicBlock *MBB) { 3921575583fSJeremy Morse if (PendingDbgValues.size() == 0) 3931575583fSJeremy Morse return; 3941575583fSJeremy Morse 3951575583fSJeremy Morse // Pick out the instruction start position. 3961575583fSJeremy Morse MachineBasicBlock::instr_iterator BundleStart; 3971575583fSJeremy Morse if (MBB && Pos == MBB->begin()) 3981575583fSJeremy Morse BundleStart = MBB->instr_begin(); 3991575583fSJeremy Morse else 4001575583fSJeremy Morse BundleStart = getBundleStart(Pos->getIterator()); 4011575583fSJeremy Morse 4021575583fSJeremy Morse Transfers.push_back({BundleStart, MBB, PendingDbgValues}); 403ae6f7882SJeremy Morse PendingDbgValues.clear(); 404ae6f7882SJeremy Morse } 4051575583fSJeremy Morse 4061575583fSJeremy Morse bool isEntryValueVariable(const DebugVariable &Var, 4071575583fSJeremy Morse const DIExpression *Expr) const { 4081575583fSJeremy Morse if (!Var.getVariable()->isParameter()) 4091575583fSJeremy Morse return false; 4101575583fSJeremy Morse 4111575583fSJeremy Morse if (Var.getInlinedAt()) 4121575583fSJeremy Morse return false; 4131575583fSJeremy Morse 4141575583fSJeremy Morse if (Expr->getNumElements() > 0) 4151575583fSJeremy Morse return false; 4161575583fSJeremy Morse 4171575583fSJeremy Morse return true; 4181575583fSJeremy Morse } 4191575583fSJeremy Morse 4201575583fSJeremy Morse bool isEntryValueValue(const ValueIDNum &Val) const { 4211575583fSJeremy Morse // Must be in entry block (block number zero), and be a PHI / live-in value. 4221575583fSJeremy Morse if (Val.getBlock() || !Val.isPHI()) 4231575583fSJeremy Morse return false; 4241575583fSJeremy Morse 4251575583fSJeremy Morse // Entry values must enter in a register. 4261575583fSJeremy Morse if (MTracker->isSpill(Val.getLoc())) 4271575583fSJeremy Morse return false; 4281575583fSJeremy Morse 4291575583fSJeremy Morse Register SP = TLI->getStackPointerRegisterToSaveRestore(); 4301575583fSJeremy Morse Register FP = TRI.getFrameRegister(MF); 4311575583fSJeremy Morse Register Reg = MTracker->LocIdxToLocID[Val.getLoc()]; 4321575583fSJeremy Morse return Reg != SP && Reg != FP; 4331575583fSJeremy Morse } 4341575583fSJeremy Morse 4350d51b6abSNikita Popov bool recoverAsEntryValue(const DebugVariable &Var, 4360d51b6abSNikita Popov const DbgValueProperties &Prop, 4371575583fSJeremy Morse const ValueIDNum &Num) { 4381575583fSJeremy Morse // Is this variable location a candidate to be an entry value. First, 4391575583fSJeremy Morse // should we be trying this at all? 4401575583fSJeremy Morse if (!ShouldEmitDebugEntryValues) 4411575583fSJeremy Morse return false; 4421575583fSJeremy Morse 4431575583fSJeremy Morse // Is the variable appropriate for entry values (i.e., is a parameter). 4441575583fSJeremy Morse if (!isEntryValueVariable(Var, Prop.DIExpr)) 4451575583fSJeremy Morse return false; 4461575583fSJeremy Morse 4471575583fSJeremy Morse // Is the value assigned to this variable still the entry value? 4481575583fSJeremy Morse if (!isEntryValueValue(Num)) 4491575583fSJeremy Morse return false; 4501575583fSJeremy Morse 4511575583fSJeremy Morse // Emit a variable location using an entry value expression. 4521575583fSJeremy Morse DIExpression *NewExpr = 4531575583fSJeremy Morse DIExpression::prepend(Prop.DIExpr, DIExpression::EntryValue); 4541575583fSJeremy Morse Register Reg = MTracker->LocIdxToLocID[Num.getLoc()]; 4551575583fSJeremy Morse MachineOperand MO = MachineOperand::CreateReg(Reg, false); 4561575583fSJeremy Morse 4571575583fSJeremy Morse PendingDbgValues.push_back(emitMOLoc(MO, Var, {NewExpr, Prop.Indirect})); 4581575583fSJeremy Morse return true; 459ae6f7882SJeremy Morse } 460ae6f7882SJeremy Morse 46168f47157SJeremy Morse /// Change a variable value after encountering a DBG_VALUE inside a block. 462ae6f7882SJeremy Morse void redefVar(const MachineInstr &MI) { 463ae6f7882SJeremy Morse DebugVariable Var(MI.getDebugVariable(), MI.getDebugExpression(), 464ae6f7882SJeremy Morse MI.getDebugLoc()->getInlinedAt()); 46568f47157SJeremy Morse DbgValueProperties Properties(MI); 46668f47157SJeremy Morse 467ae6f7882SJeremy Morse const MachineOperand &MO = MI.getOperand(0); 468ae6f7882SJeremy Morse 46968f47157SJeremy Morse // Ignore non-register locations, we don't transfer those. 47068f47157SJeremy Morse if (!MO.isReg() || MO.getReg() == 0) { 471ae6f7882SJeremy Morse auto It = ActiveVLocs.find(Var); 472ae6f7882SJeremy Morse if (It != ActiveVLocs.end()) { 473ae6f7882SJeremy Morse ActiveMLocs[It->second.Loc].erase(Var); 474ae6f7882SJeremy Morse ActiveVLocs.erase(It); 47568f47157SJeremy Morse } 476b1b2c6abSJeremy Morse // Any use-before-defs no longer apply. 477b1b2c6abSJeremy Morse UseBeforeDefVariables.erase(Var); 478ae6f7882SJeremy Morse return; 479ae6f7882SJeremy Morse } 480ae6f7882SJeremy Morse 481ae6f7882SJeremy Morse Register Reg = MO.getReg(); 48268f47157SJeremy Morse LocIdx NewLoc = MTracker->getRegMLoc(Reg); 48368f47157SJeremy Morse redefVar(MI, Properties, NewLoc); 48468f47157SJeremy Morse } 48568f47157SJeremy Morse 48668f47157SJeremy Morse /// Handle a change in variable location within a block. Terminate the 48768f47157SJeremy Morse /// variables current location, and record the value it now refers to, so 48868f47157SJeremy Morse /// that we can detect location transfers later on. 48968f47157SJeremy Morse void redefVar(const MachineInstr &MI, const DbgValueProperties &Properties, 49068f47157SJeremy Morse Optional<LocIdx> OptNewLoc) { 49168f47157SJeremy Morse DebugVariable Var(MI.getDebugVariable(), MI.getDebugExpression(), 49268f47157SJeremy Morse MI.getDebugLoc()->getInlinedAt()); 493b1b2c6abSJeremy Morse // Any use-before-defs no longer apply. 494b1b2c6abSJeremy Morse UseBeforeDefVariables.erase(Var); 49568f47157SJeremy Morse 49668f47157SJeremy Morse // Erase any previous location, 49768f47157SJeremy Morse auto It = ActiveVLocs.find(Var); 49868f47157SJeremy Morse if (It != ActiveVLocs.end()) 49968f47157SJeremy Morse ActiveMLocs[It->second.Loc].erase(Var); 50068f47157SJeremy Morse 50168f47157SJeremy Morse // If there _is_ no new location, all we had to do was erase. 50268f47157SJeremy Morse if (!OptNewLoc) 50368f47157SJeremy Morse return; 50468f47157SJeremy Morse LocIdx NewLoc = *OptNewLoc; 505ae6f7882SJeremy Morse 506ae6f7882SJeremy Morse // Check whether our local copy of values-by-location in #VarLocs is out of 507ae6f7882SJeremy Morse // date. Wipe old tracking data for the location if it's been clobbered in 508ae6f7882SJeremy Morse // the meantime. 509d9eebe3cSJeremy Morse if (MTracker->readMLoc(NewLoc) != VarLocs[NewLoc.asU64()]) { 51068f47157SJeremy Morse for (auto &P : ActiveMLocs[NewLoc]) { 511ae6f7882SJeremy Morse ActiveVLocs.erase(P); 512ae6f7882SJeremy Morse } 51368f47157SJeremy Morse ActiveMLocs[NewLoc.asU64()].clear(); 514d9eebe3cSJeremy Morse VarLocs[NewLoc.asU64()] = MTracker->readMLoc(NewLoc); 515ae6f7882SJeremy Morse } 516ae6f7882SJeremy Morse 51768f47157SJeremy Morse ActiveMLocs[NewLoc].insert(Var); 518ae6f7882SJeremy Morse if (It == ActiveVLocs.end()) { 51968f47157SJeremy Morse ActiveVLocs.insert( 52068f47157SJeremy Morse std::make_pair(Var, LocAndProperties{NewLoc, Properties})); 521ae6f7882SJeremy Morse } else { 52268f47157SJeremy Morse It->second.Loc = NewLoc; 523ae6f7882SJeremy Morse It->second.Properties = Properties; 524ae6f7882SJeremy Morse } 525ae6f7882SJeremy Morse } 526ae6f7882SJeremy Morse 52749555441SJeremy Morse /// Account for a location \p mloc being clobbered. Examine the variable 52849555441SJeremy Morse /// locations that will be terminated: and try to recover them by using 52949555441SJeremy Morse /// another location. Optionally, given \p MakeUndef, emit a DBG_VALUE to 53049555441SJeremy Morse /// explicitly terminate a location if it can't be recovered. 53149555441SJeremy Morse void clobberMloc(LocIdx MLoc, MachineBasicBlock::iterator Pos, 53249555441SJeremy Morse bool MakeUndef = true) { 533ae6f7882SJeremy Morse auto ActiveMLocIt = ActiveMLocs.find(MLoc); 534ae6f7882SJeremy Morse if (ActiveMLocIt == ActiveMLocs.end()) 535ae6f7882SJeremy Morse return; 536ae6f7882SJeremy Morse 53749555441SJeremy Morse // What was the old variable value? 53849555441SJeremy Morse ValueIDNum OldValue = VarLocs[MLoc.asU64()]; 539ae6f7882SJeremy Morse VarLocs[MLoc.asU64()] = ValueIDNum::EmptyValue; 540ae6f7882SJeremy Morse 54149555441SJeremy Morse // Examine the remaining variable locations: if we can find the same value 54249555441SJeremy Morse // again, we can recover the location. 54349555441SJeremy Morse Optional<LocIdx> NewLoc = None; 54449555441SJeremy Morse for (auto Loc : MTracker->locations()) 54549555441SJeremy Morse if (Loc.Value == OldValue) 54649555441SJeremy Morse NewLoc = Loc.Idx; 54749555441SJeremy Morse 54849555441SJeremy Morse // If there is no location, and we weren't asked to make the variable 54949555441SJeremy Morse // explicitly undef, then stop here. 5501575583fSJeremy Morse if (!NewLoc && !MakeUndef) { 5511575583fSJeremy Morse // Try and recover a few more locations with entry values. 5521575583fSJeremy Morse for (auto &Var : ActiveMLocIt->second) { 5531575583fSJeremy Morse auto &Prop = ActiveVLocs.find(Var)->second.Properties; 5541575583fSJeremy Morse recoverAsEntryValue(Var, Prop, OldValue); 5551575583fSJeremy Morse } 5561575583fSJeremy Morse flushDbgValues(Pos, nullptr); 55749555441SJeremy Morse return; 5581575583fSJeremy Morse } 55949555441SJeremy Morse 56049555441SJeremy Morse // Examine all the variables based on this location. 56149555441SJeremy Morse DenseSet<DebugVariable> NewMLocs; 562ae6f7882SJeremy Morse for (auto &Var : ActiveMLocIt->second) { 563ae6f7882SJeremy Morse auto ActiveVLocIt = ActiveVLocs.find(Var); 56449555441SJeremy Morse // Re-state the variable location: if there's no replacement then NewLoc 56549555441SJeremy Morse // is None and a $noreg DBG_VALUE will be created. Otherwise, a DBG_VALUE 56649555441SJeremy Morse // identifying the alternative location will be emitted. 5679cf31b8dSJeremy Morse const DbgValueProperties &Properties = ActiveVLocIt->second.Properties; 56849555441SJeremy Morse PendingDbgValues.push_back(MTracker->emitLoc(NewLoc, Var, Properties)); 56949555441SJeremy Morse 57049555441SJeremy Morse // Update machine locations <=> variable locations maps. Defer updating 57149555441SJeremy Morse // ActiveMLocs to avoid invalidaing the ActiveMLocIt iterator. 57249555441SJeremy Morse if (!NewLoc) { 573ae6f7882SJeremy Morse ActiveVLocs.erase(ActiveVLocIt); 57449555441SJeremy Morse } else { 57549555441SJeremy Morse ActiveVLocIt->second.Loc = *NewLoc; 57649555441SJeremy Morse NewMLocs.insert(Var); 577ae6f7882SJeremy Morse } 57849555441SJeremy Morse } 57949555441SJeremy Morse 58049555441SJeremy Morse // Commit any deferred ActiveMLoc changes. 58149555441SJeremy Morse if (!NewMLocs.empty()) 58249555441SJeremy Morse for (auto &Var : NewMLocs) 58349555441SJeremy Morse ActiveMLocs[*NewLoc].insert(Var); 58449555441SJeremy Morse 58549555441SJeremy Morse // We lazily track what locations have which values; if we've found a new 58649555441SJeremy Morse // location for the clobbered value, remember it. 58749555441SJeremy Morse if (NewLoc) 58849555441SJeremy Morse VarLocs[NewLoc->asU64()] = OldValue; 58949555441SJeremy Morse 590ae6f7882SJeremy Morse flushDbgValues(Pos, nullptr); 591ae6f7882SJeremy Morse 5924136897bSJeremy Morse // Re-find ActiveMLocIt, iterator could have been invalidated. 5934136897bSJeremy Morse ActiveMLocIt = ActiveMLocs.find(MLoc); 594ae6f7882SJeremy Morse ActiveMLocIt->second.clear(); 595ae6f7882SJeremy Morse } 596ae6f7882SJeremy Morse 597ae6f7882SJeremy Morse /// Transfer variables based on \p Src to be based on \p Dst. This handles 598ae6f7882SJeremy Morse /// both register copies as well as spills and restores. Creates DBG_VALUEs 599ae6f7882SJeremy Morse /// describing the movement. 600ae6f7882SJeremy Morse void transferMlocs(LocIdx Src, LocIdx Dst, MachineBasicBlock::iterator Pos) { 601ae6f7882SJeremy Morse // Does Src still contain the value num we expect? If not, it's been 602ae6f7882SJeremy Morse // clobbered in the meantime, and our variable locations are stale. 603d9eebe3cSJeremy Morse if (VarLocs[Src.asU64()] != MTracker->readMLoc(Src)) 604ae6f7882SJeremy Morse return; 605ae6f7882SJeremy Morse 606ae6f7882SJeremy Morse // assert(ActiveMLocs[Dst].size() == 0); 607ae6f7882SJeremy Morse //^^^ Legitimate scenario on account of un-clobbered slot being assigned to? 6084136897bSJeremy Morse 6094136897bSJeremy Morse // Move set of active variables from one location to another. 6104136897bSJeremy Morse auto MovingVars = ActiveMLocs[Src]; 6114136897bSJeremy Morse ActiveMLocs[Dst] = MovingVars; 612ae6f7882SJeremy Morse VarLocs[Dst.asU64()] = VarLocs[Src.asU64()]; 613ae6f7882SJeremy Morse 614ae6f7882SJeremy Morse // For each variable based on Src; create a location at Dst. 6154136897bSJeremy Morse for (auto &Var : MovingVars) { 616ae6f7882SJeremy Morse auto ActiveVLocIt = ActiveVLocs.find(Var); 617ae6f7882SJeremy Morse assert(ActiveVLocIt != ActiveVLocs.end()); 618ae6f7882SJeremy Morse ActiveVLocIt->second.Loc = Dst; 619ae6f7882SJeremy Morse 620ae6f7882SJeremy Morse MachineInstr *MI = 621ae6f7882SJeremy Morse MTracker->emitLoc(Dst, Var, ActiveVLocIt->second.Properties); 622ae6f7882SJeremy Morse PendingDbgValues.push_back(MI); 623ae6f7882SJeremy Morse } 624ae6f7882SJeremy Morse ActiveMLocs[Src].clear(); 625ae6f7882SJeremy Morse flushDbgValues(Pos, nullptr); 626ae6f7882SJeremy Morse 627ae6f7882SJeremy Morse // XXX XXX XXX "pretend to be old LDV" means dropping all tracking data 628ae6f7882SJeremy Morse // about the old location. 629ae6f7882SJeremy Morse if (EmulateOldLDV) 630ae6f7882SJeremy Morse VarLocs[Src.asU64()] = ValueIDNum::EmptyValue; 631ae6f7882SJeremy Morse } 632ae6f7882SJeremy Morse 633ae6f7882SJeremy Morse MachineInstrBuilder emitMOLoc(const MachineOperand &MO, 634ae6f7882SJeremy Morse const DebugVariable &Var, 635ae6f7882SJeremy Morse const DbgValueProperties &Properties) { 636b5ad32efSFangrui Song DebugLoc DL = DILocation::get(Var.getVariable()->getContext(), 0, 0, 637b5ad32efSFangrui Song Var.getVariable()->getScope(), 638b5ad32efSFangrui Song const_cast<DILocation *>(Var.getInlinedAt())); 639ae6f7882SJeremy Morse auto MIB = BuildMI(MF, DL, TII->get(TargetOpcode::DBG_VALUE)); 640ae6f7882SJeremy Morse MIB.add(MO); 641ae6f7882SJeremy Morse if (Properties.Indirect) 642ae6f7882SJeremy Morse MIB.addImm(0); 643ae6f7882SJeremy Morse else 644ae6f7882SJeremy Morse MIB.addReg(0); 645ae6f7882SJeremy Morse MIB.addMetadata(Var.getVariable()); 646ae6f7882SJeremy Morse MIB.addMetadata(Properties.DIExpr); 647ae6f7882SJeremy Morse return MIB; 648ae6f7882SJeremy Morse } 649ae6f7882SJeremy Morse }; 650ae6f7882SJeremy Morse 651ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 652ae6f7882SJeremy Morse // Implementation 653ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 654ae6f7882SJeremy Morse 655ae6f7882SJeremy Morse ValueIDNum ValueIDNum::EmptyValue = {UINT_MAX, UINT_MAX, UINT_MAX}; 6564136897bSJeremy Morse ValueIDNum ValueIDNum::TombstoneValue = {UINT_MAX, UINT_MAX, UINT_MAX - 1}; 657ae6f7882SJeremy Morse 658d9fa186aSJeremy Morse #ifndef NDEBUG 659838b4a53SJeremy Morse void DbgValue::dump(const MLocTracker *MTrack) const { 660838b4a53SJeremy Morse if (Kind == Const) { 661b5426cedSJeremy Morse MO->dump(); 662838b4a53SJeremy Morse } else if (Kind == NoVal) { 663838b4a53SJeremy Morse dbgs() << "NoVal(" << BlockNo << ")"; 664b5426cedSJeremy Morse } else if (Kind == VPHI) { 665b5426cedSJeremy Morse dbgs() << "VPHI(" << BlockNo << "," << MTrack->IDAsString(ID) << ")"; 666838b4a53SJeremy Morse } else { 667838b4a53SJeremy Morse assert(Kind == Def); 668838b4a53SJeremy Morse dbgs() << MTrack->IDAsString(ID); 669838b4a53SJeremy Morse } 670838b4a53SJeremy Morse if (Properties.Indirect) 671838b4a53SJeremy Morse dbgs() << " indir"; 672838b4a53SJeremy Morse if (Properties.DIExpr) 673838b4a53SJeremy Morse dbgs() << " " << *Properties.DIExpr; 674838b4a53SJeremy Morse } 675d9fa186aSJeremy Morse #endif 676838b4a53SJeremy Morse 677838b4a53SJeremy Morse MLocTracker::MLocTracker(MachineFunction &MF, const TargetInstrInfo &TII, 678838b4a53SJeremy Morse const TargetRegisterInfo &TRI, 679838b4a53SJeremy Morse const TargetLowering &TLI) 680838b4a53SJeremy Morse : MF(MF), TII(TII), TRI(TRI), TLI(TLI), 681838b4a53SJeremy Morse LocIdxToIDNum(ValueIDNum::EmptyValue), LocIdxToLocID(0) { 682838b4a53SJeremy Morse NumRegs = TRI.getNumRegs(); 683838b4a53SJeremy Morse reset(); 684838b4a53SJeremy Morse LocIDToLocIdx.resize(NumRegs, LocIdx::MakeIllegalLoc()); 685838b4a53SJeremy Morse assert(NumRegs < (1u << NUM_LOC_BITS)); // Detect bit packing failure 686838b4a53SJeremy Morse 687838b4a53SJeremy Morse // Always track SP. This avoids the implicit clobbering caused by regmasks 688838b4a53SJeremy Morse // from affectings its values. (LiveDebugValues disbelieves calls and 689838b4a53SJeremy Morse // regmasks that claim to clobber SP). 690838b4a53SJeremy Morse Register SP = TLI.getStackPointerRegisterToSaveRestore(); 691838b4a53SJeremy Morse if (SP) { 692e7084ceaSJeremy Morse unsigned ID = getLocID(SP); 693838b4a53SJeremy Morse (void)lookupOrTrackRegister(ID); 694fbf269c7SJeremy Morse 695fbf269c7SJeremy Morse for (MCRegAliasIterator RAI(SP, &TRI, true); RAI.isValid(); ++RAI) 696fbf269c7SJeremy Morse SPAliases.insert(*RAI); 697838b4a53SJeremy Morse } 698e7084ceaSJeremy Morse 699e7084ceaSJeremy Morse // Build some common stack positions -- full registers being spilt to the 700e7084ceaSJeremy Morse // stack. 701e7084ceaSJeremy Morse StackSlotIdxes.insert({{8, 0}, 0}); 702e7084ceaSJeremy Morse StackSlotIdxes.insert({{16, 0}, 1}); 703e7084ceaSJeremy Morse StackSlotIdxes.insert({{32, 0}, 2}); 704e7084ceaSJeremy Morse StackSlotIdxes.insert({{64, 0}, 3}); 705e7084ceaSJeremy Morse StackSlotIdxes.insert({{128, 0}, 4}); 706e7084ceaSJeremy Morse StackSlotIdxes.insert({{256, 0}, 5}); 707e7084ceaSJeremy Morse StackSlotIdxes.insert({{512, 0}, 6}); 708e7084ceaSJeremy Morse 709e7084ceaSJeremy Morse // Traverse all the subregister idxes, and ensure there's an index for them. 710e7084ceaSJeremy Morse // Duplicates are no problem: we're interested in their position in the 711e7084ceaSJeremy Morse // stack slot, we don't want to type the slot. 712e7084ceaSJeremy Morse for (unsigned int I = 1; I < TRI.getNumSubRegIndices(); ++I) { 713e7084ceaSJeremy Morse unsigned Size = TRI.getSubRegIdxSize(I); 714e7084ceaSJeremy Morse unsigned Offs = TRI.getSubRegIdxOffset(I); 715e7084ceaSJeremy Morse unsigned Idx = StackSlotIdxes.size(); 716e7084ceaSJeremy Morse 717e7084ceaSJeremy Morse // Some subregs have -1, -2 and so forth fed into their fields, to mean 718e7084ceaSJeremy Morse // special backend things. Ignore those. 719e7084ceaSJeremy Morse if (Size > 60000 || Offs > 60000) 720e7084ceaSJeremy Morse continue; 721e7084ceaSJeremy Morse 722e7084ceaSJeremy Morse StackSlotIdxes.insert({{Size, Offs}, Idx}); 723e7084ceaSJeremy Morse } 724e7084ceaSJeremy Morse 725e7084ceaSJeremy Morse for (auto &Idx : StackSlotIdxes) 726e7084ceaSJeremy Morse StackIdxesToPos[Idx.second] = Idx.first; 727e7084ceaSJeremy Morse 728e7084ceaSJeremy Morse NumSlotIdxes = StackSlotIdxes.size(); 729838b4a53SJeremy Morse } 730838b4a53SJeremy Morse 731838b4a53SJeremy Morse LocIdx MLocTracker::trackRegister(unsigned ID) { 732838b4a53SJeremy Morse assert(ID != 0); 733838b4a53SJeremy Morse LocIdx NewIdx = LocIdx(LocIdxToIDNum.size()); 734838b4a53SJeremy Morse LocIdxToIDNum.grow(NewIdx); 735838b4a53SJeremy Morse LocIdxToLocID.grow(NewIdx); 736838b4a53SJeremy Morse 737838b4a53SJeremy Morse // Default: it's an mphi. 738838b4a53SJeremy Morse ValueIDNum ValNum = {CurBB, 0, NewIdx}; 739838b4a53SJeremy Morse // Was this reg ever touched by a regmask? 740838b4a53SJeremy Morse for (const auto &MaskPair : reverse(Masks)) { 741838b4a53SJeremy Morse if (MaskPair.first->clobbersPhysReg(ID)) { 742838b4a53SJeremy Morse // There was an earlier def we skipped. 743838b4a53SJeremy Morse ValNum = {CurBB, MaskPair.second, NewIdx}; 744838b4a53SJeremy Morse break; 745838b4a53SJeremy Morse } 746838b4a53SJeremy Morse } 747838b4a53SJeremy Morse 748838b4a53SJeremy Morse LocIdxToIDNum[NewIdx] = ValNum; 749838b4a53SJeremy Morse LocIdxToLocID[NewIdx] = ID; 750838b4a53SJeremy Morse return NewIdx; 751838b4a53SJeremy Morse } 752838b4a53SJeremy Morse 753838b4a53SJeremy Morse void MLocTracker::writeRegMask(const MachineOperand *MO, unsigned CurBB, 754838b4a53SJeremy Morse unsigned InstID) { 755838b4a53SJeremy Morse // Def any register we track have that isn't preserved. The regmask 756838b4a53SJeremy Morse // terminates the liveness of a register, meaning its value can't be 757838b4a53SJeremy Morse // relied upon -- we represent this by giving it a new value. 758838b4a53SJeremy Morse for (auto Location : locations()) { 759838b4a53SJeremy Morse unsigned ID = LocIdxToLocID[Location.Idx]; 760838b4a53SJeremy Morse // Don't clobber SP, even if the mask says it's clobbered. 761fbf269c7SJeremy Morse if (ID < NumRegs && !SPAliases.count(ID) && MO->clobbersPhysReg(ID)) 762838b4a53SJeremy Morse defReg(ID, CurBB, InstID); 763838b4a53SJeremy Morse } 764838b4a53SJeremy Morse Masks.push_back(std::make_pair(MO, InstID)); 765838b4a53SJeremy Morse } 766838b4a53SJeremy Morse 76714aaaa12SJeremy Morse Optional<SpillLocationNo> MLocTracker::getOrTrackSpillLoc(SpillLoc L) { 768e7084ceaSJeremy Morse SpillLocationNo SpillID(SpillLocs.idFor(L)); 76914aaaa12SJeremy Morse 770e7084ceaSJeremy Morse if (SpillID.id() == 0) { 77114aaaa12SJeremy Morse // If there is no location, and we have reached the limit of how many stack 77214aaaa12SJeremy Morse // slots to track, then don't track this one. 77314aaaa12SJeremy Morse if (SpillLocs.size() >= StackWorkingSetLimit) 77414aaaa12SJeremy Morse return None; 77514aaaa12SJeremy Morse 776e7084ceaSJeremy Morse // Spill location is untracked: create record for this one, and all 777e7084ceaSJeremy Morse // subregister slots too. 778e7084ceaSJeremy Morse SpillID = SpillLocationNo(SpillLocs.insert(L)); 779e7084ceaSJeremy Morse for (unsigned StackIdx = 0; StackIdx < NumSlotIdxes; ++StackIdx) { 780e7084ceaSJeremy Morse unsigned L = getSpillIDWithIdx(SpillID, StackIdx); 781838b4a53SJeremy Morse LocIdx Idx = LocIdx(LocIdxToIDNum.size()); // New idx 782838b4a53SJeremy Morse LocIdxToIDNum.grow(Idx); 783838b4a53SJeremy Morse LocIdxToLocID.grow(Idx); 784838b4a53SJeremy Morse LocIDToLocIdx.push_back(Idx); 785838b4a53SJeremy Morse LocIdxToLocID[Idx] = L; 786e7084ceaSJeremy Morse // Initialize to PHI value; corresponds to the location's live-in value 787e7084ceaSJeremy Morse // during transfer function construction. 788e7084ceaSJeremy Morse LocIdxToIDNum[Idx] = ValueIDNum(CurBB, 0, Idx); 789838b4a53SJeremy Morse } 790838b4a53SJeremy Morse } 791e7084ceaSJeremy Morse return SpillID; 792e7084ceaSJeremy Morse } 793838b4a53SJeremy Morse 794838b4a53SJeremy Morse std::string MLocTracker::LocIdxToName(LocIdx Idx) const { 795838b4a53SJeremy Morse unsigned ID = LocIdxToLocID[Idx]; 796e7084ceaSJeremy Morse if (ID >= NumRegs) { 797e7084ceaSJeremy Morse StackSlotPos Pos = locIDToSpillIdx(ID); 798e7084ceaSJeremy Morse ID -= NumRegs; 799e7084ceaSJeremy Morse unsigned Slot = ID / NumSlotIdxes; 800e7084ceaSJeremy Morse return Twine("slot ") 801e7084ceaSJeremy Morse .concat(Twine(Slot).concat(Twine(" sz ").concat(Twine(Pos.first) 802e7084ceaSJeremy Morse .concat(Twine(" offs ").concat(Twine(Pos.second)))))) 803e7084ceaSJeremy Morse .str(); 804e7084ceaSJeremy Morse } else { 805838b4a53SJeremy Morse return TRI.getRegAsmName(ID).str(); 806838b4a53SJeremy Morse } 807e7084ceaSJeremy Morse } 808838b4a53SJeremy Morse 809838b4a53SJeremy Morse std::string MLocTracker::IDAsString(const ValueIDNum &Num) const { 810838b4a53SJeremy Morse std::string DefName = LocIdxToName(Num.getLoc()); 811838b4a53SJeremy Morse return Num.asString(DefName); 812838b4a53SJeremy Morse } 813838b4a53SJeremy Morse 814d9fa186aSJeremy Morse #ifndef NDEBUG 815838b4a53SJeremy Morse LLVM_DUMP_METHOD void MLocTracker::dump() { 816838b4a53SJeremy Morse for (auto Location : locations()) { 817838b4a53SJeremy Morse std::string MLocName = LocIdxToName(Location.Value.getLoc()); 818838b4a53SJeremy Morse std::string DefName = Location.Value.asString(MLocName); 819838b4a53SJeremy Morse dbgs() << LocIdxToName(Location.Idx) << " --> " << DefName << "\n"; 820838b4a53SJeremy Morse } 821838b4a53SJeremy Morse } 822838b4a53SJeremy Morse 823838b4a53SJeremy Morse LLVM_DUMP_METHOD void MLocTracker::dump_mloc_map() { 824838b4a53SJeremy Morse for (auto Location : locations()) { 825838b4a53SJeremy Morse std::string foo = LocIdxToName(Location.Idx); 826838b4a53SJeremy Morse dbgs() << "Idx " << Location.Idx.asU64() << " " << foo << "\n"; 827838b4a53SJeremy Morse } 828838b4a53SJeremy Morse } 829d9fa186aSJeremy Morse #endif 830838b4a53SJeremy Morse 831838b4a53SJeremy Morse MachineInstrBuilder MLocTracker::emitLoc(Optional<LocIdx> MLoc, 832838b4a53SJeremy Morse const DebugVariable &Var, 833838b4a53SJeremy Morse const DbgValueProperties &Properties) { 834838b4a53SJeremy Morse DebugLoc DL = DILocation::get(Var.getVariable()->getContext(), 0, 0, 835838b4a53SJeremy Morse Var.getVariable()->getScope(), 836838b4a53SJeremy Morse const_cast<DILocation *>(Var.getInlinedAt())); 837838b4a53SJeremy Morse auto MIB = BuildMI(MF, DL, TII.get(TargetOpcode::DBG_VALUE)); 838838b4a53SJeremy Morse 839838b4a53SJeremy Morse const DIExpression *Expr = Properties.DIExpr; 840838b4a53SJeremy Morse if (!MLoc) { 841838b4a53SJeremy Morse // No location -> DBG_VALUE $noreg 842838b4a53SJeremy Morse MIB.addReg(0); 843838b4a53SJeremy Morse MIB.addReg(0); 844838b4a53SJeremy Morse } else if (LocIdxToLocID[*MLoc] >= NumRegs) { 845838b4a53SJeremy Morse unsigned LocID = LocIdxToLocID[*MLoc]; 846e7084ceaSJeremy Morse SpillLocationNo SpillID = locIDToSpill(LocID); 847e7084ceaSJeremy Morse StackSlotPos StackIdx = locIDToSpillIdx(LocID); 848e7084ceaSJeremy Morse unsigned short Offset = StackIdx.second; 849838b4a53SJeremy Morse 850e7084ceaSJeremy Morse // TODO: support variables that are located in spill slots, with non-zero 851e7084ceaSJeremy Morse // offsets from the start of the spill slot. It would require some more 852e7084ceaSJeremy Morse // complex DIExpression calculations. This doesn't seem to be produced by 853e7084ceaSJeremy Morse // LLVM right now, so don't try and support it. 854e7084ceaSJeremy Morse // Accept no-subregister slots and subregisters where the offset is zero. 855e7084ceaSJeremy Morse // The consumer should already have type information to work out how large 856e7084ceaSJeremy Morse // the variable is. 857e7084ceaSJeremy Morse if (Offset == 0) { 858e7084ceaSJeremy Morse const SpillLoc &Spill = SpillLocs[SpillID.id()]; 859e7084ceaSJeremy Morse Expr = TRI.prependOffsetExpression(Expr, DIExpression::ApplyOffset, 860838b4a53SJeremy Morse Spill.SpillOffset); 861838b4a53SJeremy Morse unsigned Base = Spill.SpillBase; 862838b4a53SJeremy Morse MIB.addReg(Base); 863838b4a53SJeremy Morse MIB.addImm(0); 864536b9eb3SJeremy Morse 865536b9eb3SJeremy Morse // Being on the stack makes this location indirect; if it was _already_ 866536b9eb3SJeremy Morse // indirect though, we need to add extra indirection. See this test for 867536b9eb3SJeremy Morse // a scenario where this happens: 868536b9eb3SJeremy Morse // llvm/test/DebugInfo/X86/spill-nontrivial-param.ll 869536b9eb3SJeremy Morse if (Properties.Indirect) { 870536b9eb3SJeremy Morse std::vector<uint64_t> Elts = {dwarf::DW_OP_deref}; 871536b9eb3SJeremy Morse Expr = DIExpression::append(Expr, Elts); 872536b9eb3SJeremy Morse } 873838b4a53SJeremy Morse } else { 874e7084ceaSJeremy Morse // This is a stack location with a weird subregister offset: emit an undef 875e7084ceaSJeremy Morse // DBG_VALUE instead. 876e7084ceaSJeremy Morse MIB.addReg(0); 877e7084ceaSJeremy Morse MIB.addReg(0); 878e7084ceaSJeremy Morse } 879e7084ceaSJeremy Morse } else { 880e7084ceaSJeremy Morse // Non-empty, non-stack slot, must be a plain register. 881838b4a53SJeremy Morse unsigned LocID = LocIdxToLocID[*MLoc]; 882838b4a53SJeremy Morse MIB.addReg(LocID); 883838b4a53SJeremy Morse if (Properties.Indirect) 884838b4a53SJeremy Morse MIB.addImm(0); 885838b4a53SJeremy Morse else 886838b4a53SJeremy Morse MIB.addReg(0); 887838b4a53SJeremy Morse } 888838b4a53SJeremy Morse 889838b4a53SJeremy Morse MIB.addMetadata(Var.getVariable()); 890838b4a53SJeremy Morse MIB.addMetadata(Expr); 891838b4a53SJeremy Morse return MIB; 892838b4a53SJeremy Morse } 893838b4a53SJeremy Morse 894ae6f7882SJeremy Morse /// Default construct and initialize the pass. 8953a8c5148SKazu Hirata InstrRefBasedLDV::InstrRefBasedLDV() = default; 896ae6f7882SJeremy Morse 897838b4a53SJeremy Morse bool InstrRefBasedLDV::isCalleeSaved(LocIdx L) const { 898838b4a53SJeremy Morse unsigned Reg = MTracker->LocIdxToLocID[L]; 899838b4a53SJeremy Morse for (MCRegAliasIterator RAI(Reg, TRI, true); RAI.isValid(); ++RAI) 900838b4a53SJeremy Morse if (CalleeSavedRegs.test(*RAI)) 901838b4a53SJeremy Morse return true; 902838b4a53SJeremy Morse return false; 903838b4a53SJeremy Morse } 904838b4a53SJeremy Morse 905ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 906ae6f7882SJeremy Morse // Debug Range Extension Implementation 907ae6f7882SJeremy Morse //===----------------------------------------------------------------------===// 908ae6f7882SJeremy Morse 909ae6f7882SJeremy Morse #ifndef NDEBUG 910ae6f7882SJeremy Morse // Something to restore in the future. 911ae6f7882SJeremy Morse // void InstrRefBasedLDV::printVarLocInMBB(..) 912ae6f7882SJeremy Morse #endif 913ae6f7882SJeremy Morse 91414aaaa12SJeremy Morse Optional<SpillLocationNo> 915ae6f7882SJeremy Morse InstrRefBasedLDV::extractSpillBaseRegAndOffset(const MachineInstr &MI) { 916ae6f7882SJeremy Morse assert(MI.hasOneMemOperand() && 917ae6f7882SJeremy Morse "Spill instruction does not have exactly one memory operand?"); 918ae6f7882SJeremy Morse auto MMOI = MI.memoperands_begin(); 919ae6f7882SJeremy Morse const PseudoSourceValue *PVal = (*MMOI)->getPseudoValue(); 920ae6f7882SJeremy Morse assert(PVal->kind() == PseudoSourceValue::FixedStack && 921ae6f7882SJeremy Morse "Inconsistent memory operand in spill instruction"); 922ae6f7882SJeremy Morse int FI = cast<FixedStackPseudoSourceValue>(PVal)->getFrameIndex(); 923ae6f7882SJeremy Morse const MachineBasicBlock *MBB = MI.getParent(); 924ae6f7882SJeremy Morse Register Reg; 925d57bba7cSSander de Smalen StackOffset Offset = TFI->getFrameIndexReference(*MBB->getParent(), FI, Reg); 926e7084ceaSJeremy Morse return MTracker->getOrTrackSpillLoc({Reg, Offset}); 927ae6f7882SJeremy Morse } 928ae6f7882SJeremy Morse 92914aaaa12SJeremy Morse Optional<LocIdx> 93014aaaa12SJeremy Morse InstrRefBasedLDV::findLocationForMemOperand(const MachineInstr &MI) { 93114aaaa12SJeremy Morse Optional<SpillLocationNo> SpillLoc = extractSpillBaseRegAndOffset(MI); 93214aaaa12SJeremy Morse if (!SpillLoc) 93314aaaa12SJeremy Morse return None; 934ee3eee71SJeremy Morse 935ee3eee71SJeremy Morse // Where in the stack slot is this value defined -- i.e., what size of value 936ee3eee71SJeremy Morse // is this? An important question, because it could be loaded into a register 937ee3eee71SJeremy Morse // from the stack at some point. Happily the memory operand will tell us 938ee3eee71SJeremy Morse // the size written to the stack. 939ee3eee71SJeremy Morse auto *MemOperand = *MI.memoperands_begin(); 940ee3eee71SJeremy Morse unsigned SizeInBits = MemOperand->getSizeInBits(); 941ee3eee71SJeremy Morse 942ee3eee71SJeremy Morse // Find that position in the stack indexes we're tracking. 943ee3eee71SJeremy Morse auto IdxIt = MTracker->StackSlotIdxes.find({SizeInBits, 0}); 944ee3eee71SJeremy Morse if (IdxIt == MTracker->StackSlotIdxes.end()) 945ee3eee71SJeremy Morse // That index is not tracked. This is suprising, and unlikely to ever 946ee3eee71SJeremy Morse // occur, but the safe action is to indicate the variable is optimised out. 947ee3eee71SJeremy Morse return None; 948ee3eee71SJeremy Morse 94914aaaa12SJeremy Morse unsigned SpillID = MTracker->getSpillIDWithIdx(*SpillLoc, IdxIt->second); 950ee3eee71SJeremy Morse return MTracker->getSpillMLoc(SpillID); 951ee3eee71SJeremy Morse } 952ee3eee71SJeremy Morse 953ae6f7882SJeremy Morse /// End all previous ranges related to @MI and start a new range from @MI 954ae6f7882SJeremy Morse /// if it is a DBG_VALUE instr. 955ae6f7882SJeremy Morse bool InstrRefBasedLDV::transferDebugValue(const MachineInstr &MI) { 956ae6f7882SJeremy Morse if (!MI.isDebugValue()) 957ae6f7882SJeremy Morse return false; 958ae6f7882SJeremy Morse 959ae6f7882SJeremy Morse const DILocalVariable *Var = MI.getDebugVariable(); 960ae6f7882SJeremy Morse const DIExpression *Expr = MI.getDebugExpression(); 961ae6f7882SJeremy Morse const DILocation *DebugLoc = MI.getDebugLoc(); 962ae6f7882SJeremy Morse const DILocation *InlinedAt = DebugLoc->getInlinedAt(); 963ae6f7882SJeremy Morse assert(Var->isValidLocationForIntrinsic(DebugLoc) && 964ae6f7882SJeremy Morse "Expected inlined-at fields to agree"); 965ae6f7882SJeremy Morse 966ae6f7882SJeremy Morse DebugVariable V(Var, Expr, InlinedAt); 96768f47157SJeremy Morse DbgValueProperties Properties(MI); 968ae6f7882SJeremy Morse 969ae6f7882SJeremy Morse // If there are no instructions in this lexical scope, do no location tracking 970ae6f7882SJeremy Morse // at all, this variable shouldn't get a legitimate location range. 971ae6f7882SJeremy Morse auto *Scope = LS.findLexicalScope(MI.getDebugLoc().get()); 972ae6f7882SJeremy Morse if (Scope == nullptr) 973ae6f7882SJeremy Morse return true; // handled it; by doing nothing 974ae6f7882SJeremy Morse 975ce8254d0SJeremy Morse // For now, ignore DBG_VALUE_LISTs when extending ranges. Allow it to 976ce8254d0SJeremy Morse // contribute to locations in this block, but don't propagate further. 977ce8254d0SJeremy Morse // Interpret it like a DBG_VALUE $noreg. 978ce8254d0SJeremy Morse if (MI.isDebugValueList()) { 979ce8254d0SJeremy Morse if (VTracker) 980ce8254d0SJeremy Morse VTracker->defVar(MI, Properties, None); 981ce8254d0SJeremy Morse if (TTracker) 982ce8254d0SJeremy Morse TTracker->redefVar(MI, Properties, None); 983ce8254d0SJeremy Morse return true; 984ce8254d0SJeremy Morse } 985ce8254d0SJeremy Morse 986ae6f7882SJeremy Morse const MachineOperand &MO = MI.getOperand(0); 987ae6f7882SJeremy Morse 988ae6f7882SJeremy Morse // MLocTracker needs to know that this register is read, even if it's only 989ae6f7882SJeremy Morse // read by a debug inst. 990ae6f7882SJeremy Morse if (MO.isReg() && MO.getReg() != 0) 991ae6f7882SJeremy Morse (void)MTracker->readReg(MO.getReg()); 992ae6f7882SJeremy Morse 993ae6f7882SJeremy Morse // If we're preparing for the second analysis (variables), the machine value 994ae6f7882SJeremy Morse // locations are already solved, and we report this DBG_VALUE and the value 995ae6f7882SJeremy Morse // it refers to to VLocTracker. 996ae6f7882SJeremy Morse if (VTracker) { 997ae6f7882SJeremy Morse if (MO.isReg()) { 998ae6f7882SJeremy Morse // Feed defVar the new variable location, or if this is a 999ae6f7882SJeremy Morse // DBG_VALUE $noreg, feed defVar None. 1000ae6f7882SJeremy Morse if (MO.getReg()) 100168f47157SJeremy Morse VTracker->defVar(MI, Properties, MTracker->readReg(MO.getReg())); 1002ae6f7882SJeremy Morse else 100368f47157SJeremy Morse VTracker->defVar(MI, Properties, None); 1004ae6f7882SJeremy Morse } else if (MI.getOperand(0).isImm() || MI.getOperand(0).isFPImm() || 1005ae6f7882SJeremy Morse MI.getOperand(0).isCImm()) { 1006ae6f7882SJeremy Morse VTracker->defVar(MI, MI.getOperand(0)); 1007ae6f7882SJeremy Morse } 1008ae6f7882SJeremy Morse } 1009ae6f7882SJeremy Morse 1010ae6f7882SJeremy Morse // If performing final tracking of transfers, report this variable definition 1011ae6f7882SJeremy Morse // to the TransferTracker too. 1012ae6f7882SJeremy Morse if (TTracker) 1013ae6f7882SJeremy Morse TTracker->redefVar(MI); 1014ae6f7882SJeremy Morse return true; 1015ae6f7882SJeremy Morse } 1016ae6f7882SJeremy Morse 1017010108bbSJeremy Morse bool InstrRefBasedLDV::transferDebugInstrRef(MachineInstr &MI, 1018ab49dce0SJeremy Morse const ValueTable *MLiveOuts, 1019ab49dce0SJeremy Morse const ValueTable *MLiveIns) { 102068f47157SJeremy Morse if (!MI.isDebugRef()) 102168f47157SJeremy Morse return false; 102268f47157SJeremy Morse 102368f47157SJeremy Morse // Only handle this instruction when we are building the variable value 102468f47157SJeremy Morse // transfer function. 1025a80181a8SJeremy Morse if (!VTracker && !TTracker) 102668f47157SJeremy Morse return false; 102768f47157SJeremy Morse 102868f47157SJeremy Morse unsigned InstNo = MI.getOperand(0).getImm(); 102968f47157SJeremy Morse unsigned OpNo = MI.getOperand(1).getImm(); 103068f47157SJeremy Morse 103168f47157SJeremy Morse const DILocalVariable *Var = MI.getDebugVariable(); 103268f47157SJeremy Morse const DIExpression *Expr = MI.getDebugExpression(); 103368f47157SJeremy Morse const DILocation *DebugLoc = MI.getDebugLoc(); 103468f47157SJeremy Morse const DILocation *InlinedAt = DebugLoc->getInlinedAt(); 103568f47157SJeremy Morse assert(Var->isValidLocationForIntrinsic(DebugLoc) && 103668f47157SJeremy Morse "Expected inlined-at fields to agree"); 103768f47157SJeremy Morse 103868f47157SJeremy Morse DebugVariable V(Var, Expr, InlinedAt); 103968f47157SJeremy Morse 104068f47157SJeremy Morse auto *Scope = LS.findLexicalScope(MI.getDebugLoc().get()); 104168f47157SJeremy Morse if (Scope == nullptr) 104268f47157SJeremy Morse return true; // Handled by doing nothing. This variable is never in scope. 104368f47157SJeremy Morse 104468f47157SJeremy Morse const MachineFunction &MF = *MI.getParent()->getParent(); 104568f47157SJeremy Morse 104668f47157SJeremy Morse // Various optimizations may have happened to the value during codegen, 104768f47157SJeremy Morse // recorded in the value substitution table. Apply any substitutions to 1048f551fb96SJeremy Morse // the instruction / operand number in this DBG_INSTR_REF, and collect 1049f551fb96SJeremy Morse // any subregister extractions performed during optimization. 1050f551fb96SJeremy Morse 1051f551fb96SJeremy Morse // Create dummy substitution with Src set, for lookup. 1052f551fb96SJeremy Morse auto SoughtSub = 1053f551fb96SJeremy Morse MachineFunction::DebugSubstitution({InstNo, OpNo}, {0, 0}, 0); 1054f551fb96SJeremy Morse 1055e9641c91SJeremy Morse SmallVector<unsigned, 4> SeenSubregs; 1056f551fb96SJeremy Morse auto LowerBoundIt = llvm::lower_bound(MF.DebugValueSubstitutions, SoughtSub); 1057f551fb96SJeremy Morse while (LowerBoundIt != MF.DebugValueSubstitutions.end() && 1058f551fb96SJeremy Morse LowerBoundIt->Src == SoughtSub.Src) { 1059f551fb96SJeremy Morse std::tie(InstNo, OpNo) = LowerBoundIt->Dest; 1060f551fb96SJeremy Morse SoughtSub.Src = LowerBoundIt->Dest; 1061f551fb96SJeremy Morse if (unsigned Subreg = LowerBoundIt->Subreg) 1062e9641c91SJeremy Morse SeenSubregs.push_back(Subreg); 1063f551fb96SJeremy Morse LowerBoundIt = llvm::lower_bound(MF.DebugValueSubstitutions, SoughtSub); 106468f47157SJeremy Morse } 106568f47157SJeremy Morse 106668f47157SJeremy Morse // Default machine value number is <None> -- if no instruction defines 106768f47157SJeremy Morse // the corresponding value, it must have been optimized out. 106868f47157SJeremy Morse Optional<ValueIDNum> NewID = None; 106968f47157SJeremy Morse 107068f47157SJeremy Morse // Try to lookup the instruction number, and find the machine value number 1071010108bbSJeremy Morse // that it defines. It could be an instruction, or a PHI. 107268f47157SJeremy Morse auto InstrIt = DebugInstrNumToInstr.find(InstNo); 1073010108bbSJeremy Morse auto PHIIt = std::lower_bound(DebugPHINumToValue.begin(), 1074010108bbSJeremy Morse DebugPHINumToValue.end(), InstNo); 107568f47157SJeremy Morse if (InstrIt != DebugInstrNumToInstr.end()) { 107668f47157SJeremy Morse const MachineInstr &TargetInstr = *InstrIt->second.first; 107768f47157SJeremy Morse uint64_t BlockNo = TargetInstr.getParent()->getNumber(); 107868f47157SJeremy Morse 1079ee3eee71SJeremy Morse // Pick out the designated operand. It might be a memory reference, if 1080ee3eee71SJeremy Morse // a register def was folded into a stack store. 1081ee3eee71SJeremy Morse if (OpNo == MachineFunction::DebugOperandMemNumber && 1082ee3eee71SJeremy Morse TargetInstr.hasOneMemOperand()) { 1083ee3eee71SJeremy Morse Optional<LocIdx> L = findLocationForMemOperand(TargetInstr); 1084ee3eee71SJeremy Morse if (L) 1085ee3eee71SJeremy Morse NewID = ValueIDNum(BlockNo, InstrIt->second.second, *L); 1086ee3eee71SJeremy Morse } else if (OpNo != MachineFunction::DebugOperandMemNumber) { 1087be5734ddSJeremy Morse // Permit the debug-info to be completely wrong: identifying a nonexistant 1088be5734ddSJeremy Morse // operand, or one that is not a register definition, means something 1089be5734ddSJeremy Morse // unexpected happened during optimisation. Broken debug-info, however, 1090be5734ddSJeremy Morse // shouldn't crash the compiler -- instead leave the variable value as 1091be5734ddSJeremy Morse // None, which will make it appear "optimised out". 1092be5734ddSJeremy Morse if (OpNo < TargetInstr.getNumOperands()) { 109368f47157SJeremy Morse const MachineOperand &MO = TargetInstr.getOperand(OpNo); 109468f47157SJeremy Morse 1095be5734ddSJeremy Morse if (MO.isReg() && MO.isDef() && MO.getReg()) { 1096e7084ceaSJeremy Morse unsigned LocID = MTracker->getLocID(MO.getReg()); 109768f47157SJeremy Morse LocIdx L = MTracker->LocIDToLocIdx[LocID]; 109868f47157SJeremy Morse NewID = ValueIDNum(BlockNo, InstrIt->second.second, L); 1099ee3eee71SJeremy Morse } 1100be5734ddSJeremy Morse } 1101be5734ddSJeremy Morse 1102be5734ddSJeremy Morse if (!NewID) { 1103be5734ddSJeremy Morse LLVM_DEBUG( 1104be5734ddSJeremy Morse { dbgs() << "Seen instruction reference to illegal operand\n"; }); 1105be5734ddSJeremy Morse } 1106be5734ddSJeremy Morse } 1107ee3eee71SJeremy Morse // else: NewID is left as None. 1108010108bbSJeremy Morse } else if (PHIIt != DebugPHINumToValue.end() && PHIIt->InstrNum == InstNo) { 1109010108bbSJeremy Morse // It's actually a PHI value. Which value it is might not be obvious, use 1110010108bbSJeremy Morse // the resolver helper to find out. 1111010108bbSJeremy Morse NewID = resolveDbgPHIs(*MI.getParent()->getParent(), MLiveOuts, MLiveIns, 1112010108bbSJeremy Morse MI, InstNo); 111368f47157SJeremy Morse } 111468f47157SJeremy Morse 1115e9641c91SJeremy Morse // Apply any subregister extractions, in reverse. We might have seen code 1116e9641c91SJeremy Morse // like this: 1117e9641c91SJeremy Morse // CALL64 @foo, implicit-def $rax 1118e9641c91SJeremy Morse // %0:gr64 = COPY $rax 1119e9641c91SJeremy Morse // %1:gr32 = COPY %0.sub_32bit 1120e9641c91SJeremy Morse // %2:gr16 = COPY %1.sub_16bit 1121e9641c91SJeremy Morse // %3:gr8 = COPY %2.sub_8bit 1122e9641c91SJeremy Morse // In which case each copy would have been recorded as a substitution with 1123e9641c91SJeremy Morse // a subregister qualifier. Apply those qualifiers now. 1124e9641c91SJeremy Morse if (NewID && !SeenSubregs.empty()) { 1125e9641c91SJeremy Morse unsigned Offset = 0; 1126e9641c91SJeremy Morse unsigned Size = 0; 1127e9641c91SJeremy Morse 1128e9641c91SJeremy Morse // Look at each subregister that we passed through, and progressively 1129e9641c91SJeremy Morse // narrow in, accumulating any offsets that occur. Substitutions should 1130e9641c91SJeremy Morse // only ever be the same or narrower width than what they read from; 1131e9641c91SJeremy Morse // iterate in reverse order so that we go from wide to small. 1132e9641c91SJeremy Morse for (unsigned Subreg : reverse(SeenSubregs)) { 1133e9641c91SJeremy Morse unsigned ThisSize = TRI->getSubRegIdxSize(Subreg); 1134e9641c91SJeremy Morse unsigned ThisOffset = TRI->getSubRegIdxOffset(Subreg); 1135e9641c91SJeremy Morse Offset += ThisOffset; 1136e9641c91SJeremy Morse Size = (Size == 0) ? ThisSize : std::min(Size, ThisSize); 1137e9641c91SJeremy Morse } 1138e9641c91SJeremy Morse 1139e9641c91SJeremy Morse // If that worked, look for an appropriate subregister with the register 1140e9641c91SJeremy Morse // where the define happens. Don't look at values that were defined during 1141e9641c91SJeremy Morse // a stack write: we can't currently express register locations within 1142e9641c91SJeremy Morse // spills. 1143e9641c91SJeremy Morse LocIdx L = NewID->getLoc(); 1144e9641c91SJeremy Morse if (NewID && !MTracker->isSpill(L)) { 1145e9641c91SJeremy Morse // Find the register class for the register where this def happened. 1146e9641c91SJeremy Morse // FIXME: no index for this? 1147e9641c91SJeremy Morse Register Reg = MTracker->LocIdxToLocID[L]; 1148e9641c91SJeremy Morse const TargetRegisterClass *TRC = nullptr; 1149e9641c91SJeremy Morse for (auto *TRCI : TRI->regclasses()) 1150e9641c91SJeremy Morse if (TRCI->contains(Reg)) 1151e9641c91SJeremy Morse TRC = TRCI; 1152e9641c91SJeremy Morse assert(TRC && "Couldn't find target register class?"); 1153e9641c91SJeremy Morse 1154e9641c91SJeremy Morse // If the register we have isn't the right size or in the right place, 1155e9641c91SJeremy Morse // Try to find a subregister inside it. 1156e9641c91SJeremy Morse unsigned MainRegSize = TRI->getRegSizeInBits(*TRC); 1157e9641c91SJeremy Morse if (Size != MainRegSize || Offset) { 1158e9641c91SJeremy Morse // Enumerate all subregisters, searching. 1159e9641c91SJeremy Morse Register NewReg = 0; 1160e9641c91SJeremy Morse for (MCSubRegIterator SRI(Reg, TRI, false); SRI.isValid(); ++SRI) { 1161e9641c91SJeremy Morse unsigned Subreg = TRI->getSubRegIndex(Reg, *SRI); 1162e9641c91SJeremy Morse unsigned SubregSize = TRI->getSubRegIdxSize(Subreg); 1163e9641c91SJeremy Morse unsigned SubregOffset = TRI->getSubRegIdxOffset(Subreg); 1164e9641c91SJeremy Morse if (SubregSize == Size && SubregOffset == Offset) { 1165e9641c91SJeremy Morse NewReg = *SRI; 1166e9641c91SJeremy Morse break; 1167e9641c91SJeremy Morse } 1168e9641c91SJeremy Morse } 1169e9641c91SJeremy Morse 1170e9641c91SJeremy Morse // If we didn't find anything: there's no way to express our value. 1171e9641c91SJeremy Morse if (!NewReg) { 1172e9641c91SJeremy Morse NewID = None; 1173e9641c91SJeremy Morse } else { 1174e9641c91SJeremy Morse // Re-state the value as being defined within the subregister 1175e9641c91SJeremy Morse // that we found. 1176e9641c91SJeremy Morse LocIdx NewLoc = MTracker->lookupOrTrackRegister(NewReg); 1177e9641c91SJeremy Morse NewID = ValueIDNum(NewID->getBlock(), NewID->getInst(), NewLoc); 1178e9641c91SJeremy Morse } 1179e9641c91SJeremy Morse } 1180e9641c91SJeremy Morse } else { 1181e9641c91SJeremy Morse // If we can't handle subregisters, unset the new value. 1182e9641c91SJeremy Morse NewID = None; 1183e9641c91SJeremy Morse } 1184e9641c91SJeremy Morse } 1185e9641c91SJeremy Morse 118668f47157SJeremy Morse // We, we have a value number or None. Tell the variable value tracker about 118768f47157SJeremy Morse // it. The rest of this LiveDebugValues implementation acts exactly the same 118868f47157SJeremy Morse // for DBG_INSTR_REFs as DBG_VALUEs (just, the former can refer to values that 118968f47157SJeremy Morse // aren't immediately available). 119068f47157SJeremy Morse DbgValueProperties Properties(Expr, false); 1191a80181a8SJeremy Morse if (VTracker) 119268f47157SJeremy Morse VTracker->defVar(MI, Properties, NewID); 119368f47157SJeremy Morse 119468f47157SJeremy Morse // If we're on the final pass through the function, decompose this INSTR_REF 119568f47157SJeremy Morse // into a plain DBG_VALUE. 119668f47157SJeremy Morse if (!TTracker) 119768f47157SJeremy Morse return true; 119868f47157SJeremy Morse 119968f47157SJeremy Morse // Pick a location for the machine value number, if such a location exists. 120068f47157SJeremy Morse // (This information could be stored in TransferTracker to make it faster). 120168f47157SJeremy Morse Optional<LocIdx> FoundLoc = None; 120268f47157SJeremy Morse for (auto Location : MTracker->locations()) { 120368f47157SJeremy Morse LocIdx CurL = Location.Idx; 1204d9eebe3cSJeremy Morse ValueIDNum ID = MTracker->readMLoc(CurL); 120568f47157SJeremy Morse if (NewID && ID == NewID) { 120668f47157SJeremy Morse // If this is the first location with that value, pick it. Otherwise, 120768f47157SJeremy Morse // consider whether it's a "longer term" location. 120868f47157SJeremy Morse if (!FoundLoc) { 120968f47157SJeremy Morse FoundLoc = CurL; 121068f47157SJeremy Morse continue; 121168f47157SJeremy Morse } 121268f47157SJeremy Morse 121368f47157SJeremy Morse if (MTracker->isSpill(CurL)) 121468f47157SJeremy Morse FoundLoc = CurL; // Spills are a longer term location. 121568f47157SJeremy Morse else if (!MTracker->isSpill(*FoundLoc) && 121668f47157SJeremy Morse !MTracker->isSpill(CurL) && 121768f47157SJeremy Morse !isCalleeSaved(*FoundLoc) && 121868f47157SJeremy Morse isCalleeSaved(CurL)) 121968f47157SJeremy Morse FoundLoc = CurL; // Callee saved regs are longer term than normal. 122068f47157SJeremy Morse } 122168f47157SJeremy Morse } 122268f47157SJeremy Morse 122368f47157SJeremy Morse // Tell transfer tracker that the variable value has changed. 122468f47157SJeremy Morse TTracker->redefVar(MI, Properties, FoundLoc); 122568f47157SJeremy Morse 1226b1b2c6abSJeremy Morse // If there was a value with no location; but the value is defined in a 1227b1b2c6abSJeremy Morse // later instruction in this block, this is a block-local use-before-def. 1228b1b2c6abSJeremy Morse if (!FoundLoc && NewID && NewID->getBlock() == CurBB && 1229b1b2c6abSJeremy Morse NewID->getInst() > CurInst) 1230b1b2c6abSJeremy Morse TTracker->addUseBeforeDef(V, {MI.getDebugExpression(), false}, *NewID); 1231b1b2c6abSJeremy Morse 123268f47157SJeremy Morse // Produce a DBG_VALUE representing what this DBG_INSTR_REF meant. 123368f47157SJeremy Morse // This DBG_VALUE is potentially a $noreg / undefined location, if 123468f47157SJeremy Morse // FoundLoc is None. 123568f47157SJeremy Morse // (XXX -- could morph the DBG_INSTR_REF in the future). 123668f47157SJeremy Morse MachineInstr *DbgMI = MTracker->emitLoc(FoundLoc, V, Properties); 123768f47157SJeremy Morse TTracker->PendingDbgValues.push_back(DbgMI); 123868f47157SJeremy Morse TTracker->flushDbgValues(MI.getIterator(), nullptr); 1239010108bbSJeremy Morse return true; 1240010108bbSJeremy Morse } 1241010108bbSJeremy Morse 1242010108bbSJeremy Morse bool InstrRefBasedLDV::transferDebugPHI(MachineInstr &MI) { 1243010108bbSJeremy Morse if (!MI.isDebugPHI()) 1244010108bbSJeremy Morse return false; 1245010108bbSJeremy Morse 1246010108bbSJeremy Morse // Analyse these only when solving the machine value location problem. 1247010108bbSJeremy Morse if (VTracker || TTracker) 1248010108bbSJeremy Morse return true; 1249010108bbSJeremy Morse 1250010108bbSJeremy Morse // First operand is the value location, either a stack slot or register. 1251010108bbSJeremy Morse // Second is the debug instruction number of the original PHI. 1252010108bbSJeremy Morse const MachineOperand &MO = MI.getOperand(0); 1253010108bbSJeremy Morse unsigned InstrNum = MI.getOperand(1).getImm(); 1254010108bbSJeremy Morse 1255be5734ddSJeremy Morse auto EmitBadPHI = [this, &MI, InstrNum](void) -> bool { 1256be5734ddSJeremy Morse // Helper lambda to do any accounting when we fail to find a location for 1257be5734ddSJeremy Morse // a DBG_PHI. This can happen if DBG_PHIs are malformed, or refer to a 1258be5734ddSJeremy Morse // dead stack slot, for example. 1259be5734ddSJeremy Morse // Record a DebugPHIRecord with an empty value + location. 1260be5734ddSJeremy Morse DebugPHINumToValue.push_back({InstrNum, MI.getParent(), None, None}); 1261be5734ddSJeremy Morse return true; 1262be5734ddSJeremy Morse }; 1263be5734ddSJeremy Morse 1264be5734ddSJeremy Morse if (MO.isReg() && MO.getReg()) { 1265010108bbSJeremy Morse // The value is whatever's currently in the register. Read and record it, 1266010108bbSJeremy Morse // to be analysed later. 1267010108bbSJeremy Morse Register Reg = MO.getReg(); 1268010108bbSJeremy Morse ValueIDNum Num = MTracker->readReg(Reg); 1269010108bbSJeremy Morse auto PHIRec = DebugPHIRecord( 1270010108bbSJeremy Morse {InstrNum, MI.getParent(), Num, MTracker->lookupOrTrackRegister(Reg)}); 1271010108bbSJeremy Morse DebugPHINumToValue.push_back(PHIRec); 1272e265644bSJeremy Morse 1273e7084ceaSJeremy Morse // Ensure this register is tracked. 1274e265644bSJeremy Morse for (MCRegAliasIterator RAI(MO.getReg(), TRI, true); RAI.isValid(); ++RAI) 1275e265644bSJeremy Morse MTracker->lookupOrTrackRegister(*RAI); 1276be5734ddSJeremy Morse } else if (MO.isFI()) { 1277010108bbSJeremy Morse // The value is whatever's in this stack slot. 1278010108bbSJeremy Morse unsigned FI = MO.getIndex(); 1279010108bbSJeremy Morse 1280010108bbSJeremy Morse // If the stack slot is dead, then this was optimized away. 1281010108bbSJeremy Morse // FIXME: stack slot colouring should account for slots that get merged. 1282010108bbSJeremy Morse if (MFI->isDeadObjectIndex(FI)) 1283be5734ddSJeremy Morse return EmitBadPHI(); 1284010108bbSJeremy Morse 1285e7084ceaSJeremy Morse // Identify this spill slot, ensure it's tracked. 1286010108bbSJeremy Morse Register Base; 1287010108bbSJeremy Morse StackOffset Offs = TFI->getFrameIndexReference(*MI.getMF(), FI, Base); 1288010108bbSJeremy Morse SpillLoc SL = {Base, Offs}; 128914aaaa12SJeremy Morse Optional<SpillLocationNo> SpillNo = MTracker->getOrTrackSpillLoc(SL); 129014aaaa12SJeremy Morse 129114aaaa12SJeremy Morse // We might be able to find a value, but have chosen not to, to avoid 129214aaaa12SJeremy Morse // tracking too much stack information. 129314aaaa12SJeremy Morse if (!SpillNo) 1294be5734ddSJeremy Morse return EmitBadPHI(); 1295010108bbSJeremy Morse 1296e7084ceaSJeremy Morse // Problem: what value should we extract from the stack? LLVM does not 1297e7084ceaSJeremy Morse // record what size the last store to the slot was, and it would become 1298e7084ceaSJeremy Morse // sketchy after stack slot colouring anyway. Take a look at what values 1299e7084ceaSJeremy Morse // are stored on the stack, and pick the largest one that wasn't def'd 1300e7084ceaSJeremy Morse // by a spill (i.e., the value most likely to have been def'd in a register 1301e7084ceaSJeremy Morse // and then spilt. 1302e7084ceaSJeremy Morse std::array<unsigned, 4> CandidateSizes = {64, 32, 16, 8}; 1303e7084ceaSJeremy Morse Optional<ValueIDNum> Result = None; 1304e7084ceaSJeremy Morse Optional<LocIdx> SpillLoc = None; 13053aed2822SKazu Hirata for (unsigned CS : CandidateSizes) { 130614aaaa12SJeremy Morse unsigned SpillID = MTracker->getLocID(*SpillNo, {CS, 0}); 1307e7084ceaSJeremy Morse SpillLoc = MTracker->getSpillMLoc(SpillID); 1308e7084ceaSJeremy Morse ValueIDNum Val = MTracker->readMLoc(*SpillLoc); 1309e7084ceaSJeremy Morse // If this value was defined in it's own position, then it was probably 1310e7084ceaSJeremy Morse // an aliasing index of a small value that was spilt. 1311e7084ceaSJeremy Morse if (Val.getLoc() != SpillLoc->asU64()) { 1312e7084ceaSJeremy Morse Result = Val; 1313e7084ceaSJeremy Morse break; 1314e7084ceaSJeremy Morse } 1315e7084ceaSJeremy Morse } 1316e7084ceaSJeremy Morse 1317e7084ceaSJeremy Morse // If we didn't find anything, we're probably looking at a PHI, or a memory 1318e7084ceaSJeremy Morse // store folded into an instruction. FIXME: Take a guess that's it's 64 1319e7084ceaSJeremy Morse // bits. This isn't ideal, but tracking the size that the spill is 1320e7084ceaSJeremy Morse // "supposed" to be is more complex, and benefits a small number of 1321e7084ceaSJeremy Morse // locations. 1322e7084ceaSJeremy Morse if (!Result) { 132314aaaa12SJeremy Morse unsigned SpillID = MTracker->getLocID(*SpillNo, {64, 0}); 1324e7084ceaSJeremy Morse SpillLoc = MTracker->getSpillMLoc(SpillID); 1325e7084ceaSJeremy Morse Result = MTracker->readMLoc(*SpillLoc); 1326e7084ceaSJeremy Morse } 1327010108bbSJeremy Morse 1328010108bbSJeremy Morse // Record this DBG_PHI for later analysis. 1329be5734ddSJeremy Morse auto DbgPHI = 1330be5734ddSJeremy Morse DebugPHIRecord({InstrNum, MI.getParent(), *Result, *SpillLoc}); 1331010108bbSJeremy Morse DebugPHINumToValue.push_back(DbgPHI); 1332be5734ddSJeremy Morse } else { 1333be5734ddSJeremy Morse // Else: if the operand is neither a legal register or a stack slot, then 1334be5734ddSJeremy Morse // we're being fed illegal debug-info. Record an empty PHI, so that any 1335be5734ddSJeremy Morse // debug users trying to read this number will be put off trying to 1336be5734ddSJeremy Morse // interpret the value. 1337be5734ddSJeremy Morse LLVM_DEBUG( 1338be5734ddSJeremy Morse { dbgs() << "Seen DBG_PHI with unrecognised operand format\n"; }); 1339be5734ddSJeremy Morse return EmitBadPHI(); 1340010108bbSJeremy Morse } 134168f47157SJeremy Morse 134268f47157SJeremy Morse return true; 134368f47157SJeremy Morse } 134468f47157SJeremy Morse 1345ae6f7882SJeremy Morse void InstrRefBasedLDV::transferRegisterDef(MachineInstr &MI) { 1346ae6f7882SJeremy Morse // Meta Instructions do not affect the debug liveness of any register they 1347ae6f7882SJeremy Morse // define. 1348ae6f7882SJeremy Morse if (MI.isImplicitDef()) { 1349ae6f7882SJeremy Morse // Except when there's an implicit def, and the location it's defining has 1350ae6f7882SJeremy Morse // no value number. The whole point of an implicit def is to announce that 1351ae6f7882SJeremy Morse // the register is live, without be specific about it's value. So define 1352ae6f7882SJeremy Morse // a value if there isn't one already. 1353ae6f7882SJeremy Morse ValueIDNum Num = MTracker->readReg(MI.getOperand(0).getReg()); 1354ae6f7882SJeremy Morse // Has a legitimate value -> ignore the implicit def. 1355ae6f7882SJeremy Morse if (Num.getLoc() != 0) 1356ae6f7882SJeremy Morse return; 1357ae6f7882SJeremy Morse // Otherwise, def it here. 1358ae6f7882SJeremy Morse } else if (MI.isMetaInstruction()) 1359ae6f7882SJeremy Morse return; 1360ae6f7882SJeremy Morse 1361bfadc5dcSJeremy Morse // We always ignore SP defines on call instructions, they don't actually 1362bfadc5dcSJeremy Morse // change the value of the stack pointer... except for win32's _chkstk. This 1363bfadc5dcSJeremy Morse // is rare: filter quickly for the common case (no stack adjustments, not a 1364bfadc5dcSJeremy Morse // call, etc). If it is a call that modifies SP, recognise the SP register 1365bfadc5dcSJeremy Morse // defs. 1366bfadc5dcSJeremy Morse bool CallChangesSP = false; 1367bfadc5dcSJeremy Morse if (AdjustsStackInCalls && MI.isCall() && MI.getOperand(0).isSymbol() && 1368bfadc5dcSJeremy Morse !strcmp(MI.getOperand(0).getSymbolName(), StackProbeSymbolName.data())) 1369bfadc5dcSJeremy Morse CallChangesSP = true; 1370bfadc5dcSJeremy Morse 1371bfadc5dcSJeremy Morse // Test whether we should ignore a def of this register due to it being part 1372bfadc5dcSJeremy Morse // of the stack pointer. 1373bfadc5dcSJeremy Morse auto IgnoreSPAlias = [this, &MI, CallChangesSP](Register R) -> bool { 1374bfadc5dcSJeremy Morse if (CallChangesSP) 1375bfadc5dcSJeremy Morse return false; 1376133e25f9SJeremy Morse return MI.isCall() && MTracker->SPAliases.count(R); 1377133e25f9SJeremy Morse }; 1378133e25f9SJeremy Morse 1379ae6f7882SJeremy Morse // Find the regs killed by MI, and find regmasks of preserved regs. 1380ae6f7882SJeremy Morse // Max out the number of statically allocated elements in `DeadRegs`, as this 1381ae6f7882SJeremy Morse // prevents fallback to std::set::count() operations. 1382ae6f7882SJeremy Morse SmallSet<uint32_t, 32> DeadRegs; 1383ae6f7882SJeremy Morse SmallVector<const uint32_t *, 4> RegMasks; 1384ae6f7882SJeremy Morse SmallVector<const MachineOperand *, 4> RegMaskPtrs; 1385ae6f7882SJeremy Morse for (const MachineOperand &MO : MI.operands()) { 1386ae6f7882SJeremy Morse // Determine whether the operand is a register def. 1387ae6f7882SJeremy Morse if (MO.isReg() && MO.isDef() && MO.getReg() && 1388ae6f7882SJeremy Morse Register::isPhysicalRegister(MO.getReg()) && 1389133e25f9SJeremy Morse !IgnoreSPAlias(MO.getReg())) { 1390ae6f7882SJeremy Morse // Remove ranges of all aliased registers. 1391ae6f7882SJeremy Morse for (MCRegAliasIterator RAI(MO.getReg(), TRI, true); RAI.isValid(); ++RAI) 1392ae6f7882SJeremy Morse // FIXME: Can we break out of this loop early if no insertion occurs? 1393ae6f7882SJeremy Morse DeadRegs.insert(*RAI); 1394ae6f7882SJeremy Morse } else if (MO.isRegMask()) { 1395ae6f7882SJeremy Morse RegMasks.push_back(MO.getRegMask()); 1396ae6f7882SJeremy Morse RegMaskPtrs.push_back(&MO); 1397ae6f7882SJeremy Morse } 1398ae6f7882SJeremy Morse } 1399ae6f7882SJeremy Morse 1400ae6f7882SJeremy Morse // Tell MLocTracker about all definitions, of regmasks and otherwise. 1401ae6f7882SJeremy Morse for (uint32_t DeadReg : DeadRegs) 1402ae6f7882SJeremy Morse MTracker->defReg(DeadReg, CurBB, CurInst); 1403ae6f7882SJeremy Morse 1404ae6f7882SJeremy Morse for (auto *MO : RegMaskPtrs) 1405ae6f7882SJeremy Morse MTracker->writeRegMask(MO, CurBB, CurInst); 140649555441SJeremy Morse 1407ee3eee71SJeremy Morse // If this instruction writes to a spill slot, def that slot. 1408ee3eee71SJeremy Morse if (hasFoldedStackStore(MI)) { 140914aaaa12SJeremy Morse if (Optional<SpillLocationNo> SpillNo = extractSpillBaseRegAndOffset(MI)) { 1410ee3eee71SJeremy Morse for (unsigned int I = 0; I < MTracker->NumSlotIdxes; ++I) { 141114aaaa12SJeremy Morse unsigned SpillID = MTracker->getSpillIDWithIdx(*SpillNo, I); 1412ee3eee71SJeremy Morse LocIdx L = MTracker->getSpillMLoc(SpillID); 1413ee3eee71SJeremy Morse MTracker->setMLoc(L, ValueIDNum(CurBB, CurInst, L)); 1414ee3eee71SJeremy Morse } 1415ee3eee71SJeremy Morse } 141614aaaa12SJeremy Morse } 1417ee3eee71SJeremy Morse 141849555441SJeremy Morse if (!TTracker) 141949555441SJeremy Morse return; 142049555441SJeremy Morse 142149555441SJeremy Morse // When committing variable values to locations: tell transfer tracker that 142249555441SJeremy Morse // we've clobbered things. It may be able to recover the variable from a 142349555441SJeremy Morse // different location. 142449555441SJeremy Morse 142549555441SJeremy Morse // Inform TTracker about any direct clobbers. 142649555441SJeremy Morse for (uint32_t DeadReg : DeadRegs) { 142749555441SJeremy Morse LocIdx Loc = MTracker->lookupOrTrackRegister(DeadReg); 142849555441SJeremy Morse TTracker->clobberMloc(Loc, MI.getIterator(), false); 142949555441SJeremy Morse } 143049555441SJeremy Morse 143149555441SJeremy Morse // Look for any clobbers performed by a register mask. Only test locations 143249555441SJeremy Morse // that are actually being tracked. 14338e75536eSJeremy Morse if (!RegMaskPtrs.empty()) { 143449555441SJeremy Morse for (auto L : MTracker->locations()) { 143549555441SJeremy Morse // Stack locations can't be clobbered by regmasks. 143649555441SJeremy Morse if (MTracker->isSpill(L.Idx)) 143749555441SJeremy Morse continue; 143849555441SJeremy Morse 143949555441SJeremy Morse Register Reg = MTracker->LocIdxToLocID[L.Idx]; 1440133e25f9SJeremy Morse if (IgnoreSPAlias(Reg)) 1441133e25f9SJeremy Morse continue; 1442133e25f9SJeremy Morse 144349555441SJeremy Morse for (auto *MO : RegMaskPtrs) 144449555441SJeremy Morse if (MO->clobbersPhysReg(Reg)) 144549555441SJeremy Morse TTracker->clobberMloc(L.Idx, MI.getIterator(), false); 144649555441SJeremy Morse } 14478e75536eSJeremy Morse } 1448ee3eee71SJeremy Morse 1449ee3eee71SJeremy Morse // Tell TTracker about any folded stack store. 1450ee3eee71SJeremy Morse if (hasFoldedStackStore(MI)) { 145114aaaa12SJeremy Morse if (Optional<SpillLocationNo> SpillNo = extractSpillBaseRegAndOffset(MI)) { 1452ee3eee71SJeremy Morse for (unsigned int I = 0; I < MTracker->NumSlotIdxes; ++I) { 145314aaaa12SJeremy Morse unsigned SpillID = MTracker->getSpillIDWithIdx(*SpillNo, I); 1454ee3eee71SJeremy Morse LocIdx L = MTracker->getSpillMLoc(SpillID); 1455ee3eee71SJeremy Morse TTracker->clobberMloc(L, MI.getIterator(), true); 1456ee3eee71SJeremy Morse } 1457ee3eee71SJeremy Morse } 1458ae6f7882SJeremy Morse } 145914aaaa12SJeremy Morse } 1460ae6f7882SJeremy Morse 1461ae6f7882SJeremy Morse void InstrRefBasedLDV::performCopy(Register SrcRegNum, Register DstRegNum) { 1462d9eebe3cSJeremy Morse // In all circumstances, re-def all aliases. It's definitely a new value now. 1463d9eebe3cSJeremy Morse for (MCRegAliasIterator RAI(DstRegNum, TRI, true); RAI.isValid(); ++RAI) 1464d9eebe3cSJeremy Morse MTracker->defReg(*RAI, CurBB, CurInst); 1465ae6f7882SJeremy Morse 1466d9eebe3cSJeremy Morse ValueIDNum SrcValue = MTracker->readReg(SrcRegNum); 1467ae6f7882SJeremy Morse MTracker->setReg(DstRegNum, SrcValue); 1468ae6f7882SJeremy Morse 1469d9eebe3cSJeremy Morse // Copy subregisters from one location to another. 1470ae6f7882SJeremy Morse for (MCSubRegIndexIterator SRI(SrcRegNum, TRI); SRI.isValid(); ++SRI) { 1471ae6f7882SJeremy Morse unsigned SrcSubReg = SRI.getSubReg(); 1472ae6f7882SJeremy Morse unsigned SubRegIdx = SRI.getSubRegIndex(); 1473ae6f7882SJeremy Morse unsigned DstSubReg = TRI->getSubReg(DstRegNum, SubRegIdx); 1474ae6f7882SJeremy Morse if (!DstSubReg) 1475ae6f7882SJeremy Morse continue; 1476ae6f7882SJeremy Morse 1477ae6f7882SJeremy Morse // Do copy. There are two matching subregisters, the source value should 1478ae6f7882SJeremy Morse // have been def'd when the super-reg was, the latter might not be tracked 1479ae6f7882SJeremy Morse // yet. 1480d9eebe3cSJeremy Morse // This will force SrcSubReg to be tracked, if it isn't yet. Will read 1481d9eebe3cSJeremy Morse // mphi values if it wasn't tracked. 1482d9eebe3cSJeremy Morse LocIdx SrcL = MTracker->lookupOrTrackRegister(SrcSubReg); 1483d9eebe3cSJeremy Morse LocIdx DstL = MTracker->lookupOrTrackRegister(DstSubReg); 1484d9eebe3cSJeremy Morse (void)SrcL; 1485ae6f7882SJeremy Morse (void)DstL; 1486d9eebe3cSJeremy Morse ValueIDNum CpyValue = MTracker->readReg(SrcSubReg); 1487ae6f7882SJeremy Morse 1488ae6f7882SJeremy Morse MTracker->setReg(DstSubReg, CpyValue); 1489ae6f7882SJeremy Morse } 1490ae6f7882SJeremy Morse } 1491ae6f7882SJeremy Morse 149214aaaa12SJeremy Morse Optional<SpillLocationNo> 149314aaaa12SJeremy Morse InstrRefBasedLDV::isSpillInstruction(const MachineInstr &MI, 1494ae6f7882SJeremy Morse MachineFunction *MF) { 1495ae6f7882SJeremy Morse // TODO: Handle multiple stores folded into one. 1496ae6f7882SJeremy Morse if (!MI.hasOneMemOperand()) 149714aaaa12SJeremy Morse return None; 1498ae6f7882SJeremy Morse 1499ee3eee71SJeremy Morse // Reject any memory operand that's aliased -- we can't guarantee its value. 1500ee3eee71SJeremy Morse auto MMOI = MI.memoperands_begin(); 1501ee3eee71SJeremy Morse const PseudoSourceValue *PVal = (*MMOI)->getPseudoValue(); 1502ee3eee71SJeremy Morse if (PVal->isAliased(MFI)) 150314aaaa12SJeremy Morse return None; 1504ee3eee71SJeremy Morse 1505ae6f7882SJeremy Morse if (!MI.getSpillSize(TII) && !MI.getFoldedSpillSize(TII)) 150614aaaa12SJeremy Morse return None; // This is not a spill instruction, since no valid size was 1507ae6f7882SJeremy Morse // returned from either function. 1508ae6f7882SJeremy Morse 150914aaaa12SJeremy Morse return extractSpillBaseRegAndOffset(MI); 1510ae6f7882SJeremy Morse } 1511ae6f7882SJeremy Morse 1512ae6f7882SJeremy Morse bool InstrRefBasedLDV::isLocationSpill(const MachineInstr &MI, 1513ae6f7882SJeremy Morse MachineFunction *MF, unsigned &Reg) { 1514ae6f7882SJeremy Morse if (!isSpillInstruction(MI, MF)) 1515ae6f7882SJeremy Morse return false; 1516ae6f7882SJeremy Morse 1517ae6f7882SJeremy Morse int FI; 1518ae6f7882SJeremy Morse Reg = TII->isStoreToStackSlotPostFE(MI, FI); 1519ae6f7882SJeremy Morse return Reg != 0; 1520ae6f7882SJeremy Morse } 1521ae6f7882SJeremy Morse 1522e7084ceaSJeremy Morse Optional<SpillLocationNo> 1523ae6f7882SJeremy Morse InstrRefBasedLDV::isRestoreInstruction(const MachineInstr &MI, 1524ae6f7882SJeremy Morse MachineFunction *MF, unsigned &Reg) { 1525ae6f7882SJeremy Morse if (!MI.hasOneMemOperand()) 1526ae6f7882SJeremy Morse return None; 1527ae6f7882SJeremy Morse 1528ae6f7882SJeremy Morse // FIXME: Handle folded restore instructions with more than one memory 1529ae6f7882SJeremy Morse // operand. 1530ae6f7882SJeremy Morse if (MI.getRestoreSize(TII)) { 1531ae6f7882SJeremy Morse Reg = MI.getOperand(0).getReg(); 1532ae6f7882SJeremy Morse return extractSpillBaseRegAndOffset(MI); 1533ae6f7882SJeremy Morse } 1534ae6f7882SJeremy Morse return None; 1535ae6f7882SJeremy Morse } 1536ae6f7882SJeremy Morse 1537ae6f7882SJeremy Morse bool InstrRefBasedLDV::transferSpillOrRestoreInst(MachineInstr &MI) { 1538ae6f7882SJeremy Morse // XXX -- it's too difficult to implement VarLocBasedImpl's stack location 1539ae6f7882SJeremy Morse // limitations under the new model. Therefore, when comparing them, compare 1540ae6f7882SJeremy Morse // versions that don't attempt spills or restores at all. 1541ae6f7882SJeremy Morse if (EmulateOldLDV) 1542ae6f7882SJeremy Morse return false; 1543ae6f7882SJeremy Morse 1544e7084ceaSJeremy Morse // Strictly limit ourselves to plain loads and stores, not all instructions 1545e7084ceaSJeremy Morse // that can access the stack. 1546e7084ceaSJeremy Morse int DummyFI = -1; 1547e7084ceaSJeremy Morse if (!TII->isStoreToStackSlotPostFE(MI, DummyFI) && 1548e7084ceaSJeremy Morse !TII->isLoadFromStackSlotPostFE(MI, DummyFI)) 1549e7084ceaSJeremy Morse return false; 1550e7084ceaSJeremy Morse 1551ae6f7882SJeremy Morse MachineFunction *MF = MI.getMF(); 1552ae6f7882SJeremy Morse unsigned Reg; 1553ae6f7882SJeremy Morse 1554ae6f7882SJeremy Morse LLVM_DEBUG(dbgs() << "Examining instruction: "; MI.dump();); 1555ae6f7882SJeremy Morse 1556ee3eee71SJeremy Morse // Strictly limit ourselves to plain loads and stores, not all instructions 1557ee3eee71SJeremy Morse // that can access the stack. 1558ee3eee71SJeremy Morse int FIDummy; 1559ee3eee71SJeremy Morse if (!TII->isStoreToStackSlotPostFE(MI, FIDummy) && 1560ee3eee71SJeremy Morse !TII->isLoadFromStackSlotPostFE(MI, FIDummy)) 1561ee3eee71SJeremy Morse return false; 1562ee3eee71SJeremy Morse 1563ae6f7882SJeremy Morse // First, if there are any DBG_VALUEs pointing at a spill slot that is 1564ae6f7882SJeremy Morse // written to, terminate that variable location. The value in memory 1565ae6f7882SJeremy Morse // will have changed. DbgEntityHistoryCalculator doesn't try to detect this. 156614aaaa12SJeremy Morse if (Optional<SpillLocationNo> Loc = isSpillInstruction(MI, MF)) { 1567e7084ceaSJeremy Morse // Un-set this location and clobber, so that earlier locations don't 1568e7084ceaSJeremy Morse // continue past this store. 1569e7084ceaSJeremy Morse for (unsigned SlotIdx = 0; SlotIdx < MTracker->NumSlotIdxes; ++SlotIdx) { 157014aaaa12SJeremy Morse unsigned SpillID = MTracker->getSpillIDWithIdx(*Loc, SlotIdx); 1571e7084ceaSJeremy Morse Optional<LocIdx> MLoc = MTracker->getSpillMLoc(SpillID); 1572e7084ceaSJeremy Morse if (!MLoc) 1573e7084ceaSJeremy Morse continue; 1574e7084ceaSJeremy Morse 1575e7084ceaSJeremy Morse // We need to over-write the stack slot with something (here, a def at 1576e7084ceaSJeremy Morse // this instruction) to ensure no values are preserved in this stack slot 1577e7084ceaSJeremy Morse // after the spill. It also prevents TTracker from trying to recover the 1578e7084ceaSJeremy Morse // location and re-installing it in the same place. 1579e7084ceaSJeremy Morse ValueIDNum Def(CurBB, CurInst, *MLoc); 1580e7084ceaSJeremy Morse MTracker->setMLoc(*MLoc, Def); 1581e7084ceaSJeremy Morse if (TTracker) 1582ae6f7882SJeremy Morse TTracker->clobberMloc(*MLoc, MI.getIterator()); 1583ae6f7882SJeremy Morse } 1584ae6f7882SJeremy Morse } 1585ae6f7882SJeremy Morse 1586ae6f7882SJeremy Morse // Try to recognise spill and restore instructions that may transfer a value. 1587ae6f7882SJeremy Morse if (isLocationSpill(MI, MF, Reg)) { 158814aaaa12SJeremy Morse // isLocationSpill returning true should guarantee we can extract a 158914aaaa12SJeremy Morse // location. 159014aaaa12SJeremy Morse SpillLocationNo Loc = *extractSpillBaseRegAndOffset(MI); 1591ae6f7882SJeremy Morse 1592e7084ceaSJeremy Morse auto DoTransfer = [&](Register SrcReg, unsigned SpillID) { 1593e7084ceaSJeremy Morse auto ReadValue = MTracker->readReg(SrcReg); 1594e7084ceaSJeremy Morse LocIdx DstLoc = MTracker->getSpillMLoc(SpillID); 1595e7084ceaSJeremy Morse MTracker->setMLoc(DstLoc, ReadValue); 1596ae6f7882SJeremy Morse 1597e7084ceaSJeremy Morse if (TTracker) { 1598e7084ceaSJeremy Morse LocIdx SrcLoc = MTracker->getRegMLoc(SrcReg); 1599e7084ceaSJeremy Morse TTracker->transferMlocs(SrcLoc, DstLoc, MI.getIterator()); 1600ae6f7882SJeremy Morse } 1601e7084ceaSJeremy Morse }; 1602e7084ceaSJeremy Morse 1603e7084ceaSJeremy Morse // Then, transfer subreg bits. 1604e7084ceaSJeremy Morse for (MCSubRegIterator SRI(Reg, TRI, false); SRI.isValid(); ++SRI) { 1605e7084ceaSJeremy Morse // Ensure this reg is tracked, 1606e7084ceaSJeremy Morse (void)MTracker->lookupOrTrackRegister(*SRI); 1607e7084ceaSJeremy Morse unsigned SubregIdx = TRI->getSubRegIndex(Reg, *SRI); 1608e7084ceaSJeremy Morse unsigned SpillID = MTracker->getLocID(Loc, SubregIdx); 1609e7084ceaSJeremy Morse DoTransfer(*SRI, SpillID); 1610e7084ceaSJeremy Morse } 1611e7084ceaSJeremy Morse 1612e7084ceaSJeremy Morse // Directly lookup size of main source reg, and transfer. 1613e7084ceaSJeremy Morse unsigned Size = TRI->getRegSizeInBits(Reg, *MRI); 1614e7084ceaSJeremy Morse unsigned SpillID = MTracker->getLocID(Loc, {Size, 0}); 1615e7084ceaSJeremy Morse DoTransfer(Reg, SpillID); 1616e7084ceaSJeremy Morse } else { 161714aaaa12SJeremy Morse Optional<SpillLocationNo> Loc = isRestoreInstruction(MI, MF, Reg); 161814aaaa12SJeremy Morse if (!Loc) 1619e7084ceaSJeremy Morse return false; 1620e7084ceaSJeremy Morse 1621e7084ceaSJeremy Morse // Assumption: we're reading from the base of the stack slot, not some 1622e7084ceaSJeremy Morse // offset into it. It seems very unlikely LLVM would ever generate 1623e7084ceaSJeremy Morse // restores where this wasn't true. This then becomes a question of what 1624e7084ceaSJeremy Morse // subregisters in the destination register line up with positions in the 1625e7084ceaSJeremy Morse // stack slot. 1626e7084ceaSJeremy Morse 1627e7084ceaSJeremy Morse // Def all registers that alias the destination. 1628e7084ceaSJeremy Morse for (MCRegAliasIterator RAI(Reg, TRI, true); RAI.isValid(); ++RAI) 1629e7084ceaSJeremy Morse MTracker->defReg(*RAI, CurBB, CurInst); 1630e7084ceaSJeremy Morse 1631e7084ceaSJeremy Morse // Now find subregisters within the destination register, and load values 1632e7084ceaSJeremy Morse // from stack slot positions. 1633e7084ceaSJeremy Morse auto DoTransfer = [&](Register DestReg, unsigned SpillID) { 1634e7084ceaSJeremy Morse LocIdx SrcIdx = MTracker->getSpillMLoc(SpillID); 1635e7084ceaSJeremy Morse auto ReadValue = MTracker->readMLoc(SrcIdx); 1636e7084ceaSJeremy Morse MTracker->setReg(DestReg, ReadValue); 1637e7084ceaSJeremy Morse 1638e7084ceaSJeremy Morse if (TTracker) { 1639e7084ceaSJeremy Morse LocIdx DstLoc = MTracker->getRegMLoc(DestReg); 1640e7084ceaSJeremy Morse TTracker->transferMlocs(SrcIdx, DstLoc, MI.getIterator()); 1641e7084ceaSJeremy Morse } 1642e7084ceaSJeremy Morse }; 1643e7084ceaSJeremy Morse 1644e7084ceaSJeremy Morse for (MCSubRegIterator SRI(Reg, TRI, false); SRI.isValid(); ++SRI) { 1645e7084ceaSJeremy Morse unsigned Subreg = TRI->getSubRegIndex(Reg, *SRI); 164614aaaa12SJeremy Morse unsigned SpillID = MTracker->getLocID(*Loc, Subreg); 1647e7084ceaSJeremy Morse DoTransfer(*SRI, SpillID); 1648e7084ceaSJeremy Morse } 1649e7084ceaSJeremy Morse 1650e7084ceaSJeremy Morse // Directly look up this registers slot idx by size, and transfer. 1651e7084ceaSJeremy Morse unsigned Size = TRI->getRegSizeInBits(Reg, *MRI); 165214aaaa12SJeremy Morse unsigned SpillID = MTracker->getLocID(*Loc, {Size, 0}); 1653e7084ceaSJeremy Morse DoTransfer(Reg, SpillID); 1654ae6f7882SJeremy Morse } 1655ae6f7882SJeremy Morse return true; 1656ae6f7882SJeremy Morse } 1657ae6f7882SJeremy Morse 1658ae6f7882SJeremy Morse bool InstrRefBasedLDV::transferRegisterCopy(MachineInstr &MI) { 1659ae6f7882SJeremy Morse auto DestSrc = TII->isCopyInstr(MI); 1660ae6f7882SJeremy Morse if (!DestSrc) 1661ae6f7882SJeremy Morse return false; 1662ae6f7882SJeremy Morse 1663ae6f7882SJeremy Morse const MachineOperand *DestRegOp = DestSrc->Destination; 1664ae6f7882SJeremy Morse const MachineOperand *SrcRegOp = DestSrc->Source; 1665ae6f7882SJeremy Morse 1666ae6f7882SJeremy Morse auto isCalleeSavedReg = [&](unsigned Reg) { 1667ae6f7882SJeremy Morse for (MCRegAliasIterator RAI(Reg, TRI, true); RAI.isValid(); ++RAI) 1668ae6f7882SJeremy Morse if (CalleeSavedRegs.test(*RAI)) 1669ae6f7882SJeremy Morse return true; 1670ae6f7882SJeremy Morse return false; 1671ae6f7882SJeremy Morse }; 1672ae6f7882SJeremy Morse 1673ae6f7882SJeremy Morse Register SrcReg = SrcRegOp->getReg(); 1674ae6f7882SJeremy Morse Register DestReg = DestRegOp->getReg(); 1675ae6f7882SJeremy Morse 1676ae6f7882SJeremy Morse // Ignore identity copies. Yep, these make it as far as LiveDebugValues. 1677ae6f7882SJeremy Morse if (SrcReg == DestReg) 1678ae6f7882SJeremy Morse return true; 1679ae6f7882SJeremy Morse 1680ae6f7882SJeremy Morse // For emulating VarLocBasedImpl: 1681ae6f7882SJeremy Morse // We want to recognize instructions where destination register is callee 1682ae6f7882SJeremy Morse // saved register. If register that could be clobbered by the call is 1683ae6f7882SJeremy Morse // included, there would be a great chance that it is going to be clobbered 1684ae6f7882SJeremy Morse // soon. It is more likely that previous register, which is callee saved, is 1685ae6f7882SJeremy Morse // going to stay unclobbered longer, even if it is killed. 1686ae6f7882SJeremy Morse // 1687ae6f7882SJeremy Morse // For InstrRefBasedImpl, we can track multiple locations per value, so 1688ae6f7882SJeremy Morse // ignore this condition. 1689ae6f7882SJeremy Morse if (EmulateOldLDV && !isCalleeSavedReg(DestReg)) 1690ae6f7882SJeremy Morse return false; 1691ae6f7882SJeremy Morse 1692ae6f7882SJeremy Morse // InstrRefBasedImpl only followed killing copies. 1693ae6f7882SJeremy Morse if (EmulateOldLDV && !SrcRegOp->isKill()) 1694ae6f7882SJeremy Morse return false; 1695ae6f7882SJeremy Morse 1696ae6f7882SJeremy Morse // Copy MTracker info, including subregs if available. 1697ae6f7882SJeremy Morse InstrRefBasedLDV::performCopy(SrcReg, DestReg); 1698ae6f7882SJeremy Morse 1699ae6f7882SJeremy Morse // Only produce a transfer of DBG_VALUE within a block where old LDV 1700ae6f7882SJeremy Morse // would have. We might make use of the additional value tracking in some 1701ae6f7882SJeremy Morse // other way, later. 1702ae6f7882SJeremy Morse if (TTracker && isCalleeSavedReg(DestReg) && SrcRegOp->isKill()) 1703ae6f7882SJeremy Morse TTracker->transferMlocs(MTracker->getRegMLoc(SrcReg), 1704ae6f7882SJeremy Morse MTracker->getRegMLoc(DestReg), MI.getIterator()); 1705ae6f7882SJeremy Morse 1706ae6f7882SJeremy Morse // VarLocBasedImpl would quit tracking the old location after copying. 1707ae6f7882SJeremy Morse if (EmulateOldLDV && SrcReg != DestReg) 1708ae6f7882SJeremy Morse MTracker->defReg(SrcReg, CurBB, CurInst); 1709ae6f7882SJeremy Morse 171049555441SJeremy Morse // Finally, the copy might have clobbered variables based on the destination 171149555441SJeremy Morse // register. Tell TTracker about it, in case a backup location exists. 171249555441SJeremy Morse if (TTracker) { 171349555441SJeremy Morse for (MCRegAliasIterator RAI(DestReg, TRI, true); RAI.isValid(); ++RAI) { 171449555441SJeremy Morse LocIdx ClobberedLoc = MTracker->getRegMLoc(*RAI); 171549555441SJeremy Morse TTracker->clobberMloc(ClobberedLoc, MI.getIterator(), false); 171649555441SJeremy Morse } 171749555441SJeremy Morse } 171849555441SJeremy Morse 1719ae6f7882SJeremy Morse return true; 1720ae6f7882SJeremy Morse } 1721ae6f7882SJeremy Morse 1722ae6f7882SJeremy Morse /// Accumulate a mapping between each DILocalVariable fragment and other 1723ae6f7882SJeremy Morse /// fragments of that DILocalVariable which overlap. This reduces work during 1724ae6f7882SJeremy Morse /// the data-flow stage from "Find any overlapping fragments" to "Check if the 1725ae6f7882SJeremy Morse /// known-to-overlap fragments are present". 17260eee8445SJeremy Morse /// \param MI A previously unprocessed debug instruction to analyze for 1727ae6f7882SJeremy Morse /// fragment usage. 1728ae6f7882SJeremy Morse void InstrRefBasedLDV::accumulateFragmentMap(MachineInstr &MI) { 17290eee8445SJeremy Morse assert(MI.isDebugValue() || MI.isDebugRef()); 1730ae6f7882SJeremy Morse DebugVariable MIVar(MI.getDebugVariable(), MI.getDebugExpression(), 1731ae6f7882SJeremy Morse MI.getDebugLoc()->getInlinedAt()); 1732ae6f7882SJeremy Morse FragmentInfo ThisFragment = MIVar.getFragmentOrDefault(); 1733ae6f7882SJeremy Morse 1734ae6f7882SJeremy Morse // If this is the first sighting of this variable, then we are guaranteed 1735ae6f7882SJeremy Morse // there are currently no overlapping fragments either. Initialize the set 1736ae6f7882SJeremy Morse // of seen fragments, record no overlaps for the current one, and return. 1737ae6f7882SJeremy Morse auto SeenIt = SeenFragments.find(MIVar.getVariable()); 1738ae6f7882SJeremy Morse if (SeenIt == SeenFragments.end()) { 1739ae6f7882SJeremy Morse SmallSet<FragmentInfo, 4> OneFragment; 1740ae6f7882SJeremy Morse OneFragment.insert(ThisFragment); 1741ae6f7882SJeremy Morse SeenFragments.insert({MIVar.getVariable(), OneFragment}); 1742ae6f7882SJeremy Morse 1743ae6f7882SJeremy Morse OverlapFragments.insert({{MIVar.getVariable(), ThisFragment}, {}}); 1744ae6f7882SJeremy Morse return; 1745ae6f7882SJeremy Morse } 1746ae6f7882SJeremy Morse 1747ae6f7882SJeremy Morse // If this particular Variable/Fragment pair already exists in the overlap 1748ae6f7882SJeremy Morse // map, it has already been accounted for. 1749ae6f7882SJeremy Morse auto IsInOLapMap = 1750ae6f7882SJeremy Morse OverlapFragments.insert({{MIVar.getVariable(), ThisFragment}, {}}); 1751ae6f7882SJeremy Morse if (!IsInOLapMap.second) 1752ae6f7882SJeremy Morse return; 1753ae6f7882SJeremy Morse 1754ae6f7882SJeremy Morse auto &ThisFragmentsOverlaps = IsInOLapMap.first->second; 1755ae6f7882SJeremy Morse auto &AllSeenFragments = SeenIt->second; 1756ae6f7882SJeremy Morse 1757ae6f7882SJeremy Morse // Otherwise, examine all other seen fragments for this variable, with "this" 1758ae6f7882SJeremy Morse // fragment being a previously unseen fragment. Record any pair of 1759ae6f7882SJeremy Morse // overlapping fragments. 1760ae6f7882SJeremy Morse for (auto &ASeenFragment : AllSeenFragments) { 1761ae6f7882SJeremy Morse // Does this previously seen fragment overlap? 1762ae6f7882SJeremy Morse if (DIExpression::fragmentsOverlap(ThisFragment, ASeenFragment)) { 1763ae6f7882SJeremy Morse // Yes: Mark the current fragment as being overlapped. 1764ae6f7882SJeremy Morse ThisFragmentsOverlaps.push_back(ASeenFragment); 1765ae6f7882SJeremy Morse // Mark the previously seen fragment as being overlapped by the current 1766ae6f7882SJeremy Morse // one. 1767ae6f7882SJeremy Morse auto ASeenFragmentsOverlaps = 1768ae6f7882SJeremy Morse OverlapFragments.find({MIVar.getVariable(), ASeenFragment}); 1769ae6f7882SJeremy Morse assert(ASeenFragmentsOverlaps != OverlapFragments.end() && 1770ae6f7882SJeremy Morse "Previously seen var fragment has no vector of overlaps"); 1771ae6f7882SJeremy Morse ASeenFragmentsOverlaps->second.push_back(ThisFragment); 1772ae6f7882SJeremy Morse } 1773ae6f7882SJeremy Morse } 1774ae6f7882SJeremy Morse 1775ae6f7882SJeremy Morse AllSeenFragments.insert(ThisFragment); 1776ae6f7882SJeremy Morse } 1777ae6f7882SJeremy Morse 1778ab49dce0SJeremy Morse void InstrRefBasedLDV::process(MachineInstr &MI, const ValueTable *MLiveOuts, 1779ab49dce0SJeremy Morse const ValueTable *MLiveIns) { 1780ae6f7882SJeremy Morse // Try to interpret an MI as a debug or transfer instruction. Only if it's 1781ae6f7882SJeremy Morse // none of these should we interpret it's register defs as new value 1782ae6f7882SJeremy Morse // definitions. 1783ae6f7882SJeremy Morse if (transferDebugValue(MI)) 1784ae6f7882SJeremy Morse return; 1785010108bbSJeremy Morse if (transferDebugInstrRef(MI, MLiveOuts, MLiveIns)) 1786010108bbSJeremy Morse return; 1787010108bbSJeremy Morse if (transferDebugPHI(MI)) 178868f47157SJeremy Morse return; 1789ae6f7882SJeremy Morse if (transferRegisterCopy(MI)) 1790ae6f7882SJeremy Morse return; 1791ae6f7882SJeremy Morse if (transferSpillOrRestoreInst(MI)) 1792ae6f7882SJeremy Morse return; 1793ae6f7882SJeremy Morse transferRegisterDef(MI); 1794ae6f7882SJeremy Morse } 1795ae6f7882SJeremy Morse 1796ab93e710SJeremy Morse void InstrRefBasedLDV::produceMLocTransferFunction( 1797ae6f7882SJeremy Morse MachineFunction &MF, SmallVectorImpl<MLocTransferMap> &MLocTransfer, 1798ab93e710SJeremy Morse unsigned MaxNumBlocks) { 1799ae6f7882SJeremy Morse // Because we try to optimize around register mask operands by ignoring regs 1800ae6f7882SJeremy Morse // that aren't currently tracked, we set up something ugly for later: RegMask 1801ae6f7882SJeremy Morse // operands that are seen earlier than the first use of a register, still need 1802ae6f7882SJeremy Morse // to clobber that register in the transfer function. But this information 1803ae6f7882SJeremy Morse // isn't actively recorded. Instead, we track each RegMask used in each block, 1804ae6f7882SJeremy Morse // and accumulated the clobbered but untracked registers in each block into 1805ae6f7882SJeremy Morse // the following bitvector. Later, if new values are tracked, we can add 1806ae6f7882SJeremy Morse // appropriate clobbers. 1807ae6f7882SJeremy Morse SmallVector<BitVector, 32> BlockMasks; 1808ae6f7882SJeremy Morse BlockMasks.resize(MaxNumBlocks); 1809ae6f7882SJeremy Morse 1810ae6f7882SJeremy Morse // Reserve one bit per register for the masks described above. 1811ae6f7882SJeremy Morse unsigned BVWords = MachineOperand::getRegMaskSize(TRI->getNumRegs()); 1812ae6f7882SJeremy Morse for (auto &BV : BlockMasks) 1813ae6f7882SJeremy Morse BV.resize(TRI->getNumRegs(), true); 1814ae6f7882SJeremy Morse 1815ae6f7882SJeremy Morse // Step through all instructions and inhale the transfer function. 1816ae6f7882SJeremy Morse for (auto &MBB : MF) { 1817ae6f7882SJeremy Morse // Object fields that are read by trackers to know where we are in the 1818ae6f7882SJeremy Morse // function. 1819ae6f7882SJeremy Morse CurBB = MBB.getNumber(); 1820ae6f7882SJeremy Morse CurInst = 1; 1821ae6f7882SJeremy Morse 1822ae6f7882SJeremy Morse // Set all machine locations to a PHI value. For transfer function 1823ae6f7882SJeremy Morse // production only, this signifies the live-in value to the block. 1824ae6f7882SJeremy Morse MTracker->reset(); 1825ae6f7882SJeremy Morse MTracker->setMPhis(CurBB); 1826ae6f7882SJeremy Morse 1827ae6f7882SJeremy Morse // Step through each instruction in this block. 1828ae6f7882SJeremy Morse for (auto &MI : MBB) { 1829ab49dce0SJeremy Morse // Pass in an empty unique_ptr for the value tables when accumulating the 1830ab49dce0SJeremy Morse // machine transfer function. 1831ab49dce0SJeremy Morse process(MI, nullptr, nullptr); 1832ab49dce0SJeremy Morse 1833ae6f7882SJeremy Morse // Also accumulate fragment map. 18340eee8445SJeremy Morse if (MI.isDebugValue() || MI.isDebugRef()) 1835ae6f7882SJeremy Morse accumulateFragmentMap(MI); 183668f47157SJeremy Morse 183768f47157SJeremy Morse // Create a map from the instruction number (if present) to the 183868f47157SJeremy Morse // MachineInstr and its position. 1839cca049adSDjordje Todorovic if (uint64_t InstrNo = MI.peekDebugInstrNum()) { 184068f47157SJeremy Morse auto InstrAndPos = std::make_pair(&MI, CurInst); 184168f47157SJeremy Morse auto InsertResult = 184268f47157SJeremy Morse DebugInstrNumToInstr.insert(std::make_pair(InstrNo, InstrAndPos)); 184368f47157SJeremy Morse 184468f47157SJeremy Morse // There should never be duplicate instruction numbers. 184568f47157SJeremy Morse assert(InsertResult.second); 184668f47157SJeremy Morse (void)InsertResult; 184768f47157SJeremy Morse } 184868f47157SJeremy Morse 1849ae6f7882SJeremy Morse ++CurInst; 1850ae6f7882SJeremy Morse } 1851ae6f7882SJeremy Morse 1852ae6f7882SJeremy Morse // Produce the transfer function, a map of machine location to new value. If 1853ae6f7882SJeremy Morse // any machine location has the live-in phi value from the start of the 1854ae6f7882SJeremy Morse // block, it's live-through and doesn't need recording in the transfer 1855ae6f7882SJeremy Morse // function. 1856ae6f7882SJeremy Morse for (auto Location : MTracker->locations()) { 1857ae6f7882SJeremy Morse LocIdx Idx = Location.Idx; 1858ae6f7882SJeremy Morse ValueIDNum &P = Location.Value; 1859ae6f7882SJeremy Morse if (P.isPHI() && P.getLoc() == Idx.asU64()) 1860ae6f7882SJeremy Morse continue; 1861ae6f7882SJeremy Morse 1862ae6f7882SJeremy Morse // Insert-or-update. 1863ae6f7882SJeremy Morse auto &TransferMap = MLocTransfer[CurBB]; 1864ae6f7882SJeremy Morse auto Result = TransferMap.insert(std::make_pair(Idx.asU64(), P)); 1865ae6f7882SJeremy Morse if (!Result.second) 1866ae6f7882SJeremy Morse Result.first->second = P; 1867ae6f7882SJeremy Morse } 1868ae6f7882SJeremy Morse 1869ae6f7882SJeremy Morse // Accumulate any bitmask operands into the clobberred reg mask for this 1870ae6f7882SJeremy Morse // block. 1871ae6f7882SJeremy Morse for (auto &P : MTracker->Masks) { 1872ae6f7882SJeremy Morse BlockMasks[CurBB].clearBitsNotInMask(P.first->getRegMask(), BVWords); 1873ae6f7882SJeremy Morse } 1874ae6f7882SJeremy Morse } 1875ae6f7882SJeremy Morse 1876ae6f7882SJeremy Morse // Compute a bitvector of all the registers that are tracked in this block. 1877ae6f7882SJeremy Morse BitVector UsedRegs(TRI->getNumRegs()); 1878ae6f7882SJeremy Morse for (auto Location : MTracker->locations()) { 1879ae6f7882SJeremy Morse unsigned ID = MTracker->LocIdxToLocID[Location.Idx]; 1880fbf269c7SJeremy Morse // Ignore stack slots, and aliases of the stack pointer. 1881fbf269c7SJeremy Morse if (ID >= TRI->getNumRegs() || MTracker->SPAliases.count(ID)) 1882ae6f7882SJeremy Morse continue; 1883ae6f7882SJeremy Morse UsedRegs.set(ID); 1884ae6f7882SJeremy Morse } 1885ae6f7882SJeremy Morse 1886ae6f7882SJeremy Morse // Check that any regmask-clobber of a register that gets tracked, is not 1887ae6f7882SJeremy Morse // live-through in the transfer function. It needs to be clobbered at the 1888ae6f7882SJeremy Morse // very least. 1889ae6f7882SJeremy Morse for (unsigned int I = 0; I < MaxNumBlocks; ++I) { 1890ae6f7882SJeremy Morse BitVector &BV = BlockMasks[I]; 1891ae6f7882SJeremy Morse BV.flip(); 1892ae6f7882SJeremy Morse BV &= UsedRegs; 1893ae6f7882SJeremy Morse // This produces all the bits that we clobber, but also use. Check that 1894ae6f7882SJeremy Morse // they're all clobbered or at least set in the designated transfer 1895ae6f7882SJeremy Morse // elem. 1896ae6f7882SJeremy Morse for (unsigned Bit : BV.set_bits()) { 1897e7084ceaSJeremy Morse unsigned ID = MTracker->getLocID(Bit); 1898ae6f7882SJeremy Morse LocIdx Idx = MTracker->LocIDToLocIdx[ID]; 1899ae6f7882SJeremy Morse auto &TransferMap = MLocTransfer[I]; 1900ae6f7882SJeremy Morse 1901ae6f7882SJeremy Morse // Install a value representing the fact that this location is effectively 1902ae6f7882SJeremy Morse // written to in this block. As there's no reserved value, instead use 1903ae6f7882SJeremy Morse // a value number that is never generated. Pick the value number for the 1904ae6f7882SJeremy Morse // first instruction in the block, def'ing this location, which we know 1905ae6f7882SJeremy Morse // this block never used anyway. 1906ae6f7882SJeremy Morse ValueIDNum NotGeneratedNum = ValueIDNum(I, 1, Idx); 1907ae6f7882SJeremy Morse auto Result = 1908ae6f7882SJeremy Morse TransferMap.insert(std::make_pair(Idx.asU64(), NotGeneratedNum)); 1909ae6f7882SJeremy Morse if (!Result.second) { 1910ae6f7882SJeremy Morse ValueIDNum &ValueID = Result.first->second; 1911ae6f7882SJeremy Morse if (ValueID.getBlock() == I && ValueID.isPHI()) 1912ae6f7882SJeremy Morse // It was left as live-through. Set it to clobbered. 1913ae6f7882SJeremy Morse ValueID = NotGeneratedNum; 1914ae6f7882SJeremy Morse } 1915ae6f7882SJeremy Morse } 1916ae6f7882SJeremy Morse } 1917ae6f7882SJeremy Morse } 1918ae6f7882SJeremy Morse 1919a3936a6cSJeremy Morse bool InstrRefBasedLDV::mlocJoin( 1920a3936a6cSJeremy Morse MachineBasicBlock &MBB, SmallPtrSet<const MachineBasicBlock *, 16> &Visited, 1921ab49dce0SJeremy Morse FuncValueTable &OutLocs, ValueTable &InLocs) { 1922ae6f7882SJeremy Morse LLVM_DEBUG(dbgs() << "join MBB: " << MBB.getNumber() << "\n"); 1923ae6f7882SJeremy Morse bool Changed = false; 1924ae6f7882SJeremy Morse 1925a3936a6cSJeremy Morse // Handle value-propagation when control flow merges on entry to a block. For 1926a3936a6cSJeremy Morse // any location without a PHI already placed, the location has the same value 1927a3936a6cSJeremy Morse // as its predecessors. If a PHI is placed, test to see whether it's now a 1928a3936a6cSJeremy Morse // redundant PHI that we can eliminate. 1929a3936a6cSJeremy Morse 1930ae6f7882SJeremy Morse SmallVector<const MachineBasicBlock *, 8> BlockOrders; 1931a3936a6cSJeremy Morse for (auto Pred : MBB.predecessors()) 1932ae6f7882SJeremy Morse BlockOrders.push_back(Pred); 1933ae6f7882SJeremy Morse 1934ae6f7882SJeremy Morse // Visit predecessors in RPOT order. 1935ae6f7882SJeremy Morse auto Cmp = [&](const MachineBasicBlock *A, const MachineBasicBlock *B) { 1936ae6f7882SJeremy Morse return BBToOrder.find(A)->second < BBToOrder.find(B)->second; 1937ae6f7882SJeremy Morse }; 19389bcc0d10SKazu Hirata llvm::sort(BlockOrders, Cmp); 1939ae6f7882SJeremy Morse 1940ae6f7882SJeremy Morse // Skip entry block. 1941ae6f7882SJeremy Morse if (BlockOrders.size() == 0) 1942a3936a6cSJeremy Morse return false; 1943ae6f7882SJeremy Morse 1944a3936a6cSJeremy Morse // Step through all machine locations, look at each predecessor and test 1945a3936a6cSJeremy Morse // whether we can eliminate redundant PHIs. 1946ae6f7882SJeremy Morse for (auto Location : MTracker->locations()) { 1947ae6f7882SJeremy Morse LocIdx Idx = Location.Idx; 1948a3936a6cSJeremy Morse 1949ae6f7882SJeremy Morse // Pick out the first predecessors live-out value for this location. It's 1950a3936a6cSJeremy Morse // guaranteed to not be a backedge, as we order by RPO. 1951a3936a6cSJeremy Morse ValueIDNum FirstVal = OutLocs[BlockOrders[0]->getNumber()][Idx.asU64()]; 1952ae6f7882SJeremy Morse 1953a3936a6cSJeremy Morse // If we've already eliminated a PHI here, do no further checking, just 1954a3936a6cSJeremy Morse // propagate the first live-in value into this block. 1955a3936a6cSJeremy Morse if (InLocs[Idx.asU64()] != ValueIDNum(MBB.getNumber(), 0, Idx)) { 1956a3936a6cSJeremy Morse if (InLocs[Idx.asU64()] != FirstVal) { 1957a3936a6cSJeremy Morse InLocs[Idx.asU64()] = FirstVal; 1958a3936a6cSJeremy Morse Changed |= true; 1959a3936a6cSJeremy Morse } 1960a3936a6cSJeremy Morse continue; 1961a3936a6cSJeremy Morse } 1962a3936a6cSJeremy Morse 1963a3936a6cSJeremy Morse // We're now examining a PHI to see whether it's un-necessary. Loop around 1964a3936a6cSJeremy Morse // the other live-in values and test whether they're all the same. 1965ae6f7882SJeremy Morse bool Disagree = false; 1966ae6f7882SJeremy Morse for (unsigned int I = 1; I < BlockOrders.size(); ++I) { 1967a3936a6cSJeremy Morse const MachineBasicBlock *PredMBB = BlockOrders[I]; 1968a3936a6cSJeremy Morse const ValueIDNum &PredLiveOut = 1969a3936a6cSJeremy Morse OutLocs[PredMBB->getNumber()][Idx.asU64()]; 1970a3936a6cSJeremy Morse 1971a3936a6cSJeremy Morse // Incoming values agree, continue trying to eliminate this PHI. 1972a3936a6cSJeremy Morse if (FirstVal == PredLiveOut) 1973a3936a6cSJeremy Morse continue; 1974a3936a6cSJeremy Morse 1975a3936a6cSJeremy Morse // We can also accept a PHI value that feeds back into itself. 1976a3936a6cSJeremy Morse if (PredLiveOut == ValueIDNum(MBB.getNumber(), 0, Idx)) 1977a3936a6cSJeremy Morse continue; 1978a3936a6cSJeremy Morse 1979ae6f7882SJeremy Morse // Live-out of a predecessor disagrees with the first predecessor. 1980ae6f7882SJeremy Morse Disagree = true; 1981ae6f7882SJeremy Morse } 1982ae6f7882SJeremy Morse 1983a3936a6cSJeremy Morse // No disagreement? No PHI. Otherwise, leave the PHI in live-ins. 1984a3936a6cSJeremy Morse if (!Disagree) { 1985a3936a6cSJeremy Morse InLocs[Idx.asU64()] = FirstVal; 1986ae6f7882SJeremy Morse Changed |= true; 1987ae6f7882SJeremy Morse } 1988ae6f7882SJeremy Morse } 1989ae6f7882SJeremy Morse 1990cca049adSDjordje Todorovic // TODO: Reimplement NumInserted and NumRemoved. 1991a3936a6cSJeremy Morse return Changed; 1992ae6f7882SJeremy Morse } 1993ae6f7882SJeremy Morse 199497ddf49eSJeremy Morse void InstrRefBasedLDV::findStackIndexInterference( 199597ddf49eSJeremy Morse SmallVectorImpl<unsigned> &Slots) { 199697ddf49eSJeremy Morse // We could spend a bit of time finding the exact, minimal, set of stack 199797ddf49eSJeremy Morse // indexes that interfere with each other, much like reg units. Or, we can 199897ddf49eSJeremy Morse // rely on the fact that: 199997ddf49eSJeremy Morse // * The smallest / lowest index will interfere with everything at zero 200097ddf49eSJeremy Morse // offset, which will be the largest set of registers, 200197ddf49eSJeremy Morse // * Most indexes with non-zero offset will end up being interference units 200297ddf49eSJeremy Morse // anyway. 200397ddf49eSJeremy Morse // So just pick those out and return them. 200497ddf49eSJeremy Morse 200597ddf49eSJeremy Morse // We can rely on a single-byte stack index existing already, because we 200697ddf49eSJeremy Morse // initialize them in MLocTracker. 200797ddf49eSJeremy Morse auto It = MTracker->StackSlotIdxes.find({8, 0}); 200897ddf49eSJeremy Morse assert(It != MTracker->StackSlotIdxes.end()); 200997ddf49eSJeremy Morse Slots.push_back(It->second); 201097ddf49eSJeremy Morse 201197ddf49eSJeremy Morse // Find anything that has a non-zero offset and add that too. 201297ddf49eSJeremy Morse for (auto &Pair : MTracker->StackSlotIdxes) { 201397ddf49eSJeremy Morse // Is offset zero? If so, ignore. 201497ddf49eSJeremy Morse if (!Pair.first.second) 201597ddf49eSJeremy Morse continue; 201697ddf49eSJeremy Morse Slots.push_back(Pair.second); 201797ddf49eSJeremy Morse } 201897ddf49eSJeremy Morse } 201997ddf49eSJeremy Morse 202097ddf49eSJeremy Morse void InstrRefBasedLDV::placeMLocPHIs( 202197ddf49eSJeremy Morse MachineFunction &MF, SmallPtrSetImpl<MachineBasicBlock *> &AllBlocks, 2022ab49dce0SJeremy Morse FuncValueTable &MInLocs, SmallVectorImpl<MLocTransferMap> &MLocTransfer) { 202397ddf49eSJeremy Morse SmallVector<unsigned, 4> StackUnits; 202497ddf49eSJeremy Morse findStackIndexInterference(StackUnits); 202597ddf49eSJeremy Morse 2026fbf269c7SJeremy Morse // To avoid repeatedly running the PHI placement algorithm, leverage the 2027fbf269c7SJeremy Morse // fact that a def of register MUST also def its register units. Find the 2028fbf269c7SJeremy Morse // units for registers, place PHIs for them, and then replicate them for 2029fbf269c7SJeremy Morse // aliasing registers. Some inputs that are never def'd (DBG_PHIs of 2030fbf269c7SJeremy Morse // arguments) don't lead to register units being tracked, just place PHIs for 203197ddf49eSJeremy Morse // those registers directly. Stack slots have their own form of "unit", 203297ddf49eSJeremy Morse // store them to one side. 2033fbf269c7SJeremy Morse SmallSet<Register, 32> RegUnitsToPHIUp; 203497ddf49eSJeremy Morse SmallSet<LocIdx, 32> NormalLocsToPHI; 203597ddf49eSJeremy Morse SmallSet<SpillLocationNo, 32> StackSlots; 2036fbf269c7SJeremy Morse for (auto Location : MTracker->locations()) { 2037fbf269c7SJeremy Morse LocIdx L = Location.Idx; 2038fbf269c7SJeremy Morse if (MTracker->isSpill(L)) { 203997ddf49eSJeremy Morse StackSlots.insert(MTracker->locIDToSpill(MTracker->LocIdxToLocID[L])); 2040fbf269c7SJeremy Morse continue; 2041fbf269c7SJeremy Morse } 2042fbf269c7SJeremy Morse 2043fbf269c7SJeremy Morse Register R = MTracker->LocIdxToLocID[L]; 2044fbf269c7SJeremy Morse SmallSet<Register, 8> FoundRegUnits; 2045fbf269c7SJeremy Morse bool AnyIllegal = false; 2046fbf269c7SJeremy Morse for (MCRegUnitIterator RUI(R.asMCReg(), TRI); RUI.isValid(); ++RUI) { 2047fbf269c7SJeremy Morse for (MCRegUnitRootIterator URoot(*RUI, TRI); URoot.isValid(); ++URoot){ 2048fbf269c7SJeremy Morse if (!MTracker->isRegisterTracked(*URoot)) { 2049fbf269c7SJeremy Morse // Not all roots were loaded into the tracking map: this register 2050fbf269c7SJeremy Morse // isn't actually def'd anywhere, we only read from it. Generate PHIs 2051fbf269c7SJeremy Morse // for this reg, but don't iterate units. 2052fbf269c7SJeremy Morse AnyIllegal = true; 2053fbf269c7SJeremy Morse } else { 2054fbf269c7SJeremy Morse FoundRegUnits.insert(*URoot); 2055fbf269c7SJeremy Morse } 2056fbf269c7SJeremy Morse } 2057fbf269c7SJeremy Morse } 2058fbf269c7SJeremy Morse 2059fbf269c7SJeremy Morse if (AnyIllegal) { 206097ddf49eSJeremy Morse NormalLocsToPHI.insert(L); 2061fbf269c7SJeremy Morse continue; 2062fbf269c7SJeremy Morse } 2063fbf269c7SJeremy Morse 2064fbf269c7SJeremy Morse RegUnitsToPHIUp.insert(FoundRegUnits.begin(), FoundRegUnits.end()); 2065fbf269c7SJeremy Morse } 2066fbf269c7SJeremy Morse 2067fbf269c7SJeremy Morse // Lambda to fetch PHIs for a given location, and write into the PHIBlocks 2068fbf269c7SJeremy Morse // collection. 2069fbf269c7SJeremy Morse SmallVector<MachineBasicBlock *, 32> PHIBlocks; 2070fbf269c7SJeremy Morse auto CollectPHIsForLoc = [&](LocIdx L) { 2071fbf269c7SJeremy Morse // Collect the set of defs. 2072fbf269c7SJeremy Morse SmallPtrSet<MachineBasicBlock *, 32> DefBlocks; 2073fbf269c7SJeremy Morse for (unsigned int I = 0; I < OrderToBB.size(); ++I) { 2074fbf269c7SJeremy Morse MachineBasicBlock *MBB = OrderToBB[I]; 2075fbf269c7SJeremy Morse const auto &TransferFunc = MLocTransfer[MBB->getNumber()]; 2076fbf269c7SJeremy Morse if (TransferFunc.find(L) != TransferFunc.end()) 2077fbf269c7SJeremy Morse DefBlocks.insert(MBB); 2078fbf269c7SJeremy Morse } 2079fbf269c7SJeremy Morse 2080fbf269c7SJeremy Morse // The entry block defs the location too: it's the live-in / argument value. 2081fbf269c7SJeremy Morse // Only insert if there are other defs though; everything is trivially live 2082fbf269c7SJeremy Morse // through otherwise. 2083fbf269c7SJeremy Morse if (!DefBlocks.empty()) 2084fbf269c7SJeremy Morse DefBlocks.insert(&*MF.begin()); 2085fbf269c7SJeremy Morse 2086fbf269c7SJeremy Morse // Ask the SSA construction algorithm where we should put PHIs. Clear 2087fbf269c7SJeremy Morse // anything that might have been hanging around from earlier. 2088fbf269c7SJeremy Morse PHIBlocks.clear(); 2089fbf269c7SJeremy Morse BlockPHIPlacement(AllBlocks, DefBlocks, PHIBlocks); 2090fbf269c7SJeremy Morse }; 2091fbf269c7SJeremy Morse 209297ddf49eSJeremy Morse auto InstallPHIsAtLoc = [&PHIBlocks, &MInLocs](LocIdx L) { 2093fbf269c7SJeremy Morse for (const MachineBasicBlock *MBB : PHIBlocks) 2094fbf269c7SJeremy Morse MInLocs[MBB->getNumber()][L.asU64()] = ValueIDNum(MBB->getNumber(), 0, L); 209597ddf49eSJeremy Morse }; 209697ddf49eSJeremy Morse 209797ddf49eSJeremy Morse // For locations with no reg units, just place PHIs. 209897ddf49eSJeremy Morse for (LocIdx L : NormalLocsToPHI) { 209997ddf49eSJeremy Morse CollectPHIsForLoc(L); 210097ddf49eSJeremy Morse // Install those PHI values into the live-in value array. 210197ddf49eSJeremy Morse InstallPHIsAtLoc(L); 210297ddf49eSJeremy Morse } 210397ddf49eSJeremy Morse 210497ddf49eSJeremy Morse // For stack slots, calculate PHIs for the equivalent of the units, then 210597ddf49eSJeremy Morse // install for each index. 210697ddf49eSJeremy Morse for (SpillLocationNo Slot : StackSlots) { 210797ddf49eSJeremy Morse for (unsigned Idx : StackUnits) { 210897ddf49eSJeremy Morse unsigned SpillID = MTracker->getSpillIDWithIdx(Slot, Idx); 210997ddf49eSJeremy Morse LocIdx L = MTracker->getSpillMLoc(SpillID); 211097ddf49eSJeremy Morse CollectPHIsForLoc(L); 211197ddf49eSJeremy Morse InstallPHIsAtLoc(L); 211297ddf49eSJeremy Morse 211397ddf49eSJeremy Morse // Find anything that aliases this stack index, install PHIs for it too. 211497ddf49eSJeremy Morse unsigned Size, Offset; 211597ddf49eSJeremy Morse std::tie(Size, Offset) = MTracker->StackIdxesToPos[Idx]; 211697ddf49eSJeremy Morse for (auto &Pair : MTracker->StackSlotIdxes) { 211797ddf49eSJeremy Morse unsigned ThisSize, ThisOffset; 211897ddf49eSJeremy Morse std::tie(ThisSize, ThisOffset) = Pair.first; 211997ddf49eSJeremy Morse if (ThisSize + ThisOffset <= Offset || Size + Offset <= ThisOffset) 212097ddf49eSJeremy Morse continue; 212197ddf49eSJeremy Morse 212297ddf49eSJeremy Morse unsigned ThisID = MTracker->getSpillIDWithIdx(Slot, Pair.second); 212397ddf49eSJeremy Morse LocIdx ThisL = MTracker->getSpillMLoc(ThisID); 212497ddf49eSJeremy Morse InstallPHIsAtLoc(ThisL); 212597ddf49eSJeremy Morse } 212697ddf49eSJeremy Morse } 2127fbf269c7SJeremy Morse } 2128fbf269c7SJeremy Morse 2129fbf269c7SJeremy Morse // For reg units, place PHIs, and then place them for any aliasing registers. 2130fbf269c7SJeremy Morse for (Register R : RegUnitsToPHIUp) { 2131fbf269c7SJeremy Morse LocIdx L = MTracker->lookupOrTrackRegister(R); 2132fbf269c7SJeremy Morse CollectPHIsForLoc(L); 2133fbf269c7SJeremy Morse 2134fbf269c7SJeremy Morse // Install those PHI values into the live-in value array. 213597ddf49eSJeremy Morse InstallPHIsAtLoc(L); 2136fbf269c7SJeremy Morse 2137fbf269c7SJeremy Morse // Now find aliases and install PHIs for those. 2138fbf269c7SJeremy Morse for (MCRegAliasIterator RAI(R, TRI, true); RAI.isValid(); ++RAI) { 2139fbf269c7SJeremy Morse // Super-registers that are "above" the largest register read/written by 2140fbf269c7SJeremy Morse // the function will alias, but will not be tracked. 2141fbf269c7SJeremy Morse if (!MTracker->isRegisterTracked(*RAI)) 2142fbf269c7SJeremy Morse continue; 2143fbf269c7SJeremy Morse 2144fbf269c7SJeremy Morse LocIdx AliasLoc = MTracker->lookupOrTrackRegister(*RAI); 214597ddf49eSJeremy Morse InstallPHIsAtLoc(AliasLoc); 2146fbf269c7SJeremy Morse } 2147fbf269c7SJeremy Morse } 2148fbf269c7SJeremy Morse } 2149fbf269c7SJeremy Morse 2150a3936a6cSJeremy Morse void InstrRefBasedLDV::buildMLocValueMap( 2151ab49dce0SJeremy Morse MachineFunction &MF, FuncValueTable &MInLocs, FuncValueTable &MOutLocs, 2152ae6f7882SJeremy Morse SmallVectorImpl<MLocTransferMap> &MLocTransfer) { 2153ae6f7882SJeremy Morse std::priority_queue<unsigned int, std::vector<unsigned int>, 2154ae6f7882SJeremy Morse std::greater<unsigned int>> 2155ae6f7882SJeremy Morse Worklist, Pending; 2156ae6f7882SJeremy Morse 2157ae6f7882SJeremy Morse // We track what is on the current and pending worklist to avoid inserting 2158ae6f7882SJeremy Morse // the same thing twice. We could avoid this with a custom priority queue, 2159ae6f7882SJeremy Morse // but this is probably not worth it. 2160ae6f7882SJeremy Morse SmallPtrSet<MachineBasicBlock *, 16> OnPending, OnWorklist; 2161ae6f7882SJeremy Morse 2162a3936a6cSJeremy Morse // Initialize worklist with every block to be visited. Also produce list of 2163a3936a6cSJeremy Morse // all blocks. 2164a3936a6cSJeremy Morse SmallPtrSet<MachineBasicBlock *, 32> AllBlocks; 2165ae6f7882SJeremy Morse for (unsigned int I = 0; I < BBToOrder.size(); ++I) { 2166ae6f7882SJeremy Morse Worklist.push(I); 2167ae6f7882SJeremy Morse OnWorklist.insert(OrderToBB[I]); 2168a3936a6cSJeremy Morse AllBlocks.insert(OrderToBB[I]); 2169ae6f7882SJeremy Morse } 2170ae6f7882SJeremy Morse 2171a3936a6cSJeremy Morse // Initialize entry block to PHIs. These represent arguments. 2172a3936a6cSJeremy Morse for (auto Location : MTracker->locations()) 2173a3936a6cSJeremy Morse MInLocs[0][Location.Idx.asU64()] = ValueIDNum(0, 0, Location.Idx); 2174a3936a6cSJeremy Morse 2175ae6f7882SJeremy Morse MTracker->reset(); 2176ae6f7882SJeremy Morse 2177a3936a6cSJeremy Morse // Start by placing PHIs, using the usual SSA constructor algorithm. Consider 2178a3936a6cSJeremy Morse // any machine-location that isn't live-through a block to be def'd in that 2179a3936a6cSJeremy Morse // block. 2180fbf269c7SJeremy Morse placeMLocPHIs(MF, AllBlocks, MInLocs, MLocTransfer); 2181a3936a6cSJeremy Morse 2182a3936a6cSJeremy Morse // Propagate values to eliminate redundant PHIs. At the same time, this 2183a3936a6cSJeremy Morse // produces the table of Block x Location => Value for the entry to each 2184a3936a6cSJeremy Morse // block. 2185a3936a6cSJeremy Morse // The kind of PHIs we can eliminate are, for example, where one path in a 2186a3936a6cSJeremy Morse // conditional spills and restores a register, and the register still has 2187a3936a6cSJeremy Morse // the same value once control flow joins, unbeknowns to the PHI placement 2188a3936a6cSJeremy Morse // code. Propagating values allows us to identify such un-necessary PHIs and 2189a3936a6cSJeremy Morse // remove them. 2190ae6f7882SJeremy Morse SmallPtrSet<const MachineBasicBlock *, 16> Visited; 2191ae6f7882SJeremy Morse while (!Worklist.empty() || !Pending.empty()) { 2192ae6f7882SJeremy Morse // Vector for storing the evaluated block transfer function. 2193ae6f7882SJeremy Morse SmallVector<std::pair<LocIdx, ValueIDNum>, 32> ToRemap; 2194ae6f7882SJeremy Morse 2195ae6f7882SJeremy Morse while (!Worklist.empty()) { 2196ae6f7882SJeremy Morse MachineBasicBlock *MBB = OrderToBB[Worklist.top()]; 2197ae6f7882SJeremy Morse CurBB = MBB->getNumber(); 2198ae6f7882SJeremy Morse Worklist.pop(); 2199ae6f7882SJeremy Morse 2200ae6f7882SJeremy Morse // Join the values in all predecessor blocks. 2201a3936a6cSJeremy Morse bool InLocsChanged; 2202a3936a6cSJeremy Morse InLocsChanged = mlocJoin(*MBB, Visited, MOutLocs, MInLocs[CurBB]); 2203ae6f7882SJeremy Morse InLocsChanged |= Visited.insert(MBB).second; 2204ae6f7882SJeremy Morse 2205ae6f7882SJeremy Morse // Don't examine transfer function if we've visited this loc at least 2206ae6f7882SJeremy Morse // once, and inlocs haven't changed. 2207ae6f7882SJeremy Morse if (!InLocsChanged) 2208ae6f7882SJeremy Morse continue; 2209ae6f7882SJeremy Morse 2210ae6f7882SJeremy Morse // Load the current set of live-ins into MLocTracker. 2211ae6f7882SJeremy Morse MTracker->loadFromArray(MInLocs[CurBB], CurBB); 2212ae6f7882SJeremy Morse 2213ae6f7882SJeremy Morse // Each element of the transfer function can be a new def, or a read of 2214ae6f7882SJeremy Morse // a live-in value. Evaluate each element, and store to "ToRemap". 2215ae6f7882SJeremy Morse ToRemap.clear(); 2216ae6f7882SJeremy Morse for (auto &P : MLocTransfer[CurBB]) { 2217ae6f7882SJeremy Morse if (P.second.getBlock() == CurBB && P.second.isPHI()) { 2218ae6f7882SJeremy Morse // This is a movement of whatever was live in. Read it. 2219d9eebe3cSJeremy Morse ValueIDNum NewID = MTracker->readMLoc(P.second.getLoc()); 2220ae6f7882SJeremy Morse ToRemap.push_back(std::make_pair(P.first, NewID)); 2221ae6f7882SJeremy Morse } else { 2222ae6f7882SJeremy Morse // It's a def. Just set it. 2223ae6f7882SJeremy Morse assert(P.second.getBlock() == CurBB); 2224ae6f7882SJeremy Morse ToRemap.push_back(std::make_pair(P.first, P.second)); 2225ae6f7882SJeremy Morse } 2226ae6f7882SJeremy Morse } 2227ae6f7882SJeremy Morse 2228ae6f7882SJeremy Morse // Commit the transfer function changes into mloc tracker, which 2229ae6f7882SJeremy Morse // transforms the contents of the MLocTracker into the live-outs. 2230ae6f7882SJeremy Morse for (auto &P : ToRemap) 2231ae6f7882SJeremy Morse MTracker->setMLoc(P.first, P.second); 2232ae6f7882SJeremy Morse 2233ae6f7882SJeremy Morse // Now copy out-locs from mloc tracker into out-loc vector, checking 2234ae6f7882SJeremy Morse // whether changes have occurred. These changes can have come from both 2235ae6f7882SJeremy Morse // the transfer function, and mlocJoin. 2236ae6f7882SJeremy Morse bool OLChanged = false; 2237ae6f7882SJeremy Morse for (auto Location : MTracker->locations()) { 2238ae6f7882SJeremy Morse OLChanged |= MOutLocs[CurBB][Location.Idx.asU64()] != Location.Value; 2239ae6f7882SJeremy Morse MOutLocs[CurBB][Location.Idx.asU64()] = Location.Value; 2240ae6f7882SJeremy Morse } 2241ae6f7882SJeremy Morse 2242ae6f7882SJeremy Morse MTracker->reset(); 2243ae6f7882SJeremy Morse 2244ae6f7882SJeremy Morse // No need to examine successors again if out-locs didn't change. 2245ae6f7882SJeremy Morse if (!OLChanged) 2246ae6f7882SJeremy Morse continue; 2247ae6f7882SJeremy Morse 2248ae6f7882SJeremy Morse // All successors should be visited: put any back-edges on the pending 2249a3936a6cSJeremy Morse // list for the next pass-through, and any other successors to be 2250a3936a6cSJeremy Morse // visited this pass, if they're not going to be already. 2251ae6f7882SJeremy Morse for (auto s : MBB->successors()) { 2252ae6f7882SJeremy Morse // Does branching to this successor represent a back-edge? 2253ae6f7882SJeremy Morse if (BBToOrder[s] > BBToOrder[MBB]) { 2254ae6f7882SJeremy Morse // No: visit it during this dataflow iteration. 2255ae6f7882SJeremy Morse if (OnWorklist.insert(s).second) 2256ae6f7882SJeremy Morse Worklist.push(BBToOrder[s]); 2257ae6f7882SJeremy Morse } else { 2258ae6f7882SJeremy Morse // Yes: visit it on the next iteration. 2259ae6f7882SJeremy Morse if (OnPending.insert(s).second) 2260ae6f7882SJeremy Morse Pending.push(BBToOrder[s]); 2261ae6f7882SJeremy Morse } 2262ae6f7882SJeremy Morse } 2263ae6f7882SJeremy Morse } 2264ae6f7882SJeremy Morse 2265ae6f7882SJeremy Morse Worklist.swap(Pending); 2266ae6f7882SJeremy Morse std::swap(OnPending, OnWorklist); 2267ae6f7882SJeremy Morse OnPending.clear(); 2268ae6f7882SJeremy Morse // At this point, pending must be empty, since it was just the empty 2269ae6f7882SJeremy Morse // worklist 2270ae6f7882SJeremy Morse assert(Pending.empty() && "Pending should be empty"); 2271ae6f7882SJeremy Morse } 2272ae6f7882SJeremy Morse 2273a3936a6cSJeremy Morse // Once all the live-ins don't change on mlocJoin(), we've eliminated all 2274a3936a6cSJeremy Morse // redundant PHIs. 2275a3936a6cSJeremy Morse } 2276a3936a6cSJeremy Morse 2277a3936a6cSJeremy Morse void InstrRefBasedLDV::BlockPHIPlacement( 2278a3936a6cSJeremy Morse const SmallPtrSetImpl<MachineBasicBlock *> &AllBlocks, 2279a3936a6cSJeremy Morse const SmallPtrSetImpl<MachineBasicBlock *> &DefBlocks, 2280a3936a6cSJeremy Morse SmallVectorImpl<MachineBasicBlock *> &PHIBlocks) { 2281a3936a6cSJeremy Morse // Apply IDF calculator to the designated set of location defs, storing 2282a3936a6cSJeremy Morse // required PHIs into PHIBlocks. Uses the dominator tree stored in the 2283a3936a6cSJeremy Morse // InstrRefBasedLDV object. 2284e0b11c76SMarkus Böck IDFCalculatorBase<MachineBasicBlock, false> IDF(DomTree->getBase()); 2285a3936a6cSJeremy Morse 2286a3936a6cSJeremy Morse IDF.setLiveInBlocks(AllBlocks); 2287a3936a6cSJeremy Morse IDF.setDefiningBlocks(DefBlocks); 2288a3936a6cSJeremy Morse IDF.calculate(PHIBlocks); 2289ae6f7882SJeremy Morse } 2290ae6f7882SJeremy Morse 2291b5426cedSJeremy Morse Optional<ValueIDNum> InstrRefBasedLDV::pickVPHILoc( 2292b5426cedSJeremy Morse const MachineBasicBlock &MBB, const DebugVariable &Var, 2293ab49dce0SJeremy Morse const LiveIdxT &LiveOuts, FuncValueTable &MOutLocs, 2294b5426cedSJeremy Morse const SmallVectorImpl<const MachineBasicBlock *> &BlockOrders) { 2295ae6f7882SJeremy Morse // Collect a set of locations from predecessor where its live-out value can 2296ae6f7882SJeremy Morse // be found. 2297ae6f7882SJeremy Morse SmallVector<SmallVector<LocIdx, 4>, 8> Locs; 2298b5426cedSJeremy Morse SmallVector<const DbgValueProperties *, 4> Properties; 2299ae6f7882SJeremy Morse unsigned NumLocs = MTracker->getNumLocs(); 2300b5426cedSJeremy Morse 2301b5426cedSJeremy Morse // No predecessors means no PHIs. 2302b5426cedSJeremy Morse if (BlockOrders.empty()) 2303b5426cedSJeremy Morse return None; 2304ae6f7882SJeremy Morse 2305ae6f7882SJeremy Morse for (auto p : BlockOrders) { 2306ae6f7882SJeremy Morse unsigned ThisBBNum = p->getNumber(); 230789950adeSJeremy Morse auto OutValIt = LiveOuts.find(p); 230889950adeSJeremy Morse if (OutValIt == LiveOuts.end()) 230989950adeSJeremy Morse // If we have a predecessor not in scope, we'll never find a PHI position. 2310b5426cedSJeremy Morse return None; 231189950adeSJeremy Morse const DbgValue &OutVal = *OutValIt->second; 2312ae6f7882SJeremy Morse 2313ae6f7882SJeremy Morse if (OutVal.Kind == DbgValue::Const || OutVal.Kind == DbgValue::NoVal) 2314ae6f7882SJeremy Morse // Consts and no-values cannot have locations we can join on. 2315b5426cedSJeremy Morse return None; 2316ae6f7882SJeremy Morse 2317b5426cedSJeremy Morse Properties.push_back(&OutVal.Properties); 2318b5426cedSJeremy Morse 2319b5426cedSJeremy Morse // Create new empty vector of locations. 2320b5426cedSJeremy Morse Locs.resize(Locs.size() + 1); 2321b5426cedSJeremy Morse 2322b5426cedSJeremy Morse // If the live-in value is a def, find the locations where that value is 2323b5426cedSJeremy Morse // present. Do the same for VPHIs where we know the VPHI value. 2324b5426cedSJeremy Morse if (OutVal.Kind == DbgValue::Def || 2325b5426cedSJeremy Morse (OutVal.Kind == DbgValue::VPHI && OutVal.BlockNo != MBB.getNumber() && 2326b5426cedSJeremy Morse OutVal.ID != ValueIDNum::EmptyValue)) { 2327ae6f7882SJeremy Morse ValueIDNum ValToLookFor = OutVal.ID; 2328ae6f7882SJeremy Morse // Search the live-outs of the predecessor for the specified value. 2329ae6f7882SJeremy Morse for (unsigned int I = 0; I < NumLocs; ++I) { 2330ae6f7882SJeremy Morse if (MOutLocs[ThisBBNum][I] == ValToLookFor) 2331ae6f7882SJeremy Morse Locs.back().push_back(LocIdx(I)); 2332ae6f7882SJeremy Morse } 2333b5426cedSJeremy Morse } else { 2334b5426cedSJeremy Morse assert(OutVal.Kind == DbgValue::VPHI); 2335b5426cedSJeremy Morse // For VPHIs where we don't know the location, we definitely can't find 2336b5426cedSJeremy Morse // a join loc. 2337b5426cedSJeremy Morse if (OutVal.BlockNo != MBB.getNumber()) 2338b5426cedSJeremy Morse return None; 2339b5426cedSJeremy Morse 2340b5426cedSJeremy Morse // Otherwise: this is a VPHI on a backedge feeding back into itself, i.e. 2341b5426cedSJeremy Morse // a value that's live-through the whole loop. (It has to be a backedge, 2342b5426cedSJeremy Morse // because a block can't dominate itself). We can accept as a PHI location 2343b5426cedSJeremy Morse // any location where the other predecessors agree, _and_ the machine 2344b5426cedSJeremy Morse // locations feed back into themselves. Therefore, add all self-looping 2345b5426cedSJeremy Morse // machine-value PHI locations. 2346b5426cedSJeremy Morse for (unsigned int I = 0; I < NumLocs; ++I) { 2347b5426cedSJeremy Morse ValueIDNum MPHI(MBB.getNumber(), 0, LocIdx(I)); 2348b5426cedSJeremy Morse if (MOutLocs[ThisBBNum][I] == MPHI) 2349b5426cedSJeremy Morse Locs.back().push_back(LocIdx(I)); 2350b5426cedSJeremy Morse } 2351b5426cedSJeremy Morse } 2352ae6f7882SJeremy Morse } 2353ae6f7882SJeremy Morse 2354b5426cedSJeremy Morse // We should have found locations for all predecessors, or returned. 2355b5426cedSJeremy Morse assert(Locs.size() == BlockOrders.size()); 2356ae6f7882SJeremy Morse 2357b5426cedSJeremy Morse // Check that all properties are the same. We can't pick a location if they're 2358b5426cedSJeremy Morse // not. 2359b5426cedSJeremy Morse const DbgValueProperties *Properties0 = Properties[0]; 2360b5426cedSJeremy Morse for (auto *Prop : Properties) 2361b5426cedSJeremy Morse if (*Prop != *Properties0) 2362b5426cedSJeremy Morse return None; 2363b5426cedSJeremy Morse 2364ae6f7882SJeremy Morse // Starting with the first set of locations, take the intersection with 2365ae6f7882SJeremy Morse // subsequent sets. 2366b5426cedSJeremy Morse SmallVector<LocIdx, 4> CandidateLocs = Locs[0]; 2367b5426cedSJeremy Morse for (unsigned int I = 1; I < Locs.size(); ++I) { 2368b5426cedSJeremy Morse auto &LocVec = Locs[I]; 2369b5426cedSJeremy Morse SmallVector<LocIdx, 4> NewCandidates; 2370b5426cedSJeremy Morse std::set_intersection(CandidateLocs.begin(), CandidateLocs.end(), 2371b5426cedSJeremy Morse LocVec.begin(), LocVec.end(), std::inserter(NewCandidates, NewCandidates.begin())); 2372b5426cedSJeremy Morse CandidateLocs = NewCandidates; 2373ae6f7882SJeremy Morse } 2374b5426cedSJeremy Morse if (CandidateLocs.empty()) 2375ae6f7882SJeremy Morse return None; 2376ae6f7882SJeremy Morse 2377ae6f7882SJeremy Morse // We now have a set of LocIdxes that contain the right output value in 2378ae6f7882SJeremy Morse // each of the predecessors. Pick the lowest; if there's a register loc, 2379ae6f7882SJeremy Morse // that'll be it. 2380b5426cedSJeremy Morse LocIdx L = *CandidateLocs.begin(); 2381ae6f7882SJeremy Morse 2382ae6f7882SJeremy Morse // Return a PHI-value-number for the found location. 2383ae6f7882SJeremy Morse ValueIDNum PHIVal = {(unsigned)MBB.getNumber(), 0, L}; 2384b5426cedSJeremy Morse return PHIVal; 2385ae6f7882SJeremy Morse } 2386ae6f7882SJeremy Morse 2387b5426cedSJeremy Morse bool InstrRefBasedLDV::vlocJoin( 2388849b1794SJeremy Morse MachineBasicBlock &MBB, LiveIdxT &VLOCOutLocs, 2389ae6f7882SJeremy Morse SmallPtrSet<const MachineBasicBlock *, 8> &BlocksToExplore, 239089950adeSJeremy Morse DbgValue &LiveIn) { 2391ae6f7882SJeremy Morse LLVM_DEBUG(dbgs() << "join MBB: " << MBB.getNumber() << "\n"); 2392ae6f7882SJeremy Morse bool Changed = false; 2393ae6f7882SJeremy Morse 2394ae6f7882SJeremy Morse // Order predecessors by RPOT order, for exploring them in that order. 23957925aa09SKazu Hirata SmallVector<MachineBasicBlock *, 8> BlockOrders(MBB.predecessors()); 2396ae6f7882SJeremy Morse 2397ae6f7882SJeremy Morse auto Cmp = [&](MachineBasicBlock *A, MachineBasicBlock *B) { 2398ae6f7882SJeremy Morse return BBToOrder[A] < BBToOrder[B]; 2399ae6f7882SJeremy Morse }; 2400ae6f7882SJeremy Morse 24019bcc0d10SKazu Hirata llvm::sort(BlockOrders, Cmp); 2402ae6f7882SJeremy Morse 2403ae6f7882SJeremy Morse unsigned CurBlockRPONum = BBToOrder[&MBB]; 2404ae6f7882SJeremy Morse 2405b5426cedSJeremy Morse // Collect all the incoming DbgValues for this variable, from predecessor 2406b5426cedSJeremy Morse // live-out values. 2407ae6f7882SJeremy Morse SmallVector<InValueT, 8> Values; 2408ae6f7882SJeremy Morse bool Bail = false; 2409ea970661SJeremy Morse int BackEdgesStart = 0; 2410ae6f7882SJeremy Morse for (auto p : BlockOrders) { 2411ae6f7882SJeremy Morse // If the predecessor isn't in scope / to be explored, we'll never be 2412ae6f7882SJeremy Morse // able to join any locations. 2413805d5959SKazu Hirata if (!BlocksToExplore.contains(p)) { 2414ae6f7882SJeremy Morse Bail = true; 2415ae6f7882SJeremy Morse break; 2416ae6f7882SJeremy Morse } 2417ae6f7882SJeremy Morse 241889950adeSJeremy Morse // All Live-outs will have been initialized. 241989950adeSJeremy Morse DbgValue &OutLoc = *VLOCOutLocs.find(p)->second; 2420ae6f7882SJeremy Morse 2421ae6f7882SJeremy Morse // Keep track of where back-edges begin in the Values vector. Relies on 2422ae6f7882SJeremy Morse // BlockOrders being sorted by RPO. 2423ae6f7882SJeremy Morse unsigned ThisBBRPONum = BBToOrder[p]; 2424ae6f7882SJeremy Morse if (ThisBBRPONum < CurBlockRPONum) 2425ae6f7882SJeremy Morse ++BackEdgesStart; 2426ae6f7882SJeremy Morse 242789950adeSJeremy Morse Values.push_back(std::make_pair(p, &OutLoc)); 2428ae6f7882SJeremy Morse } 2429ae6f7882SJeremy Morse 2430ae6f7882SJeremy Morse // If there were no values, or one of the predecessors couldn't have a 2431ae6f7882SJeremy Morse // value, then give up immediately. It's not safe to produce a live-in 2432b5426cedSJeremy Morse // value. Leave as whatever it was before. 243389950adeSJeremy Morse if (Bail || Values.size() == 0) 243489950adeSJeremy Morse return false; 2435ae6f7882SJeremy Morse 2436ae6f7882SJeremy Morse // All (non-entry) blocks have at least one non-backedge predecessor. 2437ae6f7882SJeremy Morse // Pick the variable value from the first of these, to compare against 2438ae6f7882SJeremy Morse // all others. 2439ae6f7882SJeremy Morse const DbgValue &FirstVal = *Values[0].second; 2440ae6f7882SJeremy Morse 2441b5426cedSJeremy Morse // If the old live-in value is not a PHI then either a) no PHI is needed 2442b5426cedSJeremy Morse // here, or b) we eliminated the PHI that was here. If so, we can just 244389950adeSJeremy Morse // propagate in the first parent's incoming value. 244489950adeSJeremy Morse if (LiveIn.Kind != DbgValue::VPHI || LiveIn.BlockNo != MBB.getNumber()) { 244589950adeSJeremy Morse Changed = LiveIn != FirstVal; 244689950adeSJeremy Morse if (Changed) 244789950adeSJeremy Morse LiveIn = FirstVal; 244889950adeSJeremy Morse return Changed; 2449ae6f7882SJeremy Morse } 2450ae6f7882SJeremy Morse 2451b5426cedSJeremy Morse // Scan for variable values that can never be resolved: if they have 2452b5426cedSJeremy Morse // different DIExpressions, different indirectness, or are mixed constants / 2453b5426cedSJeremy Morse // non-constants. 2454b5426cedSJeremy Morse for (auto &V : Values) { 2455b5426cedSJeremy Morse if (V.second->Properties != FirstVal.Properties) 245689950adeSJeremy Morse return false; 2457b5426cedSJeremy Morse if (V.second->Kind == DbgValue::NoVal) 245889950adeSJeremy Morse return false; 2459b5426cedSJeremy Morse if (V.second->Kind == DbgValue::Const && FirstVal.Kind != DbgValue::Const) 246089950adeSJeremy Morse return false; 2461b5426cedSJeremy Morse } 2462b5426cedSJeremy Morse 2463b5426cedSJeremy Morse // Try to eliminate this PHI. Do the incoming values all agree? 2464ae6f7882SJeremy Morse bool Disagree = false; 2465ae6f7882SJeremy Morse for (auto &V : Values) { 2466ae6f7882SJeremy Morse if (*V.second == FirstVal) 2467ae6f7882SJeremy Morse continue; // No disagreement. 2468ae6f7882SJeremy Morse 2469b5426cedSJeremy Morse // Eliminate if a backedge feeds a VPHI back into itself. 2470b5426cedSJeremy Morse if (V.second->Kind == DbgValue::VPHI && 2471b5426cedSJeremy Morse V.second->BlockNo == MBB.getNumber() && 2472b5426cedSJeremy Morse // Is this a backedge? 2473b5426cedSJeremy Morse std::distance(Values.begin(), &V) >= BackEdgesStart) 2474b5426cedSJeremy Morse continue; 2475b5426cedSJeremy Morse 2476ae6f7882SJeremy Morse Disagree = true; 2477ae6f7882SJeremy Morse } 2478ae6f7882SJeremy Morse 2479b5426cedSJeremy Morse // No disagreement -> live-through value. 2480b5426cedSJeremy Morse if (!Disagree) { 248189950adeSJeremy Morse Changed = LiveIn != FirstVal; 248289950adeSJeremy Morse if (Changed) 248389950adeSJeremy Morse LiveIn = FirstVal; 248489950adeSJeremy Morse return Changed; 2485ae6f7882SJeremy Morse } else { 2486b5426cedSJeremy Morse // Otherwise use a VPHI. 248789950adeSJeremy Morse DbgValue VPHI(MBB.getNumber(), FirstVal.Properties, DbgValue::VPHI); 248889950adeSJeremy Morse Changed = LiveIn != VPHI; 248989950adeSJeremy Morse if (Changed) 249089950adeSJeremy Morse LiveIn = VPHI; 2491b5426cedSJeremy Morse return Changed; 2492ae6f7882SJeremy Morse } 249389950adeSJeremy Morse } 2494ae6f7882SJeremy Morse 24954a2cb013SJeremy Morse void InstrRefBasedLDV::getBlocksForScope( 24964a2cb013SJeremy Morse const DILocation *DILoc, 24974a2cb013SJeremy Morse SmallPtrSetImpl<const MachineBasicBlock *> &BlocksToExplore, 24984a2cb013SJeremy Morse const SmallPtrSetImpl<MachineBasicBlock *> &AssignBlocks) { 24994a2cb013SJeremy Morse // Get the set of "normal" in-lexical-scope blocks. 25004a2cb013SJeremy Morse LS.getMachineBasicBlocks(DILoc, BlocksToExplore); 25014a2cb013SJeremy Morse 25024a2cb013SJeremy Morse // VarLoc LiveDebugValues tracks variable locations that are defined in 25034a2cb013SJeremy Morse // blocks not in scope. This is something we could legitimately ignore, but 25044a2cb013SJeremy Morse // lets allow it for now for the sake of coverage. 25054a2cb013SJeremy Morse BlocksToExplore.insert(AssignBlocks.begin(), AssignBlocks.end()); 25064a2cb013SJeremy Morse 25074a2cb013SJeremy Morse // Storage for artificial blocks we intend to add to BlocksToExplore. 25084a2cb013SJeremy Morse DenseSet<const MachineBasicBlock *> ToAdd; 25094a2cb013SJeremy Morse 25104a2cb013SJeremy Morse // To avoid needlessly dropping large volumes of variable locations, propagate 25114a2cb013SJeremy Morse // variables through aritifical blocks, i.e. those that don't have any 25124a2cb013SJeremy Morse // instructions in scope at all. To accurately replicate VarLoc 25134a2cb013SJeremy Morse // LiveDebugValues, this means exploring all artificial successors too. 25144a2cb013SJeremy Morse // Perform a depth-first-search to enumerate those blocks. 25154a2cb013SJeremy Morse for (auto *MBB : BlocksToExplore) { 25164a2cb013SJeremy Morse // Depth-first-search state: each node is a block and which successor 25174a2cb013SJeremy Morse // we're currently exploring. 25184a2cb013SJeremy Morse SmallVector<std::pair<const MachineBasicBlock *, 25194a2cb013SJeremy Morse MachineBasicBlock::const_succ_iterator>, 25204a2cb013SJeremy Morse 8> 25214a2cb013SJeremy Morse DFS; 25224a2cb013SJeremy Morse 25234a2cb013SJeremy Morse // Find any artificial successors not already tracked. 25244a2cb013SJeremy Morse for (auto *succ : MBB->successors()) { 25254a2cb013SJeremy Morse if (BlocksToExplore.count(succ)) 25264a2cb013SJeremy Morse continue; 25274a2cb013SJeremy Morse if (!ArtificialBlocks.count(succ)) 25284a2cb013SJeremy Morse continue; 25294a2cb013SJeremy Morse ToAdd.insert(succ); 25304a2cb013SJeremy Morse DFS.push_back({succ, succ->succ_begin()}); 25314a2cb013SJeremy Morse } 25324a2cb013SJeremy Morse 25334a2cb013SJeremy Morse // Search all those blocks, depth first. 25344a2cb013SJeremy Morse while (!DFS.empty()) { 25354a2cb013SJeremy Morse const MachineBasicBlock *CurBB = DFS.back().first; 25364a2cb013SJeremy Morse MachineBasicBlock::const_succ_iterator &CurSucc = DFS.back().second; 25374a2cb013SJeremy Morse // Walk back if we've explored this blocks successors to the end. 25384a2cb013SJeremy Morse if (CurSucc == CurBB->succ_end()) { 25394a2cb013SJeremy Morse DFS.pop_back(); 25404a2cb013SJeremy Morse continue; 25414a2cb013SJeremy Morse } 25424a2cb013SJeremy Morse 25434a2cb013SJeremy Morse // If the current successor is artificial and unexplored, descend into 25444a2cb013SJeremy Morse // it. 25454a2cb013SJeremy Morse if (!ToAdd.count(*CurSucc) && ArtificialBlocks.count(*CurSucc)) { 25464a2cb013SJeremy Morse ToAdd.insert(*CurSucc); 25474a2cb013SJeremy Morse DFS.push_back({*CurSucc, (*CurSucc)->succ_begin()}); 25484a2cb013SJeremy Morse continue; 25494a2cb013SJeremy Morse } 25504a2cb013SJeremy Morse 25514a2cb013SJeremy Morse ++CurSucc; 25524a2cb013SJeremy Morse } 25534a2cb013SJeremy Morse }; 25544a2cb013SJeremy Morse 25554a2cb013SJeremy Morse BlocksToExplore.insert(ToAdd.begin(), ToAdd.end()); 25564a2cb013SJeremy Morse } 25574a2cb013SJeremy Morse 25584a2cb013SJeremy Morse void InstrRefBasedLDV::buildVLocValueMap( 25594a2cb013SJeremy Morse const DILocation *DILoc, const SmallSet<DebugVariable, 4> &VarsWeCareAbout, 2560ae6f7882SJeremy Morse SmallPtrSetImpl<MachineBasicBlock *> &AssignBlocks, LiveInsT &Output, 2561ab49dce0SJeremy Morse FuncValueTable &MOutLocs, FuncValueTable &MInLocs, 2562ae6f7882SJeremy Morse SmallVectorImpl<VLocTracker> &AllTheVLocs) { 2563b5426cedSJeremy Morse // This method is much like buildMLocValueMap: but focuses on a single 2564ae6f7882SJeremy Morse // LexicalScope at a time. Pick out a set of blocks and variables that are 2565ae6f7882SJeremy Morse // to have their value assignments solved, then run our dataflow algorithm 2566ae6f7882SJeremy Morse // until a fixedpoint is reached. 2567ae6f7882SJeremy Morse std::priority_queue<unsigned int, std::vector<unsigned int>, 2568ae6f7882SJeremy Morse std::greater<unsigned int>> 2569ae6f7882SJeremy Morse Worklist, Pending; 2570ae6f7882SJeremy Morse SmallPtrSet<MachineBasicBlock *, 16> OnWorklist, OnPending; 2571ae6f7882SJeremy Morse 2572ae6f7882SJeremy Morse // The set of blocks we'll be examining. 2573ae6f7882SJeremy Morse SmallPtrSet<const MachineBasicBlock *, 8> BlocksToExplore; 2574ae6f7882SJeremy Morse 2575ae6f7882SJeremy Morse // The order in which to examine them (RPO). 2576ae6f7882SJeremy Morse SmallVector<MachineBasicBlock *, 8> BlockOrders; 2577ae6f7882SJeremy Morse 2578ae6f7882SJeremy Morse // RPO ordering function. 2579ae6f7882SJeremy Morse auto Cmp = [&](MachineBasicBlock *A, MachineBasicBlock *B) { 2580ae6f7882SJeremy Morse return BBToOrder[A] < BBToOrder[B]; 2581ae6f7882SJeremy Morse }; 2582ae6f7882SJeremy Morse 25834a2cb013SJeremy Morse getBlocksForScope(DILoc, BlocksToExplore, AssignBlocks); 2584ae6f7882SJeremy Morse 2585ae6f7882SJeremy Morse // Single block scope: not interesting! No propagation at all. Note that 2586ae6f7882SJeremy Morse // this could probably go above ArtificialBlocks without damage, but 2587ae6f7882SJeremy Morse // that then produces output differences from original-live-debug-values, 2588ae6f7882SJeremy Morse // which propagates from a single block into many artificial ones. 2589ae6f7882SJeremy Morse if (BlocksToExplore.size() == 1) 2590ae6f7882SJeremy Morse return; 2591ae6f7882SJeremy Morse 259289950adeSJeremy Morse // Convert a const set to a non-const set. LexicalScopes 259389950adeSJeremy Morse // getMachineBasicBlocks returns const MBB pointers, IDF wants mutable ones. 259489950adeSJeremy Morse // (Neither of them mutate anything). 259589950adeSJeremy Morse SmallPtrSet<MachineBasicBlock *, 8> MutBlocksToExplore; 259689950adeSJeremy Morse for (const auto *MBB : BlocksToExplore) 259789950adeSJeremy Morse MutBlocksToExplore.insert(const_cast<MachineBasicBlock *>(MBB)); 259889950adeSJeremy Morse 2599ae6f7882SJeremy Morse // Picks out relevants blocks RPO order and sort them. 2600ae6f7882SJeremy Morse for (auto *MBB : BlocksToExplore) 2601ae6f7882SJeremy Morse BlockOrders.push_back(const_cast<MachineBasicBlock *>(MBB)); 2602ae6f7882SJeremy Morse 26039bcc0d10SKazu Hirata llvm::sort(BlockOrders, Cmp); 2604ae6f7882SJeremy Morse unsigned NumBlocks = BlockOrders.size(); 2605ae6f7882SJeremy Morse 2606ae6f7882SJeremy Morse // Allocate some vectors for storing the live ins and live outs. Large. 260789950adeSJeremy Morse SmallVector<DbgValue, 32> LiveIns, LiveOuts; 260889950adeSJeremy Morse LiveIns.reserve(NumBlocks); 260989950adeSJeremy Morse LiveOuts.reserve(NumBlocks); 2610ae6f7882SJeremy Morse 2611b5426cedSJeremy Morse // Initialize all values to start as NoVals. This signifies "it's live 2612b5426cedSJeremy Morse // through, but we don't know what it is". 2613b5426cedSJeremy Morse DbgValueProperties EmptyProperties(EmptyExpr, false); 261489950adeSJeremy Morse for (unsigned int I = 0; I < NumBlocks; ++I) { 261589950adeSJeremy Morse DbgValue EmptyDbgValue(I, EmptyProperties, DbgValue::NoVal); 261689950adeSJeremy Morse LiveIns.push_back(EmptyDbgValue); 261789950adeSJeremy Morse LiveOuts.push_back(EmptyDbgValue); 2618b5426cedSJeremy Morse } 2619b5426cedSJeremy Morse 2620ae6f7882SJeremy Morse // Produce by-MBB indexes of live-in/live-outs, to ease lookup within 2621ae6f7882SJeremy Morse // vlocJoin. 2622ae6f7882SJeremy Morse LiveIdxT LiveOutIdx, LiveInIdx; 2623ae6f7882SJeremy Morse LiveOutIdx.reserve(NumBlocks); 2624ae6f7882SJeremy Morse LiveInIdx.reserve(NumBlocks); 2625ae6f7882SJeremy Morse for (unsigned I = 0; I < NumBlocks; ++I) { 2626ae6f7882SJeremy Morse LiveOutIdx[BlockOrders[I]] = &LiveOuts[I]; 2627ae6f7882SJeremy Morse LiveInIdx[BlockOrders[I]] = &LiveIns[I]; 2628ae6f7882SJeremy Morse } 2629ae6f7882SJeremy Morse 263089950adeSJeremy Morse // Loop over each variable and place PHIs for it, then propagate values 263189950adeSJeremy Morse // between blocks. This keeps the locality of working on one lexical scope at 263289950adeSJeremy Morse // at time, but avoids re-processing variable values because some other 263389950adeSJeremy Morse // variable has been assigned. 263489950adeSJeremy Morse for (auto &Var : VarsWeCareAbout) { 263589950adeSJeremy Morse // Re-initialize live-ins and live-outs, to clear the remains of previous 263689950adeSJeremy Morse // variables live-ins / live-outs. 263789950adeSJeremy Morse for (unsigned int I = 0; I < NumBlocks; ++I) { 263889950adeSJeremy Morse DbgValue EmptyDbgValue(I, EmptyProperties, DbgValue::NoVal); 263989950adeSJeremy Morse LiveIns[I] = EmptyDbgValue; 264089950adeSJeremy Morse LiveOuts[I] = EmptyDbgValue; 2641849b1794SJeremy Morse } 2642849b1794SJeremy Morse 2643b5426cedSJeremy Morse // Place PHIs for variable values, using the LLVM IDF calculator. 2644b5426cedSJeremy Morse // Collect the set of blocks where variables are def'd. 2645b5426cedSJeremy Morse SmallPtrSet<MachineBasicBlock *, 32> DefBlocks; 2646b5426cedSJeremy Morse for (const MachineBasicBlock *ExpMBB : BlocksToExplore) { 2647b5426cedSJeremy Morse auto &TransferFunc = AllTheVLocs[ExpMBB->getNumber()].Vars; 2648b5426cedSJeremy Morse if (TransferFunc.find(Var) != TransferFunc.end()) 2649b5426cedSJeremy Morse DefBlocks.insert(const_cast<MachineBasicBlock *>(ExpMBB)); 2650b5426cedSJeremy Morse } 2651b5426cedSJeremy Morse 2652b5426cedSJeremy Morse SmallVector<MachineBasicBlock *, 32> PHIBlocks; 2653b5426cedSJeremy Morse 2654c703d77aSJeremy Morse // Request the set of PHIs we should insert for this variable. If there's 2655c703d77aSJeremy Morse // only one value definition, things are very simple. 2656c703d77aSJeremy Morse if (DefBlocks.size() == 1) { 2657c703d77aSJeremy Morse placePHIsForSingleVarDefinition(MutBlocksToExplore, *DefBlocks.begin(), 2658c703d77aSJeremy Morse AllTheVLocs, Var, Output); 2659c703d77aSJeremy Morse continue; 2660c703d77aSJeremy Morse } 2661c703d77aSJeremy Morse 2662c703d77aSJeremy Morse // Otherwise: we need to place PHIs through SSA and propagate values. 2663b5426cedSJeremy Morse BlockPHIPlacement(MutBlocksToExplore, DefBlocks, PHIBlocks); 2664b5426cedSJeremy Morse 2665b5426cedSJeremy Morse // Insert PHIs into the per-block live-in tables for this variable. 2666b5426cedSJeremy Morse for (MachineBasicBlock *PHIMBB : PHIBlocks) { 2667b5426cedSJeremy Morse unsigned BlockNo = PHIMBB->getNumber(); 266889950adeSJeremy Morse DbgValue *LiveIn = LiveInIdx[PHIMBB]; 266989950adeSJeremy Morse *LiveIn = DbgValue(BlockNo, EmptyProperties, DbgValue::VPHI); 2670b5426cedSJeremy Morse } 2671b5426cedSJeremy Morse 2672ae6f7882SJeremy Morse for (auto *MBB : BlockOrders) { 2673ae6f7882SJeremy Morse Worklist.push(BBToOrder[MBB]); 2674ae6f7882SJeremy Morse OnWorklist.insert(MBB); 2675ae6f7882SJeremy Morse } 2676ae6f7882SJeremy Morse 267789950adeSJeremy Morse // Iterate over all the blocks we selected, propagating the variables value. 267889950adeSJeremy Morse // This loop does two things: 2679b5426cedSJeremy Morse // * Eliminates un-necessary VPHIs in vlocJoin, 2680b5426cedSJeremy Morse // * Evaluates the blocks transfer function (i.e. variable assignments) and 2681b5426cedSJeremy Morse // stores the result to the blocks live-outs. 268289950adeSJeremy Morse // Always evaluate the transfer function on the first iteration, and when 268389950adeSJeremy Morse // the live-ins change thereafter. 2684ae6f7882SJeremy Morse bool FirstTrip = true; 2685ae6f7882SJeremy Morse while (!Worklist.empty() || !Pending.empty()) { 2686ae6f7882SJeremy Morse while (!Worklist.empty()) { 2687ae6f7882SJeremy Morse auto *MBB = OrderToBB[Worklist.top()]; 2688ae6f7882SJeremy Morse CurBB = MBB->getNumber(); 2689ae6f7882SJeremy Morse Worklist.pop(); 2690ae6f7882SJeremy Morse 269189950adeSJeremy Morse auto LiveInsIt = LiveInIdx.find(MBB); 269289950adeSJeremy Morse assert(LiveInsIt != LiveInIdx.end()); 269389950adeSJeremy Morse DbgValue *LiveIn = LiveInsIt->second; 2694ae6f7882SJeremy Morse 2695ae6f7882SJeremy Morse // Join values from predecessors. Updates LiveInIdx, and writes output 2696ae6f7882SJeremy Morse // into JoinedInLocs. 2697849b1794SJeremy Morse bool InLocsChanged = 26988dda516bSJeremy Morse vlocJoin(*MBB, LiveOutIdx, BlocksToExplore, *LiveIn); 2699ae6f7882SJeremy Morse 2700b5426cedSJeremy Morse SmallVector<const MachineBasicBlock *, 8> Preds; 2701b5426cedSJeremy Morse for (const auto *Pred : MBB->predecessors()) 2702b5426cedSJeremy Morse Preds.push_back(Pred); 2703ae6f7882SJeremy Morse 270489950adeSJeremy Morse // If this block's live-in value is a VPHI, try to pick a machine-value 270589950adeSJeremy Morse // for it. This makes the machine-value available and propagated 270689950adeSJeremy Morse // through all blocks by the time value propagation finishes. We can't 270789950adeSJeremy Morse // do this any earlier as it needs to read the block live-outs. 270889950adeSJeremy Morse if (LiveIn->Kind == DbgValue::VPHI && LiveIn->BlockNo == (int)CurBB) { 2709b5426cedSJeremy Morse // There's a small possibility that on a preceeding path, a VPHI is 2710b5426cedSJeremy Morse // eliminated and transitions from VPHI-with-location to 2711b5426cedSJeremy Morse // live-through-value. As a result, the selected location of any VPHI 2712b5426cedSJeremy Morse // might change, so we need to re-compute it on each iteration. 271389950adeSJeremy Morse Optional<ValueIDNum> ValueNum = 271489950adeSJeremy Morse pickVPHILoc(*MBB, Var, LiveOutIdx, MOutLocs, Preds); 2715ae6f7882SJeremy Morse 2716b5426cedSJeremy Morse if (ValueNum) { 271789950adeSJeremy Morse InLocsChanged |= LiveIn->ID != *ValueNum; 271889950adeSJeremy Morse LiveIn->ID = *ValueNum; 2719b5426cedSJeremy Morse } 2720b5426cedSJeremy Morse } 2721b5426cedSJeremy Morse 2722b5426cedSJeremy Morse if (!InLocsChanged && !FirstTrip) 2723ae6f7882SJeremy Morse continue; 2724ae6f7882SJeremy Morse 272589950adeSJeremy Morse DbgValue *LiveOut = LiveOutIdx[MBB]; 2726849b1794SJeremy Morse bool OLChanged = false; 2727849b1794SJeremy Morse 2728ae6f7882SJeremy Morse // Do transfer function. 2729ab93e710SJeremy Morse auto &VTracker = AllTheVLocs[MBB->getNumber()]; 273089950adeSJeremy Morse auto TransferIt = VTracker.Vars.find(Var); 273189950adeSJeremy Morse if (TransferIt != VTracker.Vars.end()) { 2732ae6f7882SJeremy Morse // Erase on empty transfer (DBG_VALUE $noreg). 273389950adeSJeremy Morse if (TransferIt->second.Kind == DbgValue::Undef) { 2734849b1794SJeremy Morse DbgValue NewVal(MBB->getNumber(), EmptyProperties, DbgValue::NoVal); 273589950adeSJeremy Morse if (*LiveOut != NewVal) { 273689950adeSJeremy Morse *LiveOut = NewVal; 2737849b1794SJeremy Morse OLChanged = true; 2738849b1794SJeremy Morse } 2739ae6f7882SJeremy Morse } else { 2740ae6f7882SJeremy Morse // Insert new variable value; or overwrite. 274189950adeSJeremy Morse if (*LiveOut != TransferIt->second) { 274289950adeSJeremy Morse *LiveOut = TransferIt->second; 2743849b1794SJeremy Morse OLChanged = true; 2744849b1794SJeremy Morse } 2745ae6f7882SJeremy Morse } 274689950adeSJeremy Morse } else { 274789950adeSJeremy Morse // Just copy live-ins to live-outs, for anything not transferred. 274889950adeSJeremy Morse if (*LiveOut != *LiveIn) { 274989950adeSJeremy Morse *LiveOut = *LiveIn; 2750849b1794SJeremy Morse OLChanged = true; 2751849b1794SJeremy Morse } 2752849b1794SJeremy Morse } 2753849b1794SJeremy Morse 2754849b1794SJeremy Morse // If no live-out value changed, there's no need to explore further. 2755849b1794SJeremy Morse if (!OLChanged) 2756849b1794SJeremy Morse continue; 2757ae6f7882SJeremy Morse 2758ae6f7882SJeremy Morse // We should visit all successors. Ensure we'll visit any non-backedge 2759ae6f7882SJeremy Morse // successors during this dataflow iteration; book backedge successors 2760ae6f7882SJeremy Morse // to be visited next time around. 2761ae6f7882SJeremy Morse for (auto s : MBB->successors()) { 2762ae6f7882SJeremy Morse // Ignore out of scope / not-to-be-explored successors. 2763ae6f7882SJeremy Morse if (LiveInIdx.find(s) == LiveInIdx.end()) 2764ae6f7882SJeremy Morse continue; 2765ae6f7882SJeremy Morse 2766ae6f7882SJeremy Morse if (BBToOrder[s] > BBToOrder[MBB]) { 2767ae6f7882SJeremy Morse if (OnWorklist.insert(s).second) 2768ae6f7882SJeremy Morse Worklist.push(BBToOrder[s]); 2769ae6f7882SJeremy Morse } else if (OnPending.insert(s).second && (FirstTrip || OLChanged)) { 2770ae6f7882SJeremy Morse Pending.push(BBToOrder[s]); 2771ae6f7882SJeremy Morse } 2772ae6f7882SJeremy Morse } 2773ae6f7882SJeremy Morse } 2774ae6f7882SJeremy Morse Worklist.swap(Pending); 2775ae6f7882SJeremy Morse std::swap(OnWorklist, OnPending); 2776ae6f7882SJeremy Morse OnPending.clear(); 2777ae6f7882SJeremy Morse assert(Pending.empty()); 2778ae6f7882SJeremy Morse FirstTrip = false; 2779ae6f7882SJeremy Morse } 2780ae6f7882SJeremy Morse 2781b5426cedSJeremy Morse // Save live-ins to output vector. Ignore any that are still marked as being 2782b5426cedSJeremy Morse // VPHIs with no location -- those are variables that we know the value of, 2783b5426cedSJeremy Morse // but are not actually available in the register file. 2784ae6f7882SJeremy Morse for (auto *MBB : BlockOrders) { 278589950adeSJeremy Morse DbgValue *BlockLiveIn = LiveInIdx[MBB]; 278689950adeSJeremy Morse if (BlockLiveIn->Kind == DbgValue::NoVal) 2787ae6f7882SJeremy Morse continue; 278889950adeSJeremy Morse if (BlockLiveIn->Kind == DbgValue::VPHI && 278989950adeSJeremy Morse BlockLiveIn->ID == ValueIDNum::EmptyValue) 2790b5426cedSJeremy Morse continue; 279189950adeSJeremy Morse if (BlockLiveIn->Kind == DbgValue::VPHI) 279289950adeSJeremy Morse BlockLiveIn->Kind = DbgValue::Def; 27938dda516bSJeremy Morse assert(BlockLiveIn->Properties.DIExpr->getFragmentInfo() == 27948dda516bSJeremy Morse Var.getFragment() && "Fragment info missing during value prop"); 279589950adeSJeremy Morse Output[MBB->getNumber()].push_back(std::make_pair(Var, *BlockLiveIn)); 2796ae6f7882SJeremy Morse } 279789950adeSJeremy Morse } // Per-variable loop. 2798ae6f7882SJeremy Morse 2799ae6f7882SJeremy Morse BlockOrders.clear(); 2800ae6f7882SJeremy Morse BlocksToExplore.clear(); 2801ae6f7882SJeremy Morse } 2802ae6f7882SJeremy Morse 2803c703d77aSJeremy Morse void InstrRefBasedLDV::placePHIsForSingleVarDefinition( 2804c703d77aSJeremy Morse const SmallPtrSetImpl<MachineBasicBlock *> &InScopeBlocks, 2805c703d77aSJeremy Morse MachineBasicBlock *AssignMBB, SmallVectorImpl<VLocTracker> &AllTheVLocs, 2806c703d77aSJeremy Morse const DebugVariable &Var, LiveInsT &Output) { 2807c703d77aSJeremy Morse // If there is a single definition of the variable, then working out it's 2808c703d77aSJeremy Morse // value everywhere is very simple: it's every block dominated by the 2809c703d77aSJeremy Morse // definition. At the dominance frontier, the usual algorithm would: 2810c703d77aSJeremy Morse // * Place PHIs, 2811c703d77aSJeremy Morse // * Propagate values into them, 2812c703d77aSJeremy Morse // * Find there's no incoming variable value from the other incoming branches 2813c703d77aSJeremy Morse // of the dominance frontier, 2814c703d77aSJeremy Morse // * Specify there's no variable value in blocks past the frontier. 2815c703d77aSJeremy Morse // This is a common case, hence it's worth special-casing it. 2816c703d77aSJeremy Morse 2817c703d77aSJeremy Morse // Pick out the variables value from the block transfer function. 2818c703d77aSJeremy Morse VLocTracker &VLocs = AllTheVLocs[AssignMBB->getNumber()]; 2819c703d77aSJeremy Morse auto ValueIt = VLocs.Vars.find(Var); 2820c703d77aSJeremy Morse const DbgValue &Value = ValueIt->second; 2821c703d77aSJeremy Morse 282243de3057SJeremy Morse // If it's an explicit assignment of "undef", that means there is no location 282343de3057SJeremy Morse // anyway, anywhere. 282443de3057SJeremy Morse if (Value.Kind == DbgValue::Undef) 282543de3057SJeremy Morse return; 282643de3057SJeremy Morse 2827c703d77aSJeremy Morse // Assign the variable value to entry to each dominated block that's in scope. 2828c703d77aSJeremy Morse // Skip the definition block -- it's assigned the variable value in the middle 2829c703d77aSJeremy Morse // of the block somewhere. 2830c703d77aSJeremy Morse for (auto *ScopeBlock : InScopeBlocks) { 2831c703d77aSJeremy Morse if (!DomTree->properlyDominates(AssignMBB, ScopeBlock)) 2832c703d77aSJeremy Morse continue; 2833c703d77aSJeremy Morse 2834c703d77aSJeremy Morse Output[ScopeBlock->getNumber()].push_back({Var, Value}); 2835c703d77aSJeremy Morse } 2836c703d77aSJeremy Morse 2837c703d77aSJeremy Morse // All blocks that aren't dominated have no live-in value, thus no variable 2838c703d77aSJeremy Morse // value will be given to them. 2839c703d77aSJeremy Morse } 2840c703d77aSJeremy Morse 28412ac06241SFangrui Song #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 2842ae6f7882SJeremy Morse void InstrRefBasedLDV::dump_mloc_transfer( 2843ae6f7882SJeremy Morse const MLocTransferMap &mloc_transfer) const { 2844ae6f7882SJeremy Morse for (auto &P : mloc_transfer) { 2845ae6f7882SJeremy Morse std::string foo = MTracker->LocIdxToName(P.first); 2846ae6f7882SJeremy Morse std::string bar = MTracker->IDAsString(P.second); 2847ae6f7882SJeremy Morse dbgs() << "Loc " << foo << " --> " << bar << "\n"; 2848ae6f7882SJeremy Morse } 2849ae6f7882SJeremy Morse } 28502ac06241SFangrui Song #endif 2851ae6f7882SJeremy Morse 2852ae6f7882SJeremy Morse void InstrRefBasedLDV::initialSetup(MachineFunction &MF) { 2853ae6f7882SJeremy Morse // Build some useful data structures. 2854b5426cedSJeremy Morse 2855b5426cedSJeremy Morse LLVMContext &Context = MF.getFunction().getContext(); 2856b5426cedSJeremy Morse EmptyExpr = DIExpression::get(Context, {}); 2857b5426cedSJeremy Morse 2858ae6f7882SJeremy Morse auto hasNonArtificialLocation = [](const MachineInstr &MI) -> bool { 2859ae6f7882SJeremy Morse if (const DebugLoc &DL = MI.getDebugLoc()) 2860ae6f7882SJeremy Morse return DL.getLine() != 0; 2861ae6f7882SJeremy Morse return false; 2862ae6f7882SJeremy Morse }; 2863ae6f7882SJeremy Morse // Collect a set of all the artificial blocks. 2864ae6f7882SJeremy Morse for (auto &MBB : MF) 2865ae6f7882SJeremy Morse if (none_of(MBB.instrs(), hasNonArtificialLocation)) 2866ae6f7882SJeremy Morse ArtificialBlocks.insert(&MBB); 2867ae6f7882SJeremy Morse 2868ae6f7882SJeremy Morse // Compute mappings of block <=> RPO order. 2869ae6f7882SJeremy Morse ReversePostOrderTraversal<MachineFunction *> RPOT(&MF); 2870ae6f7882SJeremy Morse unsigned int RPONumber = 0; 2871d5adba10SKazu Hirata for (MachineBasicBlock *MBB : RPOT) { 2872d5adba10SKazu Hirata OrderToBB[RPONumber] = MBB; 2873d5adba10SKazu Hirata BBToOrder[MBB] = RPONumber; 2874d5adba10SKazu Hirata BBNumToRPO[MBB->getNumber()] = RPONumber; 2875ae6f7882SJeremy Morse ++RPONumber; 2876ae6f7882SJeremy Morse } 2877f551fb96SJeremy Morse 2878f551fb96SJeremy Morse // Order value substitutions by their "source" operand pair, for quick lookup. 2879f551fb96SJeremy Morse llvm::sort(MF.DebugValueSubstitutions); 2880f551fb96SJeremy Morse 2881f551fb96SJeremy Morse #ifdef EXPENSIVE_CHECKS 2882f551fb96SJeremy Morse // As an expensive check, test whether there are any duplicate substitution 2883f551fb96SJeremy Morse // sources in the collection. 2884f551fb96SJeremy Morse if (MF.DebugValueSubstitutions.size() > 2) { 2885f551fb96SJeremy Morse for (auto It = MF.DebugValueSubstitutions.begin(); 2886f551fb96SJeremy Morse It != std::prev(MF.DebugValueSubstitutions.end()); ++It) { 2887f551fb96SJeremy Morse assert(It->Src != std::next(It)->Src && "Duplicate variable location " 2888f551fb96SJeremy Morse "substitution seen"); 2889f551fb96SJeremy Morse } 2890f551fb96SJeremy Morse } 2891f551fb96SJeremy Morse #endif 2892ae6f7882SJeremy Morse } 2893ae6f7882SJeremy Morse 28949fd9d56dSJeremy Morse // Produce an "ejection map" for blocks, i.e., what's the highest-numbered 28959fd9d56dSJeremy Morse // lexical scope it's used in. When exploring in DFS order and we pass that 28969fd9d56dSJeremy Morse // scope, the block can be processed and any tracking information freed. 28979fd9d56dSJeremy Morse void InstrRefBasedLDV::makeDepthFirstEjectionMap( 28989fd9d56dSJeremy Morse SmallVectorImpl<unsigned> &EjectionMap, 28999fd9d56dSJeremy Morse const ScopeToDILocT &ScopeToDILocation, 29009fd9d56dSJeremy Morse ScopeToAssignBlocksT &ScopeToAssignBlocks) { 29019fd9d56dSJeremy Morse SmallPtrSet<const MachineBasicBlock *, 8> BlocksToExplore; 29029fd9d56dSJeremy Morse SmallVector<std::pair<LexicalScope *, ssize_t>, 4> WorkStack; 29039fd9d56dSJeremy Morse auto *TopScope = LS.getCurrentFunctionScope(); 29049fd9d56dSJeremy Morse 29059fd9d56dSJeremy Morse // Unlike lexical scope explorers, we explore in reverse order, to find the 29069fd9d56dSJeremy Morse // "last" lexical scope used for each block early. 29079fd9d56dSJeremy Morse WorkStack.push_back({TopScope, TopScope->getChildren().size() - 1}); 29089fd9d56dSJeremy Morse 29099fd9d56dSJeremy Morse while (!WorkStack.empty()) { 29109fd9d56dSJeremy Morse auto &ScopePosition = WorkStack.back(); 29119fd9d56dSJeremy Morse LexicalScope *WS = ScopePosition.first; 29129fd9d56dSJeremy Morse ssize_t ChildNum = ScopePosition.second--; 29139fd9d56dSJeremy Morse 29149fd9d56dSJeremy Morse const SmallVectorImpl<LexicalScope *> &Children = WS->getChildren(); 29159fd9d56dSJeremy Morse if (ChildNum >= 0) { 29169fd9d56dSJeremy Morse // If ChildNum is positive, there are remaining children to explore. 29179fd9d56dSJeremy Morse // Push the child and its children-count onto the stack. 29189fd9d56dSJeremy Morse auto &ChildScope = Children[ChildNum]; 29199fd9d56dSJeremy Morse WorkStack.push_back( 29209fd9d56dSJeremy Morse std::make_pair(ChildScope, ChildScope->getChildren().size() - 1)); 29219fd9d56dSJeremy Morse } else { 29229fd9d56dSJeremy Morse WorkStack.pop_back(); 29239fd9d56dSJeremy Morse 29249fd9d56dSJeremy Morse // We've explored all children and any later blocks: examine all blocks 29259fd9d56dSJeremy Morse // in our scope. If they haven't yet had an ejection number set, then 29269fd9d56dSJeremy Morse // this scope will be the last to use that block. 29279fd9d56dSJeremy Morse auto DILocationIt = ScopeToDILocation.find(WS); 29289fd9d56dSJeremy Morse if (DILocationIt != ScopeToDILocation.end()) { 29299fd9d56dSJeremy Morse getBlocksForScope(DILocationIt->second, BlocksToExplore, 29309fd9d56dSJeremy Morse ScopeToAssignBlocks.find(WS)->second); 29319fd9d56dSJeremy Morse for (auto *MBB : BlocksToExplore) { 29329fd9d56dSJeremy Morse unsigned BBNum = MBB->getNumber(); 29339fd9d56dSJeremy Morse if (EjectionMap[BBNum] == 0) 29349fd9d56dSJeremy Morse EjectionMap[BBNum] = WS->getDFSOut(); 29359fd9d56dSJeremy Morse } 29369fd9d56dSJeremy Morse 29379fd9d56dSJeremy Morse BlocksToExplore.clear(); 29389fd9d56dSJeremy Morse } 29399fd9d56dSJeremy Morse } 29409fd9d56dSJeremy Morse } 29419fd9d56dSJeremy Morse } 29429fd9d56dSJeremy Morse 29439fd9d56dSJeremy Morse bool InstrRefBasedLDV::depthFirstVLocAndEmit( 29449fd9d56dSJeremy Morse unsigned MaxNumBlocks, const ScopeToDILocT &ScopeToDILocation, 29459fd9d56dSJeremy Morse const ScopeToVarsT &ScopeToVars, ScopeToAssignBlocksT &ScopeToAssignBlocks, 2946ab49dce0SJeremy Morse LiveInsT &Output, FuncValueTable &MOutLocs, FuncValueTable &MInLocs, 29479fd9d56dSJeremy Morse SmallVectorImpl<VLocTracker> &AllTheVLocs, MachineFunction &MF, 29489fd9d56dSJeremy Morse DenseMap<DebugVariable, unsigned> &AllVarsNumbering, 29499fd9d56dSJeremy Morse const TargetPassConfig &TPC) { 29509fd9d56dSJeremy Morse TTracker = new TransferTracker(TII, MTracker, MF, *TRI, CalleeSavedRegs, TPC); 29519fd9d56dSJeremy Morse unsigned NumLocs = MTracker->getNumLocs(); 29529fd9d56dSJeremy Morse VTracker = nullptr; 29539fd9d56dSJeremy Morse 29549fd9d56dSJeremy Morse // No scopes? No variable locations. 2955ab49dce0SJeremy Morse if (!LS.getCurrentFunctionScope()) 29569fd9d56dSJeremy Morse return false; 29579fd9d56dSJeremy Morse 29589fd9d56dSJeremy Morse // Build map from block number to the last scope that uses the block. 29599fd9d56dSJeremy Morse SmallVector<unsigned, 16> EjectionMap; 29609fd9d56dSJeremy Morse EjectionMap.resize(MaxNumBlocks, 0); 29619fd9d56dSJeremy Morse makeDepthFirstEjectionMap(EjectionMap, ScopeToDILocation, 29629fd9d56dSJeremy Morse ScopeToAssignBlocks); 29639fd9d56dSJeremy Morse 29649fd9d56dSJeremy Morse // Helper lambda for ejecting a block -- if nothing is going to use the block, 29659fd9d56dSJeremy Morse // we can translate the variable location information into DBG_VALUEs and then 29669fd9d56dSJeremy Morse // free all of InstrRefBasedLDV's data structures. 29679fd9d56dSJeremy Morse auto EjectBlock = [&](MachineBasicBlock &MBB) -> void { 29689fd9d56dSJeremy Morse unsigned BBNum = MBB.getNumber(); 29699fd9d56dSJeremy Morse AllTheVLocs[BBNum].clear(); 29709fd9d56dSJeremy Morse 29719fd9d56dSJeremy Morse // Prime the transfer-tracker, and then step through all the block 29729fd9d56dSJeremy Morse // instructions, installing transfers. 29739fd9d56dSJeremy Morse MTracker->reset(); 29749fd9d56dSJeremy Morse MTracker->loadFromArray(MInLocs[BBNum], BBNum); 29759fd9d56dSJeremy Morse TTracker->loadInlocs(MBB, MInLocs[BBNum], Output[BBNum], NumLocs); 29769fd9d56dSJeremy Morse 29779fd9d56dSJeremy Morse CurBB = BBNum; 29789fd9d56dSJeremy Morse CurInst = 1; 29799fd9d56dSJeremy Morse for (auto &MI : MBB) { 2980ab49dce0SJeremy Morse process(MI, MOutLocs.get(), MInLocs.get()); 29819fd9d56dSJeremy Morse TTracker->checkInstForNewValues(CurInst, MI.getIterator()); 29829fd9d56dSJeremy Morse ++CurInst; 29839fd9d56dSJeremy Morse } 29849fd9d56dSJeremy Morse 29859fd9d56dSJeremy Morse // Free machine-location tables for this block. 2986ab49dce0SJeremy Morse MInLocs[BBNum].reset(); 2987ab49dce0SJeremy Morse MOutLocs[BBNum].reset(); 29889fd9d56dSJeremy Morse // We don't need live-in variable values for this block either. 29899fd9d56dSJeremy Morse Output[BBNum].clear(); 29909fd9d56dSJeremy Morse AllTheVLocs[BBNum].clear(); 29919fd9d56dSJeremy Morse }; 29929fd9d56dSJeremy Morse 29939fd9d56dSJeremy Morse SmallPtrSet<const MachineBasicBlock *, 8> BlocksToExplore; 29949fd9d56dSJeremy Morse SmallVector<std::pair<LexicalScope *, ssize_t>, 4> WorkStack; 29959fd9d56dSJeremy Morse WorkStack.push_back({LS.getCurrentFunctionScope(), 0}); 29969fd9d56dSJeremy Morse unsigned HighestDFSIn = 0; 29979fd9d56dSJeremy Morse 29989fd9d56dSJeremy Morse // Proceed to explore in depth first order. 29999fd9d56dSJeremy Morse while (!WorkStack.empty()) { 30009fd9d56dSJeremy Morse auto &ScopePosition = WorkStack.back(); 30019fd9d56dSJeremy Morse LexicalScope *WS = ScopePosition.first; 30029fd9d56dSJeremy Morse ssize_t ChildNum = ScopePosition.second++; 30039fd9d56dSJeremy Morse 30049fd9d56dSJeremy Morse // We obesrve scopes with children twice here, once descending in, once 30059fd9d56dSJeremy Morse // ascending out of the scope nest. Use HighestDFSIn as a ratchet to ensure 30069fd9d56dSJeremy Morse // we don't process a scope twice. Additionally, ignore scopes that don't 30079fd9d56dSJeremy Morse // have a DILocation -- by proxy, this means we never tracked any variable 30089fd9d56dSJeremy Morse // assignments in that scope. 30099fd9d56dSJeremy Morse auto DILocIt = ScopeToDILocation.find(WS); 30109fd9d56dSJeremy Morse if (HighestDFSIn <= WS->getDFSIn() && DILocIt != ScopeToDILocation.end()) { 30119fd9d56dSJeremy Morse const DILocation *DILoc = DILocIt->second; 30129fd9d56dSJeremy Morse auto &VarsWeCareAbout = ScopeToVars.find(WS)->second; 30139fd9d56dSJeremy Morse auto &BlocksInScope = ScopeToAssignBlocks.find(WS)->second; 30149fd9d56dSJeremy Morse 30159fd9d56dSJeremy Morse buildVLocValueMap(DILoc, VarsWeCareAbout, BlocksInScope, Output, MOutLocs, 30169fd9d56dSJeremy Morse MInLocs, AllTheVLocs); 30179fd9d56dSJeremy Morse } 30189fd9d56dSJeremy Morse 30199fd9d56dSJeremy Morse HighestDFSIn = std::max(HighestDFSIn, WS->getDFSIn()); 30209fd9d56dSJeremy Morse 30219fd9d56dSJeremy Morse // Descend into any scope nests. 30229fd9d56dSJeremy Morse const SmallVectorImpl<LexicalScope *> &Children = WS->getChildren(); 30239fd9d56dSJeremy Morse if (ChildNum < (ssize_t)Children.size()) { 30249fd9d56dSJeremy Morse // There are children to explore -- push onto stack and continue. 30259fd9d56dSJeremy Morse auto &ChildScope = Children[ChildNum]; 30269fd9d56dSJeremy Morse WorkStack.push_back(std::make_pair(ChildScope, 0)); 30279fd9d56dSJeremy Morse } else { 30289fd9d56dSJeremy Morse WorkStack.pop_back(); 30299fd9d56dSJeremy Morse 30309fd9d56dSJeremy Morse // We've explored a leaf, or have explored all the children of a scope. 30319fd9d56dSJeremy Morse // Try to eject any blocks where this is the last scope it's relevant to. 30329fd9d56dSJeremy Morse auto DILocationIt = ScopeToDILocation.find(WS); 30339fd9d56dSJeremy Morse if (DILocationIt == ScopeToDILocation.end()) 30349fd9d56dSJeremy Morse continue; 30359fd9d56dSJeremy Morse 30369fd9d56dSJeremy Morse getBlocksForScope(DILocationIt->second, BlocksToExplore, 30379fd9d56dSJeremy Morse ScopeToAssignBlocks.find(WS)->second); 30389fd9d56dSJeremy Morse for (auto *MBB : BlocksToExplore) 30399fd9d56dSJeremy Morse if (WS->getDFSOut() == EjectionMap[MBB->getNumber()]) 30409fd9d56dSJeremy Morse EjectBlock(const_cast<MachineBasicBlock &>(*MBB)); 30419fd9d56dSJeremy Morse 30429fd9d56dSJeremy Morse BlocksToExplore.clear(); 30439fd9d56dSJeremy Morse } 30449fd9d56dSJeremy Morse } 30459fd9d56dSJeremy Morse 30469fd9d56dSJeremy Morse // Some artificial blocks may not have been ejected, meaning they're not 30479fd9d56dSJeremy Morse // connected to an actual legitimate scope. This can technically happen 30489fd9d56dSJeremy Morse // with things like the entry block. In theory, we shouldn't need to do 30499fd9d56dSJeremy Morse // anything for such out-of-scope blocks, but for the sake of being similar 30509fd9d56dSJeremy Morse // to VarLocBasedLDV, eject these too. 30519fd9d56dSJeremy Morse for (auto *MBB : ArtificialBlocks) 30529fd9d56dSJeremy Morse if (MOutLocs[MBB->getNumber()]) 30539fd9d56dSJeremy Morse EjectBlock(*MBB); 30549fd9d56dSJeremy Morse 30559fd9d56dSJeremy Morse return emitTransfers(AllVarsNumbering); 30569fd9d56dSJeremy Morse } 30579fd9d56dSJeremy Morse 30584a2cb013SJeremy Morse bool InstrRefBasedLDV::emitTransfers( 30594a2cb013SJeremy Morse DenseMap<DebugVariable, unsigned> &AllVarsNumbering) { 30604a2cb013SJeremy Morse // Go through all the transfers recorded in the TransferTracker -- this is 30614a2cb013SJeremy Morse // both the live-ins to a block, and any movements of values that happen 30624a2cb013SJeremy Morse // in the middle. 30634a2cb013SJeremy Morse for (const auto &P : TTracker->Transfers) { 30644a2cb013SJeremy Morse // We have to insert DBG_VALUEs in a consistent order, otherwise they 30654a2cb013SJeremy Morse // appear in DWARF in different orders. Use the order that they appear 30664a2cb013SJeremy Morse // when walking through each block / each instruction, stored in 30674a2cb013SJeremy Morse // AllVarsNumbering. 30684a2cb013SJeremy Morse SmallVector<std::pair<unsigned, MachineInstr *>> Insts; 30694a2cb013SJeremy Morse for (MachineInstr *MI : P.Insts) { 30704a2cb013SJeremy Morse DebugVariable Var(MI->getDebugVariable(), MI->getDebugExpression(), 30714a2cb013SJeremy Morse MI->getDebugLoc()->getInlinedAt()); 30724a2cb013SJeremy Morse Insts.emplace_back(AllVarsNumbering.find(Var)->second, MI); 30734a2cb013SJeremy Morse } 30744a2cb013SJeremy Morse llvm::sort(Insts, 30754a2cb013SJeremy Morse [](const auto &A, const auto &B) { return A.first < B.first; }); 30764a2cb013SJeremy Morse 30774a2cb013SJeremy Morse // Insert either before or after the designated point... 30784a2cb013SJeremy Morse if (P.MBB) { 30794a2cb013SJeremy Morse MachineBasicBlock &MBB = *P.MBB; 30804a2cb013SJeremy Morse for (const auto &Pair : Insts) 30814a2cb013SJeremy Morse MBB.insert(P.Pos, Pair.second); 30824a2cb013SJeremy Morse } else { 30834a2cb013SJeremy Morse // Terminators, like tail calls, can clobber things. Don't try and place 30844a2cb013SJeremy Morse // transfers after them. 30854a2cb013SJeremy Morse if (P.Pos->isTerminator()) 30864a2cb013SJeremy Morse continue; 30874a2cb013SJeremy Morse 30884a2cb013SJeremy Morse MachineBasicBlock &MBB = *P.Pos->getParent(); 30894a2cb013SJeremy Morse for (const auto &Pair : Insts) 30904a2cb013SJeremy Morse MBB.insertAfterBundle(P.Pos, Pair.second); 30914a2cb013SJeremy Morse } 30924a2cb013SJeremy Morse } 30934a2cb013SJeremy Morse 30944a2cb013SJeremy Morse return TTracker->Transfers.size() != 0; 30954a2cb013SJeremy Morse } 30964a2cb013SJeremy Morse 3097ae6f7882SJeremy Morse /// Calculate the liveness information for the given machine function and 3098ae6f7882SJeremy Morse /// extend ranges across basic blocks. 3099a3936a6cSJeremy Morse bool InstrRefBasedLDV::ExtendRanges(MachineFunction &MF, 3100a3936a6cSJeremy Morse MachineDominatorTree *DomTree, 3101a3936a6cSJeremy Morse TargetPassConfig *TPC, 3102708cbda5SJeremy Morse unsigned InputBBLimit, 3103708cbda5SJeremy Morse unsigned InputDbgValLimit) { 3104ae6f7882SJeremy Morse // No subprogram means this function contains no debuginfo. 3105ae6f7882SJeremy Morse if (!MF.getFunction().getSubprogram()) 3106ae6f7882SJeremy Morse return false; 3107ae6f7882SJeremy Morse 3108ae6f7882SJeremy Morse LLVM_DEBUG(dbgs() << "\nDebug Range Extension\n"); 3109ae6f7882SJeremy Morse this->TPC = TPC; 3110ae6f7882SJeremy Morse 3111a3936a6cSJeremy Morse this->DomTree = DomTree; 3112ae6f7882SJeremy Morse TRI = MF.getSubtarget().getRegisterInfo(); 3113e7084ceaSJeremy Morse MRI = &MF.getRegInfo(); 3114ae6f7882SJeremy Morse TII = MF.getSubtarget().getInstrInfo(); 3115ae6f7882SJeremy Morse TFI = MF.getSubtarget().getFrameLowering(); 3116ae6f7882SJeremy Morse TFI->getCalleeSaves(MF, CalleeSavedRegs); 3117010108bbSJeremy Morse MFI = &MF.getFrameInfo(); 3118ae6f7882SJeremy Morse LS.initialize(MF); 3119ae6f7882SJeremy Morse 3120bfadc5dcSJeremy Morse const auto &STI = MF.getSubtarget(); 3121bfadc5dcSJeremy Morse AdjustsStackInCalls = MFI->adjustsStack() && 3122bfadc5dcSJeremy Morse STI.getFrameLowering()->stackProbeFunctionModifiesSP(); 3123bfadc5dcSJeremy Morse if (AdjustsStackInCalls) 3124bfadc5dcSJeremy Morse StackProbeSymbolName = STI.getTargetLowering()->getStackProbeSymbolName(MF); 3125bfadc5dcSJeremy Morse 3126ae6f7882SJeremy Morse MTracker = 3127ae6f7882SJeremy Morse new MLocTracker(MF, *TII, *TRI, *MF.getSubtarget().getTargetLowering()); 3128ae6f7882SJeremy Morse VTracker = nullptr; 3129ae6f7882SJeremy Morse TTracker = nullptr; 3130ae6f7882SJeremy Morse 3131ae6f7882SJeremy Morse SmallVector<MLocTransferMap, 32> MLocTransfer; 3132ae6f7882SJeremy Morse SmallVector<VLocTracker, 8> vlocs; 3133ae6f7882SJeremy Morse LiveInsT SavedLiveIns; 3134ae6f7882SJeremy Morse 3135ae6f7882SJeremy Morse int MaxNumBlocks = -1; 3136ae6f7882SJeremy Morse for (auto &MBB : MF) 3137ae6f7882SJeremy Morse MaxNumBlocks = std::max(MBB.getNumber(), MaxNumBlocks); 3138ae6f7882SJeremy Morse assert(MaxNumBlocks >= 0); 3139ae6f7882SJeremy Morse ++MaxNumBlocks; 3140ae6f7882SJeremy Morse 3141ae6f7882SJeremy Morse MLocTransfer.resize(MaxNumBlocks); 31420eee8445SJeremy Morse vlocs.resize(MaxNumBlocks, VLocTracker(OverlapFragments, EmptyExpr)); 3143ae6f7882SJeremy Morse SavedLiveIns.resize(MaxNumBlocks); 3144ae6f7882SJeremy Morse 3145ae6f7882SJeremy Morse initialSetup(MF); 3146ae6f7882SJeremy Morse 3147ab93e710SJeremy Morse produceMLocTransferFunction(MF, MLocTransfer, MaxNumBlocks); 3148ae6f7882SJeremy Morse 3149ae6f7882SJeremy Morse // Allocate and initialize two array-of-arrays for the live-in and live-out 3150ae6f7882SJeremy Morse // machine values. The outer dimension is the block number; while the inner 3151ae6f7882SJeremy Morse // dimension is a LocIdx from MLocTracker. 3152ab49dce0SJeremy Morse FuncValueTable MOutLocs = std::make_unique<ValueTable[]>(MaxNumBlocks); 3153ab49dce0SJeremy Morse FuncValueTable MInLocs = std::make_unique<ValueTable[]>(MaxNumBlocks); 3154ae6f7882SJeremy Morse unsigned NumLocs = MTracker->getNumLocs(); 3155ae6f7882SJeremy Morse for (int i = 0; i < MaxNumBlocks; ++i) { 3156a3936a6cSJeremy Morse // These all auto-initialize to ValueIDNum::EmptyValue 3157ab49dce0SJeremy Morse MOutLocs[i] = std::make_unique<ValueIDNum[]>(NumLocs); 3158ab49dce0SJeremy Morse MInLocs[i] = std::make_unique<ValueIDNum[]>(NumLocs); 3159ae6f7882SJeremy Morse } 3160ae6f7882SJeremy Morse 3161ae6f7882SJeremy Morse // Solve the machine value dataflow problem using the MLocTransfer function, 3162ae6f7882SJeremy Morse // storing the computed live-ins / live-outs into the array-of-arrays. We use 3163ae6f7882SJeremy Morse // both live-ins and live-outs for decision making in the variable value 3164ae6f7882SJeremy Morse // dataflow problem. 3165a3936a6cSJeremy Morse buildMLocValueMap(MF, MInLocs, MOutLocs, MLocTransfer); 3166ae6f7882SJeremy Morse 3167010108bbSJeremy Morse // Patch up debug phi numbers, turning unknown block-live-in values into 3168010108bbSJeremy Morse // either live-through machine values, or PHIs. 3169010108bbSJeremy Morse for (auto &DBG_PHI : DebugPHINumToValue) { 3170010108bbSJeremy Morse // Identify unresolved block-live-ins. 3171be5734ddSJeremy Morse if (!DBG_PHI.ValueRead) 3172be5734ddSJeremy Morse continue; 3173be5734ddSJeremy Morse 3174be5734ddSJeremy Morse ValueIDNum &Num = *DBG_PHI.ValueRead; 3175010108bbSJeremy Morse if (!Num.isPHI()) 3176010108bbSJeremy Morse continue; 3177010108bbSJeremy Morse 3178010108bbSJeremy Morse unsigned BlockNo = Num.getBlock(); 3179010108bbSJeremy Morse LocIdx LocNo = Num.getLoc(); 3180010108bbSJeremy Morse Num = MInLocs[BlockNo][LocNo.asU64()]; 3181010108bbSJeremy Morse } 3182010108bbSJeremy Morse // Later, we'll be looking up ranges of instruction numbers. 3183010108bbSJeremy Morse llvm::sort(DebugPHINumToValue); 3184010108bbSJeremy Morse 3185ab93e710SJeremy Morse // Walk back through each block / instruction, collecting DBG_VALUE 3186ab93e710SJeremy Morse // instructions and recording what machine value their operands refer to. 3187ab93e710SJeremy Morse for (auto &OrderPair : OrderToBB) { 3188ab93e710SJeremy Morse MachineBasicBlock &MBB = *OrderPair.second; 3189ab93e710SJeremy Morse CurBB = MBB.getNumber(); 3190ab93e710SJeremy Morse VTracker = &vlocs[CurBB]; 3191ab93e710SJeremy Morse VTracker->MBB = &MBB; 3192ab93e710SJeremy Morse MTracker->loadFromArray(MInLocs[CurBB], CurBB); 3193ab93e710SJeremy Morse CurInst = 1; 3194ab93e710SJeremy Morse for (auto &MI : MBB) { 3195ab49dce0SJeremy Morse process(MI, MOutLocs.get(), MInLocs.get()); 3196ab93e710SJeremy Morse ++CurInst; 3197ab93e710SJeremy Morse } 3198ab93e710SJeremy Morse MTracker->reset(); 3199ab93e710SJeremy Morse } 3200ab93e710SJeremy Morse 3201ae6f7882SJeremy Morse // Number all variables in the order that they appear, to be used as a stable 3202ae6f7882SJeremy Morse // insertion order later. 3203ae6f7882SJeremy Morse DenseMap<DebugVariable, unsigned> AllVarsNumbering; 3204ae6f7882SJeremy Morse 3205ae6f7882SJeremy Morse // Map from one LexicalScope to all the variables in that scope. 32064a2cb013SJeremy Morse ScopeToVarsT ScopeToVars; 3207ae6f7882SJeremy Morse 32084a2cb013SJeremy Morse // Map from One lexical scope to all blocks where assignments happen for 32094a2cb013SJeremy Morse // that scope. 32104a2cb013SJeremy Morse ScopeToAssignBlocksT ScopeToAssignBlocks; 3211ae6f7882SJeremy Morse 32124a2cb013SJeremy Morse // Store map of DILocations that describes scopes. 32134a2cb013SJeremy Morse ScopeToDILocT ScopeToDILocation; 3214ae6f7882SJeremy Morse 3215ae6f7882SJeremy Morse // To mirror old LiveDebugValues, enumerate variables in RPOT order. Otherwise 3216ae6f7882SJeremy Morse // the order is unimportant, it just has to be stable. 3217708cbda5SJeremy Morse unsigned VarAssignCount = 0; 3218ae6f7882SJeremy Morse for (unsigned int I = 0; I < OrderToBB.size(); ++I) { 3219ae6f7882SJeremy Morse auto *MBB = OrderToBB[I]; 3220ae6f7882SJeremy Morse auto *VTracker = &vlocs[MBB->getNumber()]; 3221ae6f7882SJeremy Morse // Collect each variable with a DBG_VALUE in this block. 3222ae6f7882SJeremy Morse for (auto &idx : VTracker->Vars) { 3223ae6f7882SJeremy Morse const auto &Var = idx.first; 3224ae6f7882SJeremy Morse const DILocation *ScopeLoc = VTracker->Scopes[Var]; 3225ae6f7882SJeremy Morse assert(ScopeLoc != nullptr); 3226ae6f7882SJeremy Morse auto *Scope = LS.findLexicalScope(ScopeLoc); 3227ae6f7882SJeremy Morse 3228ae6f7882SJeremy Morse // No insts in scope -> shouldn't have been recorded. 3229ae6f7882SJeremy Morse assert(Scope != nullptr); 3230ae6f7882SJeremy Morse 3231ae6f7882SJeremy Morse AllVarsNumbering.insert(std::make_pair(Var, AllVarsNumbering.size())); 3232ae6f7882SJeremy Morse ScopeToVars[Scope].insert(Var); 32334a2cb013SJeremy Morse ScopeToAssignBlocks[Scope].insert(VTracker->MBB); 3234ae6f7882SJeremy Morse ScopeToDILocation[Scope] = ScopeLoc; 3235708cbda5SJeremy Morse ++VarAssignCount; 3236ae6f7882SJeremy Morse } 3237ae6f7882SJeremy Morse } 3238ae6f7882SJeremy Morse 3239708cbda5SJeremy Morse bool Changed = false; 3240708cbda5SJeremy Morse 3241708cbda5SJeremy Morse // If we have an extremely large number of variable assignments and blocks, 3242708cbda5SJeremy Morse // bail out at this point. We've burnt some time doing analysis already, 3243708cbda5SJeremy Morse // however we should cut our losses. 3244708cbda5SJeremy Morse if ((unsigned)MaxNumBlocks > InputBBLimit && 3245708cbda5SJeremy Morse VarAssignCount > InputDbgValLimit) { 3246708cbda5SJeremy Morse LLVM_DEBUG(dbgs() << "Disabling InstrRefBasedLDV: " << MF.getName() 3247708cbda5SJeremy Morse << " has " << MaxNumBlocks << " basic blocks and " 3248708cbda5SJeremy Morse << VarAssignCount 3249708cbda5SJeremy Morse << " variable assignments, exceeding limits.\n"); 3250708cbda5SJeremy Morse } else { 32519fd9d56dSJeremy Morse // Optionally, solve the variable value problem and emit to blocks by using 32529fd9d56dSJeremy Morse // a lexical-scope-depth search. It should be functionally identical to 32539fd9d56dSJeremy Morse // the "else" block of this condition. 32549fd9d56dSJeremy Morse Changed = depthFirstVLocAndEmit( 32559fd9d56dSJeremy Morse MaxNumBlocks, ScopeToDILocation, ScopeToVars, ScopeToAssignBlocks, 32569fd9d56dSJeremy Morse SavedLiveIns, MOutLocs, MInLocs, vlocs, MF, AllVarsNumbering, *TPC); 3257708cbda5SJeremy Morse } 3258708cbda5SJeremy Morse 3259ae6f7882SJeremy Morse delete MTracker; 32600caeaff1SJeremy Morse delete TTracker; 32610caeaff1SJeremy Morse MTracker = nullptr; 3262ae6f7882SJeremy Morse VTracker = nullptr; 3263ae6f7882SJeremy Morse TTracker = nullptr; 3264ae6f7882SJeremy Morse 3265ae6f7882SJeremy Morse ArtificialBlocks.clear(); 3266ae6f7882SJeremy Morse OrderToBB.clear(); 3267ae6f7882SJeremy Morse BBToOrder.clear(); 3268ae6f7882SJeremy Morse BBNumToRPO.clear(); 326968f47157SJeremy Morse DebugInstrNumToInstr.clear(); 3270010108bbSJeremy Morse DebugPHINumToValue.clear(); 32710eee8445SJeremy Morse OverlapFragments.clear(); 32720eee8445SJeremy Morse SeenFragments.clear(); 3273d556eb7eSJeremy Morse SeenDbgPHIs.clear(); 3274ae6f7882SJeremy Morse 3275ae6f7882SJeremy Morse return Changed; 3276ae6f7882SJeremy Morse } 3277ae6f7882SJeremy Morse 3278ae6f7882SJeremy Morse LDVImpl *llvm::makeInstrRefBasedLiveDebugValues() { 3279ae6f7882SJeremy Morse return new InstrRefBasedLDV(); 3280ae6f7882SJeremy Morse } 3281010108bbSJeremy Morse 3282010108bbSJeremy Morse namespace { 3283010108bbSJeremy Morse class LDVSSABlock; 3284010108bbSJeremy Morse class LDVSSAUpdater; 3285010108bbSJeremy Morse 3286010108bbSJeremy Morse // Pick a type to identify incoming block values as we construct SSA. We 3287010108bbSJeremy Morse // can't use anything more robust than an integer unfortunately, as SSAUpdater 3288010108bbSJeremy Morse // expects to zero-initialize the type. 3289010108bbSJeremy Morse typedef uint64_t BlockValueNum; 3290010108bbSJeremy Morse 3291010108bbSJeremy Morse /// Represents an SSA PHI node for the SSA updater class. Contains the block 3292010108bbSJeremy Morse /// this PHI is in, the value number it would have, and the expected incoming 3293010108bbSJeremy Morse /// values from parent blocks. 3294010108bbSJeremy Morse class LDVSSAPhi { 3295010108bbSJeremy Morse public: 3296010108bbSJeremy Morse SmallVector<std::pair<LDVSSABlock *, BlockValueNum>, 4> IncomingValues; 3297010108bbSJeremy Morse LDVSSABlock *ParentBlock; 3298010108bbSJeremy Morse BlockValueNum PHIValNum; 3299010108bbSJeremy Morse LDVSSAPhi(BlockValueNum PHIValNum, LDVSSABlock *ParentBlock) 3300010108bbSJeremy Morse : ParentBlock(ParentBlock), PHIValNum(PHIValNum) {} 3301010108bbSJeremy Morse 3302010108bbSJeremy Morse LDVSSABlock *getParent() { return ParentBlock; } 3303010108bbSJeremy Morse }; 3304010108bbSJeremy Morse 3305010108bbSJeremy Morse /// Thin wrapper around a block predecessor iterator. Only difference from a 3306010108bbSJeremy Morse /// normal block iterator is that it dereferences to an LDVSSABlock. 3307010108bbSJeremy Morse class LDVSSABlockIterator { 3308010108bbSJeremy Morse public: 3309010108bbSJeremy Morse MachineBasicBlock::pred_iterator PredIt; 3310010108bbSJeremy Morse LDVSSAUpdater &Updater; 3311010108bbSJeremy Morse 3312010108bbSJeremy Morse LDVSSABlockIterator(MachineBasicBlock::pred_iterator PredIt, 3313010108bbSJeremy Morse LDVSSAUpdater &Updater) 3314010108bbSJeremy Morse : PredIt(PredIt), Updater(Updater) {} 3315010108bbSJeremy Morse 3316010108bbSJeremy Morse bool operator!=(const LDVSSABlockIterator &OtherIt) const { 3317010108bbSJeremy Morse return OtherIt.PredIt != PredIt; 3318010108bbSJeremy Morse } 3319010108bbSJeremy Morse 3320010108bbSJeremy Morse LDVSSABlockIterator &operator++() { 3321010108bbSJeremy Morse ++PredIt; 3322010108bbSJeremy Morse return *this; 3323010108bbSJeremy Morse } 3324010108bbSJeremy Morse 3325010108bbSJeremy Morse LDVSSABlock *operator*(); 3326010108bbSJeremy Morse }; 3327010108bbSJeremy Morse 3328010108bbSJeremy Morse /// Thin wrapper around a block for SSA Updater interface. Necessary because 3329010108bbSJeremy Morse /// we need to track the PHI value(s) that we may have observed as necessary 3330010108bbSJeremy Morse /// in this block. 3331010108bbSJeremy Morse class LDVSSABlock { 3332010108bbSJeremy Morse public: 3333010108bbSJeremy Morse MachineBasicBlock &BB; 3334010108bbSJeremy Morse LDVSSAUpdater &Updater; 3335010108bbSJeremy Morse using PHIListT = SmallVector<LDVSSAPhi, 1>; 3336010108bbSJeremy Morse /// List of PHIs in this block. There should only ever be one. 3337010108bbSJeremy Morse PHIListT PHIList; 3338010108bbSJeremy Morse 3339010108bbSJeremy Morse LDVSSABlock(MachineBasicBlock &BB, LDVSSAUpdater &Updater) 3340010108bbSJeremy Morse : BB(BB), Updater(Updater) {} 3341010108bbSJeremy Morse 3342010108bbSJeremy Morse LDVSSABlockIterator succ_begin() { 3343010108bbSJeremy Morse return LDVSSABlockIterator(BB.succ_begin(), Updater); 3344010108bbSJeremy Morse } 3345010108bbSJeremy Morse 3346010108bbSJeremy Morse LDVSSABlockIterator succ_end() { 3347010108bbSJeremy Morse return LDVSSABlockIterator(BB.succ_end(), Updater); 3348010108bbSJeremy Morse } 3349010108bbSJeremy Morse 3350010108bbSJeremy Morse /// SSAUpdater has requested a PHI: create that within this block record. 3351010108bbSJeremy Morse LDVSSAPhi *newPHI(BlockValueNum Value) { 3352010108bbSJeremy Morse PHIList.emplace_back(Value, this); 3353010108bbSJeremy Morse return &PHIList.back(); 3354010108bbSJeremy Morse } 3355010108bbSJeremy Morse 3356010108bbSJeremy Morse /// SSAUpdater wishes to know what PHIs already exist in this block. 3357010108bbSJeremy Morse PHIListT &phis() { return PHIList; } 3358010108bbSJeremy Morse }; 3359010108bbSJeremy Morse 3360010108bbSJeremy Morse /// Utility class for the SSAUpdater interface: tracks blocks, PHIs and values 3361010108bbSJeremy Morse /// while SSAUpdater is exploring the CFG. It's passed as a handle / baton to 3362010108bbSJeremy Morse // SSAUpdaterTraits<LDVSSAUpdater>. 3363010108bbSJeremy Morse class LDVSSAUpdater { 3364010108bbSJeremy Morse public: 3365010108bbSJeremy Morse /// Map of value numbers to PHI records. 3366010108bbSJeremy Morse DenseMap<BlockValueNum, LDVSSAPhi *> PHIs; 3367010108bbSJeremy Morse /// Map of which blocks generate Undef values -- blocks that are not 3368010108bbSJeremy Morse /// dominated by any Def. 3369010108bbSJeremy Morse DenseMap<MachineBasicBlock *, BlockValueNum> UndefMap; 3370010108bbSJeremy Morse /// Map of machine blocks to our own records of them. 3371010108bbSJeremy Morse DenseMap<MachineBasicBlock *, LDVSSABlock *> BlockMap; 3372010108bbSJeremy Morse /// Machine location where any PHI must occur. 3373010108bbSJeremy Morse LocIdx Loc; 3374010108bbSJeremy Morse /// Table of live-in machine value numbers for blocks / locations. 3375ab49dce0SJeremy Morse const ValueTable *MLiveIns; 3376010108bbSJeremy Morse 3377ab49dce0SJeremy Morse LDVSSAUpdater(LocIdx L, const ValueTable *MLiveIns) 3378ab49dce0SJeremy Morse : Loc(L), MLiveIns(MLiveIns) {} 3379010108bbSJeremy Morse 3380010108bbSJeremy Morse void reset() { 3381e63b18bcSJeremy Morse for (auto &Block : BlockMap) 3382e63b18bcSJeremy Morse delete Block.second; 3383e63b18bcSJeremy Morse 3384010108bbSJeremy Morse PHIs.clear(); 3385010108bbSJeremy Morse UndefMap.clear(); 3386010108bbSJeremy Morse BlockMap.clear(); 3387010108bbSJeremy Morse } 3388010108bbSJeremy Morse 3389010108bbSJeremy Morse ~LDVSSAUpdater() { reset(); } 3390010108bbSJeremy Morse 3391010108bbSJeremy Morse /// For a given MBB, create a wrapper block for it. Stores it in the 3392010108bbSJeremy Morse /// LDVSSAUpdater block map. 3393010108bbSJeremy Morse LDVSSABlock *getSSALDVBlock(MachineBasicBlock *BB) { 3394010108bbSJeremy Morse auto it = BlockMap.find(BB); 3395010108bbSJeremy Morse if (it == BlockMap.end()) { 3396010108bbSJeremy Morse BlockMap[BB] = new LDVSSABlock(*BB, *this); 3397010108bbSJeremy Morse it = BlockMap.find(BB); 3398010108bbSJeremy Morse } 3399010108bbSJeremy Morse return it->second; 3400010108bbSJeremy Morse } 3401010108bbSJeremy Morse 3402010108bbSJeremy Morse /// Find the live-in value number for the given block. Looks up the value at 3403010108bbSJeremy Morse /// the PHI location on entry. 3404010108bbSJeremy Morse BlockValueNum getValue(LDVSSABlock *LDVBB) { 3405010108bbSJeremy Morse return MLiveIns[LDVBB->BB.getNumber()][Loc.asU64()].asU64(); 3406010108bbSJeremy Morse } 3407010108bbSJeremy Morse }; 3408010108bbSJeremy Morse 3409010108bbSJeremy Morse LDVSSABlock *LDVSSABlockIterator::operator*() { 3410010108bbSJeremy Morse return Updater.getSSALDVBlock(*PredIt); 3411010108bbSJeremy Morse } 3412010108bbSJeremy Morse 3413632e15e7SDavid Blaikie #ifndef NDEBUG 3414010108bbSJeremy Morse 3415010108bbSJeremy Morse raw_ostream &operator<<(raw_ostream &out, const LDVSSAPhi &PHI) { 3416010108bbSJeremy Morse out << "SSALDVPHI " << PHI.PHIValNum; 3417010108bbSJeremy Morse return out; 3418010108bbSJeremy Morse } 3419010108bbSJeremy Morse 3420632e15e7SDavid Blaikie #endif 3421632e15e7SDavid Blaikie 3422632e15e7SDavid Blaikie } // namespace 3423632e15e7SDavid Blaikie 3424632e15e7SDavid Blaikie namespace llvm { 3425632e15e7SDavid Blaikie 3426010108bbSJeremy Morse /// Template specialization to give SSAUpdater access to CFG and value 3427010108bbSJeremy Morse /// information. SSAUpdater calls methods in these traits, passing in the 3428010108bbSJeremy Morse /// LDVSSAUpdater object, to learn about blocks and the values they define. 3429010108bbSJeremy Morse /// It also provides methods to create PHI nodes and track them. 3430010108bbSJeremy Morse template <> class SSAUpdaterTraits<LDVSSAUpdater> { 3431010108bbSJeremy Morse public: 3432010108bbSJeremy Morse using BlkT = LDVSSABlock; 3433010108bbSJeremy Morse using ValT = BlockValueNum; 3434010108bbSJeremy Morse using PhiT = LDVSSAPhi; 3435010108bbSJeremy Morse using BlkSucc_iterator = LDVSSABlockIterator; 3436010108bbSJeremy Morse 3437010108bbSJeremy Morse // Methods to access block successors -- dereferencing to our wrapper class. 3438010108bbSJeremy Morse static BlkSucc_iterator BlkSucc_begin(BlkT *BB) { return BB->succ_begin(); } 3439010108bbSJeremy Morse static BlkSucc_iterator BlkSucc_end(BlkT *BB) { return BB->succ_end(); } 3440010108bbSJeremy Morse 3441010108bbSJeremy Morse /// Iterator for PHI operands. 3442010108bbSJeremy Morse class PHI_iterator { 3443010108bbSJeremy Morse private: 3444010108bbSJeremy Morse LDVSSAPhi *PHI; 3445010108bbSJeremy Morse unsigned Idx; 3446010108bbSJeremy Morse 3447010108bbSJeremy Morse public: 3448010108bbSJeremy Morse explicit PHI_iterator(LDVSSAPhi *P) // begin iterator 3449010108bbSJeremy Morse : PHI(P), Idx(0) {} 3450010108bbSJeremy Morse PHI_iterator(LDVSSAPhi *P, bool) // end iterator 3451010108bbSJeremy Morse : PHI(P), Idx(PHI->IncomingValues.size()) {} 3452010108bbSJeremy Morse 3453010108bbSJeremy Morse PHI_iterator &operator++() { 3454010108bbSJeremy Morse Idx++; 3455010108bbSJeremy Morse return *this; 3456010108bbSJeremy Morse } 3457010108bbSJeremy Morse bool operator==(const PHI_iterator &X) const { return Idx == X.Idx; } 3458010108bbSJeremy Morse bool operator!=(const PHI_iterator &X) const { return !operator==(X); } 3459010108bbSJeremy Morse 3460010108bbSJeremy Morse BlockValueNum getIncomingValue() { return PHI->IncomingValues[Idx].second; } 3461010108bbSJeremy Morse 3462010108bbSJeremy Morse LDVSSABlock *getIncomingBlock() { return PHI->IncomingValues[Idx].first; } 3463010108bbSJeremy Morse }; 3464010108bbSJeremy Morse 3465010108bbSJeremy Morse static inline PHI_iterator PHI_begin(PhiT *PHI) { return PHI_iterator(PHI); } 3466010108bbSJeremy Morse 3467010108bbSJeremy Morse static inline PHI_iterator PHI_end(PhiT *PHI) { 3468010108bbSJeremy Morse return PHI_iterator(PHI, true); 3469010108bbSJeremy Morse } 3470010108bbSJeremy Morse 3471010108bbSJeremy Morse /// FindPredecessorBlocks - Put the predecessors of BB into the Preds 3472010108bbSJeremy Morse /// vector. 3473010108bbSJeremy Morse static void FindPredecessorBlocks(LDVSSABlock *BB, 3474010108bbSJeremy Morse SmallVectorImpl<LDVSSABlock *> *Preds) { 3475ef2d0e0fSKazu Hirata for (MachineBasicBlock *Pred : BB->BB.predecessors()) 3476ef2d0e0fSKazu Hirata Preds->push_back(BB->Updater.getSSALDVBlock(Pred)); 3477010108bbSJeremy Morse } 3478010108bbSJeremy Morse 3479010108bbSJeremy Morse /// GetUndefVal - Normally creates an IMPLICIT_DEF instruction with a new 3480010108bbSJeremy Morse /// register. For LiveDebugValues, represents a block identified as not having 3481010108bbSJeremy Morse /// any DBG_PHI predecessors. 3482010108bbSJeremy Morse static BlockValueNum GetUndefVal(LDVSSABlock *BB, LDVSSAUpdater *Updater) { 3483010108bbSJeremy Morse // Create a value number for this block -- it needs to be unique and in the 3484010108bbSJeremy Morse // "undef" collection, so that we know it's not real. Use a number 3485010108bbSJeremy Morse // representing a PHI into this block. 3486010108bbSJeremy Morse BlockValueNum Num = ValueIDNum(BB->BB.getNumber(), 0, Updater->Loc).asU64(); 3487010108bbSJeremy Morse Updater->UndefMap[&BB->BB] = Num; 3488010108bbSJeremy Morse return Num; 3489010108bbSJeremy Morse } 3490010108bbSJeremy Morse 3491010108bbSJeremy Morse /// CreateEmptyPHI - Create a (representation of a) PHI in the given block. 3492010108bbSJeremy Morse /// SSAUpdater will populate it with information about incoming values. The 3493010108bbSJeremy Morse /// value number of this PHI is whatever the machine value number problem 3494010108bbSJeremy Morse /// solution determined it to be. This includes non-phi values if SSAUpdater 3495010108bbSJeremy Morse /// tries to create a PHI where the incoming values are identical. 3496010108bbSJeremy Morse static BlockValueNum CreateEmptyPHI(LDVSSABlock *BB, unsigned NumPreds, 3497010108bbSJeremy Morse LDVSSAUpdater *Updater) { 3498010108bbSJeremy Morse BlockValueNum PHIValNum = Updater->getValue(BB); 3499010108bbSJeremy Morse LDVSSAPhi *PHI = BB->newPHI(PHIValNum); 3500010108bbSJeremy Morse Updater->PHIs[PHIValNum] = PHI; 3501010108bbSJeremy Morse return PHIValNum; 3502010108bbSJeremy Morse } 3503010108bbSJeremy Morse 3504010108bbSJeremy Morse /// AddPHIOperand - Add the specified value as an operand of the PHI for 3505010108bbSJeremy Morse /// the specified predecessor block. 3506010108bbSJeremy Morse static void AddPHIOperand(LDVSSAPhi *PHI, BlockValueNum Val, LDVSSABlock *Pred) { 3507010108bbSJeremy Morse PHI->IncomingValues.push_back(std::make_pair(Pred, Val)); 3508010108bbSJeremy Morse } 3509010108bbSJeremy Morse 3510010108bbSJeremy Morse /// ValueIsPHI - Check if the instruction that defines the specified value 3511010108bbSJeremy Morse /// is a PHI instruction. 3512010108bbSJeremy Morse static LDVSSAPhi *ValueIsPHI(BlockValueNum Val, LDVSSAUpdater *Updater) { 3513010108bbSJeremy Morse auto PHIIt = Updater->PHIs.find(Val); 3514010108bbSJeremy Morse if (PHIIt == Updater->PHIs.end()) 3515010108bbSJeremy Morse return nullptr; 3516010108bbSJeremy Morse return PHIIt->second; 3517010108bbSJeremy Morse } 3518010108bbSJeremy Morse 3519010108bbSJeremy Morse /// ValueIsNewPHI - Like ValueIsPHI but also check if the PHI has no source 3520010108bbSJeremy Morse /// operands, i.e., it was just added. 3521010108bbSJeremy Morse static LDVSSAPhi *ValueIsNewPHI(BlockValueNum Val, LDVSSAUpdater *Updater) { 3522010108bbSJeremy Morse LDVSSAPhi *PHI = ValueIsPHI(Val, Updater); 3523010108bbSJeremy Morse if (PHI && PHI->IncomingValues.size() == 0) 3524010108bbSJeremy Morse return PHI; 3525010108bbSJeremy Morse return nullptr; 3526010108bbSJeremy Morse } 3527010108bbSJeremy Morse 3528010108bbSJeremy Morse /// GetPHIValue - For the specified PHI instruction, return the value 3529010108bbSJeremy Morse /// that it defines. 3530010108bbSJeremy Morse static BlockValueNum GetPHIValue(LDVSSAPhi *PHI) { return PHI->PHIValNum; } 3531010108bbSJeremy Morse }; 3532010108bbSJeremy Morse 3533010108bbSJeremy Morse } // end namespace llvm 3534010108bbSJeremy Morse 3535ab49dce0SJeremy Morse Optional<ValueIDNum> InstrRefBasedLDV::resolveDbgPHIs( 3536ab49dce0SJeremy Morse MachineFunction &MF, const ValueTable *MLiveOuts, 3537ab49dce0SJeremy Morse const ValueTable *MLiveIns, MachineInstr &Here, uint64_t InstrNum) { 3538ab49dce0SJeremy Morse assert(MLiveOuts && MLiveIns && 3539ab49dce0SJeremy Morse "Tried to resolve DBG_PHI before location " 3540ab49dce0SJeremy Morse "tables allocated?"); 3541ab49dce0SJeremy Morse 3542d556eb7eSJeremy Morse // This function will be called twice per DBG_INSTR_REF, and might end up 3543d556eb7eSJeremy Morse // computing lots of SSA information: memoize it. 3544d556eb7eSJeremy Morse auto SeenDbgPHIIt = SeenDbgPHIs.find(&Here); 3545d556eb7eSJeremy Morse if (SeenDbgPHIIt != SeenDbgPHIs.end()) 3546d556eb7eSJeremy Morse return SeenDbgPHIIt->second; 3547d556eb7eSJeremy Morse 3548d556eb7eSJeremy Morse Optional<ValueIDNum> Result = 3549d556eb7eSJeremy Morse resolveDbgPHIsImpl(MF, MLiveOuts, MLiveIns, Here, InstrNum); 3550d556eb7eSJeremy Morse SeenDbgPHIs.insert({&Here, Result}); 3551d556eb7eSJeremy Morse return Result; 3552d556eb7eSJeremy Morse } 3553d556eb7eSJeremy Morse 3554d556eb7eSJeremy Morse Optional<ValueIDNum> InstrRefBasedLDV::resolveDbgPHIsImpl( 3555ab49dce0SJeremy Morse MachineFunction &MF, const ValueTable *MLiveOuts, 3556ab49dce0SJeremy Morse const ValueTable *MLiveIns, MachineInstr &Here, uint64_t InstrNum) { 3557010108bbSJeremy Morse // Pick out records of DBG_PHI instructions that have been observed. If there 3558010108bbSJeremy Morse // are none, then we cannot compute a value number. 3559010108bbSJeremy Morse auto RangePair = std::equal_range(DebugPHINumToValue.begin(), 3560010108bbSJeremy Morse DebugPHINumToValue.end(), InstrNum); 3561010108bbSJeremy Morse auto LowerIt = RangePair.first; 3562010108bbSJeremy Morse auto UpperIt = RangePair.second; 3563010108bbSJeremy Morse 3564010108bbSJeremy Morse // No DBG_PHI means there can be no location. 3565010108bbSJeremy Morse if (LowerIt == UpperIt) 3566010108bbSJeremy Morse return None; 3567010108bbSJeremy Morse 3568be5734ddSJeremy Morse // If any DBG_PHIs referred to a location we didn't understand, don't try to 3569be5734ddSJeremy Morse // compute a value. There might be scenarios where we could recover a value 3570be5734ddSJeremy Morse // for some range of DBG_INSTR_REFs, but at this point we can have high 3571be5734ddSJeremy Morse // confidence that we've seen a bug. 3572be5734ddSJeremy Morse auto DBGPHIRange = make_range(LowerIt, UpperIt); 3573be5734ddSJeremy Morse for (const DebugPHIRecord &DBG_PHI : DBGPHIRange) 3574be5734ddSJeremy Morse if (!DBG_PHI.ValueRead) 3575be5734ddSJeremy Morse return None; 3576be5734ddSJeremy Morse 3577010108bbSJeremy Morse // If there's only one DBG_PHI, then that is our value number. 3578010108bbSJeremy Morse if (std::distance(LowerIt, UpperIt) == 1) 3579be5734ddSJeremy Morse return *LowerIt->ValueRead; 3580010108bbSJeremy Morse 3581010108bbSJeremy Morse // Pick out the location (physreg, slot) where any PHIs must occur. It's 3582010108bbSJeremy Morse // technically possible for us to merge values in different registers in each 3583010108bbSJeremy Morse // block, but highly unlikely that LLVM will generate such code after register 3584010108bbSJeremy Morse // allocation. 3585be5734ddSJeremy Morse LocIdx Loc = *LowerIt->ReadLoc; 3586010108bbSJeremy Morse 3587010108bbSJeremy Morse // We have several DBG_PHIs, and a use position (the Here inst). All each 3588010108bbSJeremy Morse // DBG_PHI does is identify a value at a program position. We can treat each 3589010108bbSJeremy Morse // DBG_PHI like it's a Def of a value, and the use position is a Use of a 3590010108bbSJeremy Morse // value, just like SSA. We use the bulk-standard LLVM SSA updater class to 3591010108bbSJeremy Morse // determine which Def is used at the Use, and any PHIs that happen along 3592010108bbSJeremy Morse // the way. 3593010108bbSJeremy Morse // Adapted LLVM SSA Updater: 3594010108bbSJeremy Morse LDVSSAUpdater Updater(Loc, MLiveIns); 3595010108bbSJeremy Morse // Map of which Def or PHI is the current value in each block. 3596010108bbSJeremy Morse DenseMap<LDVSSABlock *, BlockValueNum> AvailableValues; 3597010108bbSJeremy Morse // Set of PHIs that we have created along the way. 3598010108bbSJeremy Morse SmallVector<LDVSSAPhi *, 8> CreatedPHIs; 3599010108bbSJeremy Morse 3600010108bbSJeremy Morse // Each existing DBG_PHI is a Def'd value under this model. Record these Defs 3601010108bbSJeremy Morse // for the SSAUpdater. 3602010108bbSJeremy Morse for (const auto &DBG_PHI : DBGPHIRange) { 3603010108bbSJeremy Morse LDVSSABlock *Block = Updater.getSSALDVBlock(DBG_PHI.MBB); 3604be5734ddSJeremy Morse const ValueIDNum &Num = *DBG_PHI.ValueRead; 3605010108bbSJeremy Morse AvailableValues.insert(std::make_pair(Block, Num.asU64())); 3606010108bbSJeremy Morse } 3607010108bbSJeremy Morse 3608010108bbSJeremy Morse LDVSSABlock *HereBlock = Updater.getSSALDVBlock(Here.getParent()); 3609010108bbSJeremy Morse const auto &AvailIt = AvailableValues.find(HereBlock); 3610010108bbSJeremy Morse if (AvailIt != AvailableValues.end()) { 3611010108bbSJeremy Morse // Actually, we already know what the value is -- the Use is in the same 3612010108bbSJeremy Morse // block as the Def. 3613010108bbSJeremy Morse return ValueIDNum::fromU64(AvailIt->second); 3614010108bbSJeremy Morse } 3615010108bbSJeremy Morse 3616010108bbSJeremy Morse // Otherwise, we must use the SSA Updater. It will identify the value number 3617010108bbSJeremy Morse // that we are to use, and the PHIs that must happen along the way. 3618010108bbSJeremy Morse SSAUpdaterImpl<LDVSSAUpdater> Impl(&Updater, &AvailableValues, &CreatedPHIs); 3619010108bbSJeremy Morse BlockValueNum ResultInt = Impl.GetValue(Updater.getSSALDVBlock(Here.getParent())); 3620010108bbSJeremy Morse ValueIDNum Result = ValueIDNum::fromU64(ResultInt); 3621010108bbSJeremy Morse 3622010108bbSJeremy Morse // We have the number for a PHI, or possibly live-through value, to be used 3623010108bbSJeremy Morse // at this Use. There are a number of things we have to check about it though: 3624010108bbSJeremy Morse // * Does any PHI use an 'Undef' (like an IMPLICIT_DEF) value? If so, this 3625010108bbSJeremy Morse // Use was not completely dominated by DBG_PHIs and we should abort. 3626010108bbSJeremy Morse // * Are the Defs or PHIs clobbered in a block? SSAUpdater isn't aware that 3627010108bbSJeremy Morse // we've left SSA form. Validate that the inputs to each PHI are the 3628010108bbSJeremy Morse // expected values. 3629010108bbSJeremy Morse // * Is a PHI we've created actually a merging of values, or are all the 3630010108bbSJeremy Morse // predecessor values the same, leading to a non-PHI machine value number? 3631010108bbSJeremy Morse // (SSAUpdater doesn't know that either). Remap validated PHIs into the 3632010108bbSJeremy Morse // the ValidatedValues collection below to sort this out. 3633010108bbSJeremy Morse DenseMap<LDVSSABlock *, ValueIDNum> ValidatedValues; 3634010108bbSJeremy Morse 3635010108bbSJeremy Morse // Define all the input DBG_PHI values in ValidatedValues. 3636010108bbSJeremy Morse for (const auto &DBG_PHI : DBGPHIRange) { 3637010108bbSJeremy Morse LDVSSABlock *Block = Updater.getSSALDVBlock(DBG_PHI.MBB); 3638be5734ddSJeremy Morse const ValueIDNum &Num = *DBG_PHI.ValueRead; 3639010108bbSJeremy Morse ValidatedValues.insert(std::make_pair(Block, Num)); 3640010108bbSJeremy Morse } 3641010108bbSJeremy Morse 3642010108bbSJeremy Morse // Sort PHIs to validate into RPO-order. 3643010108bbSJeremy Morse SmallVector<LDVSSAPhi *, 8> SortedPHIs; 3644010108bbSJeremy Morse for (auto &PHI : CreatedPHIs) 3645010108bbSJeremy Morse SortedPHIs.push_back(PHI); 3646010108bbSJeremy Morse 3647010108bbSJeremy Morse std::sort( 3648010108bbSJeremy Morse SortedPHIs.begin(), SortedPHIs.end(), [&](LDVSSAPhi *A, LDVSSAPhi *B) { 3649010108bbSJeremy Morse return BBToOrder[&A->getParent()->BB] < BBToOrder[&B->getParent()->BB]; 3650010108bbSJeremy Morse }); 3651010108bbSJeremy Morse 3652010108bbSJeremy Morse for (auto &PHI : SortedPHIs) { 3653010108bbSJeremy Morse ValueIDNum ThisBlockValueNum = 3654010108bbSJeremy Morse MLiveIns[PHI->ParentBlock->BB.getNumber()][Loc.asU64()]; 3655010108bbSJeremy Morse 3656010108bbSJeremy Morse // Are all these things actually defined? 3657010108bbSJeremy Morse for (auto &PHIIt : PHI->IncomingValues) { 3658010108bbSJeremy Morse // Any undef input means DBG_PHIs didn't dominate the use point. 3659010108bbSJeremy Morse if (Updater.UndefMap.find(&PHIIt.first->BB) != Updater.UndefMap.end()) 3660010108bbSJeremy Morse return None; 3661010108bbSJeremy Morse 3662010108bbSJeremy Morse ValueIDNum ValueToCheck; 3663ab49dce0SJeremy Morse const ValueTable &BlockLiveOuts = MLiveOuts[PHIIt.first->BB.getNumber()]; 3664010108bbSJeremy Morse 3665010108bbSJeremy Morse auto VVal = ValidatedValues.find(PHIIt.first); 3666010108bbSJeremy Morse if (VVal == ValidatedValues.end()) { 3667010108bbSJeremy Morse // We cross a loop, and this is a backedge. LLVMs tail duplication 3668010108bbSJeremy Morse // happens so late that DBG_PHI instructions should not be able to 3669010108bbSJeremy Morse // migrate into loops -- meaning we can only be live-through this 3670010108bbSJeremy Morse // loop. 3671010108bbSJeremy Morse ValueToCheck = ThisBlockValueNum; 3672010108bbSJeremy Morse } else { 3673010108bbSJeremy Morse // Does the block have as a live-out, in the location we're examining, 3674010108bbSJeremy Morse // the value that we expect? If not, it's been moved or clobbered. 3675010108bbSJeremy Morse ValueToCheck = VVal->second; 3676010108bbSJeremy Morse } 3677010108bbSJeremy Morse 3678010108bbSJeremy Morse if (BlockLiveOuts[Loc.asU64()] != ValueToCheck) 3679010108bbSJeremy Morse return None; 3680010108bbSJeremy Morse } 3681010108bbSJeremy Morse 3682010108bbSJeremy Morse // Record this value as validated. 3683010108bbSJeremy Morse ValidatedValues.insert({PHI->ParentBlock, ThisBlockValueNum}); 3684010108bbSJeremy Morse } 3685010108bbSJeremy Morse 3686010108bbSJeremy Morse // All the PHIs are valid: we can return what the SSAUpdater said our value 3687010108bbSJeremy Morse // number was. 3688010108bbSJeremy Morse return Result; 3689010108bbSJeremy Morse } 3690