blob: 3372e2619524ac7bb8dbd8cce9c1b912143968fc [file] [log] [blame]
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00001//===-- RegAllocGreedy.cpp - greedy register allocator --------------------===//
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// This file defines the RAGreedy function pass for register allocation in
11// optimized builds.
12//
13//===----------------------------------------------------------------------===//
14
15#define DEBUG_TYPE "regalloc"
Chandler Carruthed0881b2012-12-03 16:50:05 +000016#include "llvm/CodeGen/Passes.h"
Jakob Stoklund Olesen4d7432e2010-12-10 22:21:05 +000017#include "AllocationOrder.h"
Jakob Stoklund Olesen91cbcaf2011-04-02 06:03:35 +000018#include "InterferenceCache.h"
Jakob Stoklund Olesen6aa0fbf2011-04-05 21:40:37 +000019#include "LiveDebugVariables.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000020#include "RegAllocBase.h"
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +000021#include "SpillPlacement.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000022#include "Spiller.h"
Jakob Stoklund Olesene7601e92010-12-15 23:46:13 +000023#include "SplitKit.h"
Jakob Stoklund Olesen99827e82011-02-17 22:53:48 +000024#include "llvm/ADT/Statistic.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000025#include "llvm/Analysis/AliasAnalysis.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000026#include "llvm/CodeGen/CalcSpillWeights.h"
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +000027#include "llvm/CodeGen/EdgeBundles.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000028#include "llvm/CodeGen/LiveIntervalAnalysis.h"
Pete Cooper3ca96f92012-04-02 22:44:18 +000029#include "llvm/CodeGen/LiveRangeEdit.h"
Jakob Stoklund Olesen26c9d702012-11-28 19:13:06 +000030#include "llvm/CodeGen/LiveRegMatrix.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000031#include "llvm/CodeGen/LiveStackAnalysis.h"
Benjamin Kramere2a1d892013-06-17 19:00:36 +000032#include "llvm/CodeGen/MachineBlockFrequencyInfo.h"
Jakob Stoklund Olesen1740e002010-12-17 23:16:32 +000033#include "llvm/CodeGen/MachineDominators.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000034#include "llvm/CodeGen/MachineFunctionPass.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000035#include "llvm/CodeGen/MachineLoopInfo.h"
36#include "llvm/CodeGen/MachineRegisterInfo.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000037#include "llvm/CodeGen/RegAllocRegistry.h"
Quentin Colombet1fb3362a2014-01-02 22:47:22 +000038#include "llvm/CodeGen/RegisterClassInfo.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000039#include "llvm/CodeGen/VirtRegMap.h"
40#include "llvm/PassAnalysisSupport.h"
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +000041#include "llvm/Support/CommandLine.h"
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000042#include "llvm/Support/Debug.h"
43#include "llvm/Support/ErrorHandling.h"
Jakob Stoklund Olesen92da7052010-12-11 00:19:56 +000044#include "llvm/Support/Timer.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000045#include "llvm/Support/raw_ostream.h"
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +000046#include <queue>
47
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000048using namespace llvm;
49
Jakob Stoklund Olesen99827e82011-02-17 22:53:48 +000050STATISTIC(NumGlobalSplits, "Number of split global live ranges");
51STATISTIC(NumLocalSplits, "Number of split local live ranges");
Jakob Stoklund Olesen99827e82011-02-17 22:53:48 +000052STATISTIC(NumEvicted, "Number of interferences evicted");
53
Jakob Stoklund Oleseneecb2fb2011-09-12 16:49:21 +000054static cl::opt<SplitEditor::ComplementSpillMode>
55SplitSpillMode("split-spill-mode", cl::Hidden,
56 cl::desc("Spill mode for splitting live ranges"),
57 cl::values(clEnumValN(SplitEditor::SM_Partition, "default", "Default"),
58 clEnumValN(SplitEditor::SM_Size, "size", "Optimize for size"),
59 clEnumValN(SplitEditor::SM_Speed, "speed", "Optimize for speed"),
60 clEnumValEnd),
61 cl::init(SplitEditor::SM_Partition));
62
Quentin Colombet87769712014-02-05 22:13:59 +000063static cl::opt<unsigned>
64LastChanceRecoloringMaxDepth("lcr-max-depth", cl::Hidden,
65 cl::desc("Last chance recoloring max depth"),
66 cl::init(5));
67
68static cl::opt<unsigned> LastChanceRecoloringMaxInterference(
69 "lcr-max-interf", cl::Hidden,
70 cl::desc("Last chance recoloring maximum number of considered"
71 " interference at a time"),
72 cl::init(8));
73
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000074static RegisterRegAlloc greedyRegAlloc("greedy", "greedy register allocator",
75 createGreedyRegisterAllocator);
76
77namespace {
Jakob Stoklund Olesen8e089642011-03-09 00:57:29 +000078class RAGreedy : public MachineFunctionPass,
79 public RegAllocBase,
80 private LiveRangeEdit::Delegate {
Quentin Colombet87769712014-02-05 22:13:59 +000081 // Convenient shortcuts.
82 typedef std::priority_queue<std::pair<unsigned, unsigned> > PQueue;
83 typedef SmallPtrSet<LiveInterval *, 4> SmallLISet;
84 typedef SmallSet<unsigned, 16> SmallVirtRegSet;
Jakob Stoklund Olesen8e089642011-03-09 00:57:29 +000085
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000086 // context
87 MachineFunction *MF;
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000088
Quentin Colombet1fb3362a2014-01-02 22:47:22 +000089 // Shortcuts to some useful interface.
90 const TargetInstrInfo *TII;
91 const TargetRegisterInfo *TRI;
92 RegisterClassInfo RCI;
93
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +000094 // analyses
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +000095 SlotIndexes *Indexes;
Benjamin Kramere2a1d892013-06-17 19:00:36 +000096 MachineBlockFrequencyInfo *MBFI;
Jakob Stoklund Olesen1740e002010-12-17 23:16:32 +000097 MachineDominatorTree *DomTree;
Jakob Stoklund Olesene7601e92010-12-15 23:46:13 +000098 MachineLoopInfo *Loops;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +000099 EdgeBundles *Bundles;
100 SpillPlacement *SpillPlacer;
Jakob Stoklund Olesenf8da0282011-05-06 18:00:02 +0000101 LiveDebugVariables *DebugVars;
Jakob Stoklund Olesen1740e002010-12-17 23:16:32 +0000102
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000103 // state
Andy Gibbs95777552013-04-12 10:56:28 +0000104 OwningPtr<Spiller> SpillerInstance;
Quentin Colombet87769712014-02-05 22:13:59 +0000105 PQueue Queue;
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000106 unsigned NextCascade;
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +0000107
108 // Live ranges pass through a number of stages as we try to allocate them.
109 // Some of the stages may also create new live ranges:
110 //
111 // - Region splitting.
112 // - Per-block splitting.
113 // - Local splitting.
114 // - Spilling.
115 //
116 // Ranges produced by one of the stages skip the previous stages when they are
117 // dequeued. This improves performance because we can skip interference checks
118 // that are unlikely to give any results. It also guarantees that the live
119 // range splitting algorithm terminates, something that is otherwise hard to
120 // ensure.
121 enum LiveRangeStage {
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000122 /// Newly created live range that has never been queued.
123 RS_New,
124
125 /// Only attempt assignment and eviction. Then requeue as RS_Split.
126 RS_Assign,
127
128 /// Attempt live range splitting if assignment is impossible.
129 RS_Split,
130
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +0000131 /// Attempt more aggressive live range splitting that is guaranteed to make
132 /// progress. This is used for split products that may not be making
133 /// progress.
134 RS_Split2,
135
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000136 /// Live range will be spilled. No more splitting will be attempted.
137 RS_Spill,
138
139 /// There is nothing more we can do to this live range. Abort compilation
140 /// if it can't be assigned.
141 RS_Done
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +0000142 };
143
Eli Friedman78bffa52013-09-10 23:18:14 +0000144#ifndef NDEBUG
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000145 static const char *const StageName[];
Eli Friedman78bffa52013-09-10 23:18:14 +0000146#endif
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000147
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000148 // RegInfo - Keep additional information about each live range.
149 struct RegInfo {
150 LiveRangeStage Stage;
151
152 // Cascade - Eviction loop prevention. See canEvictInterference().
153 unsigned Cascade;
154
155 RegInfo() : Stage(RS_New), Cascade(0) {}
156 };
157
158 IndexedMap<RegInfo, VirtReg2IndexFunctor> ExtraRegInfo;
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +0000159
160 LiveRangeStage getStage(const LiveInterval &VirtReg) const {
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000161 return ExtraRegInfo[VirtReg.reg].Stage;
162 }
163
164 void setStage(const LiveInterval &VirtReg, LiveRangeStage Stage) {
165 ExtraRegInfo.resize(MRI->getNumVirtRegs());
166 ExtraRegInfo[VirtReg.reg].Stage = Stage;
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +0000167 }
168
169 template<typename Iterator>
170 void setStage(Iterator Begin, Iterator End, LiveRangeStage NewStage) {
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000171 ExtraRegInfo.resize(MRI->getNumVirtRegs());
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000172 for (;Begin != End; ++Begin) {
Mark Laceyf9ea8852013-08-14 23:50:04 +0000173 unsigned Reg = *Begin;
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000174 if (ExtraRegInfo[Reg].Stage == RS_New)
175 ExtraRegInfo[Reg].Stage = NewStage;
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000176 }
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +0000177 }
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000178
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000179 /// Cost of evicting interference.
180 struct EvictionCost {
181 unsigned BrokenHints; ///< Total number of broken hints.
182 float MaxWeight; ///< Maximum spill weight evicted.
183
Andrew Trick3621b8a2013-11-22 19:07:38 +0000184 EvictionCost(): BrokenHints(0), MaxWeight(0) {}
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000185
Andrew Trick84852572013-07-25 18:35:14 +0000186 bool isMax() const { return BrokenHints == ~0u; }
187
Andrew Trick3621b8a2013-11-22 19:07:38 +0000188 void setMax() { BrokenHints = ~0u; }
189
190 void setBrokenHints(unsigned NHints) { BrokenHints = NHints; }
191
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000192 bool operator<(const EvictionCost &O) const {
193 if (BrokenHints != O.BrokenHints)
194 return BrokenHints < O.BrokenHints;
195 return MaxWeight < O.MaxWeight;
196 }
197 };
198
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000199 // splitting state.
Andy Gibbs95777552013-04-12 10:56:28 +0000200 OwningPtr<SplitAnalysis> SA;
201 OwningPtr<SplitEditor> SE;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000202
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000203 /// Cached per-block interference maps
204 InterferenceCache IntfCache;
205
Jakob Stoklund Olesen6d2bbc12011-04-07 17:27:46 +0000206 /// All basic blocks where the current register has uses.
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000207 SmallVector<SpillPlacement::BlockConstraint, 8> SplitConstraints;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000208
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000209 /// Global live range splitting candidate info.
210 struct GlobalSplitCandidate {
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000211 // Register intended for assignment, or 0.
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000212 unsigned PhysReg;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000213
214 // SplitKit interval index for this candidate.
215 unsigned IntvIdx;
216
217 // Interference for PhysReg.
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +0000218 InterferenceCache::Cursor Intf;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000219
220 // Bundles where this candidate should be live.
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000221 BitVector LiveBundles;
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000222 SmallVector<unsigned, 8> ActiveBlocks;
223
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +0000224 void reset(InterferenceCache &Cache, unsigned Reg) {
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000225 PhysReg = Reg;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000226 IntvIdx = 0;
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +0000227 Intf.setPhysReg(Cache, Reg);
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000228 LiveBundles.clear();
229 ActiveBlocks.clear();
230 }
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000231
232 // Set B[i] = C for every live bundle where B[i] was NoCand.
233 unsigned getBundles(SmallVectorImpl<unsigned> &B, unsigned C) {
234 unsigned Count = 0;
235 for (int i = LiveBundles.find_first(); i >= 0;
236 i = LiveBundles.find_next(i))
237 if (B[i] == NoCand) {
238 B[i] = C;
239 Count++;
240 }
241 return Count;
242 }
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000243 };
244
Aditya Nandakumarc1fd0dd2013-11-19 23:51:32 +0000245 /// Candidate info for each PhysReg in AllocationOrder.
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000246 /// This vector never shrinks, but grows to the size of the largest register
247 /// class.
248 SmallVector<GlobalSplitCandidate, 32> GlobalCand;
249
Reid Klecknercd4a25d2013-10-08 20:15:11 +0000250 enum LLVM_ENUM_INT_TYPE(unsigned) { NoCand = ~0u };
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000251
252 /// Candidate map. Each edge bundle is assigned to a GlobalCand entry, or to
253 /// NoCand which indicates the stack interval.
254 SmallVector<unsigned, 32> BundleCand;
255
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000256public:
257 RAGreedy();
258
259 /// Return the pass name.
260 virtual const char* getPassName() const {
Jakob Stoklund Olesen92da7052010-12-11 00:19:56 +0000261 return "Greedy Register Allocator";
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000262 }
263
264 /// RAGreedy analysis usage.
265 virtual void getAnalysisUsage(AnalysisUsage &AU) const;
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000266 virtual void releaseMemory();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000267 virtual Spiller &spiller() { return *SpillerInstance; }
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +0000268 virtual void enqueue(LiveInterval *LI);
269 virtual LiveInterval *dequeue();
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +0000270 virtual unsigned selectOrSplit(LiveInterval&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000271 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000272
273 /// Perform register allocation.
274 virtual bool runOnMachineFunction(MachineFunction &mf);
275
276 static char ID;
Andrew Trickccef0982010-12-09 18:15:21 +0000277
278private:
Quentin Colombet87769712014-02-05 22:13:59 +0000279 unsigned selectOrSplitImpl(LiveInterval &, SmallVectorImpl<unsigned> &,
280 SmallVirtRegSet &, unsigned = 0);
281
Jakob Stoklund Olesen43a87502011-03-13 01:23:11 +0000282 bool LRE_CanEraseVirtReg(unsigned);
Jakob Stoklund Olesene14b2b22011-03-16 22:56:16 +0000283 void LRE_WillShrinkVirtReg(unsigned);
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000284 void LRE_DidCloneVirtReg(unsigned, unsigned);
Quentin Colombet87769712014-02-05 22:13:59 +0000285 void enqueue(PQueue &CurQueue, LiveInterval *LI);
286 LiveInterval *dequeue(PQueue &CurQueue);
Jakob Stoklund Olesen8e089642011-03-09 00:57:29 +0000287
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000288 BlockFrequency calcSpillCost();
289 bool addSplitConstraints(InterferenceCache::Cursor, BlockFrequency&);
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000290 void addThroughConstraints(InterferenceCache::Cursor, ArrayRef<unsigned>);
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +0000291 void growRegion(GlobalSplitCandidate &Cand);
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000292 BlockFrequency calcGlobalSplitCost(GlobalSplitCandidate&);
Jakob Stoklund Olesenecad62f2011-07-23 03:41:57 +0000293 bool calcCompactRegion(GlobalSplitCandidate&);
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +0000294 void splitAroundRegion(LiveRangeEdit&, ArrayRef<unsigned>);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +0000295 void calcGapWeights(unsigned, SmallVectorImpl<float>&);
Andrew Trick8bb0a252013-07-25 18:35:19 +0000296 unsigned canReassign(LiveInterval &VirtReg, unsigned PhysReg);
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000297 bool shouldEvict(LiveInterval &A, bool, LiveInterval &B, bool);
298 bool canEvictInterference(LiveInterval&, unsigned, bool, EvictionCost&);
299 void evictInterference(LiveInterval&, unsigned,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000300 SmallVectorImpl<unsigned>&);
Quentin Colombet87769712014-02-05 22:13:59 +0000301 bool mayRecolorAllInterferences(unsigned PhysReg, LiveInterval &VirtReg,
302 SmallLISet &RecoloringCandidates,
303 const SmallVirtRegSet &FixedRegisters);
Jakob Stoklund Olesen3d7b8062010-12-14 00:37:44 +0000304
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000305 unsigned tryAssign(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000306 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000307 unsigned tryEvict(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000308 SmallVectorImpl<unsigned>&, unsigned = ~0u);
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000309 unsigned tryRegionSplit(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000310 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +0000311 unsigned tryBlockSplit(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000312 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +0000313 unsigned tryInstructionSplit(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000314 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +0000315 unsigned tryLocalSplit(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000316 SmallVectorImpl<unsigned>&);
Jakob Stoklund Olesen3d7b8062010-12-14 00:37:44 +0000317 unsigned trySplit(LiveInterval&, AllocationOrder&,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000318 SmallVectorImpl<unsigned>&);
Quentin Colombet87769712014-02-05 22:13:59 +0000319 unsigned tryLastChanceRecoloring(LiveInterval &, AllocationOrder &,
320 SmallVectorImpl<unsigned> &,
321 SmallVirtRegSet &, unsigned);
322 bool tryRecoloringCandidates(PQueue &, SmallVectorImpl<unsigned> &,
323 SmallVirtRegSet &, unsigned);
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000324};
325} // end anonymous namespace
326
327char RAGreedy::ID = 0;
328
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000329#ifndef NDEBUG
330const char *const RAGreedy::StageName[] = {
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000331 "RS_New",
332 "RS_Assign",
333 "RS_Split",
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +0000334 "RS_Split2",
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000335 "RS_Spill",
336 "RS_Done"
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000337};
338#endif
339
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +0000340// Hysteresis to use when comparing floats.
341// This helps stabilize decisions based on float comparisons.
NAKAMURA Takumia71003a2014-02-04 06:29:38 +0000342const float Hysteresis = (2007 / 2048.0f); // 0.97998046875
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +0000343
344
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000345FunctionPass* llvm::createGreedyRegisterAllocator() {
346 return new RAGreedy();
347}
348
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000349RAGreedy::RAGreedy(): MachineFunctionPass(ID) {
Jakob Stoklund Olesen6aa0fbf2011-04-05 21:40:37 +0000350 initializeLiveDebugVariablesPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000351 initializeSlotIndexesPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000352 initializeLiveIntervalsPass(*PassRegistry::getPassRegistry());
353 initializeSlotIndexesPass(*PassRegistry::getPassRegistry());
Rafael Espindola676c4052011-06-26 22:34:10 +0000354 initializeRegisterCoalescerPass(*PassRegistry::getPassRegistry());
Andrew Tricke1c034f2012-01-17 06:55:03 +0000355 initializeMachineSchedulerPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000356 initializeLiveStacksPass(*PassRegistry::getPassRegistry());
357 initializeMachineDominatorTreePass(*PassRegistry::getPassRegistry());
358 initializeMachineLoopInfoPass(*PassRegistry::getPassRegistry());
359 initializeVirtRegMapPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000360 initializeLiveRegMatrixPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000361 initializeEdgeBundlesPass(*PassRegistry::getPassRegistry());
362 initializeSpillPlacementPass(*PassRegistry::getPassRegistry());
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000363}
364
365void RAGreedy::getAnalysisUsage(AnalysisUsage &AU) const {
366 AU.setPreservesCFG();
Benjamin Kramere2a1d892013-06-17 19:00:36 +0000367 AU.addRequired<MachineBlockFrequencyInfo>();
368 AU.addPreserved<MachineBlockFrequencyInfo>();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000369 AU.addRequired<AliasAnalysis>();
370 AU.addPreserved<AliasAnalysis>();
371 AU.addRequired<LiveIntervals>();
Jakob Stoklund Olesen12243122012-06-08 23:44:45 +0000372 AU.addPreserved<LiveIntervals>();
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000373 AU.addRequired<SlotIndexes>();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000374 AU.addPreserved<SlotIndexes>();
Jakob Stoklund Olesen6aa0fbf2011-04-05 21:40:37 +0000375 AU.addRequired<LiveDebugVariables>();
376 AU.addPreserved<LiveDebugVariables>();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000377 AU.addRequired<LiveStacks>();
378 AU.addPreserved<LiveStacks>();
Jakob Stoklund Olesen1740e002010-12-17 23:16:32 +0000379 AU.addRequired<MachineDominatorTree>();
380 AU.addPreserved<MachineDominatorTree>();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000381 AU.addRequired<MachineLoopInfo>();
382 AU.addPreserved<MachineLoopInfo>();
383 AU.addRequired<VirtRegMap>();
384 AU.addPreserved<VirtRegMap>();
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000385 AU.addRequired<LiveRegMatrix>();
386 AU.addPreserved<LiveRegMatrix>();
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000387 AU.addRequired<EdgeBundles>();
388 AU.addRequired<SpillPlacement>();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000389 MachineFunctionPass::getAnalysisUsage(AU);
390}
391
Jakob Stoklund Olesen8e089642011-03-09 00:57:29 +0000392
393//===----------------------------------------------------------------------===//
394// LiveRangeEdit delegate methods
395//===----------------------------------------------------------------------===//
396
Jakob Stoklund Olesen43a87502011-03-13 01:23:11 +0000397bool RAGreedy::LRE_CanEraseVirtReg(unsigned VirtReg) {
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000398 if (VRM->hasPhys(VirtReg)) {
399 Matrix->unassign(LIS->getInterval(VirtReg));
Jakob Stoklund Olesen43a87502011-03-13 01:23:11 +0000400 return true;
401 }
402 // Unassigned virtreg is probably in the priority queue.
403 // RegAllocBase will erase it after dequeueing.
404 return false;
405}
Jakob Stoklund Olesen8e089642011-03-09 00:57:29 +0000406
Jakob Stoklund Olesene14b2b22011-03-16 22:56:16 +0000407void RAGreedy::LRE_WillShrinkVirtReg(unsigned VirtReg) {
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000408 if (!VRM->hasPhys(VirtReg))
Jakob Stoklund Olesene14b2b22011-03-16 22:56:16 +0000409 return;
410
411 // Register is assigned, put it back on the queue for reassignment.
412 LiveInterval &LI = LIS->getInterval(VirtReg);
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000413 Matrix->unassign(LI);
Jakob Stoklund Olesene14b2b22011-03-16 22:56:16 +0000414 enqueue(&LI);
415}
416
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000417void RAGreedy::LRE_DidCloneVirtReg(unsigned New, unsigned Old) {
Jakob Stoklund Olesen811b9c42011-09-14 17:34:37 +0000418 // Cloning a register we haven't even heard about yet? Just ignore it.
419 if (!ExtraRegInfo.inBounds(Old))
420 return;
421
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000422 // LRE may clone a virtual register because dead code elimination causes it to
Jakob Stoklund Olesen5387bd32011-07-26 00:54:56 +0000423 // be split into connected components. The new components are much smaller
424 // than the original, so they should get a new chance at being assigned.
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000425 // same stage as the parent.
Jakob Stoklund Olesen5387bd32011-07-26 00:54:56 +0000426 ExtraRegInfo[Old].Stage = RS_Assign;
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000427 ExtraRegInfo.grow(New);
428 ExtraRegInfo[New] = ExtraRegInfo[Old];
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000429}
430
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000431void RAGreedy::releaseMemory() {
432 SpillerInstance.reset(0);
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000433 ExtraRegInfo.clear();
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000434 GlobalCand.clear();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +0000435}
436
Quentin Colombet87769712014-02-05 22:13:59 +0000437void RAGreedy::enqueue(LiveInterval *LI) { enqueue(Queue, LI); }
438
439void RAGreedy::enqueue(PQueue &CurQueue, LiveInterval *LI) {
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +0000440 // Prioritize live ranges by size, assigning larger ranges first.
441 // The queue holds (size, reg) pairs.
Jakob Stoklund Olesene68a27e2011-02-24 23:21:36 +0000442 const unsigned Size = LI->getSize();
443 const unsigned Reg = LI->reg;
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +0000444 assert(TargetRegisterInfo::isVirtualRegister(Reg) &&
445 "Can only enqueue virtual registers");
Jakob Stoklund Olesene68a27e2011-02-24 23:21:36 +0000446 unsigned Prio;
Jakob Stoklund Oleseneaa650a2010-12-08 22:57:16 +0000447
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000448 ExtraRegInfo.grow(Reg);
449 if (ExtraRegInfo[Reg].Stage == RS_New)
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000450 ExtraRegInfo[Reg].Stage = RS_Assign;
Jakob Stoklund Olesendd9a2ec2011-03-30 02:52:39 +0000451
Jakob Stoklund Olesencad845f2011-07-28 20:48:23 +0000452 if (ExtraRegInfo[Reg].Stage == RS_Split) {
Jakob Stoklund Olesen28d79cd2011-03-27 22:49:21 +0000453 // Unsplit ranges that couldn't be allocated immediately are deferred until
Jakob Stoklund Olesen45df7e02011-09-12 16:54:42 +0000454 // everything else has been allocated.
455 Prio = Size;
Jakob Stoklund Olesencad845f2011-07-28 20:48:23 +0000456 } else {
Andrew Trick52a00932014-02-26 22:07:26 +0000457 // Giant live ranges fall back to the global assignment heuristic, which
458 // prevents excessive spilling in pathological cases.
459 bool ReverseLocal = TRI->reverseLocalAssignment();
Andrew Trickb1531e52014-02-27 21:37:33 +0000460 bool ForceGlobal = !ReverseLocal && TRI->mayOverrideLocalAssignment() &&
Andrew Trick52a00932014-02-26 22:07:26 +0000461 (Size / SlotIndex::InstrDist) > (2 * MRI->getRegClass(Reg)->getNumRegs());
462
463 if (ExtraRegInfo[Reg].Stage == RS_Assign && !ForceGlobal && !LI->empty() &&
Andrew Trick84852572013-07-25 18:35:14 +0000464 LIS->intervalIsInOneMBB(*LI)) {
465 // Allocate original local ranges in linear instruction order. Since they
466 // are singly defined, this produces optimal coloring in the absence of
467 // global interference and other constraints.
Andrew Trick52a00932014-02-26 22:07:26 +0000468 if (!ReverseLocal)
Andrew Trick2d8826a2013-12-11 03:40:15 +0000469 Prio = LI->beginIndex().getInstrDistance(Indexes->getLastIndex());
470 else {
471 // Allocating bottom up may allow many short LRGs to be assigned first
472 // to one of the cheap registers. This could be much faster for very
473 // large blocks on targets with many physical registers.
474 Prio = Indexes->getZeroIndex().getInstrDistance(LI->beginIndex());
475 }
Andrew Trick84852572013-07-25 18:35:14 +0000476 }
477 else {
478 // Allocate global and split ranges in long->short order. Long ranges that
479 // don't fit should be spilled (or split) ASAP so they don't create
480 // interference. Mark a bit to prioritize global above local ranges.
481 Prio = (1u << 29) + Size;
482 }
483 // Mark a higher bit to prioritize global and local above RS_Split.
484 Prio |= (1u << 31);
Jakob Stoklund Olesenb51f65c2011-02-23 00:56:56 +0000485
Jakob Stoklund Olesen28d79cd2011-03-27 22:49:21 +0000486 // Boost ranges that have a physical register hint.
Jakob Stoklund Olesen74052b02012-12-03 23:23:50 +0000487 if (VRM->hasKnownPreference(Reg))
Jakob Stoklund Olesen28d79cd2011-03-27 22:49:21 +0000488 Prio |= (1u << 30);
489 }
Andrew Trickf4b1ee32013-07-25 18:35:22 +0000490 // The virtual register number is a tie breaker for same-sized ranges.
491 // Give lower vreg numbers higher priority to assign them first.
Quentin Colombet87769712014-02-05 22:13:59 +0000492 CurQueue.push(std::make_pair(Prio, ~Reg));
Jakob Stoklund Oleseneaa650a2010-12-08 22:57:16 +0000493}
494
Quentin Colombet87769712014-02-05 22:13:59 +0000495LiveInterval *RAGreedy::dequeue() { return dequeue(Queue); }
496
497LiveInterval *RAGreedy::dequeue(PQueue &CurQueue) {
498 if (CurQueue.empty())
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +0000499 return 0;
Quentin Colombet87769712014-02-05 22:13:59 +0000500 LiveInterval *LI = &LIS->getInterval(~CurQueue.top().second);
501 CurQueue.pop();
Jakob Stoklund Olesen2329c542011-02-22 23:01:52 +0000502 return LI;
503}
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +0000504
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000505
506//===----------------------------------------------------------------------===//
507// Direct Assignment
508//===----------------------------------------------------------------------===//
509
510/// tryAssign - Try to assign VirtReg to an available register.
511unsigned RAGreedy::tryAssign(LiveInterval &VirtReg,
512 AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000513 SmallVectorImpl<unsigned> &NewVRegs) {
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000514 Order.rewind();
515 unsigned PhysReg;
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000516 while ((PhysReg = Order.next()))
517 if (!Matrix->checkInterference(VirtReg, PhysReg))
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000518 break;
Jakob Stoklund Olesen3cb2cb82012-12-04 22:25:16 +0000519 if (!PhysReg || Order.isHint())
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000520 return PhysReg;
521
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000522 // PhysReg is available, but there may be a better choice.
523
524 // If we missed a simple hint, try to cheaply evict interference from the
525 // preferred register.
526 if (unsigned Hint = MRI->getSimpleHint(VirtReg.reg))
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000527 if (Order.isHint(Hint)) {
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000528 DEBUG(dbgs() << "missed hint " << PrintReg(Hint, TRI) << '\n');
Andrew Trick3621b8a2013-11-22 19:07:38 +0000529 EvictionCost MaxCost;
530 MaxCost.setBrokenHints(1);
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000531 if (canEvictInterference(VirtReg, Hint, true, MaxCost)) {
532 evictInterference(VirtReg, Hint, NewVRegs);
533 return Hint;
534 }
535 }
536
537 // Try to evict interference from a cheaper alternative.
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000538 unsigned Cost = TRI->getCostPerUse(PhysReg);
539
540 // Most registers have 0 additional cost.
541 if (!Cost)
542 return PhysReg;
543
544 DEBUG(dbgs() << PrintReg(PhysReg, TRI) << " is available at cost " << Cost
545 << '\n');
546 unsigned CheapReg = tryEvict(VirtReg, Order, NewVRegs, Cost);
547 return CheapReg ? CheapReg : PhysReg;
548}
549
550
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +0000551//===----------------------------------------------------------------------===//
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000552// Interference eviction
553//===----------------------------------------------------------------------===//
554
Andrew Trick8bb0a252013-07-25 18:35:19 +0000555unsigned RAGreedy::canReassign(LiveInterval &VirtReg, unsigned PrevReg) {
556 AllocationOrder Order(VirtReg.reg, *VRM, RegClassInfo);
557 unsigned PhysReg;
558 while ((PhysReg = Order.next())) {
559 if (PhysReg == PrevReg)
560 continue;
561
562 MCRegUnitIterator Units(PhysReg, TRI);
563 for (; Units.isValid(); ++Units) {
564 // Instantiate a "subquery", not to be confused with the Queries array.
565 LiveIntervalUnion::Query subQ(&VirtReg, &Matrix->getLiveUnions()[*Units]);
566 if (subQ.checkInterference())
567 break;
568 }
569 // If no units have interference, break out with the current PhysReg.
570 if (!Units.isValid())
571 break;
572 }
573 if (PhysReg)
574 DEBUG(dbgs() << "can reassign: " << VirtReg << " from "
575 << PrintReg(PrevReg, TRI) << " to " << PrintReg(PhysReg, TRI)
576 << '\n');
577 return PhysReg;
578}
579
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000580/// shouldEvict - determine if A should evict the assigned live range B. The
581/// eviction policy defined by this function together with the allocation order
582/// defined by enqueue() decides which registers ultimately end up being split
583/// and spilled.
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000584///
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000585/// Cascade numbers are used to prevent infinite loops if this function is a
586/// cyclic relation.
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000587///
588/// @param A The live range to be assigned.
589/// @param IsHint True when A is about to be assigned to its preferred
590/// register.
591/// @param B The live range to be evicted.
592/// @param BreaksHint True when B is already assigned to its preferred register.
593bool RAGreedy::shouldEvict(LiveInterval &A, bool IsHint,
594 LiveInterval &B, bool BreaksHint) {
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +0000595 bool CanSplit = getStage(B) < RS_Spill;
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000596
597 // Be fairly aggressive about following hints as long as the evictee can be
598 // split.
599 if (CanSplit && IsHint && !BreaksHint)
600 return true;
601
Andrew Trick059e8002013-11-22 19:07:42 +0000602 if (A.weight > B.weight) {
603 DEBUG(dbgs() << "should evict: " << B << " w= " << B.weight << '\n');
604 return true;
605 }
606 return false;
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +0000607}
608
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000609/// canEvictInterference - Return true if all interferences between VirtReg and
Manman Renfa32ca12014-02-25 19:47:15 +0000610/// PhysReg can be evicted.
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000611///
612/// @param VirtReg Live range that is about to be assigned.
613/// @param PhysReg Desired register for assignment.
Dmitri Gribenko881929c2012-09-12 16:59:47 +0000614/// @param IsHint True when PhysReg is VirtReg's preferred register.
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000615/// @param MaxCost Only look for cheaper candidates and update with new cost
616/// when returning true.
617/// @returns True when interference can be evicted cheaper than MaxCost.
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000618bool RAGreedy::canEvictInterference(LiveInterval &VirtReg, unsigned PhysReg,
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000619 bool IsHint, EvictionCost &MaxCost) {
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000620 // It is only possible to evict virtual register interference.
621 if (Matrix->checkInterference(VirtReg, PhysReg) > LiveRegMatrix::IK_VirtReg)
622 return false;
623
Andrew Trick84852572013-07-25 18:35:14 +0000624 bool IsLocal = LIS->intervalIsInOneMBB(VirtReg);
625
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000626 // Find VirtReg's cascade number. This will be unassigned if VirtReg was never
627 // involved in an eviction before. If a cascade number was assigned, deny
628 // evicting anything with the same or a newer cascade number. This prevents
629 // infinite eviction loops.
630 //
631 // This works out so a register without a cascade number is allowed to evict
632 // anything, and it can be evicted by anything.
633 unsigned Cascade = ExtraRegInfo[VirtReg.reg].Cascade;
634 if (!Cascade)
635 Cascade = NextCascade;
636
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000637 EvictionCost Cost;
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000638 for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
639 LiveIntervalUnion::Query &Q = Matrix->query(VirtReg, *Units);
Jakob Stoklund Olesen0f175eb2011-04-11 21:47:01 +0000640 // If there is 10 or more interferences, chances are one is heavier.
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000641 if (Q.collectInterferingVRegs(10) >= 10)
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000642 return false;
643
Jakob Stoklund Olesen0f175eb2011-04-11 21:47:01 +0000644 // Check if any interfering live range is heavier than MaxWeight.
645 for (unsigned i = Q.interferingVRegs().size(); i; --i) {
646 LiveInterval *Intf = Q.interferingVRegs()[i - 1];
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000647 assert(TargetRegisterInfo::isVirtualRegister(Intf->reg) &&
648 "Only expecting virtual register interference from query");
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000649 // Never evict spill products. They cannot split or spill.
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +0000650 if (getStage(*Intf) == RS_Done)
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +0000651 return false;
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000652 // Once a live range becomes small enough, it is urgent that we find a
653 // register for it. This is indicated by an infinite spill weight. These
654 // urgent live ranges get to evict almost anything.
Jakob Stoklund Olesen05e22452012-05-30 21:46:58 +0000655 //
656 // Also allow urgent evictions of unspillable ranges from a strictly
657 // larger allocation order.
658 bool Urgent = !VirtReg.isSpillable() &&
659 (Intf->isSpillable() ||
660 RegClassInfo.getNumAllocatableRegs(MRI->getRegClass(VirtReg.reg)) <
661 RegClassInfo.getNumAllocatableRegs(MRI->getRegClass(Intf->reg)));
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000662 // Only evict older cascades or live ranges without a cascade.
663 unsigned IntfCascade = ExtraRegInfo[Intf->reg].Cascade;
664 if (Cascade <= IntfCascade) {
665 if (!Urgent)
666 return false;
667 // We permit breaking cascades for urgent evictions. It should be the
668 // last resort, though, so make it really expensive.
669 Cost.BrokenHints += 10;
670 }
671 // Would this break a satisfied hint?
672 bool BreaksHint = VRM->hasPreferredPhys(Intf->reg);
673 // Update eviction cost.
674 Cost.BrokenHints += BreaksHint;
675 Cost.MaxWeight = std::max(Cost.MaxWeight, Intf->weight);
676 // Abort if this would be too expensive.
677 if (!(Cost < MaxCost))
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000678 return false;
Andrew Trick84852572013-07-25 18:35:14 +0000679 if (Urgent)
680 continue;
Andrew Trickc2ab53a2013-11-29 23:49:38 +0000681 // Apply the eviction policy for non-urgent evictions.
682 if (!shouldEvict(VirtReg, IsHint, *Intf, BreaksHint))
683 return false;
Andrew Trick84852572013-07-25 18:35:14 +0000684 // If !MaxCost.isMax(), then we're just looking for a cheap register.
685 // Evicting another local live range in this case could lead to suboptimal
686 // coloring.
Andrew Trick8bb0a252013-07-25 18:35:19 +0000687 if (!MaxCost.isMax() && IsLocal && LIS->intervalIsInOneMBB(*Intf) &&
688 !canReassign(*Intf, PhysReg)) {
Andrew Trick84852572013-07-25 18:35:14 +0000689 return false;
Andrew Trick8bb0a252013-07-25 18:35:19 +0000690 }
Jakob Stoklund Olesen1305bc02011-02-09 01:14:03 +0000691 }
692 }
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000693 MaxCost = Cost;
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000694 return true;
695}
Jakob Stoklund Olesen1305bc02011-02-09 01:14:03 +0000696
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000697/// evictInterference - Evict any interferring registers that prevent VirtReg
698/// from being assigned to Physreg. This assumes that canEvictInterference
699/// returned true.
700void RAGreedy::evictInterference(LiveInterval &VirtReg, unsigned PhysReg,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000701 SmallVectorImpl<unsigned> &NewVRegs) {
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000702 // Make sure that VirtReg has a cascade number, and assign that cascade
703 // number to every evicted register. These live ranges than then only be
704 // evicted by a newer cascade, preventing infinite loops.
705 unsigned Cascade = ExtraRegInfo[VirtReg.reg].Cascade;
706 if (!Cascade)
707 Cascade = ExtraRegInfo[VirtReg.reg].Cascade = NextCascade++;
708
709 DEBUG(dbgs() << "evicting " << PrintReg(PhysReg, TRI)
710 << " interference: Cascade " << Cascade << '\n');
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000711
712 // Collect all interfering virtregs first.
713 SmallVector<LiveInterval*, 8> Intfs;
714 for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
715 LiveIntervalUnion::Query &Q = Matrix->query(VirtReg, *Units);
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000716 assert(Q.seenAllInterferences() && "Didn't check all interfererences.");
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +0000717 ArrayRef<LiveInterval*> IVR = Q.interferingVRegs();
718 Intfs.append(IVR.begin(), IVR.end());
719 }
720
721 // Evict them second. This will invalidate the queries.
722 for (unsigned i = 0, e = Intfs.size(); i != e; ++i) {
723 LiveInterval *Intf = Intfs[i];
724 // The same VirtReg may be present in multiple RegUnits. Skip duplicates.
725 if (!VRM->hasPhys(Intf->reg))
726 continue;
727 Matrix->unassign(*Intf);
728 assert((ExtraRegInfo[Intf->reg].Cascade < Cascade ||
729 VirtReg.isSpillable() < Intf->isSpillable()) &&
730 "Cannot decrease cascade number, illegal eviction");
731 ExtraRegInfo[Intf->reg].Cascade = Cascade;
732 ++NumEvicted;
Mark Laceyf9ea8852013-08-14 23:50:04 +0000733 NewVRegs.push_back(Intf->reg);
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000734 }
735}
736
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000737/// tryEvict - Try to evict all interferences for a physreg.
Jakob Stoklund Olesene9cc8e92011-06-01 18:45:02 +0000738/// @param VirtReg Currently unassigned virtual register.
739/// @param Order Physregs to try.
740/// @return Physreg to assign VirtReg, or 0.
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000741unsigned RAGreedy::tryEvict(LiveInterval &VirtReg,
742 AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +0000743 SmallVectorImpl<unsigned> &NewVRegs,
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000744 unsigned CostPerUseLimit) {
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000745 NamedRegionTimer T("Evict", TimerGroupName, TimePassesIsEnabled);
746
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000747 // Keep track of the cheapest interference seen so far.
Andrew Trick3621b8a2013-11-22 19:07:38 +0000748 EvictionCost BestCost;
749 BestCost.setMax();
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000750 unsigned BestPhys = 0;
Jakob Stoklund Olesen3dd236c2013-01-12 00:57:44 +0000751 unsigned OrderLimit = Order.getOrder().size();
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000752
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000753 // When we are just looking for a reduced cost per use, don't break any
754 // hints, and only evict smaller spill weights.
755 if (CostPerUseLimit < ~0u) {
756 BestCost.BrokenHints = 0;
757 BestCost.MaxWeight = VirtReg.weight;
Jakob Stoklund Olesen3dd236c2013-01-12 00:57:44 +0000758
759 // Check of any registers in RC are below CostPerUseLimit.
760 const TargetRegisterClass *RC = MRI->getRegClass(VirtReg.reg);
761 unsigned MinCost = RegClassInfo.getMinCost(RC);
762 if (MinCost >= CostPerUseLimit) {
763 DEBUG(dbgs() << RC->getName() << " minimum cost = " << MinCost
764 << ", no cheaper registers to be found.\n");
765 return 0;
766 }
767
768 // It is normal for register classes to have a long tail of registers with
769 // the same cost. We don't need to look at them if they're too expensive.
770 if (TRI->getCostPerUse(Order.getOrder().back()) >= CostPerUseLimit) {
771 OrderLimit = RegClassInfo.getLastCostChange(RC);
772 DEBUG(dbgs() << "Only trying the first " << OrderLimit << " regs.\n");
773 }
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000774 }
775
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000776 Order.rewind();
Aditya Nandakumar73f3d332013-12-05 21:18:40 +0000777 while (unsigned PhysReg = Order.next(OrderLimit)) {
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000778 if (TRI->getCostPerUse(PhysReg) >= CostPerUseLimit)
779 continue;
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000780 // The first use of a callee-saved register in a function has cost 1.
781 // Don't start using a CSR when the CostPerUseLimit is low.
782 if (CostPerUseLimit == 1)
783 if (unsigned CSR = RegClassInfo.getLastCalleeSavedAlias(PhysReg))
784 if (!MRI->isPhysRegUsed(CSR)) {
785 DEBUG(dbgs() << PrintReg(PhysReg, TRI) << " would clobber CSR "
786 << PrintReg(CSR, TRI) << '\n');
787 continue;
788 }
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +0000789
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000790 if (!canEvictInterference(VirtReg, PhysReg, false, BestCost))
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000791 continue;
792
793 // Best so far.
794 BestPhys = PhysReg;
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000795
Jakob Stoklund Olesen9918b332011-02-25 01:04:22 +0000796 // Stop if the hint can be used.
Jakob Stoklund Olesen3cb2cb82012-12-04 22:25:16 +0000797 if (Order.isHint())
Jakob Stoklund Olesen9918b332011-02-25 01:04:22 +0000798 break;
Jakob Stoklund Olesen1305bc02011-02-09 01:14:03 +0000799 }
800
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000801 if (!BestPhys)
802 return 0;
803
Jakob Stoklund Olesen4931bbc2011-07-08 20:46:18 +0000804 evictInterference(VirtReg, BestPhys, NewVRegs);
Jakob Stoklund Olesen6bd68cd2011-02-23 00:29:52 +0000805 return BestPhys;
Andrew Trickccef0982010-12-09 18:15:21 +0000806}
807
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +0000808
809//===----------------------------------------------------------------------===//
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000810// Region Splitting
811//===----------------------------------------------------------------------===//
812
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000813/// addSplitConstraints - Fill out the SplitConstraints vector based on the
814/// interference pattern in Physreg and its aliases. Add the constraints to
815/// SpillPlacement and return the static cost of this split in Cost, assuming
816/// that all preferences in SplitConstraints are met.
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000817/// Return false if there are no bundles with positive bias.
818bool RAGreedy::addSplitConstraints(InterferenceCache::Cursor Intf,
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000819 BlockFrequency &Cost) {
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000820 ArrayRef<SplitAnalysis::BlockInfo> UseBlocks = SA->getUseBlocks();
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000821
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000822 // Reset interference dependent info.
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000823 SplitConstraints.resize(UseBlocks.size());
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000824 BlockFrequency StaticCost = 0;
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000825 for (unsigned i = 0; i != UseBlocks.size(); ++i) {
826 const SplitAnalysis::BlockInfo &BI = UseBlocks[i];
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000827 SpillPlacement::BlockConstraint &BC = SplitConstraints[i];
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000828
Jakob Stoklund Olesenb1b76ad2011-02-09 22:50:26 +0000829 BC.Number = BI.MBB->getNumber();
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000830 Intf.moveToBlock(BC.Number);
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000831 BC.Entry = BI.LiveIn ? SpillPlacement::PrefReg : SpillPlacement::DontCare;
832 BC.Exit = BI.LiveOut ? SpillPlacement::PrefReg : SpillPlacement::DontCare;
David Blaikie041f1aa2013-05-15 07:36:59 +0000833 BC.ChangesValue = BI.FirstDef.isValid();
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000834
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000835 if (!Intf.hasInterference())
836 continue;
837
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000838 // Number of spill code instructions to insert.
839 unsigned Ins = 0;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000840
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000841 // Interference for the live-in value.
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000842 if (BI.LiveIn) {
Jakob Stoklund Olesen89339072011-04-04 15:32:15 +0000843 if (Intf.first() <= Indexes->getMBBStartIdx(BC.Number))
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000844 BC.Entry = SpillPlacement::MustSpill, ++Ins;
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +0000845 else if (Intf.first() < BI.FirstInstr)
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000846 BC.Entry = SpillPlacement::PrefSpill, ++Ins;
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +0000847 else if (Intf.first() < BI.LastInstr)
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000848 ++Ins;
Jakob Stoklund Olesenf248b202011-02-08 23:02:58 +0000849 }
850
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000851 // Interference for the live-out value.
Jakob Stoklund Olesenca26e0a2011-04-02 06:03:38 +0000852 if (BI.LiveOut) {
Jakob Stoklund Olesend93b0e32011-04-05 04:20:29 +0000853 if (Intf.last() >= SA->getLastSplitPoint(BC.Number))
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000854 BC.Exit = SpillPlacement::MustSpill, ++Ins;
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +0000855 else if (Intf.last() > BI.LastInstr)
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000856 BC.Exit = SpillPlacement::PrefSpill, ++Ins;
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +0000857 else if (Intf.last() > BI.FirstInstr)
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000858 ++Ins;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000859 }
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000860
861 // Accumulate the total frequency of inserted spill code.
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000862 while (Ins--)
863 StaticCost += SpillPlacer->getBlockFrequency(BC.Number);
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000864 }
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000865 Cost = StaticCost;
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000866
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000867 // Add constraints for use-blocks. Note that these are the only constraints
868 // that may add a positive bias, it is downhill from here.
869 SpillPlacer->addConstraints(SplitConstraints);
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000870 return SpillPlacer->scanActiveBundles();
871}
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000872
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000873
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000874/// addThroughConstraints - Add constraints and links to SpillPlacer from the
875/// live-through blocks in Blocks.
876void RAGreedy::addThroughConstraints(InterferenceCache::Cursor Intf,
877 ArrayRef<unsigned> Blocks) {
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000878 const unsigned GroupSize = 8;
879 SpillPlacement::BlockConstraint BCS[GroupSize];
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000880 unsigned TBS[GroupSize];
881 unsigned B = 0, T = 0;
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000882
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000883 for (unsigned i = 0; i != Blocks.size(); ++i) {
884 unsigned Number = Blocks[i];
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000885 Intf.moveToBlock(Number);
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000886
Jakob Stoklund Olesen6d2bbc12011-04-07 17:27:46 +0000887 if (!Intf.hasInterference()) {
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000888 assert(T < GroupSize && "Array overflow");
889 TBS[T] = Number;
890 if (++T == GroupSize) {
Frits van Bommel717d7ed2011-07-18 12:00:32 +0000891 SpillPlacer->addLinks(makeArrayRef(TBS, T));
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000892 T = 0;
893 }
Jakob Stoklund Olesen6d2bbc12011-04-07 17:27:46 +0000894 continue;
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000895 }
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000896
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000897 assert(B < GroupSize && "Array overflow");
898 BCS[B].Number = Number;
899
Jakob Stoklund Olesen6d2bbc12011-04-07 17:27:46 +0000900 // Interference for the live-in value.
901 if (Intf.first() <= Indexes->getMBBStartIdx(Number))
902 BCS[B].Entry = SpillPlacement::MustSpill;
903 else
904 BCS[B].Entry = SpillPlacement::PrefSpill;
905
906 // Interference for the live-out value.
907 if (Intf.last() >= SA->getLastSplitPoint(Number))
908 BCS[B].Exit = SpillPlacement::MustSpill;
909 else
910 BCS[B].Exit = SpillPlacement::PrefSpill;
911
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000912 if (++B == GroupSize) {
913 ArrayRef<SpillPlacement::BlockConstraint> Array(BCS, B);
914 SpillPlacer->addConstraints(Array);
915 B = 0;
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000916 }
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +0000917 }
918
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +0000919 ArrayRef<SpillPlacement::BlockConstraint> Array(BCS, B);
920 SpillPlacer->addConstraints(Array);
Frits van Bommel717d7ed2011-07-18 12:00:32 +0000921 SpillPlacer->addLinks(makeArrayRef(TBS, T));
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +0000922}
923
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +0000924void RAGreedy::growRegion(GlobalSplitCandidate &Cand) {
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000925 // Keep track of through blocks that have not been added to SpillPlacer.
926 BitVector Todo = SA->getThroughBlocks();
927 SmallVectorImpl<unsigned> &ActiveBlocks = Cand.ActiveBlocks;
928 unsigned AddedTo = 0;
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000929#ifndef NDEBUG
930 unsigned Visited = 0;
931#endif
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000932
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000933 for (;;) {
934 ArrayRef<unsigned> NewBundles = SpillPlacer->getRecentPositive();
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000935 // Find new through blocks in the periphery of PrefRegBundles.
936 for (int i = 0, e = NewBundles.size(); i != e; ++i) {
937 unsigned Bundle = NewBundles[i];
938 // Look at all blocks connected to Bundle in the full graph.
939 ArrayRef<unsigned> Blocks = Bundles->getBlocks(Bundle);
940 for (ArrayRef<unsigned>::iterator I = Blocks.begin(), E = Blocks.end();
941 I != E; ++I) {
942 unsigned Block = *I;
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000943 if (!Todo.test(Block))
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000944 continue;
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000945 Todo.reset(Block);
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000946 // This is a new through block. Add it to SpillPlacer later.
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +0000947 ActiveBlocks.push_back(Block);
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000948#ifndef NDEBUG
949 ++Visited;
950#endif
951 }
952 }
953 // Any new blocks to add?
Jakob Stoklund Olesen91f3a302011-07-05 18:46:42 +0000954 if (ActiveBlocks.size() == AddedTo)
955 break;
Jakob Stoklund Olesena953bf12011-07-23 03:22:33 +0000956
957 // Compute through constraints from the interference, or assume that all
958 // through blocks prefer spilling when forming compact regions.
959 ArrayRef<unsigned> NewBlocks = makeArrayRef(ActiveBlocks).slice(AddedTo);
960 if (Cand.PhysReg)
961 addThroughConstraints(Cand.Intf, NewBlocks);
962 else
Jakob Stoklund Olesen86954522011-08-03 23:09:38 +0000963 // Provide a strong negative bias on through blocks to prevent unwanted
964 // liveness on loop backedges.
965 SpillPlacer->addPrefSpill(NewBlocks, /* Strong= */ true);
Jakob Stoklund Olesen91f3a302011-07-05 18:46:42 +0000966 AddedTo = ActiveBlocks.size();
967
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000968 // Perhaps iterating can enable more bundles?
969 SpillPlacer->iterate();
970 }
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +0000971 DEBUG(dbgs() << ", v=" << Visited);
972}
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +0000973
Jakob Stoklund Olesenecad62f2011-07-23 03:41:57 +0000974/// calcCompactRegion - Compute the set of edge bundles that should be live
975/// when splitting the current live range into compact regions. Compact
976/// regions can be computed without looking at interference. They are the
977/// regions formed by removing all the live-through blocks from the live range.
978///
979/// Returns false if the current live range is already compact, or if the
980/// compact regions would form single block regions anyway.
981bool RAGreedy::calcCompactRegion(GlobalSplitCandidate &Cand) {
982 // Without any through blocks, the live range is already compact.
983 if (!SA->getNumThroughBlocks())
984 return false;
985
986 // Compact regions don't correspond to any physreg.
987 Cand.reset(IntfCache, 0);
988
989 DEBUG(dbgs() << "Compact region bundles");
990
991 // Use the spill placer to determine the live bundles. GrowRegion pretends
992 // that all the through blocks have interference when PhysReg is unset.
993 SpillPlacer->prepare(Cand.LiveBundles);
994
995 // The static split cost will be zero since Cand.Intf reports no interference.
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +0000996 BlockFrequency Cost;
Jakob Stoklund Olesenecad62f2011-07-23 03:41:57 +0000997 if (!addSplitConstraints(Cand.Intf, Cost)) {
998 DEBUG(dbgs() << ", none.\n");
999 return false;
1000 }
1001
1002 growRegion(Cand);
1003 SpillPlacer->finish();
1004
1005 if (!Cand.LiveBundles.any()) {
1006 DEBUG(dbgs() << ", none.\n");
1007 return false;
1008 }
1009
1010 DEBUG({
1011 for (int i = Cand.LiveBundles.find_first(); i>=0;
1012 i = Cand.LiveBundles.find_next(i))
1013 dbgs() << " EB#" << i;
1014 dbgs() << ".\n";
1015 });
1016 return true;
1017}
1018
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +00001019/// calcSpillCost - Compute how expensive it would be to split the live range in
1020/// SA around all use blocks instead of forming bundle regions.
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001021BlockFrequency RAGreedy::calcSpillCost() {
1022 BlockFrequency Cost = 0;
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +00001023 ArrayRef<SplitAnalysis::BlockInfo> UseBlocks = SA->getUseBlocks();
1024 for (unsigned i = 0; i != UseBlocks.size(); ++i) {
1025 const SplitAnalysis::BlockInfo &BI = UseBlocks[i];
1026 unsigned Number = BI.MBB->getNumber();
1027 // We normally only need one spill instruction - a load or a store.
1028 Cost += SpillPlacer->getBlockFrequency(Number);
1029
1030 // Unless the value is redefined in the block.
Jakob Stoklund Olesen3c145052011-08-02 23:04:08 +00001031 if (BI.LiveIn && BI.LiveOut && BI.FirstDef)
1032 Cost += SpillPlacer->getBlockFrequency(Number);
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +00001033 }
1034 return Cost;
1035}
1036
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001037/// calcGlobalSplitCost - Return the global split cost of following the split
1038/// pattern in LiveBundles. This cost should be added to the local cost of the
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +00001039/// interference pattern in SplitConstraints.
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001040///
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001041BlockFrequency RAGreedy::calcGlobalSplitCost(GlobalSplitCandidate &Cand) {
1042 BlockFrequency GlobalCost = 0;
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +00001043 const BitVector &LiveBundles = Cand.LiveBundles;
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001044 ArrayRef<SplitAnalysis::BlockInfo> UseBlocks = SA->getUseBlocks();
1045 for (unsigned i = 0; i != UseBlocks.size(); ++i) {
1046 const SplitAnalysis::BlockInfo &BI = UseBlocks[i];
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +00001047 SpillPlacement::BlockConstraint &BC = SplitConstraints[i];
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001048 bool RegIn = LiveBundles[Bundles->getBundle(BC.Number, 0)];
1049 bool RegOut = LiveBundles[Bundles->getBundle(BC.Number, 1)];
1050 unsigned Ins = 0;
1051
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001052 if (BI.LiveIn)
1053 Ins += RegIn != (BC.Entry == SpillPlacement::PrefReg);
1054 if (BI.LiveOut)
1055 Ins += RegOut != (BC.Exit == SpillPlacement::PrefReg);
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001056 while (Ins--)
1057 GlobalCost += SpillPlacer->getBlockFrequency(BC.Number);
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001058 }
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001059
Jakob Stoklund Olesenc49df2c2011-04-12 21:30:53 +00001060 for (unsigned i = 0, e = Cand.ActiveBlocks.size(); i != e; ++i) {
1061 unsigned Number = Cand.ActiveBlocks[i];
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001062 bool RegIn = LiveBundles[Bundles->getBundle(Number, 0)];
1063 bool RegOut = LiveBundles[Bundles->getBundle(Number, 1)];
Jakob Stoklund Olesen8ce2f432011-04-06 21:32:41 +00001064 if (!RegIn && !RegOut)
1065 continue;
1066 if (RegIn && RegOut) {
1067 // We need double spill code if this block has interference.
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001068 Cand.Intf.moveToBlock(Number);
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001069 if (Cand.Intf.hasInterference()) {
1070 GlobalCost += SpillPlacer->getBlockFrequency(Number);
1071 GlobalCost += SpillPlacer->getBlockFrequency(Number);
1072 }
Jakob Stoklund Olesen8ce2f432011-04-06 21:32:41 +00001073 continue;
1074 }
1075 // live-in / stack-out or stack-in live-out.
1076 GlobalCost += SpillPlacer->getBlockFrequency(Number);
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001077 }
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001078 return GlobalCost;
1079}
1080
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001081/// splitAroundRegion - Split the current live range around the regions
1082/// determined by BundleCand and GlobalCand.
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001083///
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001084/// Before calling this function, GlobalCand and BundleCand must be initialized
1085/// so each bundle is assigned to a valid candidate, or NoCand for the
1086/// stack-bound bundles. The shared SA/SE SplitAnalysis and SplitEditor
1087/// objects must be initialized for the current live range, and intervals
1088/// created for the used candidates.
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001089///
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001090/// @param LREdit The LiveRangeEdit object handling the current split.
1091/// @param UsedCands List of used GlobalCand entries. Every BundleCand value
1092/// must appear in this list.
1093void RAGreedy::splitAroundRegion(LiveRangeEdit &LREdit,
1094 ArrayRef<unsigned> UsedCands) {
1095 // These are the intervals created for new global ranges. We may create more
1096 // intervals for local ranges.
1097 const unsigned NumGlobalIntvs = LREdit.size();
1098 DEBUG(dbgs() << "splitAroundRegion with " << NumGlobalIntvs << " globals.\n");
1099 assert(NumGlobalIntvs && "No global intervals configured");
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001100
Jakob Stoklund Olesen8627ea92011-08-05 22:20:45 +00001101 // Isolate even single instructions when dealing with a proper sub-class.
Jakob Stoklund Olesen22f37a12011-08-06 18:20:24 +00001102 // That guarantees register class inflation for the stack interval because it
Jakob Stoklund Olesen8627ea92011-08-05 22:20:45 +00001103 // is all copies.
1104 unsigned Reg = SA->getParent().reg;
1105 bool SingleInstrs = RegClassInfo.isProperSubClass(MRI->getRegClass(Reg));
1106
Jakob Stoklund Olesenadc6a4c2011-06-30 01:30:39 +00001107 // First handle all the blocks with uses.
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001108 ArrayRef<SplitAnalysis::BlockInfo> UseBlocks = SA->getUseBlocks();
1109 for (unsigned i = 0; i != UseBlocks.size(); ++i) {
1110 const SplitAnalysis::BlockInfo &BI = UseBlocks[i];
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001111 unsigned Number = BI.MBB->getNumber();
1112 unsigned IntvIn = 0, IntvOut = 0;
1113 SlotIndex IntfIn, IntfOut;
1114 if (BI.LiveIn) {
1115 unsigned CandIn = BundleCand[Bundles->getBundle(Number, 0)];
1116 if (CandIn != NoCand) {
1117 GlobalSplitCandidate &Cand = GlobalCand[CandIn];
1118 IntvIn = Cand.IntvIdx;
1119 Cand.Intf.moveToBlock(Number);
1120 IntfIn = Cand.Intf.first();
1121 }
1122 }
1123 if (BI.LiveOut) {
1124 unsigned CandOut = BundleCand[Bundles->getBundle(Number, 1)];
1125 if (CandOut != NoCand) {
1126 GlobalSplitCandidate &Cand = GlobalCand[CandOut];
1127 IntvOut = Cand.IntvIdx;
1128 Cand.Intf.moveToBlock(Number);
1129 IntfOut = Cand.Intf.last();
1130 }
1131 }
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001132
Jakob Stoklund Olesenc70b6972011-04-12 19:32:53 +00001133 // Create separate intervals for isolated blocks with multiple uses.
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001134 if (!IntvIn && !IntvOut) {
Jakob Stoklund Olesenc70b6972011-04-12 19:32:53 +00001135 DEBUG(dbgs() << "BB#" << BI.MBB->getNumber() << " isolated.\n");
Jakob Stoklund Olesen8627ea92011-08-05 22:20:45 +00001136 if (SA->shouldSplitSingleBlock(BI, SingleInstrs))
Jakob Stoklund Olesenadc6a4c2011-06-30 01:30:39 +00001137 SE->splitSingleBlock(BI);
Jakob Stoklund Olesenc70b6972011-04-12 19:32:53 +00001138 continue;
1139 }
1140
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001141 if (IntvIn && IntvOut)
1142 SE->splitLiveThroughBlock(Number, IntvIn, IntfIn, IntvOut, IntfOut);
1143 else if (IntvIn)
1144 SE->splitRegInBlock(BI, IntvIn, IntfIn);
Jakob Stoklund Olesen795da1c2011-07-15 21:47:57 +00001145 else
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001146 SE->splitRegOutBlock(BI, IntvOut, IntfOut);
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001147 }
1148
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001149 // Handle live-through blocks. The relevant live-through blocks are stored in
1150 // the ActiveBlocks list with each candidate. We need to filter out
1151 // duplicates.
1152 BitVector Todo = SA->getThroughBlocks();
1153 for (unsigned c = 0; c != UsedCands.size(); ++c) {
1154 ArrayRef<unsigned> Blocks = GlobalCand[UsedCands[c]].ActiveBlocks;
1155 for (unsigned i = 0, e = Blocks.size(); i != e; ++i) {
1156 unsigned Number = Blocks[i];
1157 if (!Todo.test(Number))
1158 continue;
1159 Todo.reset(Number);
1160
1161 unsigned IntvIn = 0, IntvOut = 0;
1162 SlotIndex IntfIn, IntfOut;
1163
1164 unsigned CandIn = BundleCand[Bundles->getBundle(Number, 0)];
1165 if (CandIn != NoCand) {
1166 GlobalSplitCandidate &Cand = GlobalCand[CandIn];
1167 IntvIn = Cand.IntvIdx;
1168 Cand.Intf.moveToBlock(Number);
1169 IntfIn = Cand.Intf.first();
1170 }
1171
1172 unsigned CandOut = BundleCand[Bundles->getBundle(Number, 1)];
1173 if (CandOut != NoCand) {
1174 GlobalSplitCandidate &Cand = GlobalCand[CandOut];
1175 IntvOut = Cand.IntvIdx;
1176 Cand.Intf.moveToBlock(Number);
1177 IntfOut = Cand.Intf.last();
1178 }
1179 if (!IntvIn && !IntvOut)
1180 continue;
1181 SE->splitLiveThroughBlock(Number, IntvIn, IntfIn, IntvOut, IntfOut);
1182 }
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001183 }
1184
Jakob Stoklund Olesen99827e82011-02-17 22:53:48 +00001185 ++NumGlobalSplits;
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001186
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001187 SmallVector<unsigned, 8> IntvMap;
1188 SE->finish(&IntvMap);
Mark Laceyf9ea8852013-08-14 23:50:04 +00001189 DebugVars->splitRegister(Reg, LREdit.regs(), *LIS);
Jakob Stoklund Olesenf8da0282011-05-06 18:00:02 +00001190
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +00001191 ExtraRegInfo.resize(MRI->getNumVirtRegs());
Jakob Stoklund Olesen5cc91b22011-05-28 02:32:57 +00001192 unsigned OrigBlocks = SA->getNumLiveBlocks();
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001193
1194 // Sort out the new intervals created by splitting. We get four kinds:
1195 // - Remainder intervals should not be split again.
1196 // - Candidate intervals can be assigned to Cand.PhysReg.
1197 // - Block-local splits are candidates for local splitting.
1198 // - DCE leftovers should go back on the queue.
1199 for (unsigned i = 0, e = LREdit.size(); i != e; ++i) {
Mark Laceyf9ea8852013-08-14 23:50:04 +00001200 LiveInterval &Reg = LIS->getInterval(LREdit.get(i));
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001201
1202 // Ignore old intervals from DCE.
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +00001203 if (getStage(Reg) != RS_New)
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001204 continue;
1205
1206 // Remainder interval. Don't try splitting again, spill if it doesn't
1207 // allocate.
1208 if (IntvMap[i] == 0) {
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00001209 setStage(Reg, RS_Spill);
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001210 continue;
1211 }
1212
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001213 // Global intervals. Allow repeated splitting as long as the number of live
1214 // blocks is strictly decreasing.
1215 if (IntvMap[i] < NumGlobalIntvs) {
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +00001216 if (SA->countLiveBlocks(&Reg) >= OrigBlocks) {
Jakob Stoklund Oleseneef23272011-04-26 22:33:12 +00001217 DEBUG(dbgs() << "Main interval covers the same " << OrigBlocks
1218 << " blocks as original.\n");
1219 // Don't allow repeated splitting as a safe guard against looping.
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001220 setStage(Reg, RS_Split2);
Jakob Stoklund Oleseneef23272011-04-26 22:33:12 +00001221 }
1222 continue;
1223 }
1224
1225 // Other intervals are treated as new. This includes local intervals created
1226 // for blocks with multiple uses, and anything created by DCE.
Jakob Stoklund Olesen6a663b82011-04-21 18:38:15 +00001227 }
1228
Jakob Stoklund Olesen28d79cd2011-03-27 22:49:21 +00001229 if (VerifyEnabled)
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001230 MF->verify(this, "After splitting live range around region");
1231}
1232
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001233unsigned RAGreedy::tryRegionSplit(LiveInterval &VirtReg, AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +00001234 SmallVectorImpl<unsigned> &NewVRegs) {
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001235 unsigned NumCands = 0;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001236 unsigned BestCand = NoCand;
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001237 BlockFrequency BestCost;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001238 SmallVector<unsigned, 8> UsedCands;
1239
1240 // Check if we can split this live range around a compact region.
Jakob Stoklund Olesen45df7e02011-09-12 16:54:42 +00001241 bool HasCompact = calcCompactRegion(GlobalCand.front());
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001242 if (HasCompact) {
1243 // Yes, keep GlobalCand[0] as the compact region candidate.
1244 NumCands = 1;
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001245 BestCost = BlockFrequency::getMaxFrequency();
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001246 } else {
1247 // No benefit from the compact region, our fallback will be per-block
1248 // splitting. Make sure we find a solution that is cheaper than spilling.
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001249 BestCost = calcSpillCost();
Michael Gottesmanb78dec82013-12-14 00:25:45 +00001250 DEBUG(dbgs() << "Cost of isolating all blocks = ";
1251 MBFI->printBlockFreq(dbgs(), BestCost) << '\n');
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001252 }
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +00001253
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001254 Order.rewind();
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001255 while (unsigned PhysReg = Order.next()) {
Jakob Stoklund Olesena153ca52011-07-14 05:35:11 +00001256 // Discard bad candidates before we run out of interference cache cursors.
1257 // This will only affect register classes with a lot of registers (>32).
1258 if (NumCands == IntfCache.getMaxCursors()) {
1259 unsigned WorstCount = ~0u;
1260 unsigned Worst = 0;
1261 for (unsigned i = 0; i != NumCands; ++i) {
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001262 if (i == BestCand || !GlobalCand[i].PhysReg)
Jakob Stoklund Olesena153ca52011-07-14 05:35:11 +00001263 continue;
1264 unsigned Count = GlobalCand[i].LiveBundles.count();
1265 if (Count < WorstCount)
1266 Worst = i, WorstCount = Count;
1267 }
1268 --NumCands;
1269 GlobalCand[Worst] = GlobalCand[NumCands];
Jakob Stoklund Olesen559d4dc2011-11-01 00:02:31 +00001270 if (BestCand == NumCands)
1271 BestCand = Worst;
Jakob Stoklund Olesena153ca52011-07-14 05:35:11 +00001272 }
1273
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001274 if (GlobalCand.size() <= NumCands)
1275 GlobalCand.resize(NumCands+1);
1276 GlobalSplitCandidate &Cand = GlobalCand[NumCands];
1277 Cand.reset(IntfCache, PhysReg);
Jakob Stoklund Olesen4b598e12011-03-05 01:10:31 +00001278
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001279 SpillPlacer->prepare(Cand.LiveBundles);
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001280 BlockFrequency Cost;
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001281 if (!addSplitConstraints(Cand.Intf, Cost)) {
Jakob Stoklund Olesened47ed42011-04-09 02:59:09 +00001282 DEBUG(dbgs() << PrintReg(PhysReg, TRI) << "\tno positive bundles\n");
Jakob Stoklund Olesen81439a82011-04-06 21:32:38 +00001283 continue;
1284 }
Michael Gottesmanb78dec82013-12-14 00:25:45 +00001285 DEBUG(dbgs() << PrintReg(PhysReg, TRI) << "\tstatic = ";
1286 MBFI->printBlockFreq(dbgs(), Cost));
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +00001287 if (Cost >= BestCost) {
1288 DEBUG({
1289 if (BestCand == NoCand)
1290 dbgs() << " worse than no bundles\n";
1291 else
1292 dbgs() << " worse than "
1293 << PrintReg(GlobalCand[BestCand].PhysReg, TRI) << '\n';
1294 });
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001295 continue;
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001296 }
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001297 growRegion(Cand);
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001298
Jakob Stoklund Olesen36b5d8a2011-04-06 19:13:57 +00001299 SpillPlacer->finish();
1300
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001301 // No live bundles, defer to splitSingleBlocks().
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001302 if (!Cand.LiveBundles.any()) {
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001303 DEBUG(dbgs() << " no bundles.\n");
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001304 continue;
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001305 }
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001306
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001307 Cost += calcGlobalSplitCost(Cand);
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001308 DEBUG({
Michael Gottesmanb78dec82013-12-14 00:25:45 +00001309 dbgs() << ", total = "; MBFI->printBlockFreq(dbgs(), Cost)
1310 << " with bundles";
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001311 for (int i = Cand.LiveBundles.find_first(); i>=0;
1312 i = Cand.LiveBundles.find_next(i))
Jakob Stoklund Olesen1a9b66c2011-03-05 03:28:51 +00001313 dbgs() << " EB#" << i;
1314 dbgs() << ".\n";
1315 });
Jakob Stoklund Olesen032891b2011-04-22 22:47:40 +00001316 if (Cost < BestCost) {
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001317 BestCand = NumCands;
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001318 BestCost = Cost;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001319 }
Jakob Stoklund Olesend7e99372011-07-14 00:17:10 +00001320 ++NumCands;
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001321 }
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001322
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001323 // No solutions found, fall back to single block splitting.
1324 if (!HasCompact && BestCand == NoCand)
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001325 return 0;
1326
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001327 // Prepare split editor.
Jakob Stoklund Olesene5bbe372012-05-19 05:25:46 +00001328 LiveRangeEdit LREdit(&VirtReg, NewVRegs, *MF, *LIS, VRM, this);
Jakob Stoklund Oleseneecb2fb2011-09-12 16:49:21 +00001329 SE->reset(LREdit, SplitSpillMode);
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001330
1331 // Assign all edge bundles to the preferred candidate, or NoCand.
1332 BundleCand.assign(Bundles->getNumBundles(), NoCand);
1333
1334 // Assign bundles for the best candidate region.
1335 if (BestCand != NoCand) {
1336 GlobalSplitCandidate &Cand = GlobalCand[BestCand];
1337 if (unsigned B = Cand.getBundles(BundleCand, BestCand)) {
1338 UsedCands.push_back(BestCand);
1339 Cand.IntvIdx = SE->openIntv();
1340 DEBUG(dbgs() << "Split for " << PrintReg(Cand.PhysReg, TRI) << " in "
1341 << B << " bundles, intv " << Cand.IntvIdx << ".\n");
Chandler Carruth77eb5a02011-08-03 23:07:27 +00001342 (void)B;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001343 }
1344 }
1345
1346 // Assign bundles for the compact region.
1347 if (HasCompact) {
1348 GlobalSplitCandidate &Cand = GlobalCand.front();
1349 assert(!Cand.PhysReg && "Compact region has no physreg");
1350 if (unsigned B = Cand.getBundles(BundleCand, 0)) {
1351 UsedCands.push_back(0);
1352 Cand.IntvIdx = SE->openIntv();
1353 DEBUG(dbgs() << "Split for compact region in " << B << " bundles, intv "
1354 << Cand.IntvIdx << ".\n");
Chandler Carruth77eb5a02011-08-03 23:07:27 +00001355 (void)B;
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00001356 }
1357 }
1358
1359 splitAroundRegion(LREdit, UsedCands);
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00001360 return 0;
1361}
1362
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001363
1364//===----------------------------------------------------------------------===//
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001365// Per-Block Splitting
1366//===----------------------------------------------------------------------===//
1367
1368/// tryBlockSplit - Split a global live range around every block with uses. This
1369/// creates a lot of local live ranges, that will be split by tryLocalSplit if
1370/// they don't allocate.
1371unsigned RAGreedy::tryBlockSplit(LiveInterval &VirtReg, AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +00001372 SmallVectorImpl<unsigned> &NewVRegs) {
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001373 assert(&SA->getParent() == &VirtReg && "Live range wasn't analyzed");
1374 unsigned Reg = VirtReg.reg;
1375 bool SingleInstrs = RegClassInfo.isProperSubClass(MRI->getRegClass(Reg));
Jakob Stoklund Olesene5bbe372012-05-19 05:25:46 +00001376 LiveRangeEdit LREdit(&VirtReg, NewVRegs, *MF, *LIS, VRM, this);
Jakob Stoklund Oleseneecb2fb2011-09-12 16:49:21 +00001377 SE->reset(LREdit, SplitSpillMode);
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001378 ArrayRef<SplitAnalysis::BlockInfo> UseBlocks = SA->getUseBlocks();
1379 for (unsigned i = 0; i != UseBlocks.size(); ++i) {
1380 const SplitAnalysis::BlockInfo &BI = UseBlocks[i];
1381 if (SA->shouldSplitSingleBlock(BI, SingleInstrs))
1382 SE->splitSingleBlock(BI);
1383 }
1384 // No blocks were split.
1385 if (LREdit.empty())
1386 return 0;
1387
1388 // We did split for some blocks.
Jakob Stoklund Olesen02cf10b2011-08-05 23:50:31 +00001389 SmallVector<unsigned, 8> IntvMap;
1390 SE->finish(&IntvMap);
Jakob Stoklund Olesen0de95ef2011-08-05 23:10:40 +00001391
1392 // Tell LiveDebugVariables about the new ranges.
Mark Laceyf9ea8852013-08-14 23:50:04 +00001393 DebugVars->splitRegister(Reg, LREdit.regs(), *LIS);
Jakob Stoklund Olesen0de95ef2011-08-05 23:10:40 +00001394
Jakob Stoklund Olesen02cf10b2011-08-05 23:50:31 +00001395 ExtraRegInfo.resize(MRI->getNumVirtRegs());
1396
1397 // Sort out the new intervals created by splitting. The remainder interval
1398 // goes straight to spilling, the new local ranges get to stay RS_New.
1399 for (unsigned i = 0, e = LREdit.size(); i != e; ++i) {
Mark Laceyf9ea8852013-08-14 23:50:04 +00001400 LiveInterval &LI = LIS->getInterval(LREdit.get(i));
Jakob Stoklund Olesen02cf10b2011-08-05 23:50:31 +00001401 if (getStage(LI) == RS_New && IntvMap[i] == 0)
1402 setStage(LI, RS_Spill);
1403 }
1404
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001405 if (VerifyEnabled)
1406 MF->verify(this, "After splitting live range around basic blocks");
1407 return 0;
1408}
1409
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001410
1411//===----------------------------------------------------------------------===//
1412// Per-Instruction Splitting
1413//===----------------------------------------------------------------------===//
1414
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00001415/// Get the number of allocatable registers that match the constraints of \p Reg
1416/// on \p MI and that are also in \p SuperRC.
1417static unsigned getNumAllocatableRegsForConstraints(
1418 const MachineInstr *MI, unsigned Reg, const TargetRegisterClass *SuperRC,
1419 const TargetInstrInfo *TII, const TargetRegisterInfo *TRI,
1420 const RegisterClassInfo &RCI) {
1421 assert(SuperRC && "Invalid register class");
1422
1423 const TargetRegisterClass *ConstrainedRC =
1424 MI->getRegClassConstraintEffectForVReg(Reg, SuperRC, TII, TRI,
1425 /* ExploreBundle */ true);
1426 if (!ConstrainedRC)
1427 return 0;
1428 return RCI.getNumAllocatableRegs(ConstrainedRC);
1429}
1430
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001431/// tryInstructionSplit - Split a live range around individual instructions.
1432/// This is normally not worthwhile since the spiller is doing essentially the
1433/// same thing. However, when the live range is in a constrained register
1434/// class, it may help to insert copies such that parts of the live range can
1435/// be moved to a larger register class.
1436///
1437/// This is similar to spilling to a larger register class.
1438unsigned
1439RAGreedy::tryInstructionSplit(LiveInterval &VirtReg, AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +00001440 SmallVectorImpl<unsigned> &NewVRegs) {
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00001441 const TargetRegisterClass *CurRC = MRI->getRegClass(VirtReg.reg);
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001442 // There is no point to this if there are no larger sub-classes.
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00001443 if (!RegClassInfo.isProperSubClass(CurRC))
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001444 return 0;
1445
1446 // Always enable split spill mode, since we're effectively spilling to a
1447 // register.
1448 LiveRangeEdit LREdit(&VirtReg, NewVRegs, *MF, *LIS, VRM, this);
1449 SE->reset(LREdit, SplitEditor::SM_Size);
1450
1451 ArrayRef<SlotIndex> Uses = SA->getUseSlots();
1452 if (Uses.size() <= 1)
1453 return 0;
1454
1455 DEBUG(dbgs() << "Split around " << Uses.size() << " individual instrs.\n");
1456
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00001457 const TargetRegisterClass *SuperRC = TRI->getLargestLegalSuperClass(CurRC);
1458 unsigned SuperRCNumAllocatableRegs = RCI.getNumAllocatableRegs(SuperRC);
1459 // Split around every non-copy instruction if this split will relax
1460 // the constraints on the virtual register.
1461 // Otherwise, splitting just inserts uncoalescable copies that do not help
1462 // the allocation.
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001463 for (unsigned i = 0; i != Uses.size(); ++i) {
1464 if (const MachineInstr *MI = Indexes->getInstructionFromIndex(Uses[i]))
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00001465 if (MI->isFullCopy() ||
1466 SuperRCNumAllocatableRegs ==
1467 getNumAllocatableRegsForConstraints(MI, VirtReg.reg, SuperRC, TII,
1468 TRI, RCI)) {
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001469 DEBUG(dbgs() << " skip:\t" << Uses[i] << '\t' << *MI);
1470 continue;
1471 }
1472 SE->openIntv();
1473 SlotIndex SegStart = SE->enterIntvBefore(Uses[i]);
1474 SlotIndex SegStop = SE->leaveIntvAfter(Uses[i]);
1475 SE->useIntv(SegStart, SegStop);
1476 }
1477
1478 if (LREdit.empty()) {
1479 DEBUG(dbgs() << "All uses were copies.\n");
1480 return 0;
1481 }
1482
1483 SmallVector<unsigned, 8> IntvMap;
1484 SE->finish(&IntvMap);
Mark Laceyf9ea8852013-08-14 23:50:04 +00001485 DebugVars->splitRegister(VirtReg.reg, LREdit.regs(), *LIS);
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001486 ExtraRegInfo.resize(MRI->getNumVirtRegs());
1487
1488 // Assign all new registers to RS_Spill. This was the last chance.
1489 setStage(LREdit.begin(), LREdit.end(), RS_Spill);
1490 return 0;
1491}
1492
1493
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001494//===----------------------------------------------------------------------===//
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001495// Local Splitting
1496//===----------------------------------------------------------------------===//
1497
1498
1499/// calcGapWeights - Compute the maximum spill weight that needs to be evicted
1500/// in order to use PhysReg between two entries in SA->UseSlots.
1501///
1502/// GapWeight[i] represents the gap between UseSlots[i] and UseSlots[i+1].
1503///
1504void RAGreedy::calcGapWeights(unsigned PhysReg,
1505 SmallVectorImpl<float> &GapWeight) {
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001506 assert(SA->getUseBlocks().size() == 1 && "Not a local interval");
1507 const SplitAnalysis::BlockInfo &BI = SA->getUseBlocks().front();
Jakob Stoklund Olesen994fed62012-01-12 17:53:44 +00001508 ArrayRef<SlotIndex> Uses = SA->getUseSlots();
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001509 const unsigned NumGaps = Uses.size()-1;
1510
1511 // Start and end points for the interference check.
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +00001512 SlotIndex StartIdx =
1513 BI.LiveIn ? BI.FirstInstr.getBaseIndex() : BI.FirstInstr;
1514 SlotIndex StopIdx =
1515 BI.LiveOut ? BI.LastInstr.getBoundaryIndex() : BI.LastInstr;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001516
1517 GapWeight.assign(NumGaps, 0.0f);
1518
1519 // Add interference from each overlapping register.
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001520 for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
1521 if (!Matrix->query(const_cast<LiveInterval&>(SA->getParent()), *Units)
1522 .checkInterference())
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001523 continue;
1524
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +00001525 // We know that VirtReg is a continuous interval from FirstInstr to
1526 // LastInstr, so we don't need InterferenceQuery.
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001527 //
1528 // Interference that overlaps an instruction is counted in both gaps
1529 // surrounding the instruction. The exception is interference before
1530 // StartIdx and after StopIdx.
1531 //
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001532 LiveIntervalUnion::SegmentIter IntI =
1533 Matrix->getLiveUnions()[*Units] .find(StartIdx);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001534 for (unsigned Gap = 0; IntI.valid() && IntI.start() < StopIdx; ++IntI) {
1535 // Skip the gaps before IntI.
1536 while (Uses[Gap+1].getBoundaryIndex() < IntI.start())
1537 if (++Gap == NumGaps)
1538 break;
1539 if (Gap == NumGaps)
1540 break;
1541
1542 // Update the gaps covered by IntI.
1543 const float weight = IntI.value()->weight;
1544 for (; Gap != NumGaps; ++Gap) {
1545 GapWeight[Gap] = std::max(GapWeight[Gap], weight);
1546 if (Uses[Gap+1].getBaseIndex() >= IntI.stop())
1547 break;
1548 }
1549 if (Gap == NumGaps)
1550 break;
1551 }
1552 }
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001553
1554 // Add fixed interference.
1555 for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
Matthias Braun34e1be92013-10-10 21:29:02 +00001556 const LiveRange &LR = LIS->getRegUnit(*Units);
1557 LiveRange::const_iterator I = LR.find(StartIdx);
1558 LiveRange::const_iterator E = LR.end();
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001559
1560 // Same loop as above. Mark any overlapped gaps as HUGE_VALF.
1561 for (unsigned Gap = 0; I != E && I->start < StopIdx; ++I) {
1562 while (Uses[Gap+1].getBoundaryIndex() < I->start)
1563 if (++Gap == NumGaps)
1564 break;
1565 if (Gap == NumGaps)
1566 break;
1567
1568 for (; Gap != NumGaps; ++Gap) {
Aaron Ballman04999042013-11-13 00:15:44 +00001569 GapWeight[Gap] = llvm::huge_valf;
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001570 if (Uses[Gap+1].getBaseIndex() >= I->end)
1571 break;
1572 }
1573 if (Gap == NumGaps)
1574 break;
1575 }
1576 }
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001577}
1578
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001579/// tryLocalSplit - Try to split VirtReg into smaller intervals inside its only
1580/// basic block.
1581///
1582unsigned RAGreedy::tryLocalSplit(LiveInterval &VirtReg, AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +00001583 SmallVectorImpl<unsigned> &NewVRegs) {
Jakob Stoklund Olesenbf91c4e2011-04-06 03:57:00 +00001584 assert(SA->getUseBlocks().size() == 1 && "Not a local interval");
1585 const SplitAnalysis::BlockInfo &BI = SA->getUseBlocks().front();
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001586
1587 // Note that it is possible to have an interval that is live-in or live-out
1588 // while only covering a single block - A phi-def can use undef values from
1589 // predecessors, and the block could be a single-block loop.
1590 // We don't bother doing anything clever about such a case, we simply assume
Jakob Stoklund Olesen43859a62011-08-02 22:54:14 +00001591 // that the interval is continuous from FirstInstr to LastInstr. We should
1592 // make sure that we don't do anything illegal to such an interval, though.
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001593
Jakob Stoklund Olesen994fed62012-01-12 17:53:44 +00001594 ArrayRef<SlotIndex> Uses = SA->getUseSlots();
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001595 if (Uses.size() <= 2)
1596 return 0;
1597 const unsigned NumGaps = Uses.size()-1;
1598
1599 DEBUG({
1600 dbgs() << "tryLocalSplit: ";
1601 for (unsigned i = 0, e = Uses.size(); i != e; ++i)
Jakob Stoklund Olesen994fed62012-01-12 17:53:44 +00001602 dbgs() << ' ' << Uses[i];
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001603 dbgs() << '\n';
1604 });
1605
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001606 // If VirtReg is live across any register mask operands, compute a list of
1607 // gaps with register masks.
1608 SmallVector<unsigned, 8> RegMaskGaps;
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001609 if (Matrix->checkRegMaskInterference(VirtReg)) {
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001610 // Get regmask slots for the whole block.
1611 ArrayRef<SlotIndex> RMS = LIS->getRegMaskSlotsInBlock(BI.MBB->getNumber());
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001612 DEBUG(dbgs() << RMS.size() << " regmasks in block:");
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001613 // Constrain to VirtReg's live range.
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001614 unsigned ri = std::lower_bound(RMS.begin(), RMS.end(),
1615 Uses.front().getRegSlot()) - RMS.begin();
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001616 unsigned re = RMS.size();
1617 for (unsigned i = 0; i != NumGaps && ri != re; ++i) {
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001618 // Look for Uses[i] <= RMS <= Uses[i+1].
1619 assert(!SlotIndex::isEarlierInstr(RMS[ri], Uses[i]));
1620 if (SlotIndex::isEarlierInstr(Uses[i+1], RMS[ri]))
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001621 continue;
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001622 // Skip a regmask on the same instruction as the last use. It doesn't
1623 // overlap the live range.
1624 if (SlotIndex::isSameInstr(Uses[i+1], RMS[ri]) && i+1 == NumGaps)
1625 break;
1626 DEBUG(dbgs() << ' ' << RMS[ri] << ':' << Uses[i] << '-' << Uses[i+1]);
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001627 RegMaskGaps.push_back(i);
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001628 // Advance ri to the next gap. A regmask on one of the uses counts in
1629 // both gaps.
1630 while (ri != re && SlotIndex::isEarlierInstr(RMS[ri], Uses[i+1]))
1631 ++ri;
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001632 }
Jakob Stoklund Olesenb0c0d342012-02-14 23:51:27 +00001633 DEBUG(dbgs() << '\n');
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001634 }
1635
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001636 // Since we allow local split results to be split again, there is a risk of
1637 // creating infinite loops. It is tempting to require that the new live
1638 // ranges have less instructions than the original. That would guarantee
1639 // convergence, but it is too strict. A live range with 3 instructions can be
1640 // split 2+3 (including the COPY), and we want to allow that.
1641 //
1642 // Instead we use these rules:
1643 //
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001644 // 1. Allow any split for ranges with getStage() < RS_Split2. (Except for the
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001645 // noop split, of course).
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001646 // 2. Require progress be made for ranges with getStage() == RS_Split2. All
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001647 // the new ranges must have fewer instructions than before the split.
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001648 // 3. New ranges with the same number of instructions are marked RS_Split2,
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001649 // smaller ranges are marked RS_New.
1650 //
1651 // These rules allow a 3 -> 2+3 split once, which we need. They also prevent
1652 // excessive splitting and infinite loops.
1653 //
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001654 bool ProgressRequired = getStage(VirtReg) >= RS_Split2;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001655
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001656 // Best split candidate.
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001657 unsigned BestBefore = NumGaps;
1658 unsigned BestAfter = 0;
1659 float BestDiff = 0;
1660
Jakob Stoklund Olesenefeb3a12013-07-16 18:26:18 +00001661 const float blockFreq =
1662 SpillPlacer->getBlockFrequency(BI.MBB->getNumber()).getFrequency() *
Michael Gottesman5e985ee2013-12-14 02:37:38 +00001663 (1.0f / MBFI->getEntryFreq());
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001664 SmallVector<float, 8> GapWeight;
1665
1666 Order.rewind();
1667 while (unsigned PhysReg = Order.next()) {
1668 // Keep track of the largest spill weight that would need to be evicted in
1669 // order to make use of PhysReg between UseSlots[i] and UseSlots[i+1].
1670 calcGapWeights(PhysReg, GapWeight);
1671
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001672 // Remove any gaps with regmask clobbers.
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001673 if (Matrix->checkRegMaskInterference(VirtReg, PhysReg))
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001674 for (unsigned i = 0, e = RegMaskGaps.size(); i != e; ++i)
Aaron Ballman04999042013-11-13 00:15:44 +00001675 GapWeight[RegMaskGaps[i]] = llvm::huge_valf;
Jakob Stoklund Olesen17402e32012-02-11 00:42:18 +00001676
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001677 // Try to find the best sequence of gaps to close.
1678 // The new spill weight must be larger than any gap interference.
1679
1680 // We will split before Uses[SplitBefore] and after Uses[SplitAfter].
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001681 unsigned SplitBefore = 0, SplitAfter = 1;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001682
1683 // MaxGap should always be max(GapWeight[SplitBefore..SplitAfter-1]).
1684 // It is the spill weight that needs to be evicted.
1685 float MaxGap = GapWeight[0];
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001686
1687 for (;;) {
1688 // Live before/after split?
1689 const bool LiveBefore = SplitBefore != 0 || BI.LiveIn;
1690 const bool LiveAfter = SplitAfter != NumGaps || BI.LiveOut;
1691
1692 DEBUG(dbgs() << PrintReg(PhysReg, TRI) << ' '
1693 << Uses[SplitBefore] << '-' << Uses[SplitAfter]
1694 << " i=" << MaxGap);
1695
1696 // Stop before the interval gets so big we wouldn't be making progress.
1697 if (!LiveBefore && !LiveAfter) {
1698 DEBUG(dbgs() << " all\n");
1699 break;
1700 }
1701 // Should the interval be extended or shrunk?
1702 bool Shrink = true;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001703
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001704 // How many gaps would the new range have?
1705 unsigned NewGaps = LiveBefore + SplitAfter - SplitBefore + LiveAfter;
1706
1707 // Legally, without causing looping?
1708 bool Legal = !ProgressRequired || NewGaps < NumGaps;
1709
Aaron Ballman04999042013-11-13 00:15:44 +00001710 if (Legal && MaxGap < llvm::huge_valf) {
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001711 // Estimate the new spill weight. Each instruction reads or writes the
1712 // register. Conservatively assume there are no read-modify-write
1713 // instructions.
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001714 //
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001715 // Try to guess the size of the new interval.
1716 const float EstWeight = normalizeSpillWeight(blockFreq * (NewGaps + 1),
1717 Uses[SplitBefore].distance(Uses[SplitAfter]) +
1718 (LiveBefore + LiveAfter)*SlotIndex::InstrDist);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001719 // Would this split be possible to allocate?
1720 // Never allocate all gaps, we wouldn't be making progress.
Jakob Stoklund Olesen357dd362011-04-30 05:07:46 +00001721 DEBUG(dbgs() << " w=" << EstWeight);
1722 if (EstWeight * Hysteresis >= MaxGap) {
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001723 Shrink = false;
Jakob Stoklund Olesen357dd362011-04-30 05:07:46 +00001724 float Diff = EstWeight - MaxGap;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001725 if (Diff > BestDiff) {
1726 DEBUG(dbgs() << " (best)");
Jakob Stoklund Olesen357dd362011-04-30 05:07:46 +00001727 BestDiff = Hysteresis * Diff;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001728 BestBefore = SplitBefore;
1729 BestAfter = SplitAfter;
1730 }
1731 }
1732 }
1733
1734 // Try to shrink.
1735 if (Shrink) {
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001736 if (++SplitBefore < SplitAfter) {
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001737 DEBUG(dbgs() << " shrink\n");
1738 // Recompute the max when necessary.
1739 if (GapWeight[SplitBefore - 1] >= MaxGap) {
1740 MaxGap = GapWeight[SplitBefore];
1741 for (unsigned i = SplitBefore + 1; i != SplitAfter; ++i)
1742 MaxGap = std::max(MaxGap, GapWeight[i]);
1743 }
1744 continue;
1745 }
1746 MaxGap = 0;
1747 }
1748
1749 // Try to extend the interval.
1750 if (SplitAfter >= NumGaps) {
1751 DEBUG(dbgs() << " end\n");
1752 break;
1753 }
1754
1755 DEBUG(dbgs() << " extend\n");
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001756 MaxGap = std::max(MaxGap, GapWeight[SplitAfter++]);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001757 }
1758 }
1759
1760 // Didn't find any candidates?
1761 if (BestBefore == NumGaps)
1762 return 0;
1763
1764 DEBUG(dbgs() << "Best local split range: " << Uses[BestBefore]
1765 << '-' << Uses[BestAfter] << ", " << BestDiff
1766 << ", " << (BestAfter - BestBefore + 1) << " instrs\n");
1767
Jakob Stoklund Olesene5bbe372012-05-19 05:25:46 +00001768 LiveRangeEdit LREdit(&VirtReg, NewVRegs, *MF, *LIS, VRM, this);
Jakob Stoklund Olesenc9601982011-03-03 01:29:13 +00001769 SE->reset(LREdit);
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001770
Jakob Stoklund Olesenc9601982011-03-03 01:29:13 +00001771 SE->openIntv();
1772 SlotIndex SegStart = SE->enterIntvBefore(Uses[BestBefore]);
1773 SlotIndex SegStop = SE->leaveIntvAfter(Uses[BestAfter]);
1774 SE->useIntv(SegStart, SegStop);
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001775 SmallVector<unsigned, 8> IntvMap;
1776 SE->finish(&IntvMap);
Mark Laceyf9ea8852013-08-14 23:50:04 +00001777 DebugVars->splitRegister(VirtReg.reg, LREdit.regs(), *LIS);
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001778
1779 // If the new range has the same number of instructions as before, mark it as
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001780 // RS_Split2 so the next split will be forced to make progress. Otherwise,
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001781 // leave the new intervals as RS_New so they can compete.
1782 bool LiveBefore = BestBefore != 0 || BI.LiveIn;
1783 bool LiveAfter = BestAfter != NumGaps || BI.LiveOut;
1784 unsigned NewGaps = LiveBefore + BestAfter - BestBefore + LiveAfter;
1785 if (NewGaps >= NumGaps) {
1786 DEBUG(dbgs() << "Tagging non-progress ranges: ");
1787 assert(!ProgressRequired && "Didn't make progress when it was required.");
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001788 for (unsigned i = 0, e = IntvMap.size(); i != e; ++i)
1789 if (IntvMap[i] == 1) {
Mark Laceyf9ea8852013-08-14 23:50:04 +00001790 setStage(LIS->getInterval(LREdit.get(i)), RS_Split2);
1791 DEBUG(dbgs() << PrintReg(LREdit.get(i)));
Jakob Stoklund Olesendf476272011-06-06 23:55:20 +00001792 }
1793 DEBUG(dbgs() << '\n');
1794 }
Jakob Stoklund Olesen99827e82011-02-17 22:53:48 +00001795 ++NumLocalSplits;
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001796
1797 return 0;
1798}
1799
1800//===----------------------------------------------------------------------===//
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001801// Live Range Splitting
1802//===----------------------------------------------------------------------===//
1803
1804/// trySplit - Try to split VirtReg or one of its interferences, making it
1805/// assignable.
1806/// @return Physreg when VirtReg may be assigned and/or new NewVRegs.
1807unsigned RAGreedy::trySplit(LiveInterval &VirtReg, AllocationOrder &Order,
Mark Laceyf9ea8852013-08-14 23:50:04 +00001808 SmallVectorImpl<unsigned>&NewVRegs) {
Jakob Stoklund Olesend4bb1d42011-08-05 23:50:33 +00001809 // Ranges must be Split2 or less.
1810 if (getStage(VirtReg) >= RS_Spill)
1811 return 0;
1812
Jakob Stoklund Olesen93c87362011-02-17 19:13:53 +00001813 // Local intervals are handled separately.
Jakob Stoklund Olesen609bc442011-02-19 00:38:40 +00001814 if (LIS->intervalIsInOneMBB(VirtReg)) {
1815 NamedRegionTimer T("Local Splitting", TimerGroupName, TimePassesIsEnabled);
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +00001816 SA->analyze(&VirtReg);
Jakob Stoklund Olesen0ce90492012-05-23 22:37:27 +00001817 unsigned PhysReg = tryLocalSplit(VirtReg, Order, NewVRegs);
1818 if (PhysReg || !NewVRegs.empty())
1819 return PhysReg;
1820 return tryInstructionSplit(VirtReg, Order, NewVRegs);
Jakob Stoklund Olesen609bc442011-02-19 00:38:40 +00001821 }
1822
1823 NamedRegionTimer T("Global Splitting", TimerGroupName, TimePassesIsEnabled);
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001824
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +00001825 SA->analyze(&VirtReg);
1826
Jakob Stoklund Oleseneaa6ed12011-05-03 20:42:13 +00001827 // FIXME: SplitAnalysis may repair broken live ranges coming from the
1828 // coalescer. That may cause the range to become allocatable which means that
1829 // tryRegionSplit won't be making progress. This check should be replaced with
1830 // an assertion when the coalescer is fixed.
1831 if (SA->didRepairRange()) {
1832 // VirtReg has changed, so all cached queries are invalid.
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00001833 Matrix->invalidateVirtRegs();
Jakob Stoklund Oleseneaa6ed12011-05-03 20:42:13 +00001834 if (unsigned PhysReg = tryAssign(VirtReg, Order, NewVRegs))
1835 return PhysReg;
1836 }
1837
Jakob Stoklund Olesen45011172011-07-25 15:25:43 +00001838 // First try to split around a region spanning multiple blocks. RS_Split2
1839 // ranges already made dubious progress with region splitting, so they go
1840 // straight to single block splitting.
1841 if (getStage(VirtReg) < RS_Split2) {
1842 unsigned PhysReg = tryRegionSplit(VirtReg, Order, NewVRegs);
1843 if (PhysReg || !NewVRegs.empty())
1844 return PhysReg;
1845 }
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001846
Jakob Stoklund Olesencef5d8f2011-08-05 23:04:18 +00001847 // Then isolate blocks.
1848 return tryBlockSplit(VirtReg, Order, NewVRegs);
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00001849}
1850
Quentin Colombet87769712014-02-05 22:13:59 +00001851//===----------------------------------------------------------------------===//
1852// Last Chance Recoloring
1853//===----------------------------------------------------------------------===//
1854
1855/// mayRecolorAllInterferences - Check if the virtual registers that
1856/// interfere with \p VirtReg on \p PhysReg (or one of its aliases) may be
1857/// recolored to free \p PhysReg.
1858/// When true is returned, \p RecoloringCandidates has been augmented with all
1859/// the live intervals that need to be recolored in order to free \p PhysReg
1860/// for \p VirtReg.
1861/// \p FixedRegisters contains all the virtual registers that cannot be
1862/// recolored.
1863bool
1864RAGreedy::mayRecolorAllInterferences(unsigned PhysReg, LiveInterval &VirtReg,
1865 SmallLISet &RecoloringCandidates,
1866 const SmallVirtRegSet &FixedRegisters) {
1867 const TargetRegisterClass *CurRC = MRI->getRegClass(VirtReg.reg);
1868
1869 for (MCRegUnitIterator Units(PhysReg, TRI); Units.isValid(); ++Units) {
1870 LiveIntervalUnion::Query &Q = Matrix->query(VirtReg, *Units);
1871 // If there is LastChanceRecoloringMaxInterference or more interferences,
1872 // chances are one would not be recolorable.
1873 if (Q.collectInterferingVRegs(LastChanceRecoloringMaxInterference) >=
1874 LastChanceRecoloringMaxInterference) {
1875 DEBUG(dbgs() << "Early abort: too many interferences.\n");
1876 return false;
1877 }
1878 for (unsigned i = Q.interferingVRegs().size(); i; --i) {
1879 LiveInterval *Intf = Q.interferingVRegs()[i - 1];
1880 // If Intf is done and sit on the same register class as VirtReg,
1881 // it would not be recolorable as it is in the same state as VirtReg.
1882 if ((getStage(*Intf) == RS_Done &&
1883 MRI->getRegClass(Intf->reg) == CurRC) ||
1884 FixedRegisters.count(Intf->reg)) {
1885 DEBUG(dbgs() << "Early abort: the inteference is not recolorable.\n");
1886 return false;
1887 }
1888 RecoloringCandidates.insert(Intf);
1889 }
1890 }
1891 return true;
1892}
1893
1894/// tryLastChanceRecoloring - Try to assign a color to \p VirtReg by recoloring
1895/// its interferences.
1896/// Last chance recoloring chooses a color for \p VirtReg and recolors every
1897/// virtual register that was using it. The recoloring process may recursively
1898/// use the last chance recoloring. Therefore, when a virtual register has been
1899/// assigned a color by this mechanism, it is marked as Fixed, i.e., it cannot
1900/// be last-chance-recolored again during this recoloring "session".
1901/// E.g.,
1902/// Let
1903/// vA can use {R1, R2 }
1904/// vB can use { R2, R3}
1905/// vC can use {R1 }
1906/// Where vA, vB, and vC cannot be split anymore (they are reloads for
1907/// instance) and they all interfere.
1908///
1909/// vA is assigned R1
1910/// vB is assigned R2
1911/// vC tries to evict vA but vA is already done.
1912/// Regular register allocation fails.
1913///
1914/// Last chance recoloring kicks in:
1915/// vC does as if vA was evicted => vC uses R1.
1916/// vC is marked as fixed.
1917/// vA needs to find a color.
1918/// None are available.
1919/// vA cannot evict vC: vC is a fixed virtual register now.
1920/// vA does as if vB was evicted => vA uses R2.
1921/// vB needs to find a color.
1922/// R3 is available.
1923/// Recoloring => vC = R1, vA = R2, vB = R3
1924///
Alp Toker70b36992014-02-25 04:21:15 +00001925/// \p Order defines the preferred allocation order for \p VirtReg.
Quentin Colombet87769712014-02-05 22:13:59 +00001926/// \p NewRegs will contain any new virtual register that have been created
1927/// (split, spill) during the process and that must be assigned.
1928/// \p FixedRegisters contains all the virtual registers that cannot be
1929/// recolored.
1930/// \p Depth gives the current depth of the last chance recoloring.
1931/// \return a physical register that can be used for VirtReg or ~0u if none
1932/// exists.
1933unsigned RAGreedy::tryLastChanceRecoloring(LiveInterval &VirtReg,
1934 AllocationOrder &Order,
1935 SmallVectorImpl<unsigned> &NewVRegs,
1936 SmallVirtRegSet &FixedRegisters,
1937 unsigned Depth) {
1938 DEBUG(dbgs() << "Try last chance recoloring for " << VirtReg << '\n');
1939 // Ranges must be Done.
Quentin Colombet0e3b5e02014-02-13 05:17:37 +00001940 assert((getStage(VirtReg) >= RS_Done || !VirtReg.isSpillable()) &&
Quentin Colombet87769712014-02-05 22:13:59 +00001941 "Last chance recoloring should really be last chance");
1942 // Set the max depth to LastChanceRecoloringMaxDepth.
1943 // We may want to reconsider that if we end up with a too large search space
1944 // for target with hundreds of registers.
1945 // Indeed, in that case we may want to cut the search space earlier.
1946 if (Depth >= LastChanceRecoloringMaxDepth) {
1947 DEBUG(dbgs() << "Abort because max depth has been reached.\n");
1948 return ~0u;
1949 }
1950
1951 // Set of Live intervals that will need to be recolored.
1952 SmallLISet RecoloringCandidates;
1953 // Record the original mapping virtual register to physical register in case
1954 // the recoloring fails.
1955 DenseMap<unsigned, unsigned> VirtRegToPhysReg;
1956 // Mark VirtReg as fixed, i.e., it will not be recolored pass this point in
1957 // this recoloring "session".
1958 FixedRegisters.insert(VirtReg.reg);
1959
1960 Order.rewind();
1961 while (unsigned PhysReg = Order.next()) {
1962 DEBUG(dbgs() << "Try to assign: " << VirtReg << " to "
1963 << PrintReg(PhysReg, TRI) << '\n');
1964 RecoloringCandidates.clear();
1965 VirtRegToPhysReg.clear();
1966
1967 // It is only possible to recolor virtual register interference.
1968 if (Matrix->checkInterference(VirtReg, PhysReg) >
1969 LiveRegMatrix::IK_VirtReg) {
1970 DEBUG(dbgs() << "Some inteferences are not with virtual registers.\n");
1971
1972 continue;
1973 }
1974
1975 // Early give up on this PhysReg if it is obvious we cannot recolor all
1976 // the interferences.
1977 if (!mayRecolorAllInterferences(PhysReg, VirtReg, RecoloringCandidates,
1978 FixedRegisters)) {
1979 DEBUG(dbgs() << "Some inteferences cannot be recolored.\n");
1980 continue;
1981 }
1982
1983 // RecoloringCandidates contains all the virtual registers that interfer
1984 // with VirtReg on PhysReg (or one of its aliases).
1985 // Enqueue them for recoloring and perform the actual recoloring.
1986 PQueue RecoloringQueue;
1987 for (SmallLISet::iterator It = RecoloringCandidates.begin(),
1988 EndIt = RecoloringCandidates.end();
1989 It != EndIt; ++It) {
1990 unsigned ItVirtReg = (*It)->reg;
1991 enqueue(RecoloringQueue, *It);
1992 assert(VRM->hasPhys(ItVirtReg) &&
1993 "Interferences are supposed to be with allocated vairables");
1994
1995 // Record the current allocation.
1996 VirtRegToPhysReg[ItVirtReg] = VRM->getPhys(ItVirtReg);
1997 // unset the related struct.
1998 Matrix->unassign(**It);
1999 }
2000
2001 // Do as if VirtReg was assigned to PhysReg so that the underlying
2002 // recoloring has the right information about the interferes and
2003 // available colors.
2004 Matrix->assign(VirtReg, PhysReg);
2005
2006 // Save the current recoloring state.
2007 // If we cannot recolor all the interferences, we will have to start again
2008 // at this point for the next physical register.
2009 SmallVirtRegSet SaveFixedRegisters(FixedRegisters);
2010 if (tryRecoloringCandidates(RecoloringQueue, NewVRegs, FixedRegisters,
2011 Depth)) {
2012 // Do not mess up with the global assignment process.
2013 // I.e., VirtReg must be unassigned.
2014 Matrix->unassign(VirtReg);
2015 return PhysReg;
2016 }
2017
2018 DEBUG(dbgs() << "Fail to assign: " << VirtReg << " to "
2019 << PrintReg(PhysReg, TRI) << '\n');
2020
2021 // The recoloring attempt failed, undo the changes.
2022 FixedRegisters = SaveFixedRegisters;
2023 Matrix->unassign(VirtReg);
2024
2025 for (SmallLISet::iterator It = RecoloringCandidates.begin(),
2026 EndIt = RecoloringCandidates.end();
2027 It != EndIt; ++It) {
2028 unsigned ItVirtReg = (*It)->reg;
2029 if (VRM->hasPhys(ItVirtReg))
2030 Matrix->unassign(**It);
2031 Matrix->assign(**It, VirtRegToPhysReg[ItVirtReg]);
2032 }
2033 }
2034
2035 // Last chance recoloring did not worked either, give up.
2036 return ~0u;
2037}
2038
2039/// tryRecoloringCandidates - Try to assign a new color to every register
2040/// in \RecoloringQueue.
2041/// \p NewRegs will contain any new virtual register created during the
2042/// recoloring process.
2043/// \p FixedRegisters[in/out] contains all the registers that have been
2044/// recolored.
2045/// \return true if all virtual registers in RecoloringQueue were successfully
2046/// recolored, false otherwise.
2047bool RAGreedy::tryRecoloringCandidates(PQueue &RecoloringQueue,
2048 SmallVectorImpl<unsigned> &NewVRegs,
2049 SmallVirtRegSet &FixedRegisters,
2050 unsigned Depth) {
2051 while (!RecoloringQueue.empty()) {
2052 LiveInterval *LI = dequeue(RecoloringQueue);
2053 DEBUG(dbgs() << "Try to recolor: " << *LI << '\n');
2054 unsigned PhysReg;
2055 PhysReg = selectOrSplitImpl(*LI, NewVRegs, FixedRegisters, Depth + 1);
2056 if (PhysReg == ~0u || !PhysReg)
2057 return false;
2058 DEBUG(dbgs() << "Recoloring of " << *LI
2059 << " succeeded with: " << PrintReg(PhysReg, TRI) << '\n');
2060 Matrix->assign(*LI, PhysReg);
2061 FixedRegisters.insert(LI->reg);
2062 }
2063 return true;
2064}
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00002065
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00002066//===----------------------------------------------------------------------===//
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +00002067// Main Entry Point
2068//===----------------------------------------------------------------------===//
2069
2070unsigned RAGreedy::selectOrSplit(LiveInterval &VirtReg,
Mark Laceyf9ea8852013-08-14 23:50:04 +00002071 SmallVectorImpl<unsigned> &NewVRegs) {
Quentin Colombet87769712014-02-05 22:13:59 +00002072 SmallVirtRegSet FixedRegisters;
2073 return selectOrSplitImpl(VirtReg, NewVRegs, FixedRegisters);
2074}
2075
2076unsigned RAGreedy::selectOrSplitImpl(LiveInterval &VirtReg,
2077 SmallVectorImpl<unsigned> &NewVRegs,
2078 SmallVirtRegSet &FixedRegisters,
2079 unsigned Depth) {
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +00002080 // First try assigning a free register.
Jakob Stoklund Olesenb8bf3c02011-06-03 20:34:53 +00002081 AllocationOrder Order(VirtReg.reg, *VRM, RegClassInfo);
Jakob Stoklund Olesen0e34c1d2011-04-20 18:19:48 +00002082 if (unsigned PhysReg = tryAssign(VirtReg, Order, NewVRegs))
2083 return PhysReg;
Andrew Trickccef0982010-12-09 18:15:21 +00002084
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +00002085 LiveRangeStage Stage = getStage(VirtReg);
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +00002086 DEBUG(dbgs() << StageName[Stage]
2087 << " Cascade " << ExtraRegInfo[VirtReg.reg].Cascade << '\n');
Jakob Stoklund Olesen25d57452011-05-25 23:58:36 +00002088
Jakob Stoklund Olesene9cc8e92011-06-01 18:45:02 +00002089 // Try to evict a less worthy live range, but only for ranges from the primary
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00002090 // queue. The RS_Split ranges already failed to do this, and they should not
Jakob Stoklund Olesene9cc8e92011-06-01 18:45:02 +00002091 // get a second chance until they have been split.
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00002092 if (Stage != RS_Split)
Jakob Stoklund Olesene9cc8e92011-06-01 18:45:02 +00002093 if (unsigned PhysReg = tryEvict(VirtReg, Order, NewVRegs))
2094 return PhysReg;
Andrew Trickccef0982010-12-09 18:15:21 +00002095
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00002096 assert(NewVRegs.empty() && "Cannot append to existing NewVRegs");
2097
Jakob Stoklund Olesene68a27e2011-02-24 23:21:36 +00002098 // The first time we see a live range, don't try to split or spill.
2099 // Wait until the second time, when all smaller ranges have been allocated.
2100 // This gives a better picture of the interference to split around.
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00002101 if (Stage < RS_Split) {
2102 setStage(VirtReg, RS_Split);
Jakob Stoklund Olesen86985072011-03-19 23:02:47 +00002103 DEBUG(dbgs() << "wait for second round\n");
Mark Laceyf9ea8852013-08-14 23:50:04 +00002104 NewVRegs.push_back(VirtReg.reg);
Jakob Stoklund Olesene68a27e2011-02-24 23:21:36 +00002105 return 0;
2106 }
2107
Jakob Stoklund Olesena5c88992011-05-06 21:58:30 +00002108 // If we couldn't allocate a register from spilling, there is probably some
2109 // invalid inline assembly. The base class wil report it.
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00002110 if (Stage >= RS_Done || !VirtReg.isSpillable())
Quentin Colombet87769712014-02-05 22:13:59 +00002111 return tryLastChanceRecoloring(VirtReg, Order, NewVRegs, FixedRegisters,
2112 Depth);
Jakob Stoklund Olesen5f9f0812011-03-01 21:10:07 +00002113
Jakob Stoklund Olesen903b6d32010-12-14 00:37:49 +00002114 // Try splitting VirtReg or interferences.
Jakob Stoklund Olesen9fb04012011-01-19 22:11:48 +00002115 unsigned PhysReg = trySplit(VirtReg, Order, NewVRegs);
2116 if (PhysReg || !NewVRegs.empty())
Jakob Stoklund Olesen3d7b8062010-12-14 00:37:44 +00002117 return PhysReg;
2118
Jakob Stoklund Olesen0acb69d2010-12-22 22:01:30 +00002119 // Finally spill VirtReg itself.
Jakob Stoklund Olesen92da7052010-12-11 00:19:56 +00002120 NamedRegionTimer T("Spiller", TimerGroupName, TimePassesIsEnabled);
Jakob Stoklund Olesene5bbe372012-05-19 05:25:46 +00002121 LiveRangeEdit LRE(&VirtReg, NewVRegs, *MF, *LIS, VRM, this);
Jakob Stoklund Olesen4d6eafa2011-03-10 01:51:42 +00002122 spiller().spill(LRE);
Jakob Stoklund Olesen3ef8cf12011-07-25 15:25:41 +00002123 setStage(NewVRegs.begin(), NewVRegs.end(), RS_Done);
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002124
Jakob Stoklund Olesen557a82c2011-03-16 22:56:08 +00002125 if (VerifyEnabled)
2126 MF->verify(this, "After spilling");
2127
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002128 // The live virtual register requesting allocation was spilled, so tell
2129 // the caller not to allocate anything during this round.
2130 return 0;
2131}
2132
2133bool RAGreedy::runOnMachineFunction(MachineFunction &mf) {
2134 DEBUG(dbgs() << "********** GREEDY REGISTER ALLOCATION **********\n"
David Blaikiec8c29202012-08-22 17:18:53 +00002135 << "********** Function: " << mf.getName() << '\n');
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002136
2137 MF = &mf;
Quentin Colombet1fb3362a2014-01-02 22:47:22 +00002138 TRI = MF->getTarget().getRegisterInfo();
2139 TII = MF->getTarget().getInstrInfo();
2140 RCI.runOnMachineFunction(mf);
Jakob Stoklund Olesen2e98ee32010-12-17 23:16:35 +00002141 if (VerifyEnabled)
Jakob Stoklund Olesenbf4550e2010-12-18 00:06:56 +00002142 MF->verify(this, "Before greedy register allocator");
Jakob Stoklund Olesen2e98ee32010-12-17 23:16:35 +00002143
Jakob Stoklund Olesen2d2dec92012-06-20 22:52:29 +00002144 RegAllocBase::init(getAnalysis<VirtRegMap>(),
2145 getAnalysis<LiveIntervals>(),
2146 getAnalysis<LiveRegMatrix>());
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00002147 Indexes = &getAnalysis<SlotIndexes>();
Benjamin Kramere2a1d892013-06-17 19:00:36 +00002148 MBFI = &getAnalysis<MachineBlockFrequencyInfo>();
Jakob Stoklund Olesen1740e002010-12-17 23:16:32 +00002149 DomTree = &getAnalysis<MachineDominatorTree>();
Jakob Stoklund Olesenadecb5e2010-12-10 22:54:44 +00002150 SpillerInstance.reset(createInlineSpiller(*this, *MF, *VRM));
Jakob Stoklund Olesene7601e92010-12-15 23:46:13 +00002151 Loops = &getAnalysis<MachineLoopInfo>();
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00002152 Bundles = &getAnalysis<EdgeBundles>();
2153 SpillPlacer = &getAnalysis<SpillPlacement>();
Jakob Stoklund Olesenf8da0282011-05-06 18:00:02 +00002154 DebugVars = &getAnalysis<LiveDebugVariables>();
Jakob Stoklund Olesen267f6c12011-01-18 21:13:27 +00002155
Arnaud A. de Grandmaisonea3ac162013-11-11 19:04:45 +00002156 calculateSpillWeightsAndHints(*LIS, mf, *Loops, *MBFI);
Arnaud A. de Grandmaison760c1e02013-11-10 17:46:31 +00002157
Andrew Trick97064962013-07-25 07:26:26 +00002158 DEBUG(LIS->dump());
2159
Jakob Stoklund Olesenf1a60a62011-02-19 00:53:42 +00002160 SA.reset(new SplitAnalysis(*VRM, *LIS, *Loops));
Benjamin Kramere2a1d892013-06-17 19:00:36 +00002161 SE.reset(new SplitEditor(*SA, *LIS, *VRM, *DomTree, *MBFI));
Jakob Stoklund Olesen30a85632011-07-02 01:37:09 +00002162 ExtraRegInfo.clear();
2163 ExtraRegInfo.resize(MRI->getNumVirtRegs());
2164 NextCascade = 1;
Jakob Stoklund Olesen96eebf02012-06-20 22:52:26 +00002165 IntfCache.init(MF, Matrix->getLiveUnions(), Indexes, LIS, TRI);
Jakob Stoklund Olesendab4b9a2011-07-26 23:41:46 +00002166 GlobalCand.resize(32); // This will grow as needed.
Jakob Stoklund Olesene7601e92010-12-15 23:46:13 +00002167
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002168 allocatePhysRegs();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002169 releaseMemory();
Jakob Stoklund Olesenb8812a12010-12-08 03:26:16 +00002170 return true;
2171}