blob: 0b31e227cbb4a25eb662a243b3aabc97a38fbef3 [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 {
40 RegisterAnalysis<LiveIntervals> X("liveintervals",
41 "Live Interval Analysis");
42
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000043 Statistic<> numIntervals
44 ("liveintervals", "Number of original intervals");
45
46 Statistic<> numIntervalsAfter
47 ("liveintervals", "Number of intervals after coalescing");
48
49 Statistic<> numJoins
50 ("liveintervals", "Number of interval joins performed");
51
52 Statistic<> numPeep
53 ("liveintervals", "Number of identity moves eliminated after coalescing");
54
55 Statistic<> numFolded
Alkis Evlogimenosd6f6d1a2004-02-21 18:07:33 +000056 ("liveintervals", "Number of loads/stores folded into instructions");
Alkis Evlogimenos007726c2004-02-20 20:53:26 +000057
Alkis Evlogimenose88280a2004-01-22 23:08:45 +000058 cl::opt<bool>
Chris Lattnere1b95362004-07-17 21:51:25 +000059 EnableJoining("join-liveintervals",
60 cl::desc("Join compatible live intervals"),
61 cl::init(true));
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000062};
63
64void LiveIntervals::getAnalysisUsage(AnalysisUsage &AU) const
65{
Alkis Evlogimenosf6f91bf2003-12-15 04:55:38 +000066 AU.addPreserved<LiveVariables>();
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000067 AU.addRequired<LiveVariables>();
Alkis Evlogimenosf6f91bf2003-12-15 04:55:38 +000068 AU.addPreservedID(PHIEliminationID);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000069 AU.addRequiredID(PHIEliminationID);
Alkis Evlogimenos4c080862003-12-18 22:40:24 +000070 AU.addRequiredID(TwoAddressInstructionPassID);
Alkis Evlogimenos6b4edba2003-12-21 20:19:10 +000071 AU.addRequired<LoopInfo>();
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000072 MachineFunctionPass::getAnalysisUsage(AU);
73}
74
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000075void LiveIntervals::releaseMemory()
76{
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000077 mi2iMap_.clear();
Alkis Evlogimenos843b1602004-02-15 10:24:21 +000078 i2miMap_.clear();
Alkis Evlogimenos08cec002004-01-31 19:59:32 +000079 r2iMap_.clear();
80 r2rMap_.clear();
81 intervals_.clear();
82}
83
84
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000085/// runOnMachineFunction - Register allocate the whole function
86///
87bool LiveIntervals::runOnMachineFunction(MachineFunction &fn) {
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000088 mf_ = &fn;
89 tm_ = &fn.getTarget();
90 mri_ = tm_->getRegisterInfo();
91 lv_ = &getAnalysis<LiveVariables>();
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000092
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000093 // number MachineInstrs
94 unsigned miIndex = 0;
95 for (MachineFunction::iterator mbb = mf_->begin(), mbbEnd = mf_->end();
Chris Lattner6097d132004-07-19 02:15:56 +000096 mbb != mbbEnd; ++mbb)
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +000097 for (MachineBasicBlock::iterator mi = mbb->begin(), miEnd = mbb->end();
98 mi != miEnd; ++mi) {
Chris Lattner6097d132004-07-19 02:15:56 +000099 bool inserted = mi2iMap_.insert(std::make_pair(mi, miIndex)).second;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000100 assert(inserted && "multiple MachineInstr -> index mappings");
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000101 i2miMap_.push_back(mi);
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000102 miIndex += InstrSlots::NUM;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000103 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000104
105 computeIntervals();
106
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000107 numIntervals += intervals_.size();
Alkis Evlogimenos7a40eaa2003-12-24 15:44:53 +0000108
Chris Lattner7ac2d312004-07-24 02:59:07 +0000109#if 1
110 DEBUG(std::cerr << "********** INTERVALS **********\n");
111 DEBUG(std::copy(intervals_.begin(), intervals_.end(),
112 std::ostream_iterator<LiveInterval>(std::cerr, "\n")));
113#endif
114
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000115 // join intervals if requested
Chris Lattnere1b95362004-07-17 21:51:25 +0000116 if (EnableJoining) joinIntervals();
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000117
Alkis Evlogimenos007726c2004-02-20 20:53:26 +0000118 numIntervalsAfter += intervals_.size();
119
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000120 // perform a final pass over the instructions and compute spill
121 // weights, coalesce virtual registers and remove identity moves
122 const LoopInfo& loopInfo = getAnalysis<LoopInfo>();
Chris Lattner9bcdcd12004-06-02 05:57:12 +0000123 const TargetInstrInfo& tii = *tm_->getInstrInfo();
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000124
125 for (MachineFunction::iterator mbbi = mf_->begin(), mbbe = mf_->end();
126 mbbi != mbbe; ++mbbi) {
127 MachineBasicBlock* mbb = mbbi;
128 unsigned loopDepth = loopInfo.getLoopDepth(mbb->getBasicBlock());
129
130 for (MachineBasicBlock::iterator mii = mbb->begin(), mie = mbb->end();
131 mii != mie; ) {
Alkis Evlogimenos43b61f72004-04-12 17:39:20 +0000132 // if the move will be an identity move delete it
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000133 unsigned srcReg, dstReg;
Alkis Evlogimenos43b61f72004-04-12 17:39:20 +0000134 if (tii.isMoveInstr(*mii, srcReg, dstReg) &&
135 rep(srcReg) == rep(dstReg)) {
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000136 // remove from def list
Chris Lattner418da552004-06-21 13:10:56 +0000137 LiveInterval& interval = getOrCreateInterval(rep(dstReg));
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000138 // remove index -> MachineInstr and
139 // MachineInstr -> index mappings
140 Mi2IndexMap::iterator mi2i = mi2iMap_.find(mii);
141 if (mi2i != mi2iMap_.end()) {
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000142 i2miMap_[mi2i->second/InstrSlots::NUM] = 0;
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000143 mi2iMap_.erase(mi2i);
144 }
145 mii = mbbi->erase(mii);
146 ++numPeep;
147 }
Alkis Evlogimenos43b61f72004-04-12 17:39:20 +0000148 else {
149 for (unsigned i = 0; i < mii->getNumOperands(); ++i) {
150 const MachineOperand& mop = mii->getOperand(i);
151 if (mop.isRegister() && mop.getReg() &&
152 MRegisterInfo::isVirtualRegister(mop.getReg())) {
153 // replace register with representative register
154 unsigned reg = rep(mop.getReg());
155 mii->SetMachineOperandReg(i, reg);
156
157 Reg2IntervalMap::iterator r2iit = r2iMap_.find(reg);
158 assert(r2iit != r2iMap_.end());
159 r2iit->second->weight +=
160 (mop.isUse() + mop.isDef()) * pow(10.0F, loopDepth);
161 }
162 }
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000163 ++mii;
Alkis Evlogimenos43b61f72004-04-12 17:39:20 +0000164 }
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000165 }
166 }
Alkis Evlogimenosd6e40a62004-01-14 10:44:29 +0000167
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000168 DEBUG(std::cerr << "********** INTERVALS **********\n");
Alkis Evlogimenos01e74a22004-02-01 02:18:31 +0000169 DEBUG(std::copy(intervals_.begin(), intervals_.end(),
Chris Lattner418da552004-06-21 13:10:56 +0000170 std::ostream_iterator<LiveInterval>(std::cerr, "\n")));
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000171 DEBUG(std::cerr << "********** MACHINEINSTRS **********\n");
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000172 DEBUG(
Alkis Evlogimenos0f338a12004-02-22 05:46:04 +0000173 for (MachineFunction::iterator mbbi = mf_->begin(), mbbe = mf_->end();
174 mbbi != mbbe; ++mbbi) {
Chris Lattner015959e2004-05-01 21:24:39 +0000175 std::cerr << ((Value*)mbbi->getBasicBlock())->getName() << ":\n";
Alkis Evlogimenos0f338a12004-02-22 05:46:04 +0000176 for (MachineBasicBlock::iterator mii = mbbi->begin(),
177 mie = mbbi->end(); mii != mie; ++mii) {
178 std::cerr << getInstructionIndex(mii) << '\t';
Tanya Lattnerb1407622004-06-25 00:13:11 +0000179 mii->print(std::cerr, tm_);
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000180 }
181 });
182
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000183 return true;
184}
185
Chris Lattner418da552004-06-21 13:10:56 +0000186std::vector<LiveInterval*> LiveIntervals::addIntervalsForSpills(
187 const LiveInterval& li,
188 VirtRegMap& vrm,
189 int slot)
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000190{
Chris Lattner418da552004-06-21 13:10:56 +0000191 std::vector<LiveInterval*> added;
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000192
Chris Lattnera19eede2004-05-06 16:25:59 +0000193 assert(li.weight != HUGE_VAL &&
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000194 "attempt to spill already spilled interval!");
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000195
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000196 DEBUG(std::cerr << "\t\t\t\tadding intervals for spills for interval: "
197 << li << '\n');
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000198
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000199 const TargetRegisterClass* rc = mf_->getSSARegMap()->getRegClass(li.reg);
200
Chris Lattner418da552004-06-21 13:10:56 +0000201 for (LiveInterval::Ranges::const_iterator
Chris Lattner8640f4e2004-07-19 15:16:53 +0000202 i = li.ranges.begin(), e = li.ranges.end(); i != e; ++i) {
Chris Lattnerec2bc642004-07-23 08:24:23 +0000203 unsigned index = getBaseIndex(i->start);
204 unsigned end = getBaseIndex(i->end-1) + InstrSlots::NUM;
Chris Lattner8640f4e2004-07-19 15:16:53 +0000205 for (; index != end; index += InstrSlots::NUM) {
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000206 // skip deleted instructions
Chris Lattner8640f4e2004-07-19 15:16:53 +0000207 while (index != end && !getInstructionFromIndex(index))
208 index += InstrSlots::NUM;
209 if (index == end) break;
210
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000211 MachineBasicBlock::iterator mi = getInstructionFromIndex(index);
212
Alkis Evlogimenos5f375022004-03-01 20:05:10 +0000213 for_operand:
Chris Lattner57eb15e2004-07-19 05:15:10 +0000214 for (unsigned i = 0; i != mi->getNumOperands(); ++i) {
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000215 MachineOperand& mop = mi->getOperand(i);
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000216 if (mop.isRegister() && mop.getReg() == li.reg) {
Alkis Evlogimenos39354c92004-03-14 07:19:51 +0000217 if (MachineInstr* fmi =
218 mri_->foldMemoryOperand(mi, i, slot)) {
219 lv_->instructionChanged(mi, fmi);
220 vrm.virtFolded(li.reg, mi, fmi);
221 mi2iMap_.erase(mi);
222 i2miMap_[index/InstrSlots::NUM] = fmi;
223 mi2iMap_[fmi] = index;
224 MachineBasicBlock& mbb = *mi->getParent();
225 mi = mbb.insert(mbb.erase(mi), fmi);
Alkis Evlogimenos5f375022004-03-01 20:05:10 +0000226 ++numFolded;
227 goto for_operand;
228 }
229 else {
230 // This is tricky. We need to add information in
231 // the interval about the spill code so we have to
232 // use our extra load/store slots.
233 //
234 // If we have a use we are going to have a load so
235 // we start the interval from the load slot
236 // onwards. Otherwise we start from the def slot.
237 unsigned start = (mop.isUse() ?
238 getLoadIndex(index) :
239 getDefIndex(index));
240 // If we have a def we are going to have a store
241 // right after it so we end the interval after the
242 // use of the next instruction. Otherwise we end
243 // after the use of this instruction.
244 unsigned end = 1 + (mop.isDef() ?
Chris Lattner8ea13c62004-07-19 05:55:50 +0000245 getStoreIndex(index) :
Alkis Evlogimenos5f375022004-03-01 20:05:10 +0000246 getUseIndex(index));
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000247
248 // create a new register for this spill
249 unsigned nReg =
250 mf_->getSSARegMap()->createVirtualRegister(rc);
251 mi->SetMachineOperandReg(i, nReg);
252 vrm.grow();
253 vrm.assignVirt2StackSlot(nReg, slot);
Chris Lattner418da552004-06-21 13:10:56 +0000254 LiveInterval& nI = getOrCreateInterval(nReg);
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000255 assert(nI.empty());
256 // the spill weight is now infinity as it
257 // cannot be spilled again
258 nI.weight = HUGE_VAL;
Chris Lattner7ac2d312004-07-24 02:59:07 +0000259 LiveRange LR(start, end, nI.getNextValue());
260 DEBUG(std::cerr << " +" << LR);
261 nI.addRange(LR);
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000262 added.push_back(&nI);
263 // update live variables
Chris Lattner472405e2004-07-19 06:55:21 +0000264 lv_->addVirtualRegisterKilled(nReg, mi);
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000265 DEBUG(std::cerr << "\t\t\t\tadded new interval: "
266 << nI << '\n');
Alkis Evlogimenos5f375022004-03-01 20:05:10 +0000267 }
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000268 }
269 }
270 }
271 }
Alkis Evlogimenos26f5a692004-05-30 07:24:39 +0000272
273 return added;
Alkis Evlogimenos843b1602004-02-15 10:24:21 +0000274}
275
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000276void LiveIntervals::printRegName(unsigned reg) const
277{
Alkis Evlogimenos4f67b862004-02-01 01:27:01 +0000278 if (MRegisterInfo::isPhysicalRegister(reg))
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000279 std::cerr << mri_->getName(reg);
280 else
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000281 std::cerr << "%reg" << reg;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000282}
283
284void LiveIntervals::handleVirtualRegisterDef(MachineBasicBlock* mbb,
285 MachineBasicBlock::iterator mi,
Chris Lattner418da552004-06-21 13:10:56 +0000286 LiveInterval& interval)
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000287{
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000288 DEBUG(std::cerr << "\t\tregister: "; printRegName(interval.reg));
289 LiveVariables::VarInfo& vi = lv_->getVarInfo(interval.reg);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000290
Chris Lattner6097d132004-07-19 02:15:56 +0000291 // Virtual registers may be defined multiple times (due to phi
Chris Lattner6beef3e2004-07-22 00:04:14 +0000292 // elimination and 2-addr elimination). Much of what we do only has to be
293 // done once for the vreg. We use an empty interval to detect the first
294 // time we see a vreg.
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000295 if (interval.empty()) {
Chris Lattner6097d132004-07-19 02:15:56 +0000296 // Get the Idx of the defining instructions.
297 unsigned defIndex = getDefIndex(getInstructionIndex(mi));
298
Chris Lattner7ac2d312004-07-24 02:59:07 +0000299 unsigned ValNum = interval.getNextValue();
300 assert(ValNum == 0 && "First value in interval is not 0?");
301 ValNum = 0; // Clue in the optimizer.
302
Chris Lattner6097d132004-07-19 02:15:56 +0000303 // Loop over all of the blocks that the vreg is defined in. There are
304 // two cases we have to handle here. The most common case is a vreg
305 // whose lifetime is contained within a basic block. In this case there
306 // will be a single kill, in MBB, which comes after the definition.
Chris Lattner74de8b12004-07-19 07:04:55 +0000307 if (vi.Kills.size() == 1 && vi.Kills[0]->getParent() == mbb) {
Chris Lattner6097d132004-07-19 02:15:56 +0000308 // FIXME: what about dead vars?
309 unsigned killIdx;
Chris Lattner74de8b12004-07-19 07:04:55 +0000310 if (vi.Kills[0] != mi)
311 killIdx = getUseIndex(getInstructionIndex(vi.Kills[0]))+1;
Chris Lattner6097d132004-07-19 02:15:56 +0000312 else
313 killIdx = defIndex+1;
314
315 // If the kill happens after the definition, we have an intra-block
316 // live range.
317 if (killIdx > defIndex) {
318 assert(vi.AliveBlocks.empty() &&
319 "Shouldn't be alive across any blocks!");
Chris Lattner7ac2d312004-07-24 02:59:07 +0000320 LiveRange LR(defIndex, killIdx, ValNum);
321 interval.addRange(LR);
322 DEBUG(std::cerr << " +" << LR << "\n");
Chris Lattner6097d132004-07-19 02:15:56 +0000323 return;
324 }
325 }
326
327 // The other case we handle is when a virtual register lives to the end
328 // of the defining block, potentially live across some blocks, then is
329 // live into some number of blocks, but gets killed. Start by adding a
330 // range that goes from this definition to the end of the defining block.
Chris Lattnerfb449b92004-07-23 17:49:16 +0000331 LiveRange NewLR(defIndex, getInstructionIndex(&mbb->back()) +
Chris Lattner7ac2d312004-07-24 02:59:07 +0000332 InstrSlots::NUM, ValNum);
Chris Lattnerfb449b92004-07-23 17:49:16 +0000333 DEBUG(std::cerr << " +" << NewLR);
334 interval.addRange(NewLR);
Chris Lattner6097d132004-07-19 02:15:56 +0000335
336 // Iterate over all of the blocks that the variable is completely
337 // live in, adding [insrtIndex(begin), instrIndex(end)+4) to the
338 // live interval.
339 for (unsigned i = 0, e = vi.AliveBlocks.size(); i != e; ++i) {
340 if (vi.AliveBlocks[i]) {
341 MachineBasicBlock* mbb = mf_->getBlockNumbered(i);
342 if (!mbb->empty()) {
Chris Lattnerfb449b92004-07-23 17:49:16 +0000343 LiveRange LR(getInstructionIndex(&mbb->front()),
Chris Lattner7ac2d312004-07-24 02:59:07 +0000344 getInstructionIndex(&mbb->back())+InstrSlots::NUM,
345 ValNum);
Chris Lattnerfb449b92004-07-23 17:49:16 +0000346 interval.addRange(LR);
347 DEBUG(std::cerr << " +" << LR);
Chris Lattner6097d132004-07-19 02:15:56 +0000348 }
349 }
350 }
351
352 // Finally, this virtual register is live from the start of any killing
353 // block to the 'use' slot of the killing instruction.
354 for (unsigned i = 0, e = vi.Kills.size(); i != e; ++i) {
Chris Lattner74de8b12004-07-19 07:04:55 +0000355 MachineInstr *Kill = vi.Kills[i];
Chris Lattnerfb449b92004-07-23 17:49:16 +0000356 LiveRange LR(getInstructionIndex(Kill->getParent()->begin()),
Chris Lattner7ac2d312004-07-24 02:59:07 +0000357 getUseIndex(getInstructionIndex(Kill))+1, ValNum);
Chris Lattnerfb449b92004-07-23 17:49:16 +0000358 interval.addRange(LR);
359 DEBUG(std::cerr << " +" << LR);
Chris Lattner6097d132004-07-19 02:15:56 +0000360 }
361
362 } else {
363 // If this is the second time we see a virtual register definition, it
Chris Lattner6beef3e2004-07-22 00:04:14 +0000364 // must be due to phi elimination or two addr elimination. If this is
365 // the result of two address elimination, then the vreg is the first
366 // operand, and is a def-and-use.
367 if (mi->getOperand(0).isRegister() &&
368 mi->getOperand(0).getReg() == interval.reg &&
369 mi->getOperand(0).isDef() && mi->getOperand(0).isUse()) {
Chris Lattner7ac2d312004-07-24 02:59:07 +0000370 // If this is a two-address definition, then we have already processed
371 // the live range. The only problem is that we didn't realize there
372 // are actually two values in the live interval. Because of this we
373 // need to take the LiveRegion that defines this register and split it
374 // into two values.
375 unsigned DefIndex = getDefIndex(getInstructionIndex(vi.DefInst));
376 unsigned RedefIndex = getDefIndex(getInstructionIndex(mi));
377
378 // Delete the initial value, which should be short and continuous,
379 // becuase the 2-addr copy must be in the same MBB as the redef.
380 interval.removeRange(DefIndex, RedefIndex);
381
382 LiveRange LR(DefIndex, RedefIndex, interval.getNextValue());
383 DEBUG(std::cerr << " replace range with " << LR);
384 interval.addRange(LR);
385
386 // If this redefinition is dead, we need to add a dummy unit live
387 // range covering the def slot.
388 for (LiveVariables::killed_iterator KI = lv_->dead_begin(mi),
389 E = lv_->dead_end(mi); KI != E; ++KI)
390 if (KI->second == interval.reg) {
391 interval.addRange(LiveRange(RedefIndex, RedefIndex+1, 0));
392 break;
393 }
394
395 DEBUG(std::cerr << "RESULT: " << interval);
396
Chris Lattner6beef3e2004-07-22 00:04:14 +0000397 } else {
398 // Otherwise, this must be because of phi elimination. In this case,
399 // the defined value will be live until the end of the basic block it
400 // is defined in.
401 unsigned defIndex = getDefIndex(getInstructionIndex(mi));
Chris Lattnerfb449b92004-07-23 17:49:16 +0000402 LiveRange LR(defIndex,
Chris Lattner7ac2d312004-07-24 02:59:07 +0000403 getInstructionIndex(&mbb->back()) + InstrSlots::NUM,
404 interval.getNextValue());
Chris Lattnerfb449b92004-07-23 17:49:16 +0000405 interval.addRange(LR);
406 DEBUG(std::cerr << " +" << LR);
Chris Lattner6beef3e2004-07-22 00:04:14 +0000407 }
Alkis Evlogimenosdd2cc652003-12-18 08:48:48 +0000408 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000409
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000410 DEBUG(std::cerr << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000411}
412
Chris Lattnerf35fef72004-07-23 21:24:19 +0000413void LiveIntervals::handlePhysicalRegisterDef(MachineBasicBlock *MBB,
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000414 MachineBasicBlock::iterator mi,
Chris Lattner418da552004-06-21 13:10:56 +0000415 LiveInterval& interval)
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000416{
Alkis Evlogimenos607baea2004-07-09 11:10:00 +0000417 // A physical register cannot be live across basic block, so its
418 // lifetime must end somewhere in its defining basic block.
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000419 DEBUG(std::cerr << "\t\tregister: "; printRegName(interval.reg));
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000420 typedef LiveVariables::killed_iterator KillIter;
421
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000422 unsigned baseIndex = getInstructionIndex(mi);
423 unsigned start = getDefIndex(baseIndex);
424 unsigned end = start;
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000425
Alkis Evlogimenos607baea2004-07-09 11:10:00 +0000426 // If it is not used after definition, it is considered dead at
427 // the instruction defining it. Hence its interval is:
428 // [defSlot(def), defSlot(def)+1)
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000429 for (KillIter ki = lv_->dead_begin(mi), ke = lv_->dead_end(mi);
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000430 ki != ke; ++ki) {
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000431 if (interval.reg == ki->second) {
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000432 DEBUG(std::cerr << " dead");
433 end = getDefIndex(start) + 1;
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000434 goto exit;
435 }
436 }
Alkis Evlogimenosaf254732004-01-13 22:26:14 +0000437
Alkis Evlogimenos607baea2004-07-09 11:10:00 +0000438 // If it is not dead on definition, it must be killed by a
439 // subsequent instruction. Hence its interval is:
Alkis Evlogimenos80b27ce2004-07-09 11:25:27 +0000440 // [defSlot(def), useSlot(kill)+1)
Chris Lattner7ac2d312004-07-24 02:59:07 +0000441 while (true) {
Chris Lattner230b4fb2004-07-02 05:52:23 +0000442 ++mi;
Chris Lattnerf35fef72004-07-23 21:24:19 +0000443 assert(mi != MBB->end() && "physreg was not killed in defining block!");
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000444 baseIndex += InstrSlots::NUM;
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000445 for (KillIter ki = lv_->killed_begin(mi), ke = lv_->killed_end(mi);
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000446 ki != ke; ++ki) {
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000447 if (interval.reg == ki->second) {
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000448 DEBUG(std::cerr << " killed");
449 end = getUseIndex(baseIndex) + 1;
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000450 goto exit;
451 }
452 }
Chris Lattnerf35fef72004-07-23 21:24:19 +0000453 }
Alkis Evlogimenos02ba13c2004-01-31 23:13:30 +0000454
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000455exit:
Chris Lattner230b4fb2004-07-02 05:52:23 +0000456 assert(start < end && "did not find end of interval?");
Chris Lattner7ac2d312004-07-24 02:59:07 +0000457 LiveRange LR(start, end, interval.getNextValue());
458 interval.addRange(LR);
459 DEBUG(std::cerr << " +" << LR << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000460}
461
Chris Lattnerf35fef72004-07-23 21:24:19 +0000462void LiveIntervals::handleRegisterDef(MachineBasicBlock *MBB,
463 MachineBasicBlock::iterator MI,
464 unsigned reg) {
465 if (MRegisterInfo::isVirtualRegister(reg))
466 handleVirtualRegisterDef(MBB, MI, getOrCreateInterval(reg));
467 else if (lv_->getAllocatablePhysicalRegisters()[reg]) {
468 handlePhysicalRegisterDef(MBB, MI, getOrCreateInterval(reg));
469 for (const unsigned* AS = mri_->getAliasSet(reg); *AS; ++AS)
470 handlePhysicalRegisterDef(MBB, MI, getOrCreateInterval(*AS));
471 }
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000472}
473
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000474/// computeIntervals - computes the live intervals for virtual
Alkis Evlogimenos4d46e1e2004-01-31 14:37:41 +0000475/// registers. for some ordering of the machine instructions [1,N] a
Alkis Evlogimenos08cec002004-01-31 19:59:32 +0000476/// live interval is an interval [i, j) where 1 <= i <= j < N for
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000477/// which a variable is live
478void LiveIntervals::computeIntervals()
479{
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000480 DEBUG(std::cerr << "********** COMPUTING LIVE INTERVALS **********\n");
481 DEBUG(std::cerr << "********** Function: "
Chris Lattner015959e2004-05-01 21:24:39 +0000482 << ((Value*)mf_->getFunction())->getName() << '\n');
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000483
Chris Lattner6097d132004-07-19 02:15:56 +0000484 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
485 I != E; ++I) {
486 MachineBasicBlock* mbb = I;
Chris Lattner015959e2004-05-01 21:24:39 +0000487 DEBUG(std::cerr << ((Value*)mbb->getBasicBlock())->getName() << ":\n");
Alkis Evlogimenos6b4edba2003-12-21 20:19:10 +0000488
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000489 for (MachineBasicBlock::iterator mi = mbb->begin(), miEnd = mbb->end();
490 mi != miEnd; ++mi) {
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000491 const TargetInstrDescriptor& tid =
Chris Lattner9bcdcd12004-06-02 05:57:12 +0000492 tm_->getInstrInfo()->get(mi->getOpcode());
Alkis Evlogimenos39a0d5c2004-02-20 06:15:40 +0000493 DEBUG(std::cerr << getInstructionIndex(mi) << "\t";
Tanya Lattnerb1407622004-06-25 00:13:11 +0000494 mi->print(std::cerr, tm_));
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000495
496 // handle implicit defs
Alkis Evlogimenos19b64862004-01-13 06:24:30 +0000497 for (const unsigned* id = tid.ImplicitDefs; *id; ++id)
498 handleRegisterDef(mbb, mi, *id);
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000499
500 // handle explicit defs
Alkis Evlogimenosc0b9dc52004-02-12 02:27:10 +0000501 for (int i = mi->getNumOperands() - 1; i >= 0; --i) {
502 MachineOperand& mop = mi->getOperand(i);
Alkis Evlogimenos08cec002004-01-31 19:59:32 +0000503 // handle register defs - build intervals
Alkis Evlogimenos71e353e2004-02-26 22:00:20 +0000504 if (mop.isRegister() && mop.getReg() && mop.isDef())
Alkis Evlogimenosbe766c72004-02-13 21:01:20 +0000505 handleRegisterDef(mbb, mi, mop.getReg());
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000506 }
507 }
508 }
Alkis Evlogimenosff0cbe12003-11-20 03:32:25 +0000509}
Alkis Evlogimenosb27ef242003-12-05 10:38:28 +0000510
Chris Lattner1c5c0442004-07-19 14:08:10 +0000511void LiveIntervals::joinIntervalsInMachineBB(MachineBasicBlock *MBB) {
Chris Lattner7ac2d312004-07-24 02:59:07 +0000512 DEBUG(std::cerr << ((Value*)MBB->getBasicBlock())->getName() << ":\n");
513 const TargetInstrInfo &TII = *tm_->getInstrInfo();
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000514
Chris Lattner7ac2d312004-07-24 02:59:07 +0000515 for (MachineBasicBlock::iterator mi = MBB->begin(), mie = MBB->end();
516 mi != mie; ++mi) {
517 DEBUG(std::cerr << getInstructionIndex(mi) << '\t' << *mi);
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000518
Chris Lattner7ac2d312004-07-24 02:59:07 +0000519 // we only join virtual registers with allocatable
520 // physical registers since we do not have liveness information
521 // on not allocatable physical registers
522 unsigned regA, regB;
523 if (TII.isMoveInstr(*mi, regA, regB) &&
524 (MRegisterInfo::isVirtualRegister(regA) ||
525 lv_->getAllocatablePhysicalRegisters()[regA]) &&
526 (MRegisterInfo::isVirtualRegister(regB) ||
527 lv_->getAllocatablePhysicalRegisters()[regB])) {
528
529 // Get representative registers.
530 regA = rep(regA);
531 regB = rep(regB);
532
533 // If they are already joined we continue.
534 if (regA == regB)
535 continue;
536
537 // If they are both physical registers, we cannot join them.
538 if (MRegisterInfo::isPhysicalRegister(regA) &&
539 MRegisterInfo::isPhysicalRegister(regB))
540 continue;
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000541
Chris Lattner7ac2d312004-07-24 02:59:07 +0000542 // If they are not of the same register class, we cannot join them.
543 if (differingRegisterClasses(regA, regB))
544 continue;
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000545
Chris Lattner7ac2d312004-07-24 02:59:07 +0000546 LiveInterval &IntA = getInterval(regA);
547 LiveInterval &IntB = getInterval(regB);
548 assert(IntA.reg == regA && IntB.reg == regB &&
549 "Register mapping is horribly broken!");
550
551 bool TriviallyJoinable =
552 IntA.containsOneValue() && IntB.containsOneValue();
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000553
Chris Lattner7ac2d312004-07-24 02:59:07 +0000554 unsigned MIDefIdx = getDefIndex(getInstructionIndex(mi));
555 if ((TriviallyJoinable || !IntB.joinable(IntA, MIDefIdx)) &&
556 !overlapsAliases(&IntA, &IntB)) {
557 IntB.join(IntA, MIDefIdx);
Chris Lattner1c5c0442004-07-19 14:08:10 +0000558
Chris Lattner7ac2d312004-07-24 02:59:07 +0000559 // FIXME: Turn 'intervals_' into an ilist so we don't need to do these
560 // map lookups!
561 intervals_.erase(r2iMap_[regA]);
562 r2iMap_[regA] = r2iMap_[regB];
Chris Lattner1c5c0442004-07-19 14:08:10 +0000563
Chris Lattner7ac2d312004-07-24 02:59:07 +0000564 if (!MRegisterInfo::isPhysicalRegister(regA)) {
565 r2rMap_[regA] = regB;
566 } else {
567 // Otherwise merge the data structures the other way so we don't lose
568 // the physreg information.
569 r2rMap_[regB] = regA;
570 IntB.reg = regA;
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000571 }
Chris Lattner7ac2d312004-07-24 02:59:07 +0000572 DEBUG(std::cerr << "Joined. Result = " << IntB << "\n");
573 ++numJoins;
574 } else {
575 DEBUG(std::cerr << "Interference!\n");
576 }
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000577 }
Chris Lattner7ac2d312004-07-24 02:59:07 +0000578 }
Alkis Evlogimenose88280a2004-01-22 23:08:45 +0000579}
580
Chris Lattnercc0d1562004-07-19 14:40:29 +0000581namespace {
582 // DepthMBBCompare - Comparison predicate that sort first based on the loop
583 // depth of the basic block (the unsigned), and then on the MBB number.
584 struct DepthMBBCompare {
585 typedef std::pair<unsigned, MachineBasicBlock*> DepthMBBPair;
586 bool operator()(const DepthMBBPair &LHS, const DepthMBBPair &RHS) const {
587 if (LHS.first > RHS.first) return true; // Deeper loops first
588 return LHS.first == RHS.first &&
589 LHS.second->getNumber() < RHS.second->getNumber();
590 }
591 };
592}
Chris Lattner1c5c0442004-07-19 14:08:10 +0000593
Chris Lattnercc0d1562004-07-19 14:40:29 +0000594void LiveIntervals::joinIntervals() {
595 DEBUG(std::cerr << "********** JOINING INTERVALS ***********\n");
596
597 const LoopInfo &LI = getAnalysis<LoopInfo>();
598 if (LI.begin() == LI.end()) {
599 // If there are no loops in the function, join intervals in function order.
Chris Lattner1c5c0442004-07-19 14:08:10 +0000600 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
601 I != E; ++I)
602 joinIntervalsInMachineBB(I);
Chris Lattnercc0d1562004-07-19 14:40:29 +0000603 } else {
604 // Otherwise, join intervals in inner loops before other intervals.
605 // Unfortunately we can't just iterate over loop hierarchy here because
606 // there may be more MBB's than BB's. Collect MBB's for sorting.
607 std::vector<std::pair<unsigned, MachineBasicBlock*> > MBBs;
608 for (MachineFunction::iterator I = mf_->begin(), E = mf_->end();
609 I != E; ++I)
610 MBBs.push_back(std::make_pair(LI.getLoopDepth(I->getBasicBlock()), I));
611
612 // Sort by loop depth.
613 std::sort(MBBs.begin(), MBBs.end(), DepthMBBCompare());
614
615 // Finally, join intervals in loop nest order.
616 for (unsigned i = 0, e = MBBs.size(); i != e; ++i)
617 joinIntervalsInMachineBB(MBBs[i].second);
618 }
Chris Lattner1c5c0442004-07-19 14:08:10 +0000619}
620
Chris Lattner7ac2d312004-07-24 02:59:07 +0000621/// Return true if the two specified registers belong to different register
622/// classes. The registers may be either phys or virt regs.
623bool LiveIntervals::differingRegisterClasses(unsigned RegA,
624 unsigned RegB) const {
625 const TargetRegisterClass *RegClass;
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000626
Chris Lattner7ac2d312004-07-24 02:59:07 +0000627 // Get the register classes for the first reg.
628 if (MRegisterInfo::isVirtualRegister(RegA))
629 RegClass = mf_->getSSARegMap()->getRegClass(RegA);
630 else
631 RegClass = mri_->getRegClass(RegA);
632
633 // Compare against the regclass for the second reg.
634 if (MRegisterInfo::isVirtualRegister(RegB))
635 return RegClass != mf_->getSSARegMap()->getRegClass(RegB);
636 else
637 return RegClass != mri_->getRegClass(RegB);
638}
639
640bool LiveIntervals::overlapsAliases(const LiveInterval *LHS,
641 const LiveInterval *RHS) const {
642 if (!MRegisterInfo::isPhysicalRegister(LHS->reg)) {
643 if (!MRegisterInfo::isPhysicalRegister(RHS->reg))
644 return false; // vreg-vreg merge has no aliases!
645 std::swap(LHS, RHS);
646 }
647
648 assert(MRegisterInfo::isPhysicalRegister(LHS->reg) &&
649 MRegisterInfo::isVirtualRegister(RHS->reg) &&
650 "first interval must describe a physical register");
651
652 for (const unsigned *AS = mri_->getAliasSet(LHS->reg); *AS; ++AS) {
653 Reg2IntervalMap::const_iterator r2i = r2iMap_.find(*AS);
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000654 assert(r2i != r2iMap_.end() && "alias does not have interval?");
Chris Lattner7ac2d312004-07-24 02:59:07 +0000655 if (RHS->overlaps(*r2i->second))
Alkis Evlogimenos79b0c3f2004-01-23 13:37:51 +0000656 return true;
657 }
658
659 return false;
660}
661
Chris Lattner418da552004-06-21 13:10:56 +0000662LiveInterval& LiveIntervals::getOrCreateInterval(unsigned reg)
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000663{
664 Reg2IntervalMap::iterator r2iit = r2iMap_.lower_bound(reg);
665 if (r2iit == r2iMap_.end() || r2iit->first != reg) {
Chris Lattnerfb449b92004-07-23 17:49:16 +0000666 float Weight = MRegisterInfo::isPhysicalRegister(reg) ? HUGE_VAL :0.0F;
667 intervals_.push_back(LiveInterval(reg, Weight));
Alkis Evlogimenos9a8b4902004-04-09 18:07:57 +0000668 r2iit = r2iMap_.insert(r2iit, std::make_pair(reg, --intervals_.end()));
669 }
670
671 return *r2iit->second;
672}
673