blob: a495d1fd832cbd05cdda55e3a0276c96e2bfb528 [file] [log] [blame]
Lang Hames233a60e2009-11-03 23:52:08 +00001//===-- SlotIndexes.cpp - Slot Indexes Pass ------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#define DEBUG_TYPE "slotindexes"
11
12#include "llvm/CodeGen/SlotIndexes.h"
Jakob Stoklund Olesen10c5f2d2011-03-04 18:08:29 +000013#include "llvm/ADT/Statistic.h"
Lang Hames233a60e2009-11-03 23:52:08 +000014#include "llvm/CodeGen/MachineFunction.h"
15#include "llvm/Support/Debug.h"
16#include "llvm/Support/raw_ostream.h"
Dale Johannesen1caedd02010-01-22 22:38:21 +000017#include "llvm/Target/TargetInstrInfo.h"
Lang Hames233a60e2009-11-03 23:52:08 +000018
19using namespace llvm;
20
Lang Hames233a60e2009-11-03 23:52:08 +000021char SlotIndexes::ID = 0;
Owen Andersond13db2c2010-07-21 22:09:45 +000022INITIALIZE_PASS(SlotIndexes, "slotindexes",
Owen Andersonce665bd2010-10-07 22:25:06 +000023 "Slot index numbering", false, false)
Lang Hames233a60e2009-11-03 23:52:08 +000024
Jakob Stoklund Olesen10c5f2d2011-03-04 18:08:29 +000025STATISTIC(NumRenumPasses, "Number of slot index renumber passes");
26
Lang Hames233a60e2009-11-03 23:52:08 +000027void SlotIndexes::getAnalysisUsage(AnalysisUsage &au) const {
28 au.setPreservesAll();
29 MachineFunctionPass::getAnalysisUsage(au);
30}
31
32void SlotIndexes::releaseMemory() {
33 mi2iMap.clear();
34 mbb2IdxMap.clear();
35 idx2MBBMap.clear();
Lang Hames233a60e2009-11-03 23:52:08 +000036 clearList();
37}
38
39bool SlotIndexes::runOnMachineFunction(MachineFunction &fn) {
40
41 // Compute numbering as follows:
42 // Grab an iterator to the start of the index list.
43 // Iterate over all MBBs, and within each MBB all MIs, keeping the MI
44 // iterator in lock-step (though skipping it over indexes which have
45 // null pointers in the instruction field).
46 // At each iteration assert that the instruction pointed to in the index
47 // is the same one pointed to by the MI iterator. This
48
49 // FIXME: This can be simplified. The mi2iMap_, Idx2MBBMap, etc. should
50 // only need to be set up once after the first numbering is computed.
51
52 mf = &fn;
53 initList();
54
Lang Hames233a60e2009-11-03 23:52:08 +000055 // Check that the list contains only the sentinal.
56 assert(indexListHead->getNext() == 0 &&
57 "Index list non-empty at initial numbering?");
58 assert(idx2MBBMap.empty() &&
59 "Index -> MBB mapping non-empty at initial numbering?");
60 assert(mbb2IdxMap.empty() &&
61 "MBB -> Index mapping non-empty at initial numbering?");
62 assert(mi2iMap.empty() &&
63 "MachineInstr -> Index mapping non-empty at initial numbering?");
64
65 functionSize = 0;
Lang Hames233a60e2009-11-03 23:52:08 +000066 unsigned index = 0;
67
Lang Hames74ab5ee2009-12-22 00:11:50 +000068 push_back(createEntry(0, index));
69
Dan Gohmanf451cb82010-02-10 16:03:48 +000070 // Iterate over the function.
Lang Hames233a60e2009-11-03 23:52:08 +000071 for (MachineFunction::iterator mbbItr = mf->begin(), mbbEnd = mf->end();
72 mbbItr != mbbEnd; ++mbbItr) {
73 MachineBasicBlock *mbb = &*mbbItr;
74
75 // Insert an index for the MBB start.
Lang Hames233a60e2009-11-03 23:52:08 +000076 SlotIndex blockStartIndex(back(), SlotIndex::LOAD);
77
Lang Hamesfbb8fa22009-11-05 22:20:57 +000078 index += SlotIndex::NUM;
Lang Hames233a60e2009-11-03 23:52:08 +000079
80 for (MachineBasicBlock::iterator miItr = mbb->begin(), miEnd = mbb->end();
81 miItr != miEnd; ++miItr) {
Chris Lattner518bb532010-02-09 19:54:29 +000082 MachineInstr *mi = miItr;
83 if (mi->isDebugValue())
Dale Johannesen1caedd02010-01-22 22:38:21 +000084 continue;
Lang Hames233a60e2009-11-03 23:52:08 +000085
Lang Hames233a60e2009-11-03 23:52:08 +000086 // Insert a store index for the instr.
87 push_back(createEntry(mi, index));
88
89 // Save this base index in the maps.
90 mi2iMap.insert(
91 std::make_pair(mi, SlotIndex(back(), SlotIndex::LOAD)));
92
93 ++functionSize;
94
95 unsigned Slots = mi->getDesc().getNumDefs();
96 if (Slots == 0)
97 Slots = 1;
98
Lang Hamesfbb8fa22009-11-05 22:20:57 +000099 index += (Slots + 1) * SlotIndex::NUM;
Lang Hames233a60e2009-11-03 23:52:08 +0000100 }
101
Jakob Stoklund Olesen1e8e72d2010-11-11 00:19:20 +0000102 // We insert two blank instructions between basic blocks.
103 // One to represent live-out registers and one to represent live-ins.
104 push_back(createEntry(0, index));
105 index += SlotIndex::NUM;
106
107 push_back(createEntry(0, index));
Lang Hames74ab5ee2009-12-22 00:11:50 +0000108
109 SlotIndex blockEndIndex(back(), SlotIndex::LOAD);
Lang Hames233a60e2009-11-03 23:52:08 +0000110 mbb2IdxMap.insert(
111 std::make_pair(mbb, std::make_pair(blockStartIndex, blockEndIndex)));
112
113 idx2MBBMap.push_back(IdxMBBPair(blockStartIndex, mbb));
114 }
115
Lang Hames233a60e2009-11-03 23:52:08 +0000116 // Sort the Idx2MBBMap
117 std::sort(idx2MBBMap.begin(), idx2MBBMap.end(), Idx2MBBCompare());
118
119 DEBUG(dump());
120
121 // And we're done!
122 return false;
123}
124
Lang Hamesb3661582009-11-14 00:02:51 +0000125void SlotIndexes::renumberIndexes() {
Lang Hames233a60e2009-11-03 23:52:08 +0000126
Lang Hamesfbb8fa22009-11-05 22:20:57 +0000127 // Renumber updates the index of every element of the index list.
128 // If all instrs in the function have been allocated an index (which has been
129 // placed in the index list in the order of instruction iteration) then the
130 // resulting numbering will match what would have been generated by the
131 // pass during the initial numbering of the function if the new instructions
132 // had been present.
Jakob Stoklund Olesen97af9862011-02-03 20:29:41 +0000133 DEBUG(dbgs() << "\n*** Renumbering SlotIndexes ***\n");
Jakob Stoklund Olesen10c5f2d2011-03-04 18:08:29 +0000134 ++NumRenumPasses;
Lang Hames233a60e2009-11-03 23:52:08 +0000135
Lang Hamesfbb8fa22009-11-05 22:20:57 +0000136 unsigned index = 0;
Lang Hames233a60e2009-11-03 23:52:08 +0000137
Lang Hamesfbb8fa22009-11-05 22:20:57 +0000138 for (IndexListEntry *curEntry = front(); curEntry != getTail();
139 curEntry = curEntry->getNext()) {
Lang Hamesfbb8fa22009-11-05 22:20:57 +0000140 curEntry->setIndex(index);
Jakob Stoklund Olesen2c11eb32011-03-03 06:29:01 +0000141 index += 4*SlotIndex::NUM;
Lang Hamesfbb8fa22009-11-05 22:20:57 +0000142 }
Lang Hames233a60e2009-11-03 23:52:08 +0000143}
144
145void SlotIndexes::dump() const {
146 for (const IndexListEntry *itr = front(); itr != getTail();
147 itr = itr->getNext()) {
David Greene87b0efc2010-01-05 01:25:50 +0000148 dbgs() << itr->getIndex() << " ";
Lang Hames233a60e2009-11-03 23:52:08 +0000149
150 if (itr->getInstr() != 0) {
David Greene87b0efc2010-01-05 01:25:50 +0000151 dbgs() << *itr->getInstr();
Lang Hames233a60e2009-11-03 23:52:08 +0000152 } else {
David Greene87b0efc2010-01-05 01:25:50 +0000153 dbgs() << "\n";
Lang Hames233a60e2009-11-03 23:52:08 +0000154 }
155 }
156
Jeffrey Yasskin81cf4322009-11-10 01:02:17 +0000157 for (MBB2IdxMap::const_iterator itr = mbb2IdxMap.begin();
Lang Hames233a60e2009-11-03 23:52:08 +0000158 itr != mbb2IdxMap.end(); ++itr) {
David Greene87b0efc2010-01-05 01:25:50 +0000159 dbgs() << "MBB " << itr->first->getNumber() << " (" << itr->first << ") - ["
Lang Hames233a60e2009-11-03 23:52:08 +0000160 << itr->second.first << ", " << itr->second.second << "]\n";
161 }
162}
163
164// Print a SlotIndex to a raw_ostream.
165void SlotIndex::print(raw_ostream &os) const {
Jakob Stoklund Olesen97af9862011-02-03 20:29:41 +0000166 if (isValid())
167 os << entry().getIndex() << "LudS"[getSlot()];
168 else
169 os << "invalid";
Lang Hames233a60e2009-11-03 23:52:08 +0000170}
171
172// Dump a SlotIndex to stderr.
173void SlotIndex::dump() const {
David Greene87b0efc2010-01-05 01:25:50 +0000174 print(dbgs());
175 dbgs() << "\n";
Lang Hames233a60e2009-11-03 23:52:08 +0000176}
177