blob: 14bccf6fa96b3de5566a447e6ade3e516d8194b2 [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//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
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 Lattnera3b8b5c2004-07-23 17:56:30 +000019#include "LiveIntervalAnalysis.h"
Chris Lattner015959e2004-05-01 21:24:39 +000020#include "llvm/Value.h"
Alkis Evlogimenos6b4edba2003-12-21 20:19:10 +000021#include "llvm/Analysis/LoopInfo.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"
25#include "llvm/CodeGen/Passes.h"
26#include "llvm/CodeGen/SSARegMap.h"
27#include "llvm/Target/MRegisterInfo.h"
28#include "llvm/Target/TargetInstrInfo.h"
29#include "llvm/Target/TargetMachine.h"
Alkis Evlogimenose88280a2004-01-22 23:08:45 +000030#include "Support/CommandLine.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000031#include "Support/Debug.h"
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000032#include "Support/Statistic.h"
Alkis Evlogimenos843b1602004-02-15 10:24:21 +000033#include "Support/STLExtras.h"
Alkis Evlogimenos5f375022004-03-01 20:05:10 +000034#include "VirtRegMap.h"
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +000035#include <cmath>
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000036
37using namespace llvm;
38
39namespace {
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000040 RegisterAnalysis<LiveIntervals> X("liveintervals", "Live Interval Analysis");
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000041
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000042 Statistic<> numIntervals
43 ("liveintervals", "Number of original intervals");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000044
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000045 Statistic<> numIntervalsAfter
46 ("liveintervals", "Number of intervals after coalescing");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000047
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000048 Statistic<> numJoins
49 ("liveintervals", "Number of interval joins performed");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000050
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000051 Statistic<> numPeep
52 ("liveintervals", "Number of identity moves eliminated after coalescing");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000053
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000054 Statistic<> numFolded
55 ("liveintervals", "Number of loads/stores folded into instructions");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000056
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000057 cl::opt<bool>
58 EnableJoining("join-liveintervals",
59 cl::desc("Join compatible live intervals"),
60 cl::init(true));
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000061};
62
63void LiveIntervals::getAnalysisUsage(AnalysisUsage &AU) const
64{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000065 AU.addPreserved<LiveVariables>();
66 AU.addRequired<LiveVariables>();
67 AU.addPreservedID(PHIEliminationID);
68 AU.addRequiredID(PHIEliminationID);
69 AU.addRequiredID(TwoAddressInstructionPassID);
70 AU.addRequired<LoopInfo>();
71 MachineFunctionPass::getAnalysisUsage(AU);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000072}
73
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000074void LiveIntervals::releaseMemory()
75{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000076 mi2iMap_.clear();
77 i2miMap_.clear();
78 r2iMap_.clear();
79 r2rMap_.clear();
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000080}
81
82
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000083/// runOnMachineFunction - Register allocate the whole function
84///
85bool LiveIntervals::runOnMachineFunction(MachineFunction &fn) {
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000086 mf_ = &fn;
87 tm_ = &fn.getTarget();
88 mri_ = tm_->getRegisterInfo();
89 lv_ = &getAnalysis<LiveVariables>();
Alkis Evlogimenos53278012004-08-26 22:22:38 +000090 allocatableRegs_ = mri_->getAllocatableSet(fn);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000091
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +000092 // number MachineInstrs
93 unsigned miIndex = 0;
94 for (MachineFunction::iterator mbb = mf_->begin(), mbbEnd = mf_->end();
95 mbb != mbbEnd; ++mbb)
96 for (MachineBasicBlock::iterator mi = mbb->begin(), miEnd = mbb->end();
97 mi != miEnd; ++mi) {
98 bool inserted = mi2iMap_.insert(std::make_pair(mi, miIndex)).second;
99 assert(inserted && "multiple MachineInstr -> index mappings");
100 i2miMap_.push_back(mi);
101 miIndex += InstrSlots::NUM;
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000102 }
Alkis Evlogimenosd6e40a62004-01-14 10:44:29 +0000103
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000104 computeIntervals();
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000105
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000106 numIntervals += getNumIntervals();
107
108#if 1
109 DEBUG(std::cerr << "********** INTERVALS **********\n");
110 DEBUG(for (iterator I = begin(), E = end(); I != E; ++I)
111 std::cerr << I->second << "\n");
112#endif
113
114 // join intervals if requested
115 if (EnableJoining) joinIntervals();
116
117 numIntervalsAfter += getNumIntervals();
118
119 // perform a final pass over the instructions and compute spill
120 // weights, coalesce virtual registers and remove identity moves
121 const LoopInfo& loopInfo = getAnalysis<LoopInfo>();
122 const TargetInstrInfo& tii = *tm_->getInstrInfo();
123
124 for (MachineFunction::iterator mbbi = mf_->begin(), mbbe = mf_->end();
125 mbbi != mbbe; ++mbbi) {
126 MachineBasicBlock* mbb = mbbi;
127 unsigned loopDepth = loopInfo.getLoopDepth(mbb->getBasicBlock());
128
129 for (MachineBasicBlock::iterator mii = mbb->begin(), mie = mbb->end();
130 mii != mie; ) {
131 // if the move will be an identity move delete it
132 unsigned srcReg, dstReg, RegRep;
133 if (tii.isMoveInstr(*mii, srcReg, dstReg) &&
134 (RegRep = rep(srcReg)) == rep(dstReg)) {
135 // remove from def list
136 LiveInterval &interval = getOrCreateInterval(RegRep);
137 // remove index -> MachineInstr and
138 // MachineInstr -> index mappings
139 Mi2IndexMap::iterator mi2i = mi2iMap_.find(mii);
140 if (mi2i != mi2iMap_.end()) {
141 i2miMap_[mi2i->second/InstrSlots::NUM] = 0;
142 mi2iMap_.erase(mi2i);
143 }
144 mii = mbbi->erase(mii);
145 ++numPeep;
146 }
147 else {
148 for (unsigned i = 0; i < mii->getNumOperands(); ++i) {
149 const MachineOperand& mop = mii->getOperand(i);
150 if (mop.isRegister() && mop.getReg() &&
151 MRegisterInfo::isVirtualRegister(mop.getReg())) {
152 // replace register with representative register
153 unsigned reg = rep(mop.getReg());
154 mii->SetMachineOperandReg(i, reg);
155
156 LiveInterval &RegInt = getInterval(reg);
157 RegInt.weight +=
158 (mop.isUse() + mop.isDef()) * pow(10.0F, loopDepth);
159 }
160 }
161 ++mii;
162 }
163 }
164 }
165
166 DEBUG(std::cerr << "********** INTERVALS **********\n");
167 DEBUG (for (iterator I = begin(), E = end(); I != E; ++I)
168 std::cerr << I->second << "\n");
169 DEBUG(std::cerr << "********** MACHINEINSTRS **********\n");
170 DEBUG(
171 for (MachineFunction::iterator mbbi = mf_->begin(), mbbe = mf_->end();
172 mbbi != mbbe; ++mbbi) {
173 std::cerr << ((Value*)mbbi->getBasicBlock())->getName() << ":\n";
174 for (MachineBasicBlock::iterator mii = mbbi->begin(),
175 mie = mbbi->end(); mii != mie; ++mii) {
176 std::cerr << getInstructionIndex(mii) << '\t';
177 mii->print(std::cerr, tm_);
178 }
179 });
180
181 return true;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000182}
183
Chris Lattner418da552004-06-21 13:10:56 +0000184std::vector<LiveInterval*> LiveIntervals::addIntervalsForSpills(
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000185 const LiveInterval& li,
186 VirtRegMap& vrm,
187 int slot)
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000188{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000189 std::vector<LiveInterval*> added;
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000190
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000191 assert(li.weight != HUGE_VAL &&
192 "attempt to spill already spilled interval!");
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000193
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000194 DEBUG(std::cerr << "\t\t\t\tadding intervals for spills for interval: "
195 << li << '\n');
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000196
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000197 const TargetRegisterClass* rc = mf_->getSSARegMap()->getRegClass(li.reg);
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000198
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000199 for (LiveInterval::Ranges::const_iterator
200 i = li.ranges.begin(), e = li.ranges.end(); i != e; ++i) {
201 unsigned index = getBaseIndex(i->start);
202 unsigned end = getBaseIndex(i->end-1) + InstrSlots::NUM;
203 for (; index != end; index += InstrSlots::NUM) {
204 // skip deleted instructions
205 while (index != end && !getInstructionFromIndex(index))
206 index += InstrSlots::NUM;
207 if (index == end) break;
Chris Lattner8640f4e2004-07-19 15:16:53 +0000208
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000209 MachineBasicBlock::iterator mi = getInstructionFromIndex(index);
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000210
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000211 for_operand:
212 for (unsigned i = 0; i != mi->getNumOperands(); ++i) {
213 MachineOperand& mop = mi->getOperand(i);
214 if (mop.isRegister() && mop.getReg() == li.reg) {
215 if (MachineInstr* fmi =
216 mri_->foldMemoryOperand(mi, i, slot)) {
217 lv_->instructionChanged(mi, fmi);
218 vrm.virtFolded(li.reg, mi, fmi);
219 mi2iMap_.erase(mi);
220 i2miMap_[index/InstrSlots::NUM] = fmi;
221 mi2iMap_[fmi] = index;
222 MachineBasicBlock& mbb = *mi->getParent();
223 mi = mbb.insert(mbb.erase(mi), fmi);
224 ++numFolded;
225 goto for_operand;
226 }
227 else {
228 // This is tricky. We need to add information in
229 // the interval about the spill code so we have to
230 // use our extra load/store slots.
231 //
232 // If we have a use we are going to have a load so
233 // we start the interval from the load slot
234 // onwards. Otherwise we start from the def slot.
235 unsigned start = (mop.isUse() ?
236 getLoadIndex(index) :
237 getDefIndex(index));
238 // If we have a def we are going to have a store
239 // right after it so we end the interval after the
240 // use of the next instruction. Otherwise we end
241 // after the use of this instruction.
242 unsigned end = 1 + (mop.isDef() ?
243 getStoreIndex(index) :
244 getUseIndex(index));
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000245
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000246 // create a new register for this spill
247 unsigned nReg =
248 mf_->getSSARegMap()->createVirtualRegister(rc);
249 mi->SetMachineOperandReg(i, nReg);
250 vrm.grow();
251 vrm.assignVirt2StackSlot(nReg, slot);
252 LiveInterval& nI = getOrCreateInterval(nReg);
253 assert(nI.empty());
254 // the spill weight is now infinity as it
255 // cannot be spilled again
256 nI.weight = HUGE_VAL;
257 LiveRange LR(start, end, nI.getNextValue());
258 DEBUG(std::cerr << " +" << LR);
259 nI.addRange(LR);
260 added.push_back(&nI);
261 // update live variables
262 lv_->addVirtualRegisterKilled(nReg, mi);
263 DEBUG(std::cerr << "\t\t\t\tadded new interval: "
264 << nI << '\n');
265 }
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000266 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000267 }
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000268 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000269 }
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000270
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000271 return added;
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000272}
273
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000274void LiveIntervals::printRegName(unsigned reg) const
275{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000276 if (MRegisterInfo::isPhysicalRegister(reg))
277 std::cerr << mri_->getName(reg);
278 else
279 std::cerr << "%reg" << reg;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000280}
281
282void LiveIntervals::handleVirtualRegisterDef(MachineBasicBlock* mbb,
283 MachineBasicBlock::iterator mi,
Chris Lattner418da552004-06-21 13:10:56 +0000284 LiveInterval& interval)
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000285{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000286 DEBUG(std::cerr << "\t\tregister: "; printRegName(interval.reg));
287 LiveVariables::VarInfo& vi = lv_->getVarInfo(interval.reg);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000288
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000289 // Virtual registers may be defined multiple times (due to phi
290 // elimination and 2-addr elimination). Much of what we do only has to be
291 // done once for the vreg. We use an empty interval to detect the first
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000292 // time we see a vreg.
293 if (interval.empty()) {
294 // Get the Idx of the defining instructions.
295 unsigned defIndex = getDefIndex(getInstructionIndex(mi));
Chris Lattner6097d132004-07-19 02:15:56 +0000296
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000297 unsigned ValNum = interval.getNextValue();
298 assert(ValNum == 0 && "First value in interval is not 0?");
299 ValNum = 0; // Clue in the optimizer.
Chris Lattner7ac2d312004-07-24 02:59:07 +0000300
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000301 // Loop over all of the blocks that the vreg is defined in. There are
302 // two cases we have to handle here. The most common case is a vreg
303 // whose lifetime is contained within a basic block. In this case there
304 // will be a single kill, in MBB, which comes after the definition.
305 if (vi.Kills.size() == 1 && vi.Kills[0]->getParent() == mbb) {
306 // FIXME: what about dead vars?
307 unsigned killIdx;
308 if (vi.Kills[0] != mi)
309 killIdx = getUseIndex(getInstructionIndex(vi.Kills[0]))+1;
310 else
311 killIdx = defIndex+1;
Chris Lattner6097d132004-07-19 02:15:56 +0000312
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000313 // If the kill happens after the definition, we have an intra-block
314 // live range.
315 if (killIdx > defIndex) {
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000316 assert(vi.AliveBlocks.empty() &&
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000317 "Shouldn't be alive across any blocks!");
318 LiveRange LR(defIndex, killIdx, ValNum);
319 interval.addRange(LR);
320 DEBUG(std::cerr << " +" << LR << "\n");
321 return;
322 }
Alkis Evlogimenosdd2cc652003-12-18 08:48:48 +0000323 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000324
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000325 // The other case we handle is when a virtual register lives to the end
326 // of the defining block, potentially live across some blocks, then is
327 // live into some number of blocks, but gets killed. Start by adding a
328 // range that goes from this definition to the end of the defining block.
329 LiveRange NewLR(defIndex, getInstructionIndex(&mbb->back()) +
330 InstrSlots::NUM, ValNum);
331 DEBUG(std::cerr << " +" << NewLR);
332 interval.addRange(NewLR);
333
334 // Iterate over all of the blocks that the variable is completely
335 // live in, adding [insrtIndex(begin), instrIndex(end)+4) to the
336 // live interval.
337 for (unsigned i = 0, e = vi.AliveBlocks.size(); i != e; ++i) {
338 if (vi.AliveBlocks[i]) {
339 MachineBasicBlock* mbb = mf_->getBlockNumbered(i);
340 if (!mbb->empty()) {
341 LiveRange LR(getInstructionIndex(&mbb->front()),
342 getInstructionIndex(&mbb->back())+InstrSlots::NUM,
343 ValNum);
344 interval.addRange(LR);
345 DEBUG(std::cerr << " +" << LR);
346 }
347 }
348 }
349
350 // Finally, this virtual register is live from the start of any killing
351 // block to the 'use' slot of the killing instruction.
352 for (unsigned i = 0, e = vi.Kills.size(); i != e; ++i) {
353 MachineInstr *Kill = vi.Kills[i];
354 LiveRange LR(getInstructionIndex(Kill->getParent()->begin()),
355 getUseIndex(getInstructionIndex(Kill))+1, ValNum);
356 interval.addRange(LR);
357 DEBUG(std::cerr << " +" << LR);
358 }
359
360 } else {
361 // If this is the second time we see a virtual register definition, it
362 // must be due to phi elimination or two addr elimination. If this is
363 // the result of two address elimination, then the vreg is the first
364 // operand, and is a def-and-use.
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000365 if (mi->getOperand(0).isRegister() &&
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000366 mi->getOperand(0).getReg() == interval.reg &&
367 mi->getOperand(0).isDef() && mi->getOperand(0).isUse()) {
368 // If this is a two-address definition, then we have already processed
369 // the live range. The only problem is that we didn't realize there
370 // are actually two values in the live interval. Because of this we
371 // need to take the LiveRegion that defines this register and split it
372 // into two values.
373 unsigned DefIndex = getDefIndex(getInstructionIndex(vi.DefInst));
374 unsigned RedefIndex = getDefIndex(getInstructionIndex(mi));
375
376 // Delete the initial value, which should be short and continuous,
377 // becuase the 2-addr copy must be in the same MBB as the redef.
378 interval.removeRange(DefIndex, RedefIndex);
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000379
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000380 LiveRange LR(DefIndex, RedefIndex, interval.getNextValue());
381 DEBUG(std::cerr << " replace range with " << LR);
382 interval.addRange(LR);
383
384 // If this redefinition is dead, we need to add a dummy unit live
385 // range covering the def slot.
386 for (LiveVariables::killed_iterator KI = lv_->dead_begin(mi),
387 E = lv_->dead_end(mi); KI != E; ++KI)
388 if (KI->second == interval.reg) {
389 interval.addRange(LiveRange(RedefIndex, RedefIndex+1, 0));
390 break;
391 }
392
393 DEBUG(std::cerr << "RESULT: " << interval);
394
395 } else {
396 // Otherwise, this must be because of phi elimination. If this is the
397 // first redefinition of the vreg that we have seen, go back and change
398 // the live range in the PHI block to be a different value number.
399 if (interval.containsOneValue()) {
400 assert(vi.Kills.size() == 1 &&
401 "PHI elimination vreg should have one kill, the PHI itself!");
402
403 // Remove the old range that we now know has an incorrect number.
404 MachineInstr *Killer = vi.Kills[0];
405 unsigned Start = getInstructionIndex(Killer->getParent()->begin());
406 unsigned End = getUseIndex(getInstructionIndex(Killer))+1;
407 DEBUG(std::cerr << "Removing [" << Start << "," << End << "] from: "
408 << interval << "\n");
409 interval.removeRange(Start, End);
410 DEBUG(std::cerr << "RESULT: " << interval);
411
412 // Replace the interval with one of a NEW value number.
413 LiveRange LR(Start, End, interval.getNextValue());
414 DEBUG(std::cerr << " replace range with " << LR);
415 interval.addRange(LR);
416 DEBUG(std::cerr << "RESULT: " << interval);
417 }
418
419 // In the case of PHI elimination, each variable definition is only
420 // live until the end of the block. We've already taken care of the
421 // rest of the live range.
422 unsigned defIndex = getDefIndex(getInstructionIndex(mi));
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000423 LiveRange LR(defIndex,
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000424 getInstructionIndex(&mbb->back()) + InstrSlots::NUM,
425 interval.getNextValue());
426 interval.addRange(LR);
427 DEBUG(std::cerr << " +" << LR);
428 }
429 }
430
431 DEBUG(std::cerr << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000432}
433
Chris Lattnerf35fef72004-07-23 21:24:19 +0000434void LiveIntervals::handlePhysicalRegisterDef(MachineBasicBlock *MBB,
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000435 MachineBasicBlock::iterator mi,
Chris Lattner418da552004-06-21 13:10:56 +0000436 LiveInterval& interval)
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000437{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000438 // A physical register cannot be live across basic block, so its
439 // lifetime must end somewhere in its defining basic block.
440 DEBUG(std::cerr << "\t\tregister: "; printRegName(interval.reg));
441 typedef LiveVariables::killed_iterator KillIter;
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000442
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000443 unsigned baseIndex = getInstructionIndex(mi);
444 unsigned start = getDefIndex(baseIndex);
445 unsigned end = start;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000446
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000447 // If it is not used after definition, it is considered dead at
448 // the instruction defining it. Hence its interval is:
449 // [defSlot(def), defSlot(def)+1)
450 for (KillIter ki = lv_->dead_begin(mi), ke = lv_->dead_end(mi);
451 ki != ke; ++ki) {
452 if (interval.reg == ki->second) {
453 DEBUG(std::cerr << " dead");
454 end = getDefIndex(start) + 1;
455 goto exit;
456 }
457 }
458
459 // If it is not dead on definition, it must be killed by a
460 // subsequent instruction. Hence its interval is:
461 // [defSlot(def), useSlot(kill)+1)
462 while (true) {
463 ++mi;
464 assert(mi != MBB->end() && "physreg was not killed in defining block!");
465 baseIndex += InstrSlots::NUM;
466 for (KillIter ki = lv_->killed_begin(mi), ke = lv_->killed_end(mi);
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000467 ki != ke; ++ki) {
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000468 if (interval.reg == ki->second) {
469 DEBUG(std::cerr << " killed");
470 end = getUseIndex(baseIndex) + 1;
471 goto exit;
472 }
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000473 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000474 }
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000475
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000476exit:
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000477 assert(start < end && "did not find end of interval?");
478 LiveRange LR(start, end, interval.getNextValue());
479 interval.addRange(LR);
480 DEBUG(std::cerr << " +" << LR << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000481}
482
Chris Lattnerf35fef72004-07-23 21:24:19 +0000483void LiveIntervals::handleRegisterDef(MachineBasicBlock *MBB,
484 MachineBasicBlock::iterator MI,
485 unsigned reg) {
486 if (MRegisterInfo::isVirtualRegister(reg))
487 handleVirtualRegisterDef(MBB, MI, getOrCreateInterval(reg));
Alkis Evlogimenos53278012004-08-26 22:22:38 +0000488 else if (allocatableRegs_[reg]) {
Chris Lattnerf35fef72004-07-23 21:24:19 +0000489 handlePhysicalRegisterDef(MBB, MI, getOrCreateInterval(reg));
490 for (const unsigned* AS = mri_->getAliasSet(reg); *AS; ++AS)
491 handlePhysicalRegisterDef(MBB, MI, getOrCreateInterval(*AS));
492 }
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000493}
494
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000495/// computeIntervals - computes the live intervals for virtual
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000496/// registers. for some ordering of the machine instructions [1,N] a
Alkis Evlogimenos08cec002004-01-31 19:59:32 +0000497/// live interval is an interval [i, j) where 1 <= i <= j < N for
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000498/// which a variable is live
499void LiveIntervals::computeIntervals()
500{
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000501 DEBUG(std::cerr << "********** COMPUTING LIVE INTERVALS **********\n");
502 DEBUG(std::cerr << "********** Function: "
503 << ((Value*)mf_->getFunction())->getName() << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000504
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000505 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000506 I != E; ++I) {
507 MachineBasicBlock* mbb = I;
508 DEBUG(std::cerr << ((Value*)mbb->getBasicBlock())->getName() << ":\n");
Alkis Evlogimenos6b4edba2003-12-21 20:19:10 +0000509
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000510 for (MachineBasicBlock::iterator mi = mbb->begin(), miEnd = mbb->end();
511 mi != miEnd; ++mi) {
512 const TargetInstrDescriptor& tid =
513 tm_->getInstrInfo()->get(mi->getOpcode());
514 DEBUG(std::cerr << getInstructionIndex(mi) << "\t";
515 mi->print(std::cerr, tm_));
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000516
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000517 // handle implicit defs
518 for (const unsigned* id = tid.ImplicitDefs; *id; ++id)
519 handleRegisterDef(mbb, mi, *id);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000520
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000521 // handle explicit defs
522 for (int i = mi->getNumOperands() - 1; i >= 0; --i) {
523 MachineOperand& mop = mi->getOperand(i);
524 // handle register defs - build intervals
525 if (mop.isRegister() && mop.getReg() && mop.isDef())
526 handleRegisterDef(mbb, mi, mop.getReg());
527 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000528 }
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000529 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000530}
Alkis Evlogimenosb27ef242003-12-05 10:38:28 +0000531
Chris Lattner1c5c0442004-07-19 14:08:10 +0000532void LiveIntervals::joinIntervalsInMachineBB(MachineBasicBlock *MBB) {
Chris Lattner7ac2d312004-07-24 02:59:07 +0000533 DEBUG(std::cerr << ((Value*)MBB->getBasicBlock())->getName() << ":\n");
534 const TargetInstrInfo &TII = *tm_->getInstrInfo();
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000535
Chris Lattner7ac2d312004-07-24 02:59:07 +0000536 for (MachineBasicBlock::iterator mi = MBB->begin(), mie = MBB->end();
537 mi != mie; ++mi) {
538 DEBUG(std::cerr << getInstructionIndex(mi) << '\t' << *mi);
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000539
Chris Lattner7ac2d312004-07-24 02:59:07 +0000540 // we only join virtual registers with allocatable
541 // physical registers since we do not have liveness information
542 // on not allocatable physical registers
543 unsigned regA, regB;
544 if (TII.isMoveInstr(*mi, regA, regB) &&
Alkis Evlogimenos53278012004-08-26 22:22:38 +0000545 (MRegisterInfo::isVirtualRegister(regA) || allocatableRegs_[regA]) &&
546 (MRegisterInfo::isVirtualRegister(regB) || allocatableRegs_[regB])) {
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000547
Chris Lattner7ac2d312004-07-24 02:59:07 +0000548 // Get representative registers.
549 regA = rep(regA);
550 regB = rep(regB);
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000551
Chris Lattner7ac2d312004-07-24 02:59:07 +0000552 // If they are already joined we continue.
553 if (regA == regB)
554 continue;
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000555
Chris Lattner7ac2d312004-07-24 02:59:07 +0000556 // If they are both physical registers, we cannot join them.
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000557 if (MRegisterInfo::isPhysicalRegister(regA) &&
Chris Lattner7ac2d312004-07-24 02:59:07 +0000558 MRegisterInfo::isPhysicalRegister(regB))
559 continue;
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000560
Chris Lattner7ac2d312004-07-24 02:59:07 +0000561 // If they are not of the same register class, we cannot join them.
562 if (differingRegisterClasses(regA, regB))
563 continue;
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000564
Chris Lattner7ac2d312004-07-24 02:59:07 +0000565 LiveInterval &IntA = getInterval(regA);
566 LiveInterval &IntB = getInterval(regB);
567 assert(IntA.reg == regA && IntB.reg == regB &&
568 "Register mapping is horribly broken!");
Chris Lattner060913c2004-07-24 04:32:22 +0000569
570 DEBUG(std::cerr << "\t\tInspecting " << IntA << " and " << IntB << ": ");
571
Chris Lattner4df98e52004-07-24 03:32:06 +0000572 // If two intervals contain a single value and are joined by a copy, it
573 // does not matter if the intervals overlap, they can always be joined.
Chris Lattner7ac2d312004-07-24 02:59:07 +0000574 bool TriviallyJoinable =
575 IntA.containsOneValue() && IntB.containsOneValue();
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000576
Chris Lattner7ac2d312004-07-24 02:59:07 +0000577 unsigned MIDefIdx = getDefIndex(getInstructionIndex(mi));
Chris Lattnerc25b55a2004-07-25 07:47:25 +0000578 if ((TriviallyJoinable || IntB.joinable(IntA, MIDefIdx)) &&
Chris Lattner7ac2d312004-07-24 02:59:07 +0000579 !overlapsAliases(&IntA, &IntB)) {
580 IntB.join(IntA, MIDefIdx);
Chris Lattner1c5c0442004-07-19 14:08:10 +0000581
Chris Lattner7ac2d312004-07-24 02:59:07 +0000582 if (!MRegisterInfo::isPhysicalRegister(regA)) {
Chris Lattner4df98e52004-07-24 03:32:06 +0000583 r2iMap_.erase(regA);
Chris Lattner7ac2d312004-07-24 02:59:07 +0000584 r2rMap_[regA] = regB;
585 } else {
586 // Otherwise merge the data structures the other way so we don't lose
587 // the physreg information.
588 r2rMap_[regB] = regA;
589 IntB.reg = regA;
Alkis Evlogimenosa1613db2004-07-24 11:44:15 +0000590 IntA.swap(IntB);
Chris Lattner4df98e52004-07-24 03:32:06 +0000591 r2iMap_.erase(regB);
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000592 }
Chris Lattner7ac2d312004-07-24 02:59:07 +0000593 DEBUG(std::cerr << "Joined. Result = " << IntB << "\n");
594 ++numJoins;
595 } else {
596 DEBUG(std::cerr << "Interference!\n");
597 }
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000598 }
Chris Lattner7ac2d312004-07-24 02:59:07 +0000599 }
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000600}
601
Chris Lattnercc0d1562004-07-19 14:40:29 +0000602namespace {
603 // DepthMBBCompare - Comparison predicate that sort first based on the loop
604 // depth of the basic block (the unsigned), and then on the MBB number.
605 struct DepthMBBCompare {
606 typedef std::pair<unsigned, MachineBasicBlock*> DepthMBBPair;
607 bool operator()(const DepthMBBPair &LHS, const DepthMBBPair &RHS) const {
608 if (LHS.first > RHS.first) return true; // Deeper loops first
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000609 return LHS.first == RHS.first &&
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000610 LHS.second->getNumber() < RHS.second->getNumber();
Chris Lattnercc0d1562004-07-19 14:40:29 +0000611 }
612 };
613}
Chris Lattner1c5c0442004-07-19 14:08:10 +0000614
Chris Lattnercc0d1562004-07-19 14:40:29 +0000615void LiveIntervals::joinIntervals() {
616 DEBUG(std::cerr << "********** JOINING INTERVALS ***********\n");
617
618 const LoopInfo &LI = getAnalysis<LoopInfo>();
619 if (LI.begin() == LI.end()) {
620 // If there are no loops in the function, join intervals in function order.
Chris Lattner1c5c0442004-07-19 14:08:10 +0000621 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
622 I != E; ++I)
623 joinIntervalsInMachineBB(I);
Chris Lattnercc0d1562004-07-19 14:40:29 +0000624 } else {
625 // Otherwise, join intervals in inner loops before other intervals.
626 // Unfortunately we can't just iterate over loop hierarchy here because
627 // there may be more MBB's than BB's. Collect MBB's for sorting.
628 std::vector<std::pair<unsigned, MachineBasicBlock*> > MBBs;
629 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
630 I != E; ++I)
631 MBBs.push_back(std::make_pair(LI.getLoopDepth(I->getBasicBlock()), I));
632
633 // Sort by loop depth.
634 std::sort(MBBs.begin(), MBBs.end(), DepthMBBCompare());
635
Alkis Evlogimenos70651572004-08-04 09:46:56 +0000636 // Finally, join intervals in loop nest order.
Chris Lattnercc0d1562004-07-19 14:40:29 +0000637 for (unsigned i = 0, e = MBBs.size(); i != e; ++i)
638 joinIntervalsInMachineBB(MBBs[i].second);
639 }
Chris Lattnerc83e40d2004-07-25 03:24:11 +0000640
641 DEBUG(std::cerr << "*** Register mapping ***\n");
642 DEBUG(for (std::map<unsigned, unsigned>::iterator I = r2rMap_.begin(),
Alkis Evlogimenos1a8ea012004-08-04 09:46:26 +0000643 E = r2rMap_.end(); I != E; ++I)
644 std::cerr << " reg " << I->first << " -> reg " << I->second << "\n";);
Chris Lattner1c5c0442004-07-19 14:08:10 +0000645}
646
Chris Lattner7ac2d312004-07-24 02:59:07 +0000647/// Return true if the two specified registers belong to different register
648/// classes. The registers may be either phys or virt regs.
649bool LiveIntervals::differingRegisterClasses(unsigned RegA,
650 unsigned RegB) const {
651 const TargetRegisterClass *RegClass;
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000652
Chris Lattner7ac2d312004-07-24 02:59:07 +0000653 // Get the register classes for the first reg.
654 if (MRegisterInfo::isVirtualRegister(RegA))
655 RegClass = mf_->getSSARegMap()->getRegClass(RegA);
656 else
657 RegClass = mri_->getRegClass(RegA);
658
659 // Compare against the regclass for the second reg.
660 if (MRegisterInfo::isVirtualRegister(RegB))
661 return RegClass != mf_->getSSARegMap()->getRegClass(RegB);
662 else
Chris Lattnerd0d0a1a2004-08-24 17:48:29 +0000663 return !RegClass->contains(RegB);
Chris Lattner7ac2d312004-07-24 02:59:07 +0000664}
665
666bool LiveIntervals::overlapsAliases(const LiveInterval *LHS,
667 const LiveInterval *RHS) const {
668 if (!MRegisterInfo::isPhysicalRegister(LHS->reg)) {
669 if (!MRegisterInfo::isPhysicalRegister(RHS->reg))
670 return false; // vreg-vreg merge has no aliases!
671 std::swap(LHS, RHS);
672 }
673
674 assert(MRegisterInfo::isPhysicalRegister(LHS->reg) &&
675 MRegisterInfo::isVirtualRegister(RHS->reg) &&
676 "first interval must describe a physical register");
677
Chris Lattner4df98e52004-07-24 03:32:06 +0000678 for (const unsigned *AS = mri_->getAliasSet(LHS->reg); *AS; ++AS)
679 if (RHS->overlaps(getInterval(*AS)))
680 return true;
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000681
Chris Lattner4df98e52004-07-24 03:32:06 +0000682 return false;
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000683}
684
Alkis Evlogimenosa1613db2004-07-24 11:44:15 +0000685LiveInterval LiveIntervals::createInterval(unsigned reg) {
Chris Lattner4df98e52004-07-24 03:32:06 +0000686 float Weight = MRegisterInfo::isPhysicalRegister(reg) ? HUGE_VAL :0.0F;
Alkis Evlogimenosa1613db2004-07-24 11:44:15 +0000687 return LiveInterval(reg, Weight);
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000688}