blob: 26f551386cf3e890ded381f45882bef6062a5761 [file] [log] [blame]
Chris Lattnera3b8b5c2004-07-23 17:56:30 +00001//===-- LiveIntervalAnalysis.cpp - Live Interval Analysis -----------------===//
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the LiveInterval analysis pass which is used
11// by the Linear Scan Register allocator. This pass linearizes the
12// basic blocks of the function in DFS order and uses the
13// LiveVariables pass to conservatively compute live intervals for
14// each virtual and physical register.
15//
16//===----------------------------------------------------------------------===//
17
18#define DEBUG_TYPE "liveintervals"
Chris Lattner3c3fe462005-09-21 04:19:09 +000019#include "llvm/CodeGen/LiveIntervalAnalysis.h"
Misha Brukman08a6c762004-09-03 18:25:53 +000020#include "VirtRegMap.h"
Chris Lattner015959e2004-05-01 21:24:39 +000021#include "llvm/Value.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000022#include "llvm/CodeGen/LiveVariables.h"
23#include "llvm/CodeGen/MachineFrameInfo.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000024#include "llvm/CodeGen/MachineInstr.h"
Evan Cheng22f07ff2007-12-11 02:09:15 +000025#include "llvm/CodeGen/MachineLoopInfo.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000026#include "llvm/CodeGen/MachineRegisterInfo.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000027#include "llvm/CodeGen/Passes.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000028#include "llvm/Target/TargetRegisterInfo.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000029#include "llvm/Target/TargetInstrInfo.h"
30#include "llvm/Target/TargetMachine.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000031#include "llvm/Support/CommandLine.h"
32#include "llvm/Support/Debug.h"
33#include "llvm/ADT/Statistic.h"
34#include "llvm/ADT/STLExtras.h"
Alkis Evlogimenos20aa4742004-09-03 18:19:51 +000035#include <algorithm>
Jeff Cohen97af7512006-12-02 02:22:01 +000036#include <cmath>
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000037using namespace llvm;
38
Dan Gohman844731a2008-05-13 00:00:25 +000039// Hidden options for help debugging.
40static cl::opt<bool> DisableReMat("disable-rematerialization",
41 cl::init(false), cl::Hidden);
Evan Cheng81a03822007-11-17 00:40:40 +000042
Dan Gohman844731a2008-05-13 00:00:25 +000043static cl::opt<bool> SplitAtBB("split-intervals-at-bb",
44 cl::init(true), cl::Hidden);
45static cl::opt<int> SplitLimit("split-limit",
46 cl::init(-1), cl::Hidden);
Evan Chengbc165e42007-08-16 07:24:22 +000047
Chris Lattnercd3245a2006-12-19 22:41:21 +000048STATISTIC(numIntervals, "Number of original intervals");
49STATISTIC(numIntervalsAfter, "Number of intervals after coalescing");
Evan Cheng0cbb1162007-11-29 01:06:25 +000050STATISTIC(numFolds , "Number of loads/stores folded into instructions");
51STATISTIC(numSplits , "Number of intervals split");
Chris Lattnercd3245a2006-12-19 22:41:21 +000052
Devang Patel19974732007-05-03 01:11:54 +000053char LiveIntervals::ID = 0;
Dan Gohman844731a2008-05-13 00:00:25 +000054static RegisterPass<LiveIntervals> X("liveintervals", "Live Interval Analysis");
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000055
Chris Lattnerf7da2c72006-08-24 22:43:55 +000056void LiveIntervals::getAnalysisUsage(AnalysisUsage &AU) const {
David Greene25133302007-06-08 17:18:56 +000057 AU.addPreserved<LiveVariables>();
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000058 AU.addRequired<LiveVariables>();
Bill Wendling67d65bb2008-01-04 20:54:55 +000059 AU.addPreservedID(MachineLoopInfoID);
60 AU.addPreservedID(MachineDominatorsID);
Owen Andersonfcc63502008-05-29 18:35:21 +000061 AU.addPreservedID(PHIEliminationID);
62 AU.addRequiredID(PHIEliminationID);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000063 AU.addRequiredID(TwoAddressInstructionPassID);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000064 MachineFunctionPass::getAnalysisUsage(AU);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000065}
66
Chris Lattnerf7da2c72006-08-24 22:43:55 +000067void LiveIntervals::releaseMemory() {
Evan Cheng3f32d652008-06-04 09:18:41 +000068 MBB2IdxMap.clear();
Evan Cheng4ca980e2007-10-17 02:10:22 +000069 Idx2MBBMap.clear();
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000070 mi2iMap_.clear();
71 i2miMap_.clear();
72 r2iMap_.clear();
Evan Chengdd199d22007-09-06 01:07:24 +000073 // Release VNInfo memroy regions after all VNInfo objects are dtor'd.
74 VNInfoAllocator.Reset();
Evan Cheng549f27d32007-08-13 23:45:17 +000075 for (unsigned i = 0, e = ClonedMIs.size(); i != e; ++i)
76 delete ClonedMIs[i];
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000077}
78
Owen Anderson80b3ce62008-05-28 20:54:50 +000079void LiveIntervals::computeNumbering() {
80 Index2MiMap OldI2MI = i2miMap_;
Owen Anderson6c5e5612008-06-19 00:10:49 +000081 std::vector<IdxMBBPair> OldI2MBB = Idx2MBBMap;
Owen Anderson80b3ce62008-05-28 20:54:50 +000082
83 Idx2MBBMap.clear();
84 MBB2IdxMap.clear();
85 mi2iMap_.clear();
86 i2miMap_.clear();
87
Chris Lattner428b92e2006-09-15 03:57:23 +000088 // Number MachineInstrs and MachineBasicBlocks.
89 // Initialize MBB indexes to a sentinal.
Evan Cheng549f27d32007-08-13 23:45:17 +000090 MBB2IdxMap.resize(mf_->getNumBlockIDs(), std::make_pair(~0U,~0U));
Chris Lattner428b92e2006-09-15 03:57:23 +000091
92 unsigned MIIndex = 0;
93 for (MachineFunction::iterator MBB = mf_->begin(), E = mf_->end();
94 MBB != E; ++MBB) {
Evan Cheng549f27d32007-08-13 23:45:17 +000095 unsigned StartIdx = MIIndex;
Evan Cheng0c9f92e2007-02-13 01:30:55 +000096
Owen Anderson6c5e5612008-06-19 00:10:49 +000097 // Insert an empty slot at the beginning of each block.
98 MIIndex += InstrSlots::NUM;
99 i2miMap_.push_back(0);
100
Chris Lattner428b92e2006-09-15 03:57:23 +0000101 for (MachineBasicBlock::iterator I = MBB->begin(), E = MBB->end();
102 I != E; ++I) {
103 bool inserted = mi2iMap_.insert(std::make_pair(I, MIIndex)).second;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000104 assert(inserted && "multiple MachineInstr -> index mappings");
Chris Lattner428b92e2006-09-15 03:57:23 +0000105 i2miMap_.push_back(I);
106 MIIndex += InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000107
108 // Insert an empty slot after every instruction.
Owen Anderson1fbb4542008-06-16 16:58:24 +0000109 MIIndex += InstrSlots::NUM;
110 i2miMap_.push_back(0);
Owen Anderson35578012008-06-16 07:10:49 +0000111 }
Owen Anderson6c5e5612008-06-19 00:10:49 +0000112
Owen Anderson1fbb4542008-06-16 16:58:24 +0000113 // Set the MBB2IdxMap entry for this MBB.
114 MBB2IdxMap[MBB->getNumber()] = std::make_pair(StartIdx, MIIndex - 1);
115 Idx2MBBMap.push_back(std::make_pair(StartIdx, MBB));
Chris Lattner428b92e2006-09-15 03:57:23 +0000116 }
Evan Cheng4ca980e2007-10-17 02:10:22 +0000117 std::sort(Idx2MBBMap.begin(), Idx2MBBMap.end(), Idx2MBBCompare());
Owen Anderson80b3ce62008-05-28 20:54:50 +0000118
119 if (!OldI2MI.empty())
Owen Anderson6c5e5612008-06-19 00:10:49 +0000120 for (iterator OI = begin(), OE = end(); OI != OE; ++OI)
121 for (LiveInterval::iterator LI = OI->second.begin(),
122 LE = OI->second.end(); LI != LE; ++LI) {
Owen Anderson4b5b2092008-05-29 18:15:49 +0000123
Owen Anderson7eec0c22008-05-29 23:01:22 +0000124 // Remap the start index of the live range to the corresponding new
125 // number, or our best guess at what it _should_ correspond to if the
126 // original instruction has been erased. This is either the following
127 // instruction or its predecessor.
Owen Anderson6c5e5612008-06-19 00:10:49 +0000128 unsigned index = LI->start / InstrSlots::NUM;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000129 unsigned offset = LI->start % InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000130 if (offset == InstrSlots::LOAD) {
131 std::vector<IdxMBBPair>::const_iterator I =
132 std::lower_bound(OldI2MBB.begin(), OldI2MBB.end(), index);
133 // Take the pair containing the index
134 std::vector<IdxMBBPair>::const_iterator J =
135 ((I != OldI2MBB.end() && I->first > index) ||
136 (I == OldI2MBB.end() && OldI2MBB.size()>0)) ? (I-1): I;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000137
Owen Anderson6c5e5612008-06-19 00:10:49 +0000138 LI->start = getMBBStartIdx(J->second);
139 } else {
140 LI->start = mi2iMap_[OldI2MI[index]] + offset;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000141 }
142
143 // Remap the ending index in the same way that we remapped the start,
144 // except for the final step where we always map to the immediately
145 // following instruction.
Owen Anderson6c5e5612008-06-19 00:10:49 +0000146 index = LI->end / InstrSlots::NUM;
147 offset = LI->end % InstrSlots::NUM;
148 if (offset == InstrSlots::STORE) {
149 std::vector<IdxMBBPair>::const_iterator I =
150 std::lower_bound(OldI2MBB.begin(), OldI2MBB.end(), index);
151 // Take the pair containing the index
152 std::vector<IdxMBBPair>::const_iterator J =
153 ((I != OldI2MBB.end() && I->first > index) ||
154 (I == OldI2MBB.end() && OldI2MBB.size()>0)) ? (I-1): I;
155
156 LI->start = getMBBEndIdx(J->second);
Owen Anderson4b5b2092008-05-29 18:15:49 +0000157 } else {
Owen Anderson6c5e5612008-06-19 00:10:49 +0000158 LI->end = mi2iMap_[OldI2MI[index]] + offset;
Owen Anderson4b5b2092008-05-29 18:15:49 +0000159 }
Owen Anderson745825f42008-05-28 22:40:08 +0000160
Owen Anderson7eec0c22008-05-29 23:01:22 +0000161 // Remap the VNInfo def index, which works the same as the
162 // start indices above.
Owen Anderson745825f42008-05-28 22:40:08 +0000163 VNInfo* vni = LI->valno;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000164 index = vni->def / InstrSlots::NUM;
Owen Anderson4b5b2092008-05-29 18:15:49 +0000165 offset = vni->def % InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000166 if (offset == InstrSlots::LOAD) {
167 std::vector<IdxMBBPair>::const_iterator I =
168 std::lower_bound(OldI2MBB.begin(), OldI2MBB.end(), index);
169 // Take the pair containing the index
170 std::vector<IdxMBBPair>::const_iterator J =
171 ((I != OldI2MBB.end() && I->first > index) ||
172 (I == OldI2MBB.end() && OldI2MBB.size()>0)) ? (I-1): I;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000173
Owen Anderson6c5e5612008-06-19 00:10:49 +0000174 vni->def = getMBBStartIdx(J->second);
175
176 } else {
177 vni->def = mi2iMap_[OldI2MI[index]] + offset;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000178 }
Owen Anderson745825f42008-05-28 22:40:08 +0000179
Owen Anderson7eec0c22008-05-29 23:01:22 +0000180 // Remap the VNInfo kill indices, which works the same as
181 // the end indices above.
Owen Anderson4b5b2092008-05-29 18:15:49 +0000182 for (size_t i = 0; i < vni->kills.size(); ++i) {
Owen Anderson6c5e5612008-06-19 00:10:49 +0000183 index = vni->kills[i] / InstrSlots::NUM;
Owen Anderson4b5b2092008-05-29 18:15:49 +0000184 offset = vni->kills[i] % InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000185 if (OldI2MI[vni->kills[i] / InstrSlots::NUM]) {
186 std::vector<IdxMBBPair>::const_iterator I =
187 std::lower_bound(OldI2MBB.begin(), OldI2MBB.end(), index);
188 // Take the pair containing the index
189 std::vector<IdxMBBPair>::const_iterator J =
190 ((I != OldI2MBB.end() && I->first > index) ||
191 (I == OldI2MBB.end() && OldI2MBB.size()>0)) ? (I-1): I;
192
193 vni->kills[i] = getMBBEndIdx(J->second);
194 } else {
195 vni->kills[i] = mi2iMap_[OldI2MI[index]] + offset;
Owen Anderson7eec0c22008-05-29 23:01:22 +0000196 }
Owen Anderson4b5b2092008-05-29 18:15:49 +0000197 }
Owen Anderson80b3ce62008-05-28 20:54:50 +0000198 }
199}
Alkis Evlogimenosd6e40a62004-01-14 10:44:29 +0000200
Owen Anderson80b3ce62008-05-28 20:54:50 +0000201/// runOnMachineFunction - Register allocate the whole function
202///
203bool LiveIntervals::runOnMachineFunction(MachineFunction &fn) {
204 mf_ = &fn;
205 mri_ = &mf_->getRegInfo();
206 tm_ = &fn.getTarget();
207 tri_ = tm_->getRegisterInfo();
208 tii_ = tm_->getInstrInfo();
209 lv_ = &getAnalysis<LiveVariables>();
210 allocatableRegs_ = tri_->getAllocatableSet(fn);
211
212 computeNumbering();
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000213 computeIntervals();
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000214
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000215 numIntervals += getNumIntervals();
216
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000217 DOUT << "********** INTERVALS **********\n";
218 for (iterator I = begin(), E = end(); I != E; ++I) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000219 I->second.print(DOUT, tri_);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000220 DOUT << "\n";
221 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000222
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000223 numIntervalsAfter += getNumIntervals();
Chris Lattner70ca3582004-09-30 15:59:17 +0000224 DEBUG(dump());
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000225 return true;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000226}
227
Chris Lattner70ca3582004-09-30 15:59:17 +0000228/// print - Implement the dump method.
Reid Spencerce9653c2004-12-07 04:03:45 +0000229void LiveIntervals::print(std::ostream &O, const Module* ) const {
Chris Lattner70ca3582004-09-30 15:59:17 +0000230 O << "********** INTERVALS **********\n";
Chris Lattner8e7a7092005-07-27 23:03:38 +0000231 for (const_iterator I = begin(), E = end(); I != E; ++I) {
Evan Cheng3f32d652008-06-04 09:18:41 +0000232 I->second.print(O, tri_);
233 O << "\n";
Chris Lattner8e7a7092005-07-27 23:03:38 +0000234 }
Chris Lattner70ca3582004-09-30 15:59:17 +0000235
236 O << "********** MACHINEINSTRS **********\n";
237 for (MachineFunction::iterator mbbi = mf_->begin(), mbbe = mf_->end();
238 mbbi != mbbe; ++mbbi) {
239 O << ((Value*)mbbi->getBasicBlock())->getName() << ":\n";
240 for (MachineBasicBlock::iterator mii = mbbi->begin(),
241 mie = mbbi->end(); mii != mie; ++mii) {
Chris Lattner477e4552004-09-30 16:10:45 +0000242 O << getInstructionIndex(mii) << '\t' << *mii;
Chris Lattner70ca3582004-09-30 15:59:17 +0000243 }
244 }
245}
246
Evan Chengc92da382007-11-03 07:20:12 +0000247/// conflictsWithPhysRegDef - Returns true if the specified register
248/// is defined during the duration of the specified interval.
249bool LiveIntervals::conflictsWithPhysRegDef(const LiveInterval &li,
250 VirtRegMap &vrm, unsigned reg) {
251 for (LiveInterval::Ranges::const_iterator
252 I = li.ranges.begin(), E = li.ranges.end(); I != E; ++I) {
253 for (unsigned index = getBaseIndex(I->start),
254 end = getBaseIndex(I->end-1) + InstrSlots::NUM; index != end;
255 index += InstrSlots::NUM) {
256 // skip deleted instructions
257 while (index != end && !getInstructionFromIndex(index))
258 index += InstrSlots::NUM;
259 if (index == end) break;
260
261 MachineInstr *MI = getInstructionFromIndex(index);
Evan Cheng5d446262007-11-15 08:13:29 +0000262 unsigned SrcReg, DstReg;
263 if (tii_->isMoveInstr(*MI, SrcReg, DstReg))
264 if (SrcReg == li.reg || DstReg == li.reg)
265 continue;
Evan Chengc92da382007-11-03 07:20:12 +0000266 for (unsigned i = 0; i != MI->getNumOperands(); ++i) {
267 MachineOperand& mop = MI->getOperand(i);
Evan Cheng5d446262007-11-15 08:13:29 +0000268 if (!mop.isRegister())
Evan Chengc92da382007-11-03 07:20:12 +0000269 continue;
270 unsigned PhysReg = mop.getReg();
Evan Cheng5d446262007-11-15 08:13:29 +0000271 if (PhysReg == 0 || PhysReg == li.reg)
Evan Chengc92da382007-11-03 07:20:12 +0000272 continue;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000273 if (TargetRegisterInfo::isVirtualRegister(PhysReg)) {
Evan Cheng5d446262007-11-15 08:13:29 +0000274 if (!vrm.hasPhys(PhysReg))
275 continue;
Evan Chengc92da382007-11-03 07:20:12 +0000276 PhysReg = vrm.getPhys(PhysReg);
Evan Cheng5d446262007-11-15 08:13:29 +0000277 }
Dan Gohman6f0d0242008-02-10 18:45:23 +0000278 if (PhysReg && tri_->regsOverlap(PhysReg, reg))
Evan Chengc92da382007-11-03 07:20:12 +0000279 return true;
280 }
281 }
282 }
283
284 return false;
285}
286
Evan Cheng549f27d32007-08-13 23:45:17 +0000287void LiveIntervals::printRegName(unsigned reg) const {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000288 if (TargetRegisterInfo::isPhysicalRegister(reg))
Bill Wendlinge6d088a2008-02-26 21:47:57 +0000289 cerr << tri_->getName(reg);
Evan Cheng549f27d32007-08-13 23:45:17 +0000290 else
291 cerr << "%reg" << reg;
292}
293
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000294void LiveIntervals::handleVirtualRegisterDef(MachineBasicBlock *mbb,
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000295 MachineBasicBlock::iterator mi,
Chris Lattner6b128bd2006-09-03 08:07:11 +0000296 unsigned MIIdx,
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000297 LiveInterval &interval) {
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000298 DOUT << "\t\tregister: "; DEBUG(printRegName(interval.reg));
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000299 LiveVariables::VarInfo& vi = lv_->getVarInfo(interval.reg);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000300
Evan Cheng419852c2008-04-03 16:39:43 +0000301 if (mi->getOpcode() == TargetInstrInfo::IMPLICIT_DEF) {
302 DOUT << "is a implicit_def\n";
303 return;
304 }
305
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000306 // Virtual registers may be defined multiple times (due to phi
307 // elimination and 2-addr elimination). Much of what we do only has to be
308 // done once for the vreg. We use an empty interval to detect the first
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000309 // time we see a vreg.
310 if (interval.empty()) {
311 // Get the Idx of the defining instructions.
Chris Lattner6b128bd2006-09-03 08:07:11 +0000312 unsigned defIndex = getDefIndex(MIIdx);
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000313 VNInfo *ValNo;
Evan Chengc8d044e2008-02-15 18:24:29 +0000314 MachineInstr *CopyMI = NULL;
Chris Lattner91725b72006-08-31 05:54:43 +0000315 unsigned SrcReg, DstReg;
Evan Chengc8d044e2008-02-15 18:24:29 +0000316 if (mi->getOpcode() == TargetInstrInfo::EXTRACT_SUBREG ||
Evan Cheng7e073ba2008-04-09 20:57:25 +0000317 mi->getOpcode() == TargetInstrInfo::INSERT_SUBREG ||
Evan Chengc8d044e2008-02-15 18:24:29 +0000318 tii_->isMoveInstr(*mi, SrcReg, DstReg))
319 CopyMI = mi;
320 ValNo = interval.getNextValue(defIndex, CopyMI, VNInfoAllocator);
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000321
322 assert(ValNo->id == 0 && "First value in interval is not 0?");
Chris Lattner7ac2d312004-07-24 02:59:07 +0000323
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000324 // Loop over all of the blocks that the vreg is defined in. There are
325 // two cases we have to handle here. The most common case is a vreg
326 // whose lifetime is contained within a basic block. In this case there
327 // will be a single kill, in MBB, which comes after the definition.
328 if (vi.Kills.size() == 1 && vi.Kills[0]->getParent() == mbb) {
329 // FIXME: what about dead vars?
330 unsigned killIdx;
331 if (vi.Kills[0] != mi)
332 killIdx = getUseIndex(getInstructionIndex(vi.Kills[0]))+1;
333 else
334 killIdx = defIndex+1;
Chris Lattner6097d132004-07-19 02:15:56 +0000335
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000336 // If the kill happens after the definition, we have an intra-block
337 // live range.
338 if (killIdx > defIndex) {
Evan Cheng61de82d2007-02-15 05:59:24 +0000339 assert(vi.AliveBlocks.none() &&
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000340 "Shouldn't be alive across any blocks!");
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000341 LiveRange LR(defIndex, killIdx, ValNo);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000342 interval.addRange(LR);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000343 DOUT << " +" << LR << "\n";
Evan Chengf3bb2e62007-09-05 21:46:51 +0000344 interval.addKill(ValNo, killIdx);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000345 return;
346 }
Alkis Evlogimenosdd2cc652003-12-18 08:48:48 +0000347 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000348
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000349 // The other case we handle is when a virtual register lives to the end
350 // of the defining block, potentially live across some blocks, then is
351 // live into some number of blocks, but gets killed. Start by adding a
352 // range that goes from this definition to the end of the defining block.
Owen Anderson6c5e5612008-06-19 00:10:49 +0000353 LiveRange NewLR(defIndex, getMBBEndIdx(mbb), ValNo);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000354 DOUT << " +" << NewLR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000355 interval.addRange(NewLR);
356
357 // Iterate over all of the blocks that the variable is completely
358 // live in, adding [insrtIndex(begin), instrIndex(end)+4) to the
359 // live interval.
360 for (unsigned i = 0, e = vi.AliveBlocks.size(); i != e; ++i) {
361 if (vi.AliveBlocks[i]) {
Owen Anderson31ec8412008-06-16 19:32:40 +0000362 LiveRange LR(getMBBStartIdx(i),
Evan Chengf26e8552008-06-17 20:13:36 +0000363 getMBBEndIdx(i)+1, // MBB ends at -1.
Owen Anderson31ec8412008-06-16 19:32:40 +0000364 ValNo);
365 interval.addRange(LR);
366 DOUT << " +" << LR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000367 }
368 }
369
370 // Finally, this virtual register is live from the start of any killing
371 // block to the 'use' slot of the killing instruction.
372 for (unsigned i = 0, e = vi.Kills.size(); i != e; ++i) {
373 MachineInstr *Kill = vi.Kills[i];
Evan Cheng8df78602007-08-08 03:00:28 +0000374 unsigned killIdx = getUseIndex(getInstructionIndex(Kill))+1;
Chris Lattner428b92e2006-09-15 03:57:23 +0000375 LiveRange LR(getMBBStartIdx(Kill->getParent()),
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000376 killIdx, ValNo);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000377 interval.addRange(LR);
Evan Chengf3bb2e62007-09-05 21:46:51 +0000378 interval.addKill(ValNo, killIdx);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000379 DOUT << " +" << LR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000380 }
381
382 } else {
383 // If this is the second time we see a virtual register definition, it
384 // must be due to phi elimination or two addr elimination. If this is
Evan Chengbf105c82006-11-03 03:04:46 +0000385 // the result of two address elimination, then the vreg is one of the
386 // def-and-use register operand.
Evan Cheng32dfbea2007-10-12 08:50:34 +0000387 if (mi->isRegReDefinedByTwoAddr(interval.reg)) {
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000388 // If this is a two-address definition, then we have already processed
389 // the live range. The only problem is that we didn't realize there
390 // are actually two values in the live interval. Because of this we
391 // need to take the LiveRegion that defines this register and split it
392 // into two values.
Evan Chenga07cec92008-01-10 08:22:10 +0000393 assert(interval.containsOneValue());
394 unsigned DefIndex = getDefIndex(interval.getValNumInfo(0)->def);
Chris Lattner6b128bd2006-09-03 08:07:11 +0000395 unsigned RedefIndex = getDefIndex(MIIdx);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000396
Evan Cheng4f8ff162007-08-11 00:59:19 +0000397 const LiveRange *OldLR = interval.getLiveRangeContaining(RedefIndex-1);
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000398 VNInfo *OldValNo = OldLR->valno;
Evan Cheng4f8ff162007-08-11 00:59:19 +0000399
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000400 // Delete the initial value, which should be short and continuous,
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000401 // because the 2-addr copy must be in the same MBB as the redef.
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000402 interval.removeRange(DefIndex, RedefIndex);
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000403
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000404 // Two-address vregs should always only be redefined once. This means
405 // that at this point, there should be exactly one value number in it.
406 assert(interval.containsOneValue() && "Unexpected 2-addr liveint!");
407
Chris Lattner91725b72006-08-31 05:54:43 +0000408 // The new value number (#1) is defined by the instruction we claimed
409 // defined value #0.
Evan Chengc8d044e2008-02-15 18:24:29 +0000410 VNInfo *ValNo = interval.getNextValue(OldValNo->def, OldValNo->copy,
411 VNInfoAllocator);
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000412
Chris Lattner91725b72006-08-31 05:54:43 +0000413 // Value#0 is now defined by the 2-addr instruction.
Evan Chengc8d044e2008-02-15 18:24:29 +0000414 OldValNo->def = RedefIndex;
415 OldValNo->copy = 0;
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000416
417 // Add the new live interval which replaces the range for the input copy.
418 LiveRange LR(DefIndex, RedefIndex, ValNo);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000419 DOUT << " replace range with " << LR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000420 interval.addRange(LR);
Evan Chengf3bb2e62007-09-05 21:46:51 +0000421 interval.addKill(ValNo, RedefIndex);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000422
423 // If this redefinition is dead, we need to add a dummy unit live
424 // range covering the def slot.
Evan Cheng6130f662008-03-05 00:59:57 +0000425 if (mi->registerDefIsDead(interval.reg, tri_))
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000426 interval.addRange(LiveRange(RedefIndex, RedefIndex+1, OldValNo));
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000427
Evan Cheng56fdd7a2007-03-15 21:19:28 +0000428 DOUT << " RESULT: ";
Dan Gohman6f0d0242008-02-10 18:45:23 +0000429 interval.print(DOUT, tri_);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000430
431 } else {
432 // Otherwise, this must be because of phi elimination. If this is the
433 // first redefinition of the vreg that we have seen, go back and change
434 // the live range in the PHI block to be a different value number.
435 if (interval.containsOneValue()) {
436 assert(vi.Kills.size() == 1 &&
437 "PHI elimination vreg should have one kill, the PHI itself!");
438
439 // Remove the old range that we now know has an incorrect number.
Evan Chengf3bb2e62007-09-05 21:46:51 +0000440 VNInfo *VNI = interval.getValNumInfo(0);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000441 MachineInstr *Killer = vi.Kills[0];
Chris Lattner428b92e2006-09-15 03:57:23 +0000442 unsigned Start = getMBBStartIdx(Killer->getParent());
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000443 unsigned End = getUseIndex(getInstructionIndex(Killer))+1;
Evan Cheng56fdd7a2007-03-15 21:19:28 +0000444 DOUT << " Removing [" << Start << "," << End << "] from: ";
Dan Gohman6f0d0242008-02-10 18:45:23 +0000445 interval.print(DOUT, tri_); DOUT << "\n";
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000446 interval.removeRange(Start, End);
Evan Chengc3fc7d92007-11-29 09:49:23 +0000447 VNI->hasPHIKill = true;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000448 DOUT << " RESULT: "; interval.print(DOUT, tri_);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000449
Chris Lattnerbe4f88a2006-08-22 18:19:46 +0000450 // Replace the interval with one of a NEW value number. Note that this
451 // value number isn't actually defined by an instruction, weird huh? :)
Evan Chengf3bb2e62007-09-05 21:46:51 +0000452 LiveRange LR(Start, End, interval.getNextValue(~0, 0, VNInfoAllocator));
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000453 DOUT << " replace range with " << LR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000454 interval.addRange(LR);
Evan Chengf3bb2e62007-09-05 21:46:51 +0000455 interval.addKill(LR.valno, End);
Dan Gohman6f0d0242008-02-10 18:45:23 +0000456 DOUT << " RESULT: "; interval.print(DOUT, tri_);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000457 }
458
459 // In the case of PHI elimination, each variable definition is only
460 // live until the end of the block. We've already taken care of the
461 // rest of the live range.
Chris Lattner6b128bd2006-09-03 08:07:11 +0000462 unsigned defIndex = getDefIndex(MIIdx);
Chris Lattner91725b72006-08-31 05:54:43 +0000463
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000464 VNInfo *ValNo;
Evan Chengc8d044e2008-02-15 18:24:29 +0000465 MachineInstr *CopyMI = NULL;
Chris Lattner91725b72006-08-31 05:54:43 +0000466 unsigned SrcReg, DstReg;
Evan Chengc8d044e2008-02-15 18:24:29 +0000467 if (mi->getOpcode() == TargetInstrInfo::EXTRACT_SUBREG ||
Evan Cheng7e073ba2008-04-09 20:57:25 +0000468 mi->getOpcode() == TargetInstrInfo::INSERT_SUBREG ||
Evan Chengc8d044e2008-02-15 18:24:29 +0000469 tii_->isMoveInstr(*mi, SrcReg, DstReg))
470 CopyMI = mi;
471 ValNo = interval.getNextValue(defIndex, CopyMI, VNInfoAllocator);
Chris Lattner91725b72006-08-31 05:54:43 +0000472
Owen Anderson6c5e5612008-06-19 00:10:49 +0000473 unsigned killIndex = getMBBEndIdx(mbb) + 1;
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000474 LiveRange LR(defIndex, killIndex, ValNo);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000475 interval.addRange(LR);
Evan Chengc3fc7d92007-11-29 09:49:23 +0000476 interval.addKill(ValNo, killIndex);
477 ValNo->hasPHIKill = true;
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000478 DOUT << " +" << LR;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000479 }
480 }
481
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000482 DOUT << '\n';
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000483}
484
Chris Lattnerf35fef72004-07-23 21:24:19 +0000485void LiveIntervals::handlePhysicalRegisterDef(MachineBasicBlock *MBB,
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000486 MachineBasicBlock::iterator mi,
Chris Lattner6b128bd2006-09-03 08:07:11 +0000487 unsigned MIIdx,
Chris Lattner91725b72006-08-31 05:54:43 +0000488 LiveInterval &interval,
Evan Chengc8d044e2008-02-15 18:24:29 +0000489 MachineInstr *CopyMI) {
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000490 // A physical register cannot be live across basic block, so its
491 // lifetime must end somewhere in its defining basic block.
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000492 DOUT << "\t\tregister: "; DEBUG(printRegName(interval.reg));
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000493
Chris Lattner6b128bd2006-09-03 08:07:11 +0000494 unsigned baseIndex = MIIdx;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000495 unsigned start = getDefIndex(baseIndex);
496 unsigned end = start;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000497
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000498 // If it is not used after definition, it is considered dead at
499 // the instruction defining it. Hence its interval is:
500 // [defSlot(def), defSlot(def)+1)
Evan Cheng6130f662008-03-05 00:59:57 +0000501 if (mi->registerDefIsDead(interval.reg, tri_)) {
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000502 DOUT << " dead";
Chris Lattnerab4b66d2005-08-23 22:51:41 +0000503 end = getDefIndex(start) + 1;
504 goto exit;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000505 }
506
507 // If it is not dead on definition, it must be killed by a
508 // subsequent instruction. Hence its interval is:
509 // [defSlot(def), useSlot(kill)+1)
Owen Anderson6c5e5612008-06-19 00:10:49 +0000510 baseIndex += InstrSlots::NUM;
Chris Lattner5ab6f5f2005-09-02 00:20:32 +0000511 while (++mi != MBB->end()) {
Owen Anderson6c5e5612008-06-19 00:10:49 +0000512 while (getInstructionFromIndex(baseIndex) == 0)
513 baseIndex += InstrSlots::NUM;
Evan Cheng6130f662008-03-05 00:59:57 +0000514 if (mi->killsRegister(interval.reg, tri_)) {
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000515 DOUT << " killed";
Chris Lattnerab4b66d2005-08-23 22:51:41 +0000516 end = getUseIndex(baseIndex) + 1;
517 goto exit;
Evan Cheng6130f662008-03-05 00:59:57 +0000518 } else if (mi->modifiesRegister(interval.reg, tri_)) {
Evan Cheng9a1956a2006-11-15 20:54:11 +0000519 // Another instruction redefines the register before it is ever read.
520 // Then the register is essentially dead at the instruction that defines
521 // it. Hence its interval is:
522 // [defSlot(def), defSlot(def)+1)
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000523 DOUT << " dead";
Evan Cheng9a1956a2006-11-15 20:54:11 +0000524 end = getDefIndex(start) + 1;
525 goto exit;
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000526 }
Owen Anderson6c5e5612008-06-19 00:10:49 +0000527
528 baseIndex += InstrSlots::NUM;
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000529 }
Chris Lattner5ab6f5f2005-09-02 00:20:32 +0000530
531 // The only case we should have a dead physreg here without a killing or
532 // instruction where we know it's dead is if it is live-in to the function
533 // and never used.
Evan Chengc8d044e2008-02-15 18:24:29 +0000534 assert(!CopyMI && "physreg was not killed in defining block!");
Chris Lattner5ab6f5f2005-09-02 00:20:32 +0000535 end = getDefIndex(start) + 1; // It's dead.
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000536
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000537exit:
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000538 assert(start < end && "did not find end of interval?");
Chris Lattnerf768bba2005-03-09 23:05:19 +0000539
Evan Cheng24a3cc42007-04-25 07:30:23 +0000540 // Already exists? Extend old live interval.
541 LiveInterval::iterator OldLR = interval.FindLiveRangeContaining(start);
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000542 VNInfo *ValNo = (OldLR != interval.end())
Evan Chengc8d044e2008-02-15 18:24:29 +0000543 ? OldLR->valno : interval.getNextValue(start, CopyMI, VNInfoAllocator);
Evan Cheng7ecb38b2007-08-29 20:45:00 +0000544 LiveRange LR(start, end, ValNo);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000545 interval.addRange(LR);
Evan Chengf3bb2e62007-09-05 21:46:51 +0000546 interval.addKill(LR.valno, end);
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000547 DOUT << " +" << LR << '\n';
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000548}
549
Chris Lattnerf35fef72004-07-23 21:24:19 +0000550void LiveIntervals::handleRegisterDef(MachineBasicBlock *MBB,
551 MachineBasicBlock::iterator MI,
Chris Lattner6b128bd2006-09-03 08:07:11 +0000552 unsigned MIIdx,
Chris Lattnerf35fef72004-07-23 21:24:19 +0000553 unsigned reg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000554 if (TargetRegisterInfo::isVirtualRegister(reg))
Chris Lattner6b128bd2006-09-03 08:07:11 +0000555 handleVirtualRegisterDef(MBB, MI, MIIdx, getOrCreateInterval(reg));
Alkis Evlogimenos53278012004-08-26 22:22:38 +0000556 else if (allocatableRegs_[reg]) {
Evan Chengc8d044e2008-02-15 18:24:29 +0000557 MachineInstr *CopyMI = NULL;
Chris Lattner91725b72006-08-31 05:54:43 +0000558 unsigned SrcReg, DstReg;
Evan Chengc8d044e2008-02-15 18:24:29 +0000559 if (MI->getOpcode() == TargetInstrInfo::EXTRACT_SUBREG ||
Evan Cheng7e073ba2008-04-09 20:57:25 +0000560 MI->getOpcode() == TargetInstrInfo::INSERT_SUBREG ||
Evan Chengc8d044e2008-02-15 18:24:29 +0000561 tii_->isMoveInstr(*MI, SrcReg, DstReg))
562 CopyMI = MI;
563 handlePhysicalRegisterDef(MBB, MI, MIIdx, getOrCreateInterval(reg), CopyMI);
Evan Cheng24a3cc42007-04-25 07:30:23 +0000564 // Def of a register also defines its sub-registers.
Dan Gohman6f0d0242008-02-10 18:45:23 +0000565 for (const unsigned* AS = tri_->getSubRegisters(reg); *AS; ++AS)
Evan Cheng6130f662008-03-05 00:59:57 +0000566 // If MI also modifies the sub-register explicitly, avoid processing it
567 // more than once. Do not pass in TRI here so it checks for exact match.
568 if (!MI->modifiesRegister(*AS))
Evan Cheng24a3cc42007-04-25 07:30:23 +0000569 handlePhysicalRegisterDef(MBB, MI, MIIdx, getOrCreateInterval(*AS), 0);
Chris Lattnerf35fef72004-07-23 21:24:19 +0000570 }
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000571}
572
Evan Chengb371f452007-02-19 21:49:54 +0000573void LiveIntervals::handleLiveInRegister(MachineBasicBlock *MBB,
Jim Laskey9b25b8c2007-02-21 22:41:17 +0000574 unsigned MIIdx,
Evan Cheng24a3cc42007-04-25 07:30:23 +0000575 LiveInterval &interval, bool isAlias) {
Evan Chengb371f452007-02-19 21:49:54 +0000576 DOUT << "\t\tlivein register: "; DEBUG(printRegName(interval.reg));
577
578 // Look for kills, if it reaches a def before it's killed, then it shouldn't
579 // be considered a livein.
580 MachineBasicBlock::iterator mi = MBB->begin();
Jim Laskey9b25b8c2007-02-21 22:41:17 +0000581 unsigned baseIndex = MIIdx;
582 unsigned start = baseIndex;
Evan Chengb371f452007-02-19 21:49:54 +0000583 unsigned end = start;
584 while (mi != MBB->end()) {
Evan Cheng6130f662008-03-05 00:59:57 +0000585 if (mi->killsRegister(interval.reg, tri_)) {
Evan Chengb371f452007-02-19 21:49:54 +0000586 DOUT << " killed";
587 end = getUseIndex(baseIndex) + 1;
588 goto exit;
Evan Cheng6130f662008-03-05 00:59:57 +0000589 } else if (mi->modifiesRegister(interval.reg, tri_)) {
Evan Chengb371f452007-02-19 21:49:54 +0000590 // Another instruction redefines the register before it is ever read.
591 // Then the register is essentially dead at the instruction that defines
592 // it. Hence its interval is:
593 // [defSlot(def), defSlot(def)+1)
594 DOUT << " dead";
595 end = getDefIndex(start) + 1;
596 goto exit;
597 }
598
599 baseIndex += InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000600 while (getInstructionFromIndex(baseIndex) == 0)
601 baseIndex += InstrSlots::NUM;
Evan Chengb371f452007-02-19 21:49:54 +0000602 ++mi;
603 }
604
605exit:
Evan Cheng75611fb2007-06-27 01:16:36 +0000606 // Live-in register might not be used at all.
607 if (end == MIIdx) {
Evan Cheng292da942007-06-27 18:47:28 +0000608 if (isAlias) {
609 DOUT << " dead";
Evan Cheng75611fb2007-06-27 01:16:36 +0000610 end = getDefIndex(MIIdx) + 1;
Evan Cheng292da942007-06-27 18:47:28 +0000611 } else {
612 DOUT << " live through";
613 end = baseIndex;
614 }
Evan Cheng24a3cc42007-04-25 07:30:23 +0000615 }
616
Evan Chengf3bb2e62007-09-05 21:46:51 +0000617 LiveRange LR(start, end, interval.getNextValue(start, 0, VNInfoAllocator));
Jim Laskey9b25b8c2007-02-21 22:41:17 +0000618 interval.addRange(LR);
Evan Chengf3bb2e62007-09-05 21:46:51 +0000619 interval.addKill(LR.valno, end);
Evan Cheng24c2e5c2007-08-08 07:03:29 +0000620 DOUT << " +" << LR << '\n';
Evan Chengb371f452007-02-19 21:49:54 +0000621}
622
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000623/// computeIntervals - computes the live intervals for virtual
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000624/// registers. for some ordering of the machine instructions [1,N] a
Alkis Evlogimenos08cec002004-01-31 19:59:32 +0000625/// live interval is an interval [i, j) where 1 <= i <= j < N for
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000626/// which a variable is live
Chris Lattnerf7da2c72006-08-24 22:43:55 +0000627void LiveIntervals::computeIntervals() {
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000628 DOUT << "********** COMPUTING LIVE INTERVALS **********\n"
629 << "********** Function: "
630 << ((Value*)mf_->getFunction())->getName() << '\n';
Chris Lattner6b128bd2006-09-03 08:07:11 +0000631 // Track the index of the current machine instr.
632 unsigned MIIndex = 0;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000633
634 // Skip over empty initial indices.
635 while (MIIndex / InstrSlots::NUM < i2miMap_.size() &&
636 getInstructionFromIndex(MIIndex) == 0)
637 MIIndex += InstrSlots::NUM;
638
Chris Lattner428b92e2006-09-15 03:57:23 +0000639 for (MachineFunction::iterator MBBI = mf_->begin(), E = mf_->end();
640 MBBI != E; ++MBBI) {
641 MachineBasicBlock *MBB = MBBI;
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000642 DOUT << ((Value*)MBB->getBasicBlock())->getName() << ":\n";
Alkis Evlogimenos6b4edba2003-12-21 20:19:10 +0000643
Chris Lattner428b92e2006-09-15 03:57:23 +0000644 MachineBasicBlock::iterator MI = MBB->begin(), miEnd = MBB->end();
Evan Cheng0c9f92e2007-02-13 01:30:55 +0000645
Dan Gohmancb406c22007-10-03 19:26:29 +0000646 // Create intervals for live-ins to this BB first.
647 for (MachineBasicBlock::const_livein_iterator LI = MBB->livein_begin(),
648 LE = MBB->livein_end(); LI != LE; ++LI) {
649 handleLiveInRegister(MBB, MIIndex, getOrCreateInterval(*LI));
650 // Multiple live-ins can alias the same register.
Dan Gohman6f0d0242008-02-10 18:45:23 +0000651 for (const unsigned* AS = tri_->getSubRegisters(*LI); *AS; ++AS)
Dan Gohmancb406c22007-10-03 19:26:29 +0000652 if (!hasInterval(*AS))
653 handleLiveInRegister(MBB, MIIndex, getOrCreateInterval(*AS),
654 true);
Chris Lattnerdffb2e82006-09-04 18:27:40 +0000655 }
656
Chris Lattner428b92e2006-09-15 03:57:23 +0000657 for (; MI != miEnd; ++MI) {
Bill Wendlingbdc679d2006-11-29 00:39:47 +0000658 DOUT << MIIndex << "\t" << *MI;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000659
Evan Cheng438f7bc2006-11-10 08:43:01 +0000660 // Handle defs.
Chris Lattner428b92e2006-09-15 03:57:23 +0000661 for (int i = MI->getNumOperands() - 1; i >= 0; --i) {
662 MachineOperand &MO = MI->getOperand(i);
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000663 // handle register defs - build intervals
Chris Lattner428b92e2006-09-15 03:57:23 +0000664 if (MO.isRegister() && MO.getReg() && MO.isDef())
665 handleRegisterDef(MBB, MI, MIIndex, MO.getReg());
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000666 }
Chris Lattner6b128bd2006-09-03 08:07:11 +0000667
668 MIIndex += InstrSlots::NUM;
Owen Anderson6c5e5612008-06-19 00:10:49 +0000669
670 // Skip over empty indices.
671 while (MIIndex / InstrSlots::NUM < i2miMap_.size() &&
672 getInstructionFromIndex(MIIndex) == 0)
673 MIIndex += InstrSlots::NUM;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000674 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000675 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000676}
Alkis Evlogimenosb27ef242003-12-05 10:38:28 +0000677
Evan Cheng4ca980e2007-10-17 02:10:22 +0000678bool LiveIntervals::findLiveInMBBs(const LiveRange &LR,
Evan Chenga5bfc972007-10-17 06:53:44 +0000679 SmallVectorImpl<MachineBasicBlock*> &MBBs) const {
Evan Cheng4ca980e2007-10-17 02:10:22 +0000680 std::vector<IdxMBBPair>::const_iterator I =
681 std::lower_bound(Idx2MBBMap.begin(), Idx2MBBMap.end(), LR.start);
682
683 bool ResVal = false;
684 while (I != Idx2MBBMap.end()) {
685 if (LR.end <= I->first)
686 break;
687 MBBs.push_back(I->second);
688 ResVal = true;
689 ++I;
690 }
691 return ResVal;
692}
693
694
Alkis Evlogimenosa1613db2004-07-24 11:44:15 +0000695LiveInterval LiveIntervals::createInterval(unsigned reg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000696 float Weight = TargetRegisterInfo::isPhysicalRegister(reg) ?
Jim Laskey7902c752006-11-07 12:25:45 +0000697 HUGE_VALF : 0.0F;
Alkis Evlogimenosa1613db2004-07-24 11:44:15 +0000698 return LiveInterval(reg, Weight);
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000699}
Evan Chengf2fbca62007-11-12 06:35:08 +0000700
Evan Chengc8d044e2008-02-15 18:24:29 +0000701/// getVNInfoSourceReg - Helper function that parses the specified VNInfo
702/// copy field and returns the source register that defines it.
703unsigned LiveIntervals::getVNInfoSourceReg(const VNInfo *VNI) const {
704 if (!VNI->copy)
705 return 0;
706
707 if (VNI->copy->getOpcode() == TargetInstrInfo::EXTRACT_SUBREG)
708 return VNI->copy->getOperand(1).getReg();
Evan Cheng7e073ba2008-04-09 20:57:25 +0000709 if (VNI->copy->getOpcode() == TargetInstrInfo::INSERT_SUBREG)
710 return VNI->copy->getOperand(2).getReg();
Evan Chengc8d044e2008-02-15 18:24:29 +0000711 unsigned SrcReg, DstReg;
712 if (tii_->isMoveInstr(*VNI->copy, SrcReg, DstReg))
713 return SrcReg;
714 assert(0 && "Unrecognized copy instruction!");
715 return 0;
716}
Evan Chengf2fbca62007-11-12 06:35:08 +0000717
718//===----------------------------------------------------------------------===//
719// Register allocator hooks.
720//
721
Evan Chengd70dbb52008-02-22 09:24:50 +0000722/// getReMatImplicitUse - If the remat definition MI has one (for now, we only
723/// allow one) virtual register operand, then its uses are implicitly using
724/// the register. Returns the virtual register.
725unsigned LiveIntervals::getReMatImplicitUse(const LiveInterval &li,
726 MachineInstr *MI) const {
727 unsigned RegOp = 0;
728 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
729 MachineOperand &MO = MI->getOperand(i);
730 if (!MO.isRegister() || !MO.isUse())
731 continue;
732 unsigned Reg = MO.getReg();
733 if (Reg == 0 || Reg == li.reg)
734 continue;
735 // FIXME: For now, only remat MI with at most one register operand.
736 assert(!RegOp &&
737 "Can't rematerialize instruction with multiple register operand!");
738 RegOp = MO.getReg();
739 break;
740 }
741 return RegOp;
742}
743
744/// isValNoAvailableAt - Return true if the val# of the specified interval
745/// which reaches the given instruction also reaches the specified use index.
746bool LiveIntervals::isValNoAvailableAt(const LiveInterval &li, MachineInstr *MI,
747 unsigned UseIdx) const {
748 unsigned Index = getInstructionIndex(MI);
749 VNInfo *ValNo = li.FindLiveRangeContaining(Index)->valno;
750 LiveInterval::const_iterator UI = li.FindLiveRangeContaining(UseIdx);
751 return UI != li.end() && UI->valno == ValNo;
752}
753
Evan Chengf2fbca62007-11-12 06:35:08 +0000754/// isReMaterializable - Returns true if the definition MI of the specified
755/// val# of the specified interval is re-materializable.
756bool LiveIntervals::isReMaterializable(const LiveInterval &li,
Evan Cheng5ef3a042007-12-06 00:01:56 +0000757 const VNInfo *ValNo, MachineInstr *MI,
758 bool &isLoad) {
Evan Chengf2fbca62007-11-12 06:35:08 +0000759 if (DisableReMat)
760 return false;
761
Evan Cheng5ef3a042007-12-06 00:01:56 +0000762 isLoad = false;
Evan Cheng20ccded2008-03-15 00:19:36 +0000763 if (MI->getOpcode() == TargetInstrInfo::IMPLICIT_DEF)
Evan Chengd70dbb52008-02-22 09:24:50 +0000764 return true;
Evan Chengdd3465e2008-02-23 01:44:27 +0000765
766 int FrameIdx = 0;
767 if (tii_->isLoadFromStackSlot(MI, FrameIdx) &&
Evan Cheng249ded32008-02-23 03:38:34 +0000768 mf_->getFrameInfo()->isImmutableObjectIndex(FrameIdx))
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000769 // FIXME: Let target specific isReallyTriviallyReMaterializable determines
770 // this but remember this is not safe to fold into a two-address
771 // instruction.
Evan Cheng249ded32008-02-23 03:38:34 +0000772 // This is a load from fixed stack slot. It can be rematerialized.
Evan Chengdd3465e2008-02-23 01:44:27 +0000773 return true;
Evan Chengdd3465e2008-02-23 01:44:27 +0000774
Evan Chengd70dbb52008-02-22 09:24:50 +0000775 if (tii_->isTriviallyReMaterializable(MI)) {
Evan Cheng20ccded2008-03-15 00:19:36 +0000776 const TargetInstrDesc &TID = MI->getDesc();
Chris Lattner749c6f62008-01-07 07:27:27 +0000777 isLoad = TID.isSimpleLoad();
Evan Chengd70dbb52008-02-22 09:24:50 +0000778
779 unsigned ImpUse = getReMatImplicitUse(li, MI);
780 if (ImpUse) {
781 const LiveInterval &ImpLi = getInterval(ImpUse);
782 for (MachineRegisterInfo::use_iterator ri = mri_->use_begin(li.reg),
783 re = mri_->use_end(); ri != re; ++ri) {
784 MachineInstr *UseMI = &*ri;
785 unsigned UseIdx = getInstructionIndex(UseMI);
786 if (li.FindLiveRangeContaining(UseIdx)->valno != ValNo)
787 continue;
Evan Cheng298bbe82008-02-23 02:14:42 +0000788 if (!isValNoAvailableAt(ImpLi, MI, UseIdx))
Evan Chengd70dbb52008-02-22 09:24:50 +0000789 return false;
790 }
791 }
Evan Chengf2fbca62007-11-12 06:35:08 +0000792 return true;
Evan Cheng5ef3a042007-12-06 00:01:56 +0000793 }
Evan Chengf2fbca62007-11-12 06:35:08 +0000794
Evan Chengdd3465e2008-02-23 01:44:27 +0000795 return false;
Evan Cheng5ef3a042007-12-06 00:01:56 +0000796}
797
798/// isReMaterializable - Returns true if every definition of MI of every
799/// val# of the specified interval is re-materializable.
800bool LiveIntervals::isReMaterializable(const LiveInterval &li, bool &isLoad) {
801 isLoad = false;
802 for (LiveInterval::const_vni_iterator i = li.vni_begin(), e = li.vni_end();
803 i != e; ++i) {
804 const VNInfo *VNI = *i;
805 unsigned DefIdx = VNI->def;
806 if (DefIdx == ~1U)
807 continue; // Dead val#.
808 // Is the def for the val# rematerializable?
809 if (DefIdx == ~0u)
810 return false;
811 MachineInstr *ReMatDefMI = getInstructionFromIndex(DefIdx);
812 bool DefIsLoad = false;
Evan Chengd70dbb52008-02-22 09:24:50 +0000813 if (!ReMatDefMI ||
814 !isReMaterializable(li, VNI, ReMatDefMI, DefIsLoad))
Evan Cheng5ef3a042007-12-06 00:01:56 +0000815 return false;
816 isLoad |= DefIsLoad;
Evan Chengf2fbca62007-11-12 06:35:08 +0000817 }
818 return true;
819}
820
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000821/// FilterFoldedOps - Filter out two-address use operands. Return
822/// true if it finds any issue with the operands that ought to prevent
823/// folding.
824static bool FilterFoldedOps(MachineInstr *MI,
825 SmallVector<unsigned, 2> &Ops,
826 unsigned &MRInfo,
827 SmallVector<unsigned, 2> &FoldOps) {
Chris Lattner749c6f62008-01-07 07:27:27 +0000828 const TargetInstrDesc &TID = MI->getDesc();
Evan Cheng6e141fd2007-12-12 23:12:09 +0000829
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000830 MRInfo = 0;
Evan Chengaee4af62007-12-02 08:30:39 +0000831 for (unsigned i = 0, e = Ops.size(); i != e; ++i) {
832 unsigned OpIdx = Ops[i];
Evan Chengd70dbb52008-02-22 09:24:50 +0000833 MachineOperand &MO = MI->getOperand(OpIdx);
Evan Chengaee4af62007-12-02 08:30:39 +0000834 // FIXME: fold subreg use.
Evan Chengd70dbb52008-02-22 09:24:50 +0000835 if (MO.getSubReg())
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000836 return true;
Evan Chengd70dbb52008-02-22 09:24:50 +0000837 if (MO.isDef())
Evan Chengaee4af62007-12-02 08:30:39 +0000838 MRInfo |= (unsigned)VirtRegMap::isMod;
839 else {
840 // Filter out two-address use operand(s).
Evan Chengd70dbb52008-02-22 09:24:50 +0000841 if (!MO.isImplicit() &&
842 TID.getOperandConstraint(OpIdx, TOI::TIED_TO) != -1) {
Evan Chengaee4af62007-12-02 08:30:39 +0000843 MRInfo = VirtRegMap::isModRef;
844 continue;
845 }
846 MRInfo |= (unsigned)VirtRegMap::isRef;
847 }
848 FoldOps.push_back(OpIdx);
Evan Chenge62f97c2007-12-01 02:07:52 +0000849 }
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000850 return false;
851}
852
853
854/// tryFoldMemoryOperand - Attempts to fold either a spill / restore from
855/// slot / to reg or any rematerialized load into ith operand of specified
856/// MI. If it is successul, MI is updated with the newly created MI and
857/// returns true.
858bool LiveIntervals::tryFoldMemoryOperand(MachineInstr* &MI,
859 VirtRegMap &vrm, MachineInstr *DefMI,
860 unsigned InstrIdx,
861 SmallVector<unsigned, 2> &Ops,
862 bool isSS, int Slot, unsigned Reg) {
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000863 // If it is an implicit def instruction, just delete it.
Evan Cheng20ccded2008-03-15 00:19:36 +0000864 if (MI->getOpcode() == TargetInstrInfo::IMPLICIT_DEF) {
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000865 RemoveMachineInstrFromMaps(MI);
866 vrm.RemoveMachineInstrFromMaps(MI);
867 MI->eraseFromParent();
868 ++numFolds;
869 return true;
870 }
871
872 // Filter the list of operand indexes that are to be folded. Abort if
873 // any operand will prevent folding.
874 unsigned MRInfo = 0;
875 SmallVector<unsigned, 2> FoldOps;
876 if (FilterFoldedOps(MI, Ops, MRInfo, FoldOps))
877 return false;
Evan Chenge62f97c2007-12-01 02:07:52 +0000878
Evan Cheng427f4c12008-03-31 23:19:51 +0000879 // The only time it's safe to fold into a two address instruction is when
880 // it's folding reload and spill from / into a spill stack slot.
881 if (DefMI && (MRInfo & VirtRegMap::isMod))
Evan Cheng249ded32008-02-23 03:38:34 +0000882 return false;
883
Evan Chengf2f8c2a2008-02-08 22:05:27 +0000884 MachineInstr *fmi = isSS ? tii_->foldMemoryOperand(*mf_, MI, FoldOps, Slot)
885 : tii_->foldMemoryOperand(*mf_, MI, FoldOps, DefMI);
Evan Chengf2fbca62007-11-12 06:35:08 +0000886 if (fmi) {
Evan Chengd3653122008-02-27 03:04:06 +0000887 // Remember this instruction uses the spill slot.
888 if (isSS) vrm.addSpillSlotUse(Slot, fmi);
889
Evan Chengf2fbca62007-11-12 06:35:08 +0000890 // Attempt to fold the memory reference into the instruction. If
891 // we can do this, we don't need to insert spill code.
892 if (lv_)
893 lv_->instructionChanged(MI, fmi);
Evan Cheng81a03822007-11-17 00:40:40 +0000894 else
Dan Gohman6f0d0242008-02-10 18:45:23 +0000895 fmi->copyKillDeadInfo(MI, tri_);
Evan Chengf2fbca62007-11-12 06:35:08 +0000896 MachineBasicBlock &MBB = *MI->getParent();
Evan Cheng84802932008-01-10 08:24:38 +0000897 if (isSS && !mf_->getFrameInfo()->isImmutableObjectIndex(Slot))
Evan Chengaee4af62007-12-02 08:30:39 +0000898 vrm.virtFolded(Reg, MI, fmi, (VirtRegMap::ModRef)MRInfo);
Evan Cheng81a03822007-11-17 00:40:40 +0000899 vrm.transferSpillPts(MI, fmi);
Evan Cheng0cbb1162007-11-29 01:06:25 +0000900 vrm.transferRestorePts(MI, fmi);
Evan Chengc1f53c72008-03-11 21:34:46 +0000901 vrm.transferEmergencySpills(MI, fmi);
Evan Chengf2fbca62007-11-12 06:35:08 +0000902 mi2iMap_.erase(MI);
Evan Chengcddbb832007-11-30 21:23:43 +0000903 i2miMap_[InstrIdx /InstrSlots::NUM] = fmi;
904 mi2iMap_[fmi] = InstrIdx;
Evan Chengf2fbca62007-11-12 06:35:08 +0000905 MI = MBB.insert(MBB.erase(MI), fmi);
Evan Cheng0cbb1162007-11-29 01:06:25 +0000906 ++numFolds;
Evan Chengf2fbca62007-11-12 06:35:08 +0000907 return true;
908 }
909 return false;
910}
911
Evan Cheng018f9b02007-12-05 03:22:34 +0000912/// canFoldMemoryOperand - Returns true if the specified load / store
913/// folding is possible.
914bool LiveIntervals::canFoldMemoryOperand(MachineInstr *MI,
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000915 SmallVector<unsigned, 2> &Ops,
Evan Cheng3c75ba82008-04-01 21:37:32 +0000916 bool ReMat) const {
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000917 // Filter the list of operand indexes that are to be folded. Abort if
918 // any operand will prevent folding.
919 unsigned MRInfo = 0;
Evan Cheng018f9b02007-12-05 03:22:34 +0000920 SmallVector<unsigned, 2> FoldOps;
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000921 if (FilterFoldedOps(MI, Ops, MRInfo, FoldOps))
922 return false;
Evan Cheng018f9b02007-12-05 03:22:34 +0000923
Evan Cheng3c75ba82008-04-01 21:37:32 +0000924 // It's only legal to remat for a use, not a def.
925 if (ReMat && (MRInfo & VirtRegMap::isMod))
Evan Cheng79a0c1e2008-02-25 08:50:41 +0000926 return false;
Evan Cheng018f9b02007-12-05 03:22:34 +0000927
Evan Chengd70dbb52008-02-22 09:24:50 +0000928 return tii_->canFoldMemoryOperand(MI, FoldOps);
929}
930
Evan Cheng81a03822007-11-17 00:40:40 +0000931bool LiveIntervals::intervalIsInOneMBB(const LiveInterval &li) const {
932 SmallPtrSet<MachineBasicBlock*, 4> MBBs;
933 for (LiveInterval::Ranges::const_iterator
934 I = li.ranges.begin(), E = li.ranges.end(); I != E; ++I) {
935 std::vector<IdxMBBPair>::const_iterator II =
936 std::lower_bound(Idx2MBBMap.begin(), Idx2MBBMap.end(), I->start);
937 if (II == Idx2MBBMap.end())
938 continue;
939 if (I->end > II->first) // crossing a MBB.
940 return false;
941 MBBs.insert(II->second);
942 if (MBBs.size() > 1)
943 return false;
944 }
945 return true;
946}
947
Evan Chengd70dbb52008-02-22 09:24:50 +0000948/// rewriteImplicitOps - Rewrite implicit use operands of MI (i.e. uses of
949/// interval on to-be re-materialized operands of MI) with new register.
950void LiveIntervals::rewriteImplicitOps(const LiveInterval &li,
951 MachineInstr *MI, unsigned NewVReg,
952 VirtRegMap &vrm) {
953 // There is an implicit use. That means one of the other operand is
954 // being remat'ed and the remat'ed instruction has li.reg as an
955 // use operand. Make sure we rewrite that as well.
956 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
957 MachineOperand &MO = MI->getOperand(i);
958 if (!MO.isRegister())
959 continue;
960 unsigned Reg = MO.getReg();
961 if (Reg == 0 || TargetRegisterInfo::isPhysicalRegister(Reg))
962 continue;
963 if (!vrm.isReMaterialized(Reg))
964 continue;
965 MachineInstr *ReMatMI = vrm.getReMaterializedMI(Reg);
Evan Cheng6130f662008-03-05 00:59:57 +0000966 MachineOperand *UseMO = ReMatMI->findRegisterUseOperand(li.reg);
967 if (UseMO)
968 UseMO->setReg(NewVReg);
Evan Chengd70dbb52008-02-22 09:24:50 +0000969 }
970}
971
Evan Chengf2fbca62007-11-12 06:35:08 +0000972/// rewriteInstructionForSpills, rewriteInstructionsForSpills - Helper functions
973/// for addIntervalsForSpills to rewrite uses / defs for the given live range.
Evan Cheng018f9b02007-12-05 03:22:34 +0000974bool LiveIntervals::
Evan Chengd70dbb52008-02-22 09:24:50 +0000975rewriteInstructionForSpills(const LiveInterval &li, const VNInfo *VNI,
976 bool TrySplit, unsigned index, unsigned end, MachineInstr *MI,
Evan Cheng81a03822007-11-17 00:40:40 +0000977 MachineInstr *ReMatOrigDefMI, MachineInstr *ReMatDefMI,
Evan Chengf2fbca62007-11-12 06:35:08 +0000978 unsigned Slot, int LdSlot,
979 bool isLoad, bool isLoadSS, bool DefIsReMat, bool CanDelete,
Evan Chengd70dbb52008-02-22 09:24:50 +0000980 VirtRegMap &vrm,
Evan Chengf2fbca62007-11-12 06:35:08 +0000981 const TargetRegisterClass* rc,
982 SmallVector<int, 4> &ReMatIds,
Evan Cheng22f07ff2007-12-11 02:09:15 +0000983 const MachineLoopInfo *loopInfo,
Evan Cheng313d4b82008-02-23 00:33:04 +0000984 unsigned &NewVReg, unsigned ImpUse, bool &HasDef, bool &HasUse,
Evan Cheng1953d0c2007-11-29 10:12:14 +0000985 std::map<unsigned,unsigned> &MBBVRegsMap,
Evan Cheng9c3c2212008-06-06 07:54:39 +0000986 std::vector<LiveInterval*> &NewLIs, float &SSWeight) {
987 MachineBasicBlock *MBB = MI->getParent();
988 unsigned loopDepth = loopInfo->getLoopDepth(MBB);
Evan Cheng018f9b02007-12-05 03:22:34 +0000989 bool CanFold = false;
Evan Chengf2fbca62007-11-12 06:35:08 +0000990 RestartInstruction:
991 for (unsigned i = 0; i != MI->getNumOperands(); ++i) {
992 MachineOperand& mop = MI->getOperand(i);
993 if (!mop.isRegister())
994 continue;
995 unsigned Reg = mop.getReg();
996 unsigned RegI = Reg;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000997 if (Reg == 0 || TargetRegisterInfo::isPhysicalRegister(Reg))
Evan Chengf2fbca62007-11-12 06:35:08 +0000998 continue;
Evan Chengf2fbca62007-11-12 06:35:08 +0000999 if (Reg != li.reg)
1000 continue;
1001
1002 bool TryFold = !DefIsReMat;
Evan Chengcb3c3302007-11-29 23:02:50 +00001003 bool FoldSS = true; // Default behavior unless it's a remat.
Evan Chengf2fbca62007-11-12 06:35:08 +00001004 int FoldSlot = Slot;
1005 if (DefIsReMat) {
1006 // If this is the rematerializable definition MI itself and
1007 // all of its uses are rematerialized, simply delete it.
Evan Cheng81a03822007-11-17 00:40:40 +00001008 if (MI == ReMatOrigDefMI && CanDelete) {
Evan Chengcddbb832007-11-30 21:23:43 +00001009 DOUT << "\t\t\t\tErasing re-materlizable def: ";
1010 DOUT << MI << '\n';
Evan Chengf2fbca62007-11-12 06:35:08 +00001011 RemoveMachineInstrFromMaps(MI);
Evan Chengcada2452007-11-28 01:28:46 +00001012 vrm.RemoveMachineInstrFromMaps(MI);
Evan Chengf2fbca62007-11-12 06:35:08 +00001013 MI->eraseFromParent();
1014 break;
1015 }
1016
1017 // If def for this use can't be rematerialized, then try folding.
Evan Cheng0cbb1162007-11-29 01:06:25 +00001018 // If def is rematerializable and it's a load, also try folding.
Evan Chengcb3c3302007-11-29 23:02:50 +00001019 TryFold = !ReMatDefMI || (ReMatDefMI && (MI == ReMatOrigDefMI || isLoad));
Evan Chengf2fbca62007-11-12 06:35:08 +00001020 if (isLoad) {
1021 // Try fold loads (from stack slot, constant pool, etc.) into uses.
1022 FoldSS = isLoadSS;
1023 FoldSlot = LdSlot;
1024 }
1025 }
1026
Evan Chengf2fbca62007-11-12 06:35:08 +00001027 // Scan all of the operands of this instruction rewriting operands
1028 // to use NewVReg instead of li.reg as appropriate. We do this for
1029 // two reasons:
1030 //
1031 // 1. If the instr reads the same spilled vreg multiple times, we
1032 // want to reuse the NewVReg.
1033 // 2. If the instr is a two-addr instruction, we are required to
1034 // keep the src/dst regs pinned.
1035 //
1036 // Keep track of whether we replace a use and/or def so that we can
1037 // create the spill interval with the appropriate range.
Evan Chengcddbb832007-11-30 21:23:43 +00001038
Evan Cheng81a03822007-11-17 00:40:40 +00001039 HasUse = mop.isUse();
1040 HasDef = mop.isDef();
Evan Chengaee4af62007-12-02 08:30:39 +00001041 SmallVector<unsigned, 2> Ops;
1042 Ops.push_back(i);
Evan Chengf2fbca62007-11-12 06:35:08 +00001043 for (unsigned j = i+1, e = MI->getNumOperands(); j != e; ++j) {
Evan Chengaee4af62007-12-02 08:30:39 +00001044 const MachineOperand &MOj = MI->getOperand(j);
1045 if (!MOj.isRegister())
Evan Chengf2fbca62007-11-12 06:35:08 +00001046 continue;
Evan Chengaee4af62007-12-02 08:30:39 +00001047 unsigned RegJ = MOj.getReg();
Dan Gohman6f0d0242008-02-10 18:45:23 +00001048 if (RegJ == 0 || TargetRegisterInfo::isPhysicalRegister(RegJ))
Evan Chengf2fbca62007-11-12 06:35:08 +00001049 continue;
1050 if (RegJ == RegI) {
Evan Chengaee4af62007-12-02 08:30:39 +00001051 Ops.push_back(j);
1052 HasUse |= MOj.isUse();
1053 HasDef |= MOj.isDef();
Evan Chengf2fbca62007-11-12 06:35:08 +00001054 }
1055 }
1056
Evan Cheng9c3c2212008-06-06 07:54:39 +00001057 // Update stack slot spill weight if we are splitting.
1058 float Weight = getSpillWeight(HasDef, HasUse, loopDepth);
1059 if (!TrySplit)
1060 SSWeight += Weight;
1061
1062 if (!TryFold)
1063 CanFold = false;
1064 else {
Evan Cheng018f9b02007-12-05 03:22:34 +00001065 // Do not fold load / store here if we are splitting. We'll find an
1066 // optimal point to insert a load / store later.
1067 if (!TrySplit) {
1068 if (tryFoldMemoryOperand(MI, vrm, ReMatDefMI, index,
1069 Ops, FoldSS, FoldSlot, Reg)) {
1070 // Folding the load/store can completely change the instruction in
1071 // unpredictable ways, rescan it from the beginning.
1072 HasUse = false;
1073 HasDef = false;
1074 CanFold = false;
Evan Cheng9c3c2212008-06-06 07:54:39 +00001075 if (isRemoved(MI)) {
1076 SSWeight -= Weight;
Evan Cheng7e073ba2008-04-09 20:57:25 +00001077 break;
Evan Cheng9c3c2212008-06-06 07:54:39 +00001078 }
Evan Cheng018f9b02007-12-05 03:22:34 +00001079 goto RestartInstruction;
1080 }
1081 } else {
Evan Cheng9c3c2212008-06-06 07:54:39 +00001082 // We'll try to fold it later if it's profitable.
Evan Cheng3c75ba82008-04-01 21:37:32 +00001083 CanFold = canFoldMemoryOperand(MI, Ops, DefIsReMat);
Evan Cheng018f9b02007-12-05 03:22:34 +00001084 }
Evan Cheng9c3c2212008-06-06 07:54:39 +00001085 }
Evan Chengcddbb832007-11-30 21:23:43 +00001086
1087 // Create a new virtual register for the spill interval.
1088 bool CreatedNewVReg = false;
1089 if (NewVReg == 0) {
Evan Chengd70dbb52008-02-22 09:24:50 +00001090 NewVReg = mri_->createVirtualRegister(rc);
Evan Chengcddbb832007-11-30 21:23:43 +00001091 vrm.grow();
1092 CreatedNewVReg = true;
1093 }
1094 mop.setReg(NewVReg);
Evan Chengd70dbb52008-02-22 09:24:50 +00001095 if (mop.isImplicit())
1096 rewriteImplicitOps(li, MI, NewVReg, vrm);
Evan Chengcddbb832007-11-30 21:23:43 +00001097
1098 // Reuse NewVReg for other reads.
Evan Chengd70dbb52008-02-22 09:24:50 +00001099 for (unsigned j = 0, e = Ops.size(); j != e; ++j) {
1100 MachineOperand &mopj = MI->getOperand(Ops[j]);
1101 mopj.setReg(NewVReg);
1102 if (mopj.isImplicit())
1103 rewriteImplicitOps(li, MI, NewVReg, vrm);
1104 }
Evan Chengcddbb832007-11-30 21:23:43 +00001105
Evan Cheng81a03822007-11-17 00:40:40 +00001106 if (CreatedNewVReg) {
1107 if (DefIsReMat) {
1108 vrm.setVirtIsReMaterialized(NewVReg, ReMatDefMI/*, CanDelete*/);
Evan Chengd70dbb52008-02-22 09:24:50 +00001109 if (ReMatIds[VNI->id] == VirtRegMap::MAX_STACK_SLOT) {
Evan Cheng81a03822007-11-17 00:40:40 +00001110 // Each valnum may have its own remat id.
Evan Chengd70dbb52008-02-22 09:24:50 +00001111 ReMatIds[VNI->id] = vrm.assignVirtReMatId(NewVReg);
Evan Cheng81a03822007-11-17 00:40:40 +00001112 } else {
Evan Chengd70dbb52008-02-22 09:24:50 +00001113 vrm.assignVirtReMatId(NewVReg, ReMatIds[VNI->id]);
Evan Cheng81a03822007-11-17 00:40:40 +00001114 }
1115 if (!CanDelete || (HasUse && HasDef)) {
1116 // If this is a two-addr instruction then its use operands are
1117 // rematerializable but its def is not. It should be assigned a
1118 // stack slot.
1119 vrm.assignVirt2StackSlot(NewVReg, Slot);
1120 }
Evan Chengf2fbca62007-11-12 06:35:08 +00001121 } else {
Evan Chengf2fbca62007-11-12 06:35:08 +00001122 vrm.assignVirt2StackSlot(NewVReg, Slot);
1123 }
Evan Chengcb3c3302007-11-29 23:02:50 +00001124 } else if (HasUse && HasDef &&
1125 vrm.getStackSlot(NewVReg) == VirtRegMap::NO_STACK_SLOT) {
1126 // If this interval hasn't been assigned a stack slot (because earlier
1127 // def is a deleted remat def), do it now.
1128 assert(Slot != VirtRegMap::NO_STACK_SLOT);
1129 vrm.assignVirt2StackSlot(NewVReg, Slot);
Evan Chengf2fbca62007-11-12 06:35:08 +00001130 }
1131
Evan Cheng313d4b82008-02-23 00:33:04 +00001132 // Re-matting an instruction with virtual register use. Add the
1133 // register as an implicit use on the use MI.
1134 if (DefIsReMat && ImpUse)
1135 MI->addOperand(MachineOperand::CreateReg(ImpUse, false, true));
1136
Evan Chengf2fbca62007-11-12 06:35:08 +00001137 // create a new register interval for this spill / remat.
1138 LiveInterval &nI = getOrCreateInterval(NewVReg);
Evan Cheng81a03822007-11-17 00:40:40 +00001139 if (CreatedNewVReg) {
1140 NewLIs.push_back(&nI);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001141 MBBVRegsMap.insert(std::make_pair(MI->getParent()->getNumber(), NewVReg));
Evan Cheng81a03822007-11-17 00:40:40 +00001142 if (TrySplit)
1143 vrm.setIsSplitFromReg(NewVReg, li.reg);
1144 }
Evan Chengf2fbca62007-11-12 06:35:08 +00001145
1146 if (HasUse) {
Evan Cheng81a03822007-11-17 00:40:40 +00001147 if (CreatedNewVReg) {
1148 LiveRange LR(getLoadIndex(index), getUseIndex(index)+1,
1149 nI.getNextValue(~0U, 0, VNInfoAllocator));
1150 DOUT << " +" << LR;
1151 nI.addRange(LR);
1152 } else {
1153 // Extend the split live interval to this def / use.
1154 unsigned End = getUseIndex(index)+1;
1155 LiveRange LR(nI.ranges[nI.ranges.size()-1].end, End,
1156 nI.getValNumInfo(nI.getNumValNums()-1));
1157 DOUT << " +" << LR;
1158 nI.addRange(LR);
1159 }
Evan Chengf2fbca62007-11-12 06:35:08 +00001160 }
1161 if (HasDef) {
1162 LiveRange LR(getDefIndex(index), getStoreIndex(index),
1163 nI.getNextValue(~0U, 0, VNInfoAllocator));
1164 DOUT << " +" << LR;
1165 nI.addRange(LR);
1166 }
Evan Cheng81a03822007-11-17 00:40:40 +00001167
Evan Chengf2fbca62007-11-12 06:35:08 +00001168 DOUT << "\t\t\t\tAdded new interval: ";
Dan Gohman6f0d0242008-02-10 18:45:23 +00001169 nI.print(DOUT, tri_);
Evan Chengf2fbca62007-11-12 06:35:08 +00001170 DOUT << '\n';
1171 }
Evan Cheng018f9b02007-12-05 03:22:34 +00001172 return CanFold;
Evan Chengf2fbca62007-11-12 06:35:08 +00001173}
Evan Cheng81a03822007-11-17 00:40:40 +00001174bool LiveIntervals::anyKillInMBBAfterIdx(const LiveInterval &li,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001175 const VNInfo *VNI,
1176 MachineBasicBlock *MBB, unsigned Idx) const {
Evan Cheng81a03822007-11-17 00:40:40 +00001177 unsigned End = getMBBEndIdx(MBB);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001178 for (unsigned j = 0, ee = VNI->kills.size(); j != ee; ++j) {
1179 unsigned KillIdx = VNI->kills[j];
1180 if (KillIdx > Idx && KillIdx < End)
1181 return true;
Evan Cheng81a03822007-11-17 00:40:40 +00001182 }
1183 return false;
1184}
1185
Evan Cheng063284c2008-02-21 00:34:19 +00001186/// RewriteInfo - Keep track of machine instrs that will be rewritten
1187/// during spilling.
Dan Gohman844731a2008-05-13 00:00:25 +00001188namespace {
1189 struct RewriteInfo {
1190 unsigned Index;
1191 MachineInstr *MI;
1192 bool HasUse;
1193 bool HasDef;
1194 RewriteInfo(unsigned i, MachineInstr *mi, bool u, bool d)
1195 : Index(i), MI(mi), HasUse(u), HasDef(d) {}
1196 };
Evan Cheng063284c2008-02-21 00:34:19 +00001197
Dan Gohman844731a2008-05-13 00:00:25 +00001198 struct RewriteInfoCompare {
1199 bool operator()(const RewriteInfo &LHS, const RewriteInfo &RHS) const {
1200 return LHS.Index < RHS.Index;
1201 }
1202 };
1203}
Evan Cheng063284c2008-02-21 00:34:19 +00001204
Evan Chengf2fbca62007-11-12 06:35:08 +00001205void LiveIntervals::
Evan Cheng81a03822007-11-17 00:40:40 +00001206rewriteInstructionsForSpills(const LiveInterval &li, bool TrySplit,
Evan Chengf2fbca62007-11-12 06:35:08 +00001207 LiveInterval::Ranges::const_iterator &I,
Evan Cheng81a03822007-11-17 00:40:40 +00001208 MachineInstr *ReMatOrigDefMI, MachineInstr *ReMatDefMI,
Evan Chengf2fbca62007-11-12 06:35:08 +00001209 unsigned Slot, int LdSlot,
1210 bool isLoad, bool isLoadSS, bool DefIsReMat, bool CanDelete,
Evan Chengd70dbb52008-02-22 09:24:50 +00001211 VirtRegMap &vrm,
Evan Chengf2fbca62007-11-12 06:35:08 +00001212 const TargetRegisterClass* rc,
1213 SmallVector<int, 4> &ReMatIds,
Evan Cheng22f07ff2007-12-11 02:09:15 +00001214 const MachineLoopInfo *loopInfo,
Evan Cheng81a03822007-11-17 00:40:40 +00001215 BitVector &SpillMBBs,
Evan Cheng1953d0c2007-11-29 10:12:14 +00001216 std::map<unsigned, std::vector<SRInfo> > &SpillIdxes,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001217 BitVector &RestoreMBBs,
Evan Cheng1953d0c2007-11-29 10:12:14 +00001218 std::map<unsigned, std::vector<SRInfo> > &RestoreIdxes,
1219 std::map<unsigned,unsigned> &MBBVRegsMap,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001220 std::vector<LiveInterval*> &NewLIs, float &SSWeight) {
Evan Cheng018f9b02007-12-05 03:22:34 +00001221 bool AllCanFold = true;
Evan Cheng81a03822007-11-17 00:40:40 +00001222 unsigned NewVReg = 0;
Evan Cheng063284c2008-02-21 00:34:19 +00001223 unsigned start = getBaseIndex(I->start);
Evan Chengf2fbca62007-11-12 06:35:08 +00001224 unsigned end = getBaseIndex(I->end-1) + InstrSlots::NUM;
Evan Chengf2fbca62007-11-12 06:35:08 +00001225
Evan Cheng063284c2008-02-21 00:34:19 +00001226 // First collect all the def / use in this live range that will be rewritten.
Evan Cheng7e073ba2008-04-09 20:57:25 +00001227 // Make sure they are sorted according to instruction index.
Evan Cheng063284c2008-02-21 00:34:19 +00001228 std::vector<RewriteInfo> RewriteMIs;
Evan Chengd70dbb52008-02-22 09:24:50 +00001229 for (MachineRegisterInfo::reg_iterator ri = mri_->reg_begin(li.reg),
1230 re = mri_->reg_end(); ri != re; ) {
Evan Cheng419852c2008-04-03 16:39:43 +00001231 MachineInstr *MI = &*ri;
Evan Cheng063284c2008-02-21 00:34:19 +00001232 MachineOperand &O = ri.getOperand();
1233 ++ri;
Evan Cheng24d2f8a2008-03-31 07:53:30 +00001234 assert(!O.isImplicit() && "Spilling register that's used as implicit use?");
Evan Cheng063284c2008-02-21 00:34:19 +00001235 unsigned index = getInstructionIndex(MI);
1236 if (index < start || index >= end)
1237 continue;
1238 RewriteMIs.push_back(RewriteInfo(index, MI, O.isUse(), O.isDef()));
1239 }
1240 std::sort(RewriteMIs.begin(), RewriteMIs.end(), RewriteInfoCompare());
1241
Evan Cheng313d4b82008-02-23 00:33:04 +00001242 unsigned ImpUse = DefIsReMat ? getReMatImplicitUse(li, ReMatDefMI) : 0;
Evan Cheng063284c2008-02-21 00:34:19 +00001243 // Now rewrite the defs and uses.
1244 for (unsigned i = 0, e = RewriteMIs.size(); i != e; ) {
1245 RewriteInfo &rwi = RewriteMIs[i];
1246 ++i;
1247 unsigned index = rwi.Index;
1248 bool MIHasUse = rwi.HasUse;
1249 bool MIHasDef = rwi.HasDef;
1250 MachineInstr *MI = rwi.MI;
1251 // If MI def and/or use the same register multiple times, then there
1252 // are multiple entries.
Evan Cheng313d4b82008-02-23 00:33:04 +00001253 unsigned NumUses = MIHasUse;
Evan Cheng063284c2008-02-21 00:34:19 +00001254 while (i != e && RewriteMIs[i].MI == MI) {
1255 assert(RewriteMIs[i].Index == index);
Evan Cheng313d4b82008-02-23 00:33:04 +00001256 bool isUse = RewriteMIs[i].HasUse;
1257 if (isUse) ++NumUses;
1258 MIHasUse |= isUse;
Evan Cheng063284c2008-02-21 00:34:19 +00001259 MIHasDef |= RewriteMIs[i].HasDef;
1260 ++i;
1261 }
Evan Cheng81a03822007-11-17 00:40:40 +00001262 MachineBasicBlock *MBB = MI->getParent();
Evan Cheng313d4b82008-02-23 00:33:04 +00001263
Evan Cheng0a891ed2008-05-23 23:00:04 +00001264 if (ImpUse && MI != ReMatDefMI) {
Evan Cheng313d4b82008-02-23 00:33:04 +00001265 // Re-matting an instruction with virtual register use. Update the
Evan Cheng24d2f8a2008-03-31 07:53:30 +00001266 // register interval's spill weight to HUGE_VALF to prevent it from
1267 // being spilled.
Evan Cheng313d4b82008-02-23 00:33:04 +00001268 LiveInterval &ImpLi = getInterval(ImpUse);
Evan Cheng24d2f8a2008-03-31 07:53:30 +00001269 ImpLi.weight = HUGE_VALF;
Evan Cheng313d4b82008-02-23 00:33:04 +00001270 }
1271
Evan Cheng063284c2008-02-21 00:34:19 +00001272 unsigned MBBId = MBB->getNumber();
Evan Cheng018f9b02007-12-05 03:22:34 +00001273 unsigned ThisVReg = 0;
Evan Cheng70306f82007-12-03 09:58:48 +00001274 if (TrySplit) {
Evan Cheng063284c2008-02-21 00:34:19 +00001275 std::map<unsigned,unsigned>::const_iterator NVI = MBBVRegsMap.find(MBBId);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001276 if (NVI != MBBVRegsMap.end()) {
Evan Cheng018f9b02007-12-05 03:22:34 +00001277 ThisVReg = NVI->second;
Evan Cheng1953d0c2007-11-29 10:12:14 +00001278 // One common case:
1279 // x = use
1280 // ...
1281 // ...
1282 // def = ...
1283 // = use
1284 // It's better to start a new interval to avoid artifically
1285 // extend the new interval.
Evan Cheng1953d0c2007-11-29 10:12:14 +00001286 if (MIHasDef && !MIHasUse) {
1287 MBBVRegsMap.erase(MBB->getNumber());
Evan Cheng018f9b02007-12-05 03:22:34 +00001288 ThisVReg = 0;
Evan Cheng1953d0c2007-11-29 10:12:14 +00001289 }
1290 }
Evan Chengcada2452007-11-28 01:28:46 +00001291 }
Evan Cheng018f9b02007-12-05 03:22:34 +00001292
1293 bool IsNew = ThisVReg == 0;
1294 if (IsNew) {
1295 // This ends the previous live interval. If all of its def / use
1296 // can be folded, give it a low spill weight.
1297 if (NewVReg && TrySplit && AllCanFold) {
1298 LiveInterval &nI = getOrCreateInterval(NewVReg);
1299 nI.weight /= 10.0F;
1300 }
1301 AllCanFold = true;
1302 }
1303 NewVReg = ThisVReg;
1304
Evan Cheng81a03822007-11-17 00:40:40 +00001305 bool HasDef = false;
1306 bool HasUse = false;
Evan Chengd70dbb52008-02-22 09:24:50 +00001307 bool CanFold = rewriteInstructionForSpills(li, I->valno, TrySplit,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001308 index, end, MI, ReMatOrigDefMI, ReMatDefMI,
1309 Slot, LdSlot, isLoad, isLoadSS, DefIsReMat,
1310 CanDelete, vrm, rc, ReMatIds, loopInfo, NewVReg,
1311 ImpUse, HasDef, HasUse, MBBVRegsMap, NewLIs, SSWeight);
Evan Cheng81a03822007-11-17 00:40:40 +00001312 if (!HasDef && !HasUse)
1313 continue;
1314
Evan Cheng018f9b02007-12-05 03:22:34 +00001315 AllCanFold &= CanFold;
1316
Evan Cheng81a03822007-11-17 00:40:40 +00001317 // Update weight of spill interval.
1318 LiveInterval &nI = getOrCreateInterval(NewVReg);
Evan Cheng70306f82007-12-03 09:58:48 +00001319 if (!TrySplit) {
Evan Cheng81a03822007-11-17 00:40:40 +00001320 // The spill weight is now infinity as it cannot be spilled again.
1321 nI.weight = HUGE_VALF;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001322 continue;
Evan Cheng81a03822007-11-17 00:40:40 +00001323 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001324
1325 // Keep track of the last def and first use in each MBB.
Evan Cheng0cbb1162007-11-29 01:06:25 +00001326 if (HasDef) {
1327 if (MI != ReMatOrigDefMI || !CanDelete) {
Evan Cheng0cbb1162007-11-29 01:06:25 +00001328 bool HasKill = false;
1329 if (!HasUse)
1330 HasKill = anyKillInMBBAfterIdx(li, I->valno, MBB, getDefIndex(index));
1331 else {
Evan Cheng1953d0c2007-11-29 10:12:14 +00001332 // If this is a two-address code, then this index starts a new VNInfo.
Evan Cheng3f32d652008-06-04 09:18:41 +00001333 const VNInfo *VNI = li.findDefinedVNInfo(getDefIndex(index));
Evan Cheng0cbb1162007-11-29 01:06:25 +00001334 if (VNI)
1335 HasKill = anyKillInMBBAfterIdx(li, VNI, MBB, getDefIndex(index));
1336 }
Evan Chenge3110d02007-12-01 04:42:39 +00001337 std::map<unsigned, std::vector<SRInfo> >::iterator SII =
1338 SpillIdxes.find(MBBId);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001339 if (!HasKill) {
Evan Cheng1953d0c2007-11-29 10:12:14 +00001340 if (SII == SpillIdxes.end()) {
1341 std::vector<SRInfo> S;
1342 S.push_back(SRInfo(index, NewVReg, true));
1343 SpillIdxes.insert(std::make_pair(MBBId, S));
1344 } else if (SII->second.back().vreg != NewVReg) {
1345 SII->second.push_back(SRInfo(index, NewVReg, true));
1346 } else if ((int)index > SII->second.back().index) {
Evan Cheng0cbb1162007-11-29 01:06:25 +00001347 // If there is an earlier def and this is a two-address
1348 // instruction, then it's not possible to fold the store (which
1349 // would also fold the load).
Evan Cheng1953d0c2007-11-29 10:12:14 +00001350 SRInfo &Info = SII->second.back();
1351 Info.index = index;
1352 Info.canFold = !HasUse;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001353 }
1354 SpillMBBs.set(MBBId);
Evan Chenge3110d02007-12-01 04:42:39 +00001355 } else if (SII != SpillIdxes.end() &&
1356 SII->second.back().vreg == NewVReg &&
1357 (int)index > SII->second.back().index) {
1358 // There is an earlier def that's not killed (must be two-address).
1359 // The spill is no longer needed.
1360 SII->second.pop_back();
1361 if (SII->second.empty()) {
1362 SpillIdxes.erase(MBBId);
1363 SpillMBBs.reset(MBBId);
1364 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001365 }
1366 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001367 }
1368
1369 if (HasUse) {
Evan Cheng1953d0c2007-11-29 10:12:14 +00001370 std::map<unsigned, std::vector<SRInfo> >::iterator SII =
Evan Cheng0cbb1162007-11-29 01:06:25 +00001371 SpillIdxes.find(MBBId);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001372 if (SII != SpillIdxes.end() &&
1373 SII->second.back().vreg == NewVReg &&
1374 (int)index > SII->second.back().index)
Evan Cheng0cbb1162007-11-29 01:06:25 +00001375 // Use(s) following the last def, it's not safe to fold the spill.
Evan Cheng1953d0c2007-11-29 10:12:14 +00001376 SII->second.back().canFold = false;
1377 std::map<unsigned, std::vector<SRInfo> >::iterator RII =
Evan Cheng0cbb1162007-11-29 01:06:25 +00001378 RestoreIdxes.find(MBBId);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001379 if (RII != RestoreIdxes.end() && RII->second.back().vreg == NewVReg)
Evan Cheng0cbb1162007-11-29 01:06:25 +00001380 // If we are splitting live intervals, only fold if it's the first
1381 // use and there isn't another use later in the MBB.
Evan Cheng1953d0c2007-11-29 10:12:14 +00001382 RII->second.back().canFold = false;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001383 else if (IsNew) {
1384 // Only need a reload if there isn't an earlier def / use.
Evan Cheng1953d0c2007-11-29 10:12:14 +00001385 if (RII == RestoreIdxes.end()) {
1386 std::vector<SRInfo> Infos;
1387 Infos.push_back(SRInfo(index, NewVReg, true));
1388 RestoreIdxes.insert(std::make_pair(MBBId, Infos));
1389 } else {
1390 RII->second.push_back(SRInfo(index, NewVReg, true));
1391 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001392 RestoreMBBs.set(MBBId);
1393 }
1394 }
1395
1396 // Update spill weight.
Evan Cheng22f07ff2007-12-11 02:09:15 +00001397 unsigned loopDepth = loopInfo->getLoopDepth(MBB);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001398 nI.weight += getSpillWeight(HasDef, HasUse, loopDepth);
Evan Chengf2fbca62007-11-12 06:35:08 +00001399 }
Evan Cheng018f9b02007-12-05 03:22:34 +00001400
1401 if (NewVReg && TrySplit && AllCanFold) {
1402 // If all of its def / use can be folded, give it a low spill weight.
1403 LiveInterval &nI = getOrCreateInterval(NewVReg);
1404 nI.weight /= 10.0F;
1405 }
Evan Chengf2fbca62007-11-12 06:35:08 +00001406}
1407
Evan Cheng1953d0c2007-11-29 10:12:14 +00001408bool LiveIntervals::alsoFoldARestore(int Id, int index, unsigned vr,
1409 BitVector &RestoreMBBs,
1410 std::map<unsigned,std::vector<SRInfo> > &RestoreIdxes) {
1411 if (!RestoreMBBs[Id])
1412 return false;
1413 std::vector<SRInfo> &Restores = RestoreIdxes[Id];
1414 for (unsigned i = 0, e = Restores.size(); i != e; ++i)
1415 if (Restores[i].index == index &&
1416 Restores[i].vreg == vr &&
1417 Restores[i].canFold)
1418 return true;
1419 return false;
1420}
1421
1422void LiveIntervals::eraseRestoreInfo(int Id, int index, unsigned vr,
1423 BitVector &RestoreMBBs,
1424 std::map<unsigned,std::vector<SRInfo> > &RestoreIdxes) {
1425 if (!RestoreMBBs[Id])
1426 return;
1427 std::vector<SRInfo> &Restores = RestoreIdxes[Id];
1428 for (unsigned i = 0, e = Restores.size(); i != e; ++i)
1429 if (Restores[i].index == index && Restores[i].vreg)
1430 Restores[i].index = -1;
1431}
Evan Cheng81a03822007-11-17 00:40:40 +00001432
Evan Cheng4cce6b42008-04-11 17:53:36 +00001433/// handleSpilledImpDefs - Remove IMPLICIT_DEF instructions which are being
1434/// spilled and create empty intervals for their uses.
1435void
1436LiveIntervals::handleSpilledImpDefs(const LiveInterval &li, VirtRegMap &vrm,
1437 const TargetRegisterClass* rc,
1438 std::vector<LiveInterval*> &NewLIs) {
Evan Cheng419852c2008-04-03 16:39:43 +00001439 for (MachineRegisterInfo::reg_iterator ri = mri_->reg_begin(li.reg),
1440 re = mri_->reg_end(); ri != re; ) {
Evan Cheng4cce6b42008-04-11 17:53:36 +00001441 MachineOperand &O = ri.getOperand();
Evan Cheng419852c2008-04-03 16:39:43 +00001442 MachineInstr *MI = &*ri;
1443 ++ri;
Evan Cheng4cce6b42008-04-11 17:53:36 +00001444 if (O.isDef()) {
1445 assert(MI->getOpcode() == TargetInstrInfo::IMPLICIT_DEF &&
1446 "Register def was not rewritten?");
1447 RemoveMachineInstrFromMaps(MI);
1448 vrm.RemoveMachineInstrFromMaps(MI);
1449 MI->eraseFromParent();
1450 } else {
1451 // This must be an use of an implicit_def so it's not part of the live
1452 // interval. Create a new empty live interval for it.
1453 // FIXME: Can we simply erase some of the instructions? e.g. Stores?
1454 unsigned NewVReg = mri_->createVirtualRegister(rc);
1455 vrm.grow();
1456 vrm.setIsImplicitlyDefined(NewVReg);
1457 NewLIs.push_back(&getOrCreateInterval(NewVReg));
1458 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
1459 MachineOperand &MO = MI->getOperand(i);
1460 if (MO.isReg() && MO.getReg() == li.reg)
1461 MO.setReg(NewVReg);
1462 }
1463 }
Evan Cheng419852c2008-04-03 16:39:43 +00001464 }
1465}
1466
Evan Cheng81a03822007-11-17 00:40:40 +00001467
Evan Chengf2fbca62007-11-12 06:35:08 +00001468std::vector<LiveInterval*> LiveIntervals::
Evan Cheng81a03822007-11-17 00:40:40 +00001469addIntervalsForSpills(const LiveInterval &li,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001470 const MachineLoopInfo *loopInfo, VirtRegMap &vrm,
1471 float &SSWeight) {
Evan Chengf2fbca62007-11-12 06:35:08 +00001472 // Since this is called after the analysis is done we don't know if
1473 // LiveVariables is available
1474 lv_ = getAnalysisToUpdate<LiveVariables>();
1475
1476 assert(li.weight != HUGE_VALF &&
1477 "attempt to spill already spilled interval!");
1478
1479 DOUT << "\t\t\t\tadding intervals for spills for interval: ";
Dan Gohman6f0d0242008-02-10 18:45:23 +00001480 li.print(DOUT, tri_);
Evan Chengf2fbca62007-11-12 06:35:08 +00001481 DOUT << '\n';
1482
Evan Cheng9c3c2212008-06-06 07:54:39 +00001483 // Spill slot weight.
1484 SSWeight = 0.0f;
1485
Evan Cheng81a03822007-11-17 00:40:40 +00001486 // Each bit specify whether it a spill is required in the MBB.
1487 BitVector SpillMBBs(mf_->getNumBlockIDs());
Evan Cheng1953d0c2007-11-29 10:12:14 +00001488 std::map<unsigned, std::vector<SRInfo> > SpillIdxes;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001489 BitVector RestoreMBBs(mf_->getNumBlockIDs());
Evan Cheng1953d0c2007-11-29 10:12:14 +00001490 std::map<unsigned, std::vector<SRInfo> > RestoreIdxes;
1491 std::map<unsigned,unsigned> MBBVRegsMap;
Evan Chengf2fbca62007-11-12 06:35:08 +00001492 std::vector<LiveInterval*> NewLIs;
Evan Chengd70dbb52008-02-22 09:24:50 +00001493 const TargetRegisterClass* rc = mri_->getRegClass(li.reg);
Evan Chengf2fbca62007-11-12 06:35:08 +00001494
1495 unsigned NumValNums = li.getNumValNums();
1496 SmallVector<MachineInstr*, 4> ReMatDefs;
1497 ReMatDefs.resize(NumValNums, NULL);
1498 SmallVector<MachineInstr*, 4> ReMatOrigDefs;
1499 ReMatOrigDefs.resize(NumValNums, NULL);
1500 SmallVector<int, 4> ReMatIds;
1501 ReMatIds.resize(NumValNums, VirtRegMap::MAX_STACK_SLOT);
1502 BitVector ReMatDelete(NumValNums);
1503 unsigned Slot = VirtRegMap::MAX_STACK_SLOT;
1504
Evan Cheng81a03822007-11-17 00:40:40 +00001505 // Spilling a split live interval. It cannot be split any further. Also,
1506 // it's also guaranteed to be a single val# / range interval.
1507 if (vrm.getPreSplitReg(li.reg)) {
1508 vrm.setIsSplitFromReg(li.reg, 0);
Evan Chengd120ffd2007-12-05 10:24:35 +00001509 // Unset the split kill marker on the last use.
1510 unsigned KillIdx = vrm.getKillPoint(li.reg);
1511 if (KillIdx) {
1512 MachineInstr *KillMI = getInstructionFromIndex(KillIdx);
1513 assert(KillMI && "Last use disappeared?");
1514 int KillOp = KillMI->findRegisterUseOperandIdx(li.reg, true);
1515 assert(KillOp != -1 && "Last use disappeared?");
Chris Lattnerf7382302007-12-30 21:56:09 +00001516 KillMI->getOperand(KillOp).setIsKill(false);
Evan Chengd120ffd2007-12-05 10:24:35 +00001517 }
Evan Chengadf85902007-12-05 09:51:10 +00001518 vrm.removeKillPoint(li.reg);
Evan Cheng81a03822007-11-17 00:40:40 +00001519 bool DefIsReMat = vrm.isReMaterialized(li.reg);
1520 Slot = vrm.getStackSlot(li.reg);
1521 assert(Slot != VirtRegMap::MAX_STACK_SLOT);
1522 MachineInstr *ReMatDefMI = DefIsReMat ?
1523 vrm.getReMaterializedMI(li.reg) : NULL;
1524 int LdSlot = 0;
1525 bool isLoadSS = DefIsReMat && tii_->isLoadFromStackSlot(ReMatDefMI, LdSlot);
1526 bool isLoad = isLoadSS ||
Chris Lattner749c6f62008-01-07 07:27:27 +00001527 (DefIsReMat && (ReMatDefMI->getDesc().isSimpleLoad()));
Evan Cheng81a03822007-11-17 00:40:40 +00001528 bool IsFirstRange = true;
1529 for (LiveInterval::Ranges::const_iterator
1530 I = li.ranges.begin(), E = li.ranges.end(); I != E; ++I) {
1531 // If this is a split live interval with multiple ranges, it means there
1532 // are two-address instructions that re-defined the value. Only the
1533 // first def can be rematerialized!
1534 if (IsFirstRange) {
Evan Chengcb3c3302007-11-29 23:02:50 +00001535 // Note ReMatOrigDefMI has already been deleted.
Evan Cheng81a03822007-11-17 00:40:40 +00001536 rewriteInstructionsForSpills(li, false, I, NULL, ReMatDefMI,
1537 Slot, LdSlot, isLoad, isLoadSS, DefIsReMat,
Evan Chengd70dbb52008-02-22 09:24:50 +00001538 false, vrm, rc, ReMatIds, loopInfo,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001539 SpillMBBs, SpillIdxes, RestoreMBBs, RestoreIdxes,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001540 MBBVRegsMap, NewLIs, SSWeight);
Evan Cheng81a03822007-11-17 00:40:40 +00001541 } else {
1542 rewriteInstructionsForSpills(li, false, I, NULL, 0,
1543 Slot, 0, false, false, false,
Evan Chengd70dbb52008-02-22 09:24:50 +00001544 false, vrm, rc, ReMatIds, loopInfo,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001545 SpillMBBs, SpillIdxes, RestoreMBBs, RestoreIdxes,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001546 MBBVRegsMap, NewLIs, SSWeight);
Evan Cheng81a03822007-11-17 00:40:40 +00001547 }
1548 IsFirstRange = false;
1549 }
Evan Cheng419852c2008-04-03 16:39:43 +00001550
Evan Cheng9c3c2212008-06-06 07:54:39 +00001551 SSWeight = 0.0f; // Already accounted for when split.
Evan Cheng4cce6b42008-04-11 17:53:36 +00001552 handleSpilledImpDefs(li, vrm, rc, NewLIs);
Evan Cheng81a03822007-11-17 00:40:40 +00001553 return NewLIs;
1554 }
1555
1556 bool TrySplit = SplitAtBB && !intervalIsInOneMBB(li);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001557 if (SplitLimit != -1 && (int)numSplits >= SplitLimit)
1558 TrySplit = false;
1559 if (TrySplit)
1560 ++numSplits;
Evan Chengf2fbca62007-11-12 06:35:08 +00001561 bool NeedStackSlot = false;
1562 for (LiveInterval::const_vni_iterator i = li.vni_begin(), e = li.vni_end();
1563 i != e; ++i) {
1564 const VNInfo *VNI = *i;
1565 unsigned VN = VNI->id;
1566 unsigned DefIdx = VNI->def;
1567 if (DefIdx == ~1U)
1568 continue; // Dead val#.
1569 // Is the def for the val# rematerializable?
Evan Cheng81a03822007-11-17 00:40:40 +00001570 MachineInstr *ReMatDefMI = (DefIdx == ~0u)
1571 ? 0 : getInstructionFromIndex(DefIdx);
Evan Cheng5ef3a042007-12-06 00:01:56 +00001572 bool dummy;
1573 if (ReMatDefMI && isReMaterializable(li, VNI, ReMatDefMI, dummy)) {
Evan Chengf2fbca62007-11-12 06:35:08 +00001574 // Remember how to remat the def of this val#.
Evan Cheng81a03822007-11-17 00:40:40 +00001575 ReMatOrigDefs[VN] = ReMatDefMI;
Evan Chengf2fbca62007-11-12 06:35:08 +00001576 // Original def may be modified so we have to make a copy here. vrm must
1577 // delete these!
Evan Cheng81a03822007-11-17 00:40:40 +00001578 ReMatDefs[VN] = ReMatDefMI = ReMatDefMI->clone();
Evan Chengf2fbca62007-11-12 06:35:08 +00001579
1580 bool CanDelete = true;
Evan Chengc3fc7d92007-11-29 09:49:23 +00001581 if (VNI->hasPHIKill) {
1582 // A kill is a phi node, not all of its uses can be rematerialized.
Evan Chengf2fbca62007-11-12 06:35:08 +00001583 // It must not be deleted.
Evan Chengc3fc7d92007-11-29 09:49:23 +00001584 CanDelete = false;
1585 // Need a stack slot if there is any live range where uses cannot be
1586 // rematerialized.
1587 NeedStackSlot = true;
Evan Chengf2fbca62007-11-12 06:35:08 +00001588 }
Evan Chengf2fbca62007-11-12 06:35:08 +00001589 if (CanDelete)
1590 ReMatDelete.set(VN);
1591 } else {
1592 // Need a stack slot if there is any live range where uses cannot be
1593 // rematerialized.
1594 NeedStackSlot = true;
1595 }
1596 }
1597
1598 // One stack slot per live interval.
Evan Cheng81a03822007-11-17 00:40:40 +00001599 if (NeedStackSlot && vrm.getPreSplitReg(li.reg) == 0)
Evan Chengf2fbca62007-11-12 06:35:08 +00001600 Slot = vrm.assignVirt2StackSlot(li.reg);
1601
1602 // Create new intervals and rewrite defs and uses.
1603 for (LiveInterval::Ranges::const_iterator
1604 I = li.ranges.begin(), E = li.ranges.end(); I != E; ++I) {
Evan Cheng81a03822007-11-17 00:40:40 +00001605 MachineInstr *ReMatDefMI = ReMatDefs[I->valno->id];
1606 MachineInstr *ReMatOrigDefMI = ReMatOrigDefs[I->valno->id];
1607 bool DefIsReMat = ReMatDefMI != NULL;
Evan Chengf2fbca62007-11-12 06:35:08 +00001608 bool CanDelete = ReMatDelete[I->valno->id];
1609 int LdSlot = 0;
Evan Cheng81a03822007-11-17 00:40:40 +00001610 bool isLoadSS = DefIsReMat && tii_->isLoadFromStackSlot(ReMatDefMI, LdSlot);
Evan Chengf2fbca62007-11-12 06:35:08 +00001611 bool isLoad = isLoadSS ||
Chris Lattner749c6f62008-01-07 07:27:27 +00001612 (DefIsReMat && ReMatDefMI->getDesc().isSimpleLoad());
Evan Cheng81a03822007-11-17 00:40:40 +00001613 rewriteInstructionsForSpills(li, TrySplit, I, ReMatOrigDefMI, ReMatDefMI,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001614 Slot, LdSlot, isLoad, isLoadSS, DefIsReMat,
Evan Chengd70dbb52008-02-22 09:24:50 +00001615 CanDelete, vrm, rc, ReMatIds, loopInfo,
Evan Cheng0cbb1162007-11-29 01:06:25 +00001616 SpillMBBs, SpillIdxes, RestoreMBBs, RestoreIdxes,
Evan Cheng9c3c2212008-06-06 07:54:39 +00001617 MBBVRegsMap, NewLIs, SSWeight);
Evan Chengf2fbca62007-11-12 06:35:08 +00001618 }
1619
Evan Cheng0cbb1162007-11-29 01:06:25 +00001620 // Insert spills / restores if we are splitting.
Evan Cheng419852c2008-04-03 16:39:43 +00001621 if (!TrySplit) {
Evan Cheng4cce6b42008-04-11 17:53:36 +00001622 handleSpilledImpDefs(li, vrm, rc, NewLIs);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001623 return NewLIs;
Evan Cheng419852c2008-04-03 16:39:43 +00001624 }
Evan Cheng1953d0c2007-11-29 10:12:14 +00001625
Evan Chengb50bb8c2007-12-05 08:16:32 +00001626 SmallPtrSet<LiveInterval*, 4> AddedKill;
Evan Chengaee4af62007-12-02 08:30:39 +00001627 SmallVector<unsigned, 2> Ops;
Evan Cheng1953d0c2007-11-29 10:12:14 +00001628 if (NeedStackSlot) {
1629 int Id = SpillMBBs.find_first();
1630 while (Id != -1) {
Evan Cheng9c3c2212008-06-06 07:54:39 +00001631 MachineBasicBlock *MBB = mf_->getBlockNumbered(Id);
1632 unsigned loopDepth = loopInfo->getLoopDepth(MBB);
Evan Cheng1953d0c2007-11-29 10:12:14 +00001633 std::vector<SRInfo> &spills = SpillIdxes[Id];
1634 for (unsigned i = 0, e = spills.size(); i != e; ++i) {
1635 int index = spills[i].index;
1636 unsigned VReg = spills[i].vreg;
Evan Cheng597d10d2007-12-04 00:32:23 +00001637 LiveInterval &nI = getOrCreateInterval(VReg);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001638 bool isReMat = vrm.isReMaterialized(VReg);
1639 MachineInstr *MI = getInstructionFromIndex(index);
Evan Chengaee4af62007-12-02 08:30:39 +00001640 bool CanFold = false;
1641 bool FoundUse = false;
1642 Ops.clear();
Evan Chengcddbb832007-11-30 21:23:43 +00001643 if (spills[i].canFold) {
Evan Chengaee4af62007-12-02 08:30:39 +00001644 CanFold = true;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001645 for (unsigned j = 0, ee = MI->getNumOperands(); j != ee; ++j) {
1646 MachineOperand &MO = MI->getOperand(j);
1647 if (!MO.isRegister() || MO.getReg() != VReg)
1648 continue;
Evan Chengaee4af62007-12-02 08:30:39 +00001649
1650 Ops.push_back(j);
1651 if (MO.isDef())
Evan Chengcddbb832007-11-30 21:23:43 +00001652 continue;
Evan Chengaee4af62007-12-02 08:30:39 +00001653 if (isReMat ||
1654 (!FoundUse && !alsoFoldARestore(Id, index, VReg,
1655 RestoreMBBs, RestoreIdxes))) {
1656 // MI has two-address uses of the same register. If the use
1657 // isn't the first and only use in the BB, then we can't fold
1658 // it. FIXME: Move this to rewriteInstructionsForSpills.
1659 CanFold = false;
Evan Chengcddbb832007-11-30 21:23:43 +00001660 break;
1661 }
Evan Chengaee4af62007-12-02 08:30:39 +00001662 FoundUse = true;
Evan Cheng0cbb1162007-11-29 01:06:25 +00001663 }
1664 }
1665 // Fold the store into the def if possible.
Evan Chengcddbb832007-11-30 21:23:43 +00001666 bool Folded = false;
Evan Chengaee4af62007-12-02 08:30:39 +00001667 if (CanFold && !Ops.empty()) {
1668 if (tryFoldMemoryOperand(MI, vrm, NULL, index, Ops, true, Slot,VReg)){
Evan Chengcddbb832007-11-30 21:23:43 +00001669 Folded = true;
Evan Chengf38d14f2007-12-05 09:05:34 +00001670 if (FoundUse > 0) {
Evan Chengaee4af62007-12-02 08:30:39 +00001671 // Also folded uses, do not issue a load.
1672 eraseRestoreInfo(Id, index, VReg, RestoreMBBs, RestoreIdxes);
Evan Chengf38d14f2007-12-05 09:05:34 +00001673 nI.removeRange(getLoadIndex(index), getUseIndex(index)+1);
1674 }
Evan Cheng597d10d2007-12-04 00:32:23 +00001675 nI.removeRange(getDefIndex(index), getStoreIndex(index));
Evan Chengcddbb832007-11-30 21:23:43 +00001676 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001677 }
1678
Evan Cheng7e073ba2008-04-09 20:57:25 +00001679 // Otherwise tell the spiller to issue a spill.
Evan Chengb50bb8c2007-12-05 08:16:32 +00001680 if (!Folded) {
1681 LiveRange *LR = &nI.ranges[nI.ranges.size()-1];
1682 bool isKill = LR->end == getStoreIndex(index);
Evan Chengb0a6f622008-05-20 08:10:37 +00001683 if (!MI->registerDefIsDead(nI.reg))
1684 // No need to spill a dead def.
1685 vrm.addSpillPoint(VReg, isKill, MI);
Evan Chengb50bb8c2007-12-05 08:16:32 +00001686 if (isKill)
1687 AddedKill.insert(&nI);
1688 }
Evan Cheng9c3c2212008-06-06 07:54:39 +00001689
1690 // Update spill slot weight.
1691 if (!isReMat)
1692 SSWeight += getSpillWeight(true, false, loopDepth);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001693 }
Evan Cheng1953d0c2007-11-29 10:12:14 +00001694 Id = SpillMBBs.find_next(Id);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001695 }
Evan Cheng1953d0c2007-11-29 10:12:14 +00001696 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001697
Evan Cheng1953d0c2007-11-29 10:12:14 +00001698 int Id = RestoreMBBs.find_first();
1699 while (Id != -1) {
Evan Cheng9c3c2212008-06-06 07:54:39 +00001700 MachineBasicBlock *MBB = mf_->getBlockNumbered(Id);
1701 unsigned loopDepth = loopInfo->getLoopDepth(MBB);
1702
Evan Cheng1953d0c2007-11-29 10:12:14 +00001703 std::vector<SRInfo> &restores = RestoreIdxes[Id];
1704 for (unsigned i = 0, e = restores.size(); i != e; ++i) {
1705 int index = restores[i].index;
1706 if (index == -1)
1707 continue;
1708 unsigned VReg = restores[i].vreg;
Evan Cheng597d10d2007-12-04 00:32:23 +00001709 LiveInterval &nI = getOrCreateInterval(VReg);
Evan Cheng9c3c2212008-06-06 07:54:39 +00001710 bool isReMat = vrm.isReMaterialized(VReg);
Evan Cheng81a03822007-11-17 00:40:40 +00001711 MachineInstr *MI = getInstructionFromIndex(index);
Evan Chengaee4af62007-12-02 08:30:39 +00001712 bool CanFold = false;
1713 Ops.clear();
Evan Chengcddbb832007-11-30 21:23:43 +00001714 if (restores[i].canFold) {
Evan Chengaee4af62007-12-02 08:30:39 +00001715 CanFold = true;
Evan Cheng81a03822007-11-17 00:40:40 +00001716 for (unsigned j = 0, ee = MI->getNumOperands(); j != ee; ++j) {
1717 MachineOperand &MO = MI->getOperand(j);
1718 if (!MO.isRegister() || MO.getReg() != VReg)
1719 continue;
Evan Chengaee4af62007-12-02 08:30:39 +00001720
Evan Cheng0cbb1162007-11-29 01:06:25 +00001721 if (MO.isDef()) {
Evan Chengaee4af62007-12-02 08:30:39 +00001722 // If this restore were to be folded, it would have been folded
1723 // already.
1724 CanFold = false;
Evan Cheng81a03822007-11-17 00:40:40 +00001725 break;
1726 }
Evan Chengaee4af62007-12-02 08:30:39 +00001727 Ops.push_back(j);
Evan Cheng81a03822007-11-17 00:40:40 +00001728 }
1729 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001730
1731 // Fold the load into the use if possible.
Evan Chengcddbb832007-11-30 21:23:43 +00001732 bool Folded = false;
Evan Chengaee4af62007-12-02 08:30:39 +00001733 if (CanFold && !Ops.empty()) {
Evan Cheng9c3c2212008-06-06 07:54:39 +00001734 if (!isReMat)
Evan Chengaee4af62007-12-02 08:30:39 +00001735 Folded = tryFoldMemoryOperand(MI, vrm, NULL,index,Ops,true,Slot,VReg);
1736 else {
Evan Cheng0cbb1162007-11-29 01:06:25 +00001737 MachineInstr *ReMatDefMI = vrm.getReMaterializedMI(VReg);
1738 int LdSlot = 0;
1739 bool isLoadSS = tii_->isLoadFromStackSlot(ReMatDefMI, LdSlot);
1740 // If the rematerializable def is a load, also try to fold it.
Chris Lattner749c6f62008-01-07 07:27:27 +00001741 if (isLoadSS || ReMatDefMI->getDesc().isSimpleLoad())
Evan Chengaee4af62007-12-02 08:30:39 +00001742 Folded = tryFoldMemoryOperand(MI, vrm, ReMatDefMI, index,
1743 Ops, isLoadSS, LdSlot, VReg);
Evan Chengd70dbb52008-02-22 09:24:50 +00001744 unsigned ImpUse = getReMatImplicitUse(li, ReMatDefMI);
1745 if (ImpUse) {
1746 // Re-matting an instruction with virtual register use. Add the
1747 // register as an implicit use on the use MI and update the register
Evan Cheng24d2f8a2008-03-31 07:53:30 +00001748 // interval's spill weight to HUGE_VALF to prevent it from being
1749 // spilled.
Evan Chengd70dbb52008-02-22 09:24:50 +00001750 LiveInterval &ImpLi = getInterval(ImpUse);
Evan Cheng24d2f8a2008-03-31 07:53:30 +00001751 ImpLi.weight = HUGE_VALF;
Evan Chengd70dbb52008-02-22 09:24:50 +00001752 MI->addOperand(MachineOperand::CreateReg(ImpUse, false, true));
1753 }
Evan Chengaee4af62007-12-02 08:30:39 +00001754 }
Evan Cheng0cbb1162007-11-29 01:06:25 +00001755 }
1756 // If folding is not possible / failed, then tell the spiller to issue a
1757 // load / rematerialization for us.
Evan Cheng597d10d2007-12-04 00:32:23 +00001758 if (Folded)
1759 nI.removeRange(getLoadIndex(index), getUseIndex(index)+1);
Evan Chengb50bb8c2007-12-05 08:16:32 +00001760 else
Evan Cheng0cbb1162007-11-29 01:06:25 +00001761 vrm.addRestorePoint(VReg, MI);
Evan Cheng9c3c2212008-06-06 07:54:39 +00001762
1763 // Update spill slot weight.
1764 if (!isReMat)
1765 SSWeight += getSpillWeight(false, true, loopDepth);
Evan Cheng81a03822007-11-17 00:40:40 +00001766 }
Evan Cheng1953d0c2007-11-29 10:12:14 +00001767 Id = RestoreMBBs.find_next(Id);
Evan Cheng81a03822007-11-17 00:40:40 +00001768 }
1769
Evan Chengb50bb8c2007-12-05 08:16:32 +00001770 // Finalize intervals: add kills, finalize spill weights, and filter out
1771 // dead intervals.
Evan Cheng597d10d2007-12-04 00:32:23 +00001772 std::vector<LiveInterval*> RetNewLIs;
1773 for (unsigned i = 0, e = NewLIs.size(); i != e; ++i) {
1774 LiveInterval *LI = NewLIs[i];
1775 if (!LI->empty()) {
1776 LI->weight /= LI->getSize();
Evan Chengb50bb8c2007-12-05 08:16:32 +00001777 if (!AddedKill.count(LI)) {
1778 LiveRange *LR = &LI->ranges[LI->ranges.size()-1];
Evan Chengd120ffd2007-12-05 10:24:35 +00001779 unsigned LastUseIdx = getBaseIndex(LR->end);
1780 MachineInstr *LastUse = getInstructionFromIndex(LastUseIdx);
Evan Cheng6130f662008-03-05 00:59:57 +00001781 int UseIdx = LastUse->findRegisterUseOperandIdx(LI->reg, false);
Evan Chengb50bb8c2007-12-05 08:16:32 +00001782 assert(UseIdx != -1);
Evan Chengd70dbb52008-02-22 09:24:50 +00001783 if (LastUse->getOperand(UseIdx).isImplicit() ||
1784 LastUse->getDesc().getOperandConstraint(UseIdx,TOI::TIED_TO) == -1){
Evan Chengb50bb8c2007-12-05 08:16:32 +00001785 LastUse->getOperand(UseIdx).setIsKill();
Evan Chengd120ffd2007-12-05 10:24:35 +00001786 vrm.addKillPoint(LI->reg, LastUseIdx);
Evan Chengadf85902007-12-05 09:51:10 +00001787 }
Evan Chengb50bb8c2007-12-05 08:16:32 +00001788 }
Evan Cheng597d10d2007-12-04 00:32:23 +00001789 RetNewLIs.push_back(LI);
1790 }
1791 }
Evan Cheng81a03822007-11-17 00:40:40 +00001792
Evan Cheng4cce6b42008-04-11 17:53:36 +00001793 handleSpilledImpDefs(li, vrm, rc, RetNewLIs);
Evan Cheng597d10d2007-12-04 00:32:23 +00001794 return RetNewLIs;
Evan Chengf2fbca62007-11-12 06:35:08 +00001795}
Evan Cheng676dd7c2008-03-11 07:19:34 +00001796
1797/// hasAllocatableSuperReg - Return true if the specified physical register has
1798/// any super register that's allocatable.
1799bool LiveIntervals::hasAllocatableSuperReg(unsigned Reg) const {
1800 for (const unsigned* AS = tri_->getSuperRegisters(Reg); *AS; ++AS)
1801 if (allocatableRegs_[*AS] && hasInterval(*AS))
1802 return true;
1803 return false;
1804}
1805
1806/// getRepresentativeReg - Find the largest super register of the specified
1807/// physical register.
1808unsigned LiveIntervals::getRepresentativeReg(unsigned Reg) const {
1809 // Find the largest super-register that is allocatable.
1810 unsigned BestReg = Reg;
1811 for (const unsigned* AS = tri_->getSuperRegisters(Reg); *AS; ++AS) {
1812 unsigned SuperReg = *AS;
1813 if (!hasAllocatableSuperReg(SuperReg) && hasInterval(SuperReg)) {
1814 BestReg = SuperReg;
1815 break;
1816 }
1817 }
1818 return BestReg;
1819}
1820
1821/// getNumConflictsWithPhysReg - Return the number of uses and defs of the
1822/// specified interval that conflicts with the specified physical register.
1823unsigned LiveIntervals::getNumConflictsWithPhysReg(const LiveInterval &li,
1824 unsigned PhysReg) const {
1825 unsigned NumConflicts = 0;
1826 const LiveInterval &pli = getInterval(getRepresentativeReg(PhysReg));
1827 for (MachineRegisterInfo::reg_iterator I = mri_->reg_begin(li.reg),
1828 E = mri_->reg_end(); I != E; ++I) {
1829 MachineOperand &O = I.getOperand();
1830 MachineInstr *MI = O.getParent();
1831 unsigned Index = getInstructionIndex(MI);
1832 if (pli.liveAt(Index))
1833 ++NumConflicts;
1834 }
1835 return NumConflicts;
1836}
1837
1838/// spillPhysRegAroundRegDefsUses - Spill the specified physical register
1839/// around all defs and uses of the specified interval.
1840void LiveIntervals::spillPhysRegAroundRegDefsUses(const LiveInterval &li,
1841 unsigned PhysReg, VirtRegMap &vrm) {
1842 unsigned SpillReg = getRepresentativeReg(PhysReg);
1843
1844 for (const unsigned *AS = tri_->getAliasSet(PhysReg); *AS; ++AS)
1845 // If there are registers which alias PhysReg, but which are not a
1846 // sub-register of the chosen representative super register. Assert
1847 // since we can't handle it yet.
1848 assert(*AS == SpillReg || !allocatableRegs_[*AS] ||
1849 tri_->isSuperRegister(*AS, SpillReg));
1850
1851 LiveInterval &pli = getInterval(SpillReg);
1852 SmallPtrSet<MachineInstr*, 8> SeenMIs;
1853 for (MachineRegisterInfo::reg_iterator I = mri_->reg_begin(li.reg),
1854 E = mri_->reg_end(); I != E; ++I) {
1855 MachineOperand &O = I.getOperand();
1856 MachineInstr *MI = O.getParent();
1857 if (SeenMIs.count(MI))
1858 continue;
1859 SeenMIs.insert(MI);
1860 unsigned Index = getInstructionIndex(MI);
1861 if (pli.liveAt(Index)) {
1862 vrm.addEmergencySpill(SpillReg, MI);
1863 pli.removeRange(getLoadIndex(Index), getStoreIndex(Index)+1);
1864 for (const unsigned* AS = tri_->getSubRegisters(SpillReg); *AS; ++AS) {
1865 if (!hasInterval(*AS))
1866 continue;
1867 LiveInterval &spli = getInterval(*AS);
1868 if (spli.liveAt(Index))
1869 spli.removeRange(getLoadIndex(Index), getStoreIndex(Index)+1);
1870 }
1871 }
1872 }
1873}
Owen Andersonc4dc1322008-06-05 17:15:43 +00001874
1875LiveRange LiveIntervals::addLiveRangeToEndOfBlock(unsigned reg,
1876 MachineInstr* startInst) {
1877 LiveInterval& Interval = getOrCreateInterval(reg);
1878 VNInfo* VN = Interval.getNextValue(
1879 getInstructionIndex(startInst) + InstrSlots::DEF,
1880 startInst, getVNInfoAllocator());
1881 VN->hasPHIKill = true;
1882 VN->kills.push_back(getMBBEndIdx(startInst->getParent()));
1883 LiveRange LR(getInstructionIndex(startInst) + InstrSlots::DEF,
1884 getMBBEndIdx(startInst->getParent()) + 1, VN);
1885 Interval.addRange(LR);
1886
1887 return LR;
1888}