blob: 335ceae2dbff3a347bf8de0acab425a5af24cad4 [file] [log] [blame]
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +00001//===-- llvm/CodeGen/VirtRegMap.cpp - Virtual Register Map ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +00007//
8//===----------------------------------------------------------------------===//
9//
Chris Lattner8c4d88d2004-09-30 01:54:45 +000010// This file implements the VirtRegMap class.
11//
12// It also contains implementations of the the Spiller interface, which, given a
13// virtual register map and a machine function, eliminates all virtual
14// references by replacing them with physical register references - adding spill
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +000015// code as necessary.
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000016//
17//===----------------------------------------------------------------------===//
18
Chris Lattner8c4d88d2004-09-30 01:54:45 +000019#define DEBUG_TYPE "spiller"
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000020#include "VirtRegMap.h"
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +000021#include "llvm/Function.h"
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000022#include "llvm/CodeGen/MachineFrameInfo.h"
Chris Lattner8c4d88d2004-09-30 01:54:45 +000023#include "llvm/CodeGen/MachineFunction.h"
Evan Cheng4cce6b42008-04-11 17:53:36 +000024#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000025#include "llvm/CodeGen/MachineRegisterInfo.h"
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000026#include "llvm/Target/TargetMachine.h"
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +000027#include "llvm/Target/TargetInstrInfo.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000028#include "llvm/Support/CommandLine.h"
29#include "llvm/Support/Debug.h"
Chris Lattnera4f0b3a2006-08-27 12:54:02 +000030#include "llvm/Support/Compiler.h"
Evan Cheng957840b2007-02-21 02:22:03 +000031#include "llvm/ADT/BitVector.h"
Evan Chengcb742662008-06-04 09:16:33 +000032#include "llvm/ADT/DenseMap.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000033#include "llvm/ADT/Statistic.h"
34#include "llvm/ADT/STLExtras.h"
Chris Lattner08a4d5a2007-01-23 00:59:48 +000035#include "llvm/ADT/SmallSet.h"
Chris Lattner27f29162004-10-26 15:35:58 +000036#include <algorithm>
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000037using namespace llvm;
38
Chris Lattnercd3245a2006-12-19 22:41:21 +000039STATISTIC(NumSpills, "Number of register spills");
Evan Chengc1f53c72008-03-11 21:34:46 +000040STATISTIC(NumPSpills,"Number of physical register spills");
Evan Cheng2638e1a2007-03-20 08:13:50 +000041STATISTIC(NumReMats, "Number of re-materialization");
Evan Chengb6ca4b32007-08-14 23:25:37 +000042STATISTIC(NumDRM , "Number of re-materializable defs elided");
Chris Lattnercd3245a2006-12-19 22:41:21 +000043STATISTIC(NumStores, "Number of stores added");
44STATISTIC(NumLoads , "Number of loads added");
45STATISTIC(NumReused, "Number of values reused");
46STATISTIC(NumDSE , "Number of dead stores elided");
47STATISTIC(NumDCE , "Number of copies elided");
Evan Chengd3653122008-02-27 03:04:06 +000048STATISTIC(NumDSS , "Number of dead spill slots removed");
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +000049
Chris Lattnercd3245a2006-12-19 22:41:21 +000050namespace {
Chris Lattner8c4d88d2004-09-30 01:54:45 +000051 enum SpillerName { simple, local };
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000052}
53
Dan Gohman844731a2008-05-13 00:00:25 +000054static cl::opt<SpillerName>
55SpillerOpt("spiller",
56 cl::desc("Spiller to use: (default: local)"),
57 cl::Prefix,
58 cl::values(clEnumVal(simple, " simple spiller"),
59 clEnumVal(local, " local spiller"),
60 clEnumValEnd),
61 cl::init(local));
62
Chris Lattner8c4d88d2004-09-30 01:54:45 +000063//===----------------------------------------------------------------------===//
64// VirtRegMap implementation
65//===----------------------------------------------------------------------===//
66
Chris Lattner29268692006-09-05 02:12:02 +000067VirtRegMap::VirtRegMap(MachineFunction &mf)
68 : TII(*mf.getTarget().getInstrInfo()), MF(mf),
Evan Cheng2638e1a2007-03-20 08:13:50 +000069 Virt2PhysMap(NO_PHYS_REG), Virt2StackSlotMap(NO_STACK_SLOT),
Evan Cheng81a03822007-11-17 00:40:40 +000070 Virt2ReMatIdMap(NO_STACK_SLOT), Virt2SplitMap(0),
Evan Chengd3653122008-02-27 03:04:06 +000071 Virt2SplitKillMap(0), ReMatMap(NULL), ReMatId(MAX_STACK_SLOT+1),
72 LowSpillSlot(NO_STACK_SLOT), HighSpillSlot(NO_STACK_SLOT) {
73 SpillSlotToUsesMap.resize(8);
Evan Cheng4cce6b42008-04-11 17:53:36 +000074 ImplicitDefed.resize(MF.getRegInfo().getLastVirtReg()+1-
75 TargetRegisterInfo::FirstVirtualRegister);
Chris Lattner29268692006-09-05 02:12:02 +000076 grow();
77}
78
Chris Lattner8c4d88d2004-09-30 01:54:45 +000079void VirtRegMap::grow() {
Chris Lattner84bc5422007-12-31 04:13:23 +000080 unsigned LastVirtReg = MF.getRegInfo().getLastVirtReg();
Evan Cheng549f27d32007-08-13 23:45:17 +000081 Virt2PhysMap.grow(LastVirtReg);
82 Virt2StackSlotMap.grow(LastVirtReg);
83 Virt2ReMatIdMap.grow(LastVirtReg);
Evan Cheng81a03822007-11-17 00:40:40 +000084 Virt2SplitMap.grow(LastVirtReg);
Evan Chengadf85902007-12-05 09:51:10 +000085 Virt2SplitKillMap.grow(LastVirtReg);
Evan Cheng549f27d32007-08-13 23:45:17 +000086 ReMatMap.grow(LastVirtReg);
Evan Cheng4cce6b42008-04-11 17:53:36 +000087 ImplicitDefed.resize(LastVirtReg-TargetRegisterInfo::FirstVirtualRegister+1);
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +000088}
89
Chris Lattner8c4d88d2004-09-30 01:54:45 +000090int VirtRegMap::assignVirt2StackSlot(unsigned virtReg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +000091 assert(TargetRegisterInfo::isVirtualRegister(virtReg));
Chris Lattner7f690e62004-09-30 02:15:18 +000092 assert(Virt2StackSlotMap[virtReg] == NO_STACK_SLOT &&
Chris Lattner8c4d88d2004-09-30 01:54:45 +000093 "attempt to assign stack slot to already spilled register");
Chris Lattner84bc5422007-12-31 04:13:23 +000094 const TargetRegisterClass* RC = MF.getRegInfo().getRegClass(virtReg);
Evan Chengd3653122008-02-27 03:04:06 +000095 int SS = MF.getFrameInfo()->CreateStackObject(RC->getSize(),
96 RC->getAlignment());
97 if (LowSpillSlot == NO_STACK_SLOT)
98 LowSpillSlot = SS;
99 if (HighSpillSlot == NO_STACK_SLOT || SS > HighSpillSlot)
100 HighSpillSlot = SS;
101 unsigned Idx = SS-LowSpillSlot;
102 while (Idx >= SpillSlotToUsesMap.size())
103 SpillSlotToUsesMap.resize(SpillSlotToUsesMap.size()*2);
104 Virt2StackSlotMap[virtReg] = SS;
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000105 ++NumSpills;
Evan Chengd3653122008-02-27 03:04:06 +0000106 return SS;
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000107}
108
Evan Chengd3653122008-02-27 03:04:06 +0000109void VirtRegMap::assignVirt2StackSlot(unsigned virtReg, int SS) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000110 assert(TargetRegisterInfo::isVirtualRegister(virtReg));
Chris Lattner7f690e62004-09-30 02:15:18 +0000111 assert(Virt2StackSlotMap[virtReg] == NO_STACK_SLOT &&
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000112 "attempt to assign stack slot to already spilled register");
Evan Chengd3653122008-02-27 03:04:06 +0000113 assert((SS >= 0 ||
114 (SS >= MF.getFrameInfo()->getObjectIndexBegin())) &&
Evan Cheng91935142007-04-04 07:40:01 +0000115 "illegal fixed frame index");
Evan Chengd3653122008-02-27 03:04:06 +0000116 Virt2StackSlotMap[virtReg] = SS;
Alkis Evlogimenos38af59a2004-05-29 20:38:05 +0000117}
118
Evan Cheng2638e1a2007-03-20 08:13:50 +0000119int VirtRegMap::assignVirtReMatId(unsigned virtReg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000120 assert(TargetRegisterInfo::isVirtualRegister(virtReg));
Evan Cheng549f27d32007-08-13 23:45:17 +0000121 assert(Virt2ReMatIdMap[virtReg] == NO_STACK_SLOT &&
Evan Cheng2638e1a2007-03-20 08:13:50 +0000122 "attempt to assign re-mat id to already spilled register");
Evan Cheng549f27d32007-08-13 23:45:17 +0000123 Virt2ReMatIdMap[virtReg] = ReMatId;
Evan Cheng2638e1a2007-03-20 08:13:50 +0000124 return ReMatId++;
125}
126
Evan Cheng549f27d32007-08-13 23:45:17 +0000127void VirtRegMap::assignVirtReMatId(unsigned virtReg, int id) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000128 assert(TargetRegisterInfo::isVirtualRegister(virtReg));
Evan Cheng549f27d32007-08-13 23:45:17 +0000129 assert(Virt2ReMatIdMap[virtReg] == NO_STACK_SLOT &&
130 "attempt to assign re-mat id to already spilled register");
131 Virt2ReMatIdMap[virtReg] = id;
132}
133
Evan Cheng676dd7c2008-03-11 07:19:34 +0000134int VirtRegMap::getEmergencySpillSlot(const TargetRegisterClass *RC) {
135 std::map<const TargetRegisterClass*, int>::iterator I =
136 EmergencySpillSlots.find(RC);
137 if (I != EmergencySpillSlots.end())
138 return I->second;
139 int SS = MF.getFrameInfo()->CreateStackObject(RC->getSize(),
140 RC->getAlignment());
141 if (LowSpillSlot == NO_STACK_SLOT)
142 LowSpillSlot = SS;
143 if (HighSpillSlot == NO_STACK_SLOT || SS > HighSpillSlot)
144 HighSpillSlot = SS;
145 I->second = SS;
146 return SS;
147}
148
Evan Chengd3653122008-02-27 03:04:06 +0000149void VirtRegMap::addSpillSlotUse(int FI, MachineInstr *MI) {
150 if (!MF.getFrameInfo()->isFixedObjectIndex(FI)) {
David Greenecff86082008-05-22 21:12:21 +0000151 // If FI < LowSpillSlot, this stack reference was produced by
152 // instruction selection and is not a spill
153 if (FI >= LowSpillSlot) {
154 assert(FI >= 0 && "Spill slot index should not be negative!");
Bill Wendlingf3061f82008-05-23 01:29:08 +0000155 assert((unsigned)FI-LowSpillSlot < SpillSlotToUsesMap.size()
David Greenecff86082008-05-22 21:12:21 +0000156 && "Invalid spill slot");
157 SpillSlotToUsesMap[FI-LowSpillSlot].insert(MI);
158 }
Evan Chengd3653122008-02-27 03:04:06 +0000159 }
160}
161
Chris Lattnerbec6a9e2004-10-01 23:15:36 +0000162void VirtRegMap::virtFolded(unsigned VirtReg, MachineInstr *OldMI,
Evan Chengaee4af62007-12-02 08:30:39 +0000163 MachineInstr *NewMI, ModRef MRInfo) {
Chris Lattnerbec6a9e2004-10-01 23:15:36 +0000164 // Move previous memory references folded to new instruction.
165 MI2VirtMapTy::iterator IP = MI2VirtMap.lower_bound(NewMI);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000166 for (MI2VirtMapTy::iterator I = MI2VirtMap.lower_bound(OldMI),
Chris Lattnerbec6a9e2004-10-01 23:15:36 +0000167 E = MI2VirtMap.end(); I != E && I->first == OldMI; ) {
168 MI2VirtMap.insert(IP, std::make_pair(NewMI, I->second));
Chris Lattnerdbea9732004-09-30 16:35:08 +0000169 MI2VirtMap.erase(I++);
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000170 }
Chris Lattnerdbea9732004-09-30 16:35:08 +0000171
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000172 // add new memory reference
Chris Lattnerbec6a9e2004-10-01 23:15:36 +0000173 MI2VirtMap.insert(IP, std::make_pair(NewMI, std::make_pair(VirtReg, MRInfo)));
Alkis Evlogimenos5f375022004-03-01 20:05:10 +0000174}
175
Evan Cheng7f566252007-10-13 02:50:24 +0000176void VirtRegMap::virtFolded(unsigned VirtReg, MachineInstr *MI, ModRef MRInfo) {
177 MI2VirtMapTy::iterator IP = MI2VirtMap.lower_bound(MI);
178 MI2VirtMap.insert(IP, std::make_pair(MI, std::make_pair(VirtReg, MRInfo)));
179}
180
Evan Chengd3653122008-02-27 03:04:06 +0000181void VirtRegMap::RemoveMachineInstrFromMaps(MachineInstr *MI) {
182 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
183 MachineOperand &MO = MI->getOperand(i);
184 if (!MO.isFrameIndex())
185 continue;
186 int FI = MO.getIndex();
187 if (MF.getFrameInfo()->isFixedObjectIndex(FI))
188 continue;
David Greenecff86082008-05-22 21:12:21 +0000189 // This stack reference was produced by instruction selection and
190 // is not a spill
191 if (FI < LowSpillSlot)
192 continue;
Bill Wendlingf3061f82008-05-23 01:29:08 +0000193 assert((unsigned)FI-LowSpillSlot < SpillSlotToUsesMap.size()
David Greenecff86082008-05-22 21:12:21 +0000194 && "Invalid spill slot");
Evan Chengd3653122008-02-27 03:04:06 +0000195 SpillSlotToUsesMap[FI-LowSpillSlot].erase(MI);
196 }
197 MI2VirtMap.erase(MI);
198 SpillPt2VirtMap.erase(MI);
199 RestorePt2VirtMap.erase(MI);
Evan Cheng676dd7c2008-03-11 07:19:34 +0000200 EmergencySpillMap.erase(MI);
Evan Chengd3653122008-02-27 03:04:06 +0000201}
202
Chris Lattner7f690e62004-09-30 02:15:18 +0000203void VirtRegMap::print(std::ostream &OS) const {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000204 const TargetRegisterInfo* TRI = MF.getTarget().getRegisterInfo();
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +0000205
Chris Lattner7f690e62004-09-30 02:15:18 +0000206 OS << "********** REGISTER MAP **********\n";
Dan Gohman6f0d0242008-02-10 18:45:23 +0000207 for (unsigned i = TargetRegisterInfo::FirstVirtualRegister,
Chris Lattner84bc5422007-12-31 04:13:23 +0000208 e = MF.getRegInfo().getLastVirtReg(); i <= e; ++i) {
Chris Lattner7f690e62004-09-30 02:15:18 +0000209 if (Virt2PhysMap[i] != (unsigned)VirtRegMap::NO_PHYS_REG)
Bill Wendlinge6d088a2008-02-26 21:47:57 +0000210 OS << "[reg" << i << " -> " << TRI->getName(Virt2PhysMap[i])
Bill Wendling74ab84c2008-02-26 21:11:01 +0000211 << "]\n";
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000212 }
213
Dan Gohman6f0d0242008-02-10 18:45:23 +0000214 for (unsigned i = TargetRegisterInfo::FirstVirtualRegister,
Chris Lattner84bc5422007-12-31 04:13:23 +0000215 e = MF.getRegInfo().getLastVirtReg(); i <= e; ++i)
Chris Lattner7f690e62004-09-30 02:15:18 +0000216 if (Virt2StackSlotMap[i] != VirtRegMap::NO_STACK_SLOT)
217 OS << "[reg" << i << " -> fi#" << Virt2StackSlotMap[i] << "]\n";
218 OS << '\n';
Alkis Evlogimenos34d9bc92004-02-23 23:08:11 +0000219}
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +0000220
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000221void VirtRegMap::dump() const {
Dan Gohmanb5769312008-03-12 20:52:10 +0000222 print(cerr);
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000223}
Alkis Evlogimenosdd420e02004-03-01 23:18:15 +0000224
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000225
226//===----------------------------------------------------------------------===//
227// Simple Spiller Implementation
228//===----------------------------------------------------------------------===//
229
230Spiller::~Spiller() {}
Alkis Evlogimenosdd420e02004-03-01 23:18:15 +0000231
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +0000232namespace {
Chris Lattnerf8c68f62006-06-28 22:17:39 +0000233 struct VISIBILITY_HIDDEN SimpleSpiller : public Spiller {
Chris Lattner35f27052006-05-01 21:16:03 +0000234 bool runOnMachineFunction(MachineFunction& mf, VirtRegMap &VRM);
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000235 };
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +0000236}
237
Chris Lattner35f27052006-05-01 21:16:03 +0000238bool SimpleSpiller::runOnMachineFunction(MachineFunction &MF, VirtRegMap &VRM) {
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000239 DOUT << "********** REWRITE MACHINE CODE **********\n";
240 DOUT << "********** Function: " << MF.getFunction()->getName() << '\n';
Chris Lattnerb0f31bf2005-01-23 22:45:13 +0000241 const TargetMachine &TM = MF.getTarget();
Owen Andersonf6372aa2008-01-01 21:11:32 +0000242 const TargetInstrInfo &TII = *TM.getInstrInfo();
243
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000244
Chris Lattner4ea1b822004-09-30 02:33:48 +0000245 // LoadedRegs - Keep track of which vregs are loaded, so that we only load
246 // each vreg once (in the case where a spilled vreg is used by multiple
247 // operands). This is always smaller than the number of operands to the
248 // current machine instr, so it should be small.
249 std::vector<unsigned> LoadedRegs;
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000250
Chris Lattner0fc27cc2004-09-30 02:59:33 +0000251 for (MachineFunction::iterator MBBI = MF.begin(), E = MF.end();
252 MBBI != E; ++MBBI) {
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000253 DOUT << MBBI->getBasicBlock()->getName() << ":\n";
Chris Lattner0fc27cc2004-09-30 02:59:33 +0000254 MachineBasicBlock &MBB = *MBBI;
255 for (MachineBasicBlock::iterator MII = MBB.begin(),
256 E = MBB.end(); MII != E; ++MII) {
257 MachineInstr &MI = *MII;
258 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
Chris Lattner7fb64342004-10-01 19:04:51 +0000259 MachineOperand &MO = MI.getOperand(i);
Anton Korobeynikov4c71dfe2008-02-20 11:10:28 +0000260 if (MO.isRegister() && MO.getReg()) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000261 if (TargetRegisterInfo::isVirtualRegister(MO.getReg())) {
Chris Lattner886dd912005-04-04 21:35:34 +0000262 unsigned VirtReg = MO.getReg();
263 unsigned PhysReg = VRM.getPhys(VirtReg);
Evan Cheng549f27d32007-08-13 23:45:17 +0000264 if (!VRM.isAssignedReg(VirtReg)) {
Chris Lattner886dd912005-04-04 21:35:34 +0000265 int StackSlot = VRM.getStackSlot(VirtReg);
Chris Lattnerbf9716b2005-09-30 01:29:00 +0000266 const TargetRegisterClass* RC =
Chris Lattner84bc5422007-12-31 04:13:23 +0000267 MF.getRegInfo().getRegClass(VirtReg);
Misha Brukmanedf128a2005-04-21 22:36:52 +0000268
Chris Lattner886dd912005-04-04 21:35:34 +0000269 if (MO.isUse() &&
270 std::find(LoadedRegs.begin(), LoadedRegs.end(), VirtReg)
271 == LoadedRegs.end()) {
Owen Andersonf6372aa2008-01-01 21:11:32 +0000272 TII.loadRegFromStackSlot(MBB, &MI, PhysReg, StackSlot, RC);
Evan Chengd3653122008-02-27 03:04:06 +0000273 MachineInstr *LoadMI = prior(MII);
274 VRM.addSpillSlotUse(StackSlot, LoadMI);
Chris Lattner886dd912005-04-04 21:35:34 +0000275 LoadedRegs.push_back(VirtReg);
276 ++NumLoads;
Evan Chengd3653122008-02-27 03:04:06 +0000277 DOUT << '\t' << *LoadMI;
Chris Lattner886dd912005-04-04 21:35:34 +0000278 }
Misha Brukmanedf128a2005-04-21 22:36:52 +0000279
Chris Lattner886dd912005-04-04 21:35:34 +0000280 if (MO.isDef()) {
Owen Andersonf6372aa2008-01-01 21:11:32 +0000281 TII.storeRegToStackSlot(MBB, next(MII), PhysReg, true,
Evan Chengd64b5c82007-12-05 03:14:33 +0000282 StackSlot, RC);
Evan Chengd3653122008-02-27 03:04:06 +0000283 MachineInstr *StoreMI = next(MII);
284 VRM.addSpillSlotUse(StackSlot, StoreMI);
Chris Lattner886dd912005-04-04 21:35:34 +0000285 ++NumStores;
286 }
Chris Lattner0fc27cc2004-09-30 02:59:33 +0000287 }
Chris Lattner84bc5422007-12-31 04:13:23 +0000288 MF.getRegInfo().setPhysRegUsed(PhysReg);
Chris Lattnere53f4a02006-05-04 17:52:23 +0000289 MI.getOperand(i).setReg(PhysReg);
Chris Lattner886dd912005-04-04 21:35:34 +0000290 } else {
Chris Lattner84bc5422007-12-31 04:13:23 +0000291 MF.getRegInfo().setPhysRegUsed(MO.getReg());
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000292 }
Anton Korobeynikov4c71dfe2008-02-20 11:10:28 +0000293 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000294 }
Chris Lattner886dd912005-04-04 21:35:34 +0000295
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000296 DOUT << '\t' << MI;
Chris Lattner4ea1b822004-09-30 02:33:48 +0000297 LoadedRegs.clear();
Alkis Evlogimenosdd420e02004-03-01 23:18:15 +0000298 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000299 }
300 return true;
301}
302
303//===----------------------------------------------------------------------===//
304// Local Spiller Implementation
305//===----------------------------------------------------------------------===//
306
307namespace {
Evan Cheng66f71632007-10-19 21:23:22 +0000308 class AvailableSpills;
309
Chris Lattner7fb64342004-10-01 19:04:51 +0000310 /// LocalSpiller - This spiller does a simple pass over the machine basic
311 /// block to attempt to keep spills in registers as much as possible for
312 /// blocks that have low register pressure (the vreg may be spilled due to
313 /// register pressure in other blocks).
Chris Lattnerf8c68f62006-06-28 22:17:39 +0000314 class VISIBILITY_HIDDEN LocalSpiller : public Spiller {
Chris Lattner84bc5422007-12-31 04:13:23 +0000315 MachineRegisterInfo *RegInfo;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000316 const TargetRegisterInfo *TRI;
Chris Lattner7fb64342004-10-01 19:04:51 +0000317 const TargetInstrInfo *TII;
Evan Cheng7a0f1852008-05-20 08:13:21 +0000318 DenseMap<MachineInstr*, unsigned> DistanceMap;
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000319 public:
Chris Lattner35f27052006-05-01 21:16:03 +0000320 bool runOnMachineFunction(MachineFunction &MF, VirtRegMap &VRM) {
Chris Lattner84bc5422007-12-31 04:13:23 +0000321 RegInfo = &MF.getRegInfo();
Dan Gohman6f0d0242008-02-10 18:45:23 +0000322 TRI = MF.getTarget().getRegisterInfo();
Chris Lattner7fb64342004-10-01 19:04:51 +0000323 TII = MF.getTarget().getInstrInfo();
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000324 DOUT << "\n**** Local spiller rewriting function '"
325 << MF.getFunction()->getName() << "':\n";
Chris Lattner84bc5422007-12-31 04:13:23 +0000326 DOUT << "**** Machine Instrs (NOTE! Does not include spills and reloads!)"
327 " ****\n";
David Greene04fa32f2007-09-06 16:36:39 +0000328 DEBUG(MF.dump());
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000329
Chris Lattner7fb64342004-10-01 19:04:51 +0000330 for (MachineFunction::iterator MBB = MF.begin(), E = MF.end();
331 MBB != E; ++MBB)
Evan Cheng549f27d32007-08-13 23:45:17 +0000332 RewriteMBB(*MBB, VRM);
David Greene04fa32f2007-09-06 16:36:39 +0000333
Evan Chengd3653122008-02-27 03:04:06 +0000334 // Mark unused spill slots.
335 MachineFrameInfo *MFI = MF.getFrameInfo();
336 int SS = VRM.getLowSpillSlot();
337 if (SS != VirtRegMap::NO_STACK_SLOT)
338 for (int e = VRM.getHighSpillSlot(); SS <= e; ++SS)
339 if (!VRM.isSpillSlotUsed(SS)) {
340 MFI->RemoveStackObject(SS);
341 ++NumDSS;
342 }
343
David Greene04fa32f2007-09-06 16:36:39 +0000344 DOUT << "**** Post Machine Instrs ****\n";
345 DEBUG(MF.dump());
346
Chris Lattner7fb64342004-10-01 19:04:51 +0000347 return true;
348 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000349 private:
Evan Cheng7a0f1852008-05-20 08:13:21 +0000350 void TransferDeadness(MachineBasicBlock *MBB, unsigned CurDist,
351 unsigned Reg, BitVector &RegKills,
352 std::vector<MachineOperand*> &KillOps);
Evan Cheng66f71632007-10-19 21:23:22 +0000353 bool PrepForUnfoldOpti(MachineBasicBlock &MBB,
354 MachineBasicBlock::iterator &MII,
355 std::vector<MachineInstr*> &MaybeDeadStores,
356 AvailableSpills &Spills, BitVector &RegKills,
357 std::vector<MachineOperand*> &KillOps,
358 VirtRegMap &VRM);
Evan Cheng81a03822007-11-17 00:40:40 +0000359 void SpillRegToStackSlot(MachineBasicBlock &MBB,
360 MachineBasicBlock::iterator &MII,
361 int Idx, unsigned PhysReg, int StackSlot,
362 const TargetRegisterClass *RC,
Evan Cheng35a3e4a2007-12-04 19:19:45 +0000363 bool isAvailable, MachineInstr *&LastStore,
Evan Cheng81a03822007-11-17 00:40:40 +0000364 AvailableSpills &Spills,
365 SmallSet<MachineInstr*, 4> &ReMatDefs,
366 BitVector &RegKills,
367 std::vector<MachineOperand*> &KillOps,
Evan Chenge4b39002007-12-03 21:31:55 +0000368 VirtRegMap &VRM);
Evan Cheng549f27d32007-08-13 23:45:17 +0000369 void RewriteMBB(MachineBasicBlock &MBB, VirtRegMap &VRM);
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000370 };
371}
372
Chris Lattner66cf80f2006-02-03 23:13:58 +0000373/// AvailableSpills - As the local spiller is scanning and rewriting an MBB from
Evan Cheng549f27d32007-08-13 23:45:17 +0000374/// top down, keep track of which spills slots or remat are available in each
375/// register.
Chris Lattner593c9582006-02-03 23:28:46 +0000376///
377/// Note that not all physregs are created equal here. In particular, some
378/// physregs are reloads that we are allowed to clobber or ignore at any time.
379/// Other physregs are values that the register allocated program is using that
380/// we cannot CHANGE, but we can read if we like. We keep track of this on a
Evan Cheng549f27d32007-08-13 23:45:17 +0000381/// per-stack-slot / remat id basis as the low bit in the value of the
382/// SpillSlotsAvailable entries. The predicate 'canClobberPhysReg()' checks
383/// this bit and addAvailable sets it if.
Chris Lattnerf8c68f62006-06-28 22:17:39 +0000384namespace {
385class VISIBILITY_HIDDEN AvailableSpills {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000386 const TargetRegisterInfo *TRI;
Chris Lattner66cf80f2006-02-03 23:13:58 +0000387 const TargetInstrInfo *TII;
388
Evan Cheng549f27d32007-08-13 23:45:17 +0000389 // SpillSlotsOrReMatsAvailable - This map keeps track of all of the spilled
390 // or remat'ed virtual register values that are still available, due to being
391 // loaded or stored to, but not invalidated yet.
392 std::map<int, unsigned> SpillSlotsOrReMatsAvailable;
Chris Lattner66cf80f2006-02-03 23:13:58 +0000393
Evan Cheng549f27d32007-08-13 23:45:17 +0000394 // PhysRegsAvailable - This is the inverse of SpillSlotsOrReMatsAvailable,
395 // indicating which stack slot values are currently held by a physreg. This
396 // is used to invalidate entries in SpillSlotsOrReMatsAvailable when a
397 // physreg is modified.
Chris Lattner66cf80f2006-02-03 23:13:58 +0000398 std::multimap<unsigned, int> PhysRegsAvailable;
399
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000400 void disallowClobberPhysRegOnly(unsigned PhysReg);
401
Chris Lattner66cf80f2006-02-03 23:13:58 +0000402 void ClobberPhysRegOnly(unsigned PhysReg);
403public:
Dan Gohman6f0d0242008-02-10 18:45:23 +0000404 AvailableSpills(const TargetRegisterInfo *tri, const TargetInstrInfo *tii)
405 : TRI(tri), TII(tii) {
Chris Lattner66cf80f2006-02-03 23:13:58 +0000406 }
407
Dan Gohman6f0d0242008-02-10 18:45:23 +0000408 const TargetRegisterInfo *getRegInfo() const { return TRI; }
Evan Cheng91e23902007-02-23 01:13:26 +0000409
Evan Cheng549f27d32007-08-13 23:45:17 +0000410 /// getSpillSlotOrReMatPhysReg - If the specified stack slot or remat is
411 /// available in a physical register, return that PhysReg, otherwise
412 /// return 0.
413 unsigned getSpillSlotOrReMatPhysReg(int Slot) const {
414 std::map<int, unsigned>::const_iterator I =
415 SpillSlotsOrReMatsAvailable.find(Slot);
416 if (I != SpillSlotsOrReMatsAvailable.end()) {
Evan Chengb9591c62007-07-11 08:47:44 +0000417 return I->second >> 1; // Remove the CanClobber bit.
Evan Cheng91e23902007-02-23 01:13:26 +0000418 }
Chris Lattner66cf80f2006-02-03 23:13:58 +0000419 return 0;
420 }
Evan Chengde4e9422007-02-25 09:51:27 +0000421
Evan Cheng549f27d32007-08-13 23:45:17 +0000422 /// addAvailable - Mark that the specified stack slot / remat is available in
423 /// the specified physreg. If CanClobber is true, the physreg can be modified
424 /// at any time without changing the semantics of the program.
425 void addAvailable(int SlotOrReMat, MachineInstr *MI, unsigned Reg,
Evan Cheng91e23902007-02-23 01:13:26 +0000426 bool CanClobber = true) {
Chris Lattner86662492006-02-03 23:50:46 +0000427 // If this stack slot is thought to be available in some other physreg,
428 // remove its record.
Evan Cheng549f27d32007-08-13 23:45:17 +0000429 ModifyStackSlotOrReMat(SlotOrReMat);
Chris Lattner86662492006-02-03 23:50:46 +0000430
Evan Cheng549f27d32007-08-13 23:45:17 +0000431 PhysRegsAvailable.insert(std::make_pair(Reg, SlotOrReMat));
Evan Cheng90a43c32007-08-15 20:20:34 +0000432 SpillSlotsOrReMatsAvailable[SlotOrReMat]= (Reg << 1) | (unsigned)CanClobber;
Chris Lattner66cf80f2006-02-03 23:13:58 +0000433
Evan Cheng549f27d32007-08-13 23:45:17 +0000434 if (SlotOrReMat > VirtRegMap::MAX_STACK_SLOT)
435 DOUT << "Remembering RM#" << SlotOrReMat-VirtRegMap::MAX_STACK_SLOT-1;
Evan Cheng2638e1a2007-03-20 08:13:50 +0000436 else
Evan Cheng549f27d32007-08-13 23:45:17 +0000437 DOUT << "Remembering SS#" << SlotOrReMat;
Bill Wendlinge6d088a2008-02-26 21:47:57 +0000438 DOUT << " in physreg " << TRI->getName(Reg) << "\n";
Chris Lattner66cf80f2006-02-03 23:13:58 +0000439 }
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000440
Chris Lattner593c9582006-02-03 23:28:46 +0000441 /// canClobberPhysReg - Return true if the spiller is allowed to change the
442 /// value of the specified stackslot register if it desires. The specified
443 /// stack slot must be available in a physreg for this query to make sense.
Evan Cheng549f27d32007-08-13 23:45:17 +0000444 bool canClobberPhysReg(int SlotOrReMat) const {
Evan Cheng90a43c32007-08-15 20:20:34 +0000445 assert(SpillSlotsOrReMatsAvailable.count(SlotOrReMat) &&
446 "Value not available!");
Evan Cheng549f27d32007-08-13 23:45:17 +0000447 return SpillSlotsOrReMatsAvailable.find(SlotOrReMat)->second & 1;
Chris Lattner593c9582006-02-03 23:28:46 +0000448 }
Evan Cheng35a3e4a2007-12-04 19:19:45 +0000449
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000450 /// disallowClobberPhysReg - Unset the CanClobber bit of the specified
451 /// stackslot register. The register is still available but is no longer
452 /// allowed to be modifed.
453 void disallowClobberPhysReg(unsigned PhysReg);
454
Chris Lattner66cf80f2006-02-03 23:13:58 +0000455 /// ClobberPhysReg - This is called when the specified physreg changes
Evan Cheng66f71632007-10-19 21:23:22 +0000456 /// value. We use this to invalidate any info about stuff that lives in
Chris Lattner66cf80f2006-02-03 23:13:58 +0000457 /// it and any of its aliases.
458 void ClobberPhysReg(unsigned PhysReg);
459
Evan Cheng90a43c32007-08-15 20:20:34 +0000460 /// ModifyStackSlotOrReMat - This method is called when the value in a stack
461 /// slot changes. This removes information about which register the previous
462 /// value for this slot lives in (as the previous value is dead now).
Evan Cheng549f27d32007-08-13 23:45:17 +0000463 void ModifyStackSlotOrReMat(int SlotOrReMat);
Chris Lattner66cf80f2006-02-03 23:13:58 +0000464};
Chris Lattnerf8c68f62006-06-28 22:17:39 +0000465}
Chris Lattner66cf80f2006-02-03 23:13:58 +0000466
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000467/// disallowClobberPhysRegOnly - Unset the CanClobber bit of the specified
468/// stackslot register. The register is still available but is no longer
469/// allowed to be modifed.
470void AvailableSpills::disallowClobberPhysRegOnly(unsigned PhysReg) {
471 std::multimap<unsigned, int>::iterator I =
472 PhysRegsAvailable.lower_bound(PhysReg);
473 while (I != PhysRegsAvailable.end() && I->first == PhysReg) {
Evan Cheng549f27d32007-08-13 23:45:17 +0000474 int SlotOrReMat = I->second;
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000475 I++;
Evan Cheng549f27d32007-08-13 23:45:17 +0000476 assert((SpillSlotsOrReMatsAvailable[SlotOrReMat] >> 1) == PhysReg &&
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000477 "Bidirectional map mismatch!");
Evan Cheng549f27d32007-08-13 23:45:17 +0000478 SpillSlotsOrReMatsAvailable[SlotOrReMat] &= ~1;
Bill Wendlinge6d088a2008-02-26 21:47:57 +0000479 DOUT << "PhysReg " << TRI->getName(PhysReg)
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000480 << " copied, it is available for use but can no longer be modified\n";
481 }
482}
483
484/// disallowClobberPhysReg - Unset the CanClobber bit of the specified
485/// stackslot register and its aliases. The register and its aliases may
486/// still available but is no longer allowed to be modifed.
487void AvailableSpills::disallowClobberPhysReg(unsigned PhysReg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000488 for (const unsigned *AS = TRI->getAliasSet(PhysReg); *AS; ++AS)
Evan Cheng7a0d51c2006-12-14 07:54:05 +0000489 disallowClobberPhysRegOnly(*AS);
490 disallowClobberPhysRegOnly(PhysReg);
491}
492
Chris Lattner66cf80f2006-02-03 23:13:58 +0000493/// ClobberPhysRegOnly - This is called when the specified physreg changes
494/// value. We use this to invalidate any info about stuff we thing lives in it.
495void AvailableSpills::ClobberPhysRegOnly(unsigned PhysReg) {
496 std::multimap<unsigned, int>::iterator I =
497 PhysRegsAvailable.lower_bound(PhysReg);
Chris Lattner07cf1412006-02-03 00:36:31 +0000498 while (I != PhysRegsAvailable.end() && I->first == PhysReg) {
Evan Cheng549f27d32007-08-13 23:45:17 +0000499 int SlotOrReMat = I->second;
Chris Lattner07cf1412006-02-03 00:36:31 +0000500 PhysRegsAvailable.erase(I++);
Evan Cheng549f27d32007-08-13 23:45:17 +0000501 assert((SpillSlotsOrReMatsAvailable[SlotOrReMat] >> 1) == PhysReg &&
Chris Lattner66cf80f2006-02-03 23:13:58 +0000502 "Bidirectional map mismatch!");
Evan Cheng549f27d32007-08-13 23:45:17 +0000503 SpillSlotsOrReMatsAvailable.erase(SlotOrReMat);
Bill Wendlinge6d088a2008-02-26 21:47:57 +0000504 DOUT << "PhysReg " << TRI->getName(PhysReg)
Evan Cheng2638e1a2007-03-20 08:13:50 +0000505 << " clobbered, invalidating ";
Evan Cheng549f27d32007-08-13 23:45:17 +0000506 if (SlotOrReMat > VirtRegMap::MAX_STACK_SLOT)
507 DOUT << "RM#" << SlotOrReMat-VirtRegMap::MAX_STACK_SLOT-1 << "\n";
Evan Cheng2638e1a2007-03-20 08:13:50 +0000508 else
Evan Cheng549f27d32007-08-13 23:45:17 +0000509 DOUT << "SS#" << SlotOrReMat << "\n";
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000510 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000511}
512
Chris Lattner66cf80f2006-02-03 23:13:58 +0000513/// ClobberPhysReg - This is called when the specified physreg changes
514/// value. We use this to invalidate any info about stuff we thing lives in
515/// it and any of its aliases.
516void AvailableSpills::ClobberPhysReg(unsigned PhysReg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000517 for (const unsigned *AS = TRI->getAliasSet(PhysReg); *AS; ++AS)
Chris Lattner66cf80f2006-02-03 23:13:58 +0000518 ClobberPhysRegOnly(*AS);
519 ClobberPhysRegOnly(PhysReg);
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000520}
521
Evan Cheng90a43c32007-08-15 20:20:34 +0000522/// ModifyStackSlotOrReMat - This method is called when the value in a stack
523/// slot changes. This removes information about which register the previous
524/// value for this slot lives in (as the previous value is dead now).
Evan Cheng549f27d32007-08-13 23:45:17 +0000525void AvailableSpills::ModifyStackSlotOrReMat(int SlotOrReMat) {
Evan Cheng90a43c32007-08-15 20:20:34 +0000526 std::map<int, unsigned>::iterator It =
527 SpillSlotsOrReMatsAvailable.find(SlotOrReMat);
Evan Cheng549f27d32007-08-13 23:45:17 +0000528 if (It == SpillSlotsOrReMatsAvailable.end()) return;
Evan Chengb9591c62007-07-11 08:47:44 +0000529 unsigned Reg = It->second >> 1;
Evan Cheng549f27d32007-08-13 23:45:17 +0000530 SpillSlotsOrReMatsAvailable.erase(It);
Chris Lattner07cf1412006-02-03 00:36:31 +0000531
532 // This register may hold the value of multiple stack slots, only remove this
533 // stack slot from the set of values the register contains.
534 std::multimap<unsigned, int>::iterator I = PhysRegsAvailable.lower_bound(Reg);
535 for (; ; ++I) {
536 assert(I != PhysRegsAvailable.end() && I->first == Reg &&
537 "Map inverse broken!");
Evan Cheng549f27d32007-08-13 23:45:17 +0000538 if (I->second == SlotOrReMat) break;
Chris Lattner07cf1412006-02-03 00:36:31 +0000539 }
540 PhysRegsAvailable.erase(I);
541}
542
543
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000544
Evan Cheng28bb4622007-07-11 19:17:18 +0000545/// InvalidateKills - MI is going to be deleted. If any of its operands are
546/// marked kill, then invalidate the information.
547static void InvalidateKills(MachineInstr &MI, BitVector &RegKills,
Evan Chengc91f0b82007-08-14 20:23:13 +0000548 std::vector<MachineOperand*> &KillOps,
Evan Cheng66f71632007-10-19 21:23:22 +0000549 SmallVector<unsigned, 2> *KillRegs = NULL) {
Evan Cheng28bb4622007-07-11 19:17:18 +0000550 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
551 MachineOperand &MO = MI.getOperand(i);
Dan Gohman92dfe202007-09-14 20:33:02 +0000552 if (!MO.isRegister() || !MO.isUse() || !MO.isKill())
Evan Cheng28bb4622007-07-11 19:17:18 +0000553 continue;
554 unsigned Reg = MO.getReg();
Evan Chengb6ca4b32007-08-14 23:25:37 +0000555 if (KillRegs)
556 KillRegs->push_back(Reg);
Evan Cheng28bb4622007-07-11 19:17:18 +0000557 if (KillOps[Reg] == &MO) {
558 RegKills.reset(Reg);
559 KillOps[Reg] = NULL;
560 }
561 }
562}
563
Evan Cheng39c883c2007-12-11 23:36:57 +0000564/// InvalidateKill - A MI that defines the specified register is being deleted,
565/// invalidate the register kill information.
566static void InvalidateKill(unsigned Reg, BitVector &RegKills,
567 std::vector<MachineOperand*> &KillOps) {
568 if (RegKills[Reg]) {
Chris Lattnerf7382302007-12-30 21:56:09 +0000569 KillOps[Reg]->setIsKill(false);
Evan Cheng39c883c2007-12-11 23:36:57 +0000570 KillOps[Reg] = NULL;
571 RegKills.reset(Reg);
572 }
573}
574
Evan Chengb6ca4b32007-08-14 23:25:37 +0000575/// InvalidateRegDef - If the def operand of the specified def MI is now dead
576/// (since it's spill instruction is removed), mark it isDead. Also checks if
577/// the def MI has other definition operands that are not dead. Returns it by
578/// reference.
579static bool InvalidateRegDef(MachineBasicBlock::iterator I,
580 MachineInstr &NewDef, unsigned Reg,
581 bool &HasLiveDef) {
582 // Due to remat, it's possible this reg isn't being reused. That is,
583 // the def of this reg (by prev MI) is now dead.
584 MachineInstr *DefMI = I;
585 MachineOperand *DefOp = NULL;
586 for (unsigned i = 0, e = DefMI->getNumOperands(); i != e; ++i) {
587 MachineOperand &MO = DefMI->getOperand(i);
Dan Gohman92dfe202007-09-14 20:33:02 +0000588 if (MO.isRegister() && MO.isDef()) {
Evan Chengb6ca4b32007-08-14 23:25:37 +0000589 if (MO.getReg() == Reg)
590 DefOp = &MO;
591 else if (!MO.isDead())
592 HasLiveDef = true;
593 }
594 }
595 if (!DefOp)
596 return false;
597
598 bool FoundUse = false, Done = false;
599 MachineBasicBlock::iterator E = NewDef;
600 ++I; ++E;
601 for (; !Done && I != E; ++I) {
602 MachineInstr *NMI = I;
603 for (unsigned j = 0, ee = NMI->getNumOperands(); j != ee; ++j) {
604 MachineOperand &MO = NMI->getOperand(j);
Dan Gohman92dfe202007-09-14 20:33:02 +0000605 if (!MO.isRegister() || MO.getReg() != Reg)
Evan Chengb6ca4b32007-08-14 23:25:37 +0000606 continue;
607 if (MO.isUse())
608 FoundUse = true;
609 Done = true; // Stop after scanning all the operands of this MI.
610 }
611 }
612 if (!FoundUse) {
613 // Def is dead!
614 DefOp->setIsDead();
615 return true;
616 }
617 return false;
618}
619
Evan Cheng28bb4622007-07-11 19:17:18 +0000620/// UpdateKills - Track and update kill info. If a MI reads a register that is
621/// marked kill, then it must be due to register reuse. Transfer the kill info
622/// over.
623static void UpdateKills(MachineInstr &MI, BitVector &RegKills,
624 std::vector<MachineOperand*> &KillOps) {
Chris Lattner749c6f62008-01-07 07:27:27 +0000625 const TargetInstrDesc &TID = MI.getDesc();
Evan Cheng28bb4622007-07-11 19:17:18 +0000626 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
627 MachineOperand &MO = MI.getOperand(i);
Dan Gohman92dfe202007-09-14 20:33:02 +0000628 if (!MO.isRegister() || !MO.isUse())
Evan Cheng28bb4622007-07-11 19:17:18 +0000629 continue;
630 unsigned Reg = MO.getReg();
631 if (Reg == 0)
632 continue;
633
Evan Cheng70366b92008-03-21 19:09:30 +0000634 if (RegKills[Reg] && KillOps[Reg]->getParent() != &MI) {
Evan Cheng28bb4622007-07-11 19:17:18 +0000635 // That can't be right. Register is killed but not re-defined and it's
636 // being reused. Let's fix that.
Chris Lattnerf7382302007-12-30 21:56:09 +0000637 KillOps[Reg]->setIsKill(false);
Evan Cheng39c883c2007-12-11 23:36:57 +0000638 KillOps[Reg] = NULL;
639 RegKills.reset(Reg);
Chris Lattner749c6f62008-01-07 07:27:27 +0000640 if (i < TID.getNumOperands() &&
641 TID.getOperandConstraint(i, TOI::TIED_TO) == -1)
Evan Cheng28bb4622007-07-11 19:17:18 +0000642 // Unless it's a two-address operand, this is the new kill.
643 MO.setIsKill();
644 }
Evan Cheng28bb4622007-07-11 19:17:18 +0000645 if (MO.isKill()) {
646 RegKills.set(Reg);
647 KillOps[Reg] = &MO;
648 }
649 }
650
651 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
652 const MachineOperand &MO = MI.getOperand(i);
Dan Gohman92dfe202007-09-14 20:33:02 +0000653 if (!MO.isRegister() || !MO.isDef())
Evan Cheng28bb4622007-07-11 19:17:18 +0000654 continue;
655 unsigned Reg = MO.getReg();
656 RegKills.reset(Reg);
657 KillOps[Reg] = NULL;
658 }
659}
660
Evan Chengd70dbb52008-02-22 09:24:50 +0000661/// ReMaterialize - Re-materialize definition for Reg targetting DestReg.
662///
663static void ReMaterialize(MachineBasicBlock &MBB,
664 MachineBasicBlock::iterator &MII,
665 unsigned DestReg, unsigned Reg,
Evan Chengca1267c2008-03-31 20:40:39 +0000666 const TargetInstrInfo *TII,
Evan Chengd70dbb52008-02-22 09:24:50 +0000667 const TargetRegisterInfo *TRI,
668 VirtRegMap &VRM) {
Evan Chengca1267c2008-03-31 20:40:39 +0000669 TII->reMaterialize(MBB, MII, DestReg, VRM.getReMaterializedMI(Reg));
Evan Chengd70dbb52008-02-22 09:24:50 +0000670 MachineInstr *NewMI = prior(MII);
671 for (unsigned i = 0, e = NewMI->getNumOperands(); i != e; ++i) {
672 MachineOperand &MO = NewMI->getOperand(i);
673 if (!MO.isRegister() || MO.getReg() == 0)
674 continue;
675 unsigned VirtReg = MO.getReg();
676 if (TargetRegisterInfo::isPhysicalRegister(VirtReg))
677 continue;
678 assert(MO.isUse());
679 unsigned SubIdx = MO.getSubReg();
680 unsigned Phys = VRM.getPhys(VirtReg);
681 assert(Phys);
682 unsigned RReg = SubIdx ? TRI->getSubReg(Phys, SubIdx) : Phys;
683 MO.setReg(RReg);
684 }
685 ++NumReMats;
686}
687
Evan Cheng28bb4622007-07-11 19:17:18 +0000688
Chris Lattner7fb64342004-10-01 19:04:51 +0000689// ReusedOp - For each reused operand, we keep track of a bit of information, in
690// case we need to rollback upon processing a new operand. See comments below.
691namespace {
692 struct ReusedOp {
693 // The MachineInstr operand that reused an available value.
694 unsigned Operand;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000695
Evan Cheng549f27d32007-08-13 23:45:17 +0000696 // StackSlotOrReMat - The spill slot or remat id of the value being reused.
697 unsigned StackSlotOrReMat;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000698
Chris Lattner7fb64342004-10-01 19:04:51 +0000699 // PhysRegReused - The physical register the value was available in.
700 unsigned PhysRegReused;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000701
Chris Lattner7fb64342004-10-01 19:04:51 +0000702 // AssignedPhysReg - The physreg that was assigned for use by the reload.
703 unsigned AssignedPhysReg;
Chris Lattner8a61a752005-10-06 17:19:06 +0000704
705 // VirtReg - The virtual register itself.
706 unsigned VirtReg;
Misha Brukmanedf128a2005-04-21 22:36:52 +0000707
Chris Lattner8a61a752005-10-06 17:19:06 +0000708 ReusedOp(unsigned o, unsigned ss, unsigned prr, unsigned apr,
709 unsigned vreg)
Evan Cheng90a43c32007-08-15 20:20:34 +0000710 : Operand(o), StackSlotOrReMat(ss), PhysRegReused(prr),
711 AssignedPhysReg(apr), VirtReg(vreg) {}
Chris Lattner7fb64342004-10-01 19:04:51 +0000712 };
Chris Lattner540fec62006-02-25 01:51:33 +0000713
714 /// ReuseInfo - This maintains a collection of ReuseOp's for each operand that
715 /// is reused instead of reloaded.
Chris Lattnerf8c68f62006-06-28 22:17:39 +0000716 class VISIBILITY_HIDDEN ReuseInfo {
Chris Lattner540fec62006-02-25 01:51:33 +0000717 MachineInstr &MI;
718 std::vector<ReusedOp> Reuses;
Evan Cheng957840b2007-02-21 02:22:03 +0000719 BitVector PhysRegsClobbered;
Chris Lattner540fec62006-02-25 01:51:33 +0000720 public:
Dan Gohman6f0d0242008-02-10 18:45:23 +0000721 ReuseInfo(MachineInstr &mi, const TargetRegisterInfo *tri) : MI(mi) {
722 PhysRegsClobbered.resize(tri->getNumRegs());
Evan Chenge077ef62006-11-04 00:21:55 +0000723 }
Chris Lattner540fec62006-02-25 01:51:33 +0000724
725 bool hasReuses() const {
726 return !Reuses.empty();
727 }
728
729 /// addReuse - If we choose to reuse a virtual register that is already
730 /// available instead of reloading it, remember that we did so.
Evan Cheng549f27d32007-08-13 23:45:17 +0000731 void addReuse(unsigned OpNo, unsigned StackSlotOrReMat,
Chris Lattner540fec62006-02-25 01:51:33 +0000732 unsigned PhysRegReused, unsigned AssignedPhysReg,
733 unsigned VirtReg) {
734 // If the reload is to the assigned register anyway, no undo will be
735 // required.
736 if (PhysRegReused == AssignedPhysReg) return;
737
738 // Otherwise, remember this.
Evan Cheng549f27d32007-08-13 23:45:17 +0000739 Reuses.push_back(ReusedOp(OpNo, StackSlotOrReMat, PhysRegReused,
Chris Lattner540fec62006-02-25 01:51:33 +0000740 AssignedPhysReg, VirtReg));
741 }
Evan Chenge077ef62006-11-04 00:21:55 +0000742
743 void markClobbered(unsigned PhysReg) {
Evan Cheng957840b2007-02-21 02:22:03 +0000744 PhysRegsClobbered.set(PhysReg);
Evan Chenge077ef62006-11-04 00:21:55 +0000745 }
746
747 bool isClobbered(unsigned PhysReg) const {
Evan Cheng957840b2007-02-21 02:22:03 +0000748 return PhysRegsClobbered.test(PhysReg);
Evan Chenge077ef62006-11-04 00:21:55 +0000749 }
Chris Lattner540fec62006-02-25 01:51:33 +0000750
751 /// GetRegForReload - We are about to emit a reload into PhysReg. If there
752 /// is some other operand that is using the specified register, either pick
753 /// a new register to use, or evict the previous reload and use this reg.
754 unsigned GetRegForReload(unsigned PhysReg, MachineInstr *MI,
755 AvailableSpills &Spills,
Evan Chengfff3e192007-08-14 09:11:18 +0000756 std::vector<MachineInstr*> &MaybeDeadStores,
Evan Cheng28bb4622007-07-11 19:17:18 +0000757 SmallSet<unsigned, 8> &Rejected,
758 BitVector &RegKills,
Evan Cheng549f27d32007-08-13 23:45:17 +0000759 std::vector<MachineOperand*> &KillOps,
760 VirtRegMap &VRM) {
Owen Andersonf6372aa2008-01-01 21:11:32 +0000761 const TargetInstrInfo* TII = MI->getParent()->getParent()->getTarget()
762 .getInstrInfo();
763
Chris Lattner540fec62006-02-25 01:51:33 +0000764 if (Reuses.empty()) return PhysReg; // This is most often empty.
765
766 for (unsigned ro = 0, e = Reuses.size(); ro != e; ++ro) {
767 ReusedOp &Op = Reuses[ro];
768 // If we find some other reuse that was supposed to use this register
769 // exactly for its reload, we can change this reload to use ITS reload
Evan Cheng3c82cab2007-01-19 22:40:14 +0000770 // register. That is, unless its reload register has already been
771 // considered and subsequently rejected because it has also been reused
772 // by another operand.
773 if (Op.PhysRegReused == PhysReg &&
774 Rejected.count(Op.AssignedPhysReg) == 0) {
Chris Lattner540fec62006-02-25 01:51:33 +0000775 // Yup, use the reload register that we didn't use before.
Evan Cheng3c82cab2007-01-19 22:40:14 +0000776 unsigned NewReg = Op.AssignedPhysReg;
777 Rejected.insert(PhysReg);
Evan Cheng28bb4622007-07-11 19:17:18 +0000778 return GetRegForReload(NewReg, MI, Spills, MaybeDeadStores, Rejected,
Evan Cheng549f27d32007-08-13 23:45:17 +0000779 RegKills, KillOps, VRM);
Chris Lattner540fec62006-02-25 01:51:33 +0000780 } else {
781 // Otherwise, we might also have a problem if a previously reused
782 // value aliases the new register. If so, codegen the previous reload
783 // and use this one.
784 unsigned PRRU = Op.PhysRegReused;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000785 const TargetRegisterInfo *TRI = Spills.getRegInfo();
786 if (TRI->areAliases(PRRU, PhysReg)) {
Chris Lattner540fec62006-02-25 01:51:33 +0000787 // Okay, we found out that an alias of a reused register
788 // was used. This isn't good because it means we have
789 // to undo a previous reuse.
790 MachineBasicBlock *MBB = MI->getParent();
791 const TargetRegisterClass *AliasRC =
Chris Lattner84bc5422007-12-31 04:13:23 +0000792 MBB->getParent()->getRegInfo().getRegClass(Op.VirtReg);
Chris Lattner28bad082006-02-25 02:17:31 +0000793
794 // Copy Op out of the vector and remove it, we're going to insert an
795 // explicit load for it.
796 ReusedOp NewOp = Op;
797 Reuses.erase(Reuses.begin()+ro);
798
799 // Ok, we're going to try to reload the assigned physreg into the
800 // slot that we were supposed to in the first place. However, that
801 // register could hold a reuse. Check to see if it conflicts or
802 // would prefer us to use a different register.
803 unsigned NewPhysReg = GetRegForReload(NewOp.AssignedPhysReg,
Evan Cheng28bb4622007-07-11 19:17:18 +0000804 MI, Spills, MaybeDeadStores,
Evan Cheng549f27d32007-08-13 23:45:17 +0000805 Rejected, RegKills, KillOps, VRM);
Chris Lattner28bad082006-02-25 02:17:31 +0000806
Evan Chengd70dbb52008-02-22 09:24:50 +0000807 MachineBasicBlock::iterator MII = MI;
Evan Cheng549f27d32007-08-13 23:45:17 +0000808 if (NewOp.StackSlotOrReMat > VirtRegMap::MAX_STACK_SLOT) {
Evan Chengca1267c2008-03-31 20:40:39 +0000809 ReMaterialize(*MBB, MII, NewPhysReg, NewOp.VirtReg, TII, TRI,VRM);
Evan Cheng549f27d32007-08-13 23:45:17 +0000810 } else {
Evan Chengd70dbb52008-02-22 09:24:50 +0000811 TII->loadRegFromStackSlot(*MBB, MII, NewPhysReg,
Evan Cheng549f27d32007-08-13 23:45:17 +0000812 NewOp.StackSlotOrReMat, AliasRC);
Evan Chengd3653122008-02-27 03:04:06 +0000813 MachineInstr *LoadMI = prior(MII);
814 VRM.addSpillSlotUse(NewOp.StackSlotOrReMat, LoadMI);
Evan Chengfff3e192007-08-14 09:11:18 +0000815 // Any stores to this stack slot are not dead anymore.
816 MaybeDeadStores[NewOp.StackSlotOrReMat] = NULL;
Evan Cheng549f27d32007-08-13 23:45:17 +0000817 ++NumLoads;
818 }
Chris Lattner28bad082006-02-25 02:17:31 +0000819 Spills.ClobberPhysReg(NewPhysReg);
820 Spills.ClobberPhysReg(NewOp.PhysRegReused);
Chris Lattner540fec62006-02-25 01:51:33 +0000821
Chris Lattnere53f4a02006-05-04 17:52:23 +0000822 MI->getOperand(NewOp.Operand).setReg(NewPhysReg);
Chris Lattner540fec62006-02-25 01:51:33 +0000823
Evan Cheng549f27d32007-08-13 23:45:17 +0000824 Spills.addAvailable(NewOp.StackSlotOrReMat, MI, NewPhysReg);
Evan Cheng28bb4622007-07-11 19:17:18 +0000825 --MII;
826 UpdateKills(*MII, RegKills, KillOps);
827 DOUT << '\t' << *MII;
Chris Lattner540fec62006-02-25 01:51:33 +0000828
Bill Wendlingb2b9c202006-11-17 02:09:07 +0000829 DOUT << "Reuse undone!\n";
Chris Lattner540fec62006-02-25 01:51:33 +0000830 --NumReused;
Chris Lattner28bad082006-02-25 02:17:31 +0000831
832 // Finally, PhysReg is now available, go ahead and use it.
Chris Lattner540fec62006-02-25 01:51:33 +0000833 return PhysReg;
834 }
835 }
836 }
837 return PhysReg;
838 }
Evan Cheng3c82cab2007-01-19 22:40:14 +0000839
840 /// GetRegForReload - Helper for the above GetRegForReload(). Add a
841 /// 'Rejected' set to remember which registers have been considered and
842 /// rejected for the reload. This avoids infinite looping in case like
843 /// this:
844 /// t1 := op t2, t3
845 /// t2 <- assigned r0 for use by the reload but ended up reuse r1
846 /// t3 <- assigned r1 for use by the reload but ended up reuse r0
847 /// t1 <- desires r1
848 /// sees r1 is taken by t2, tries t2's reload register r0
849 /// sees r0 is taken by t3, tries t3's reload register r1
850 /// sees r1 is taken by t2, tries t2's reload register r0 ...
851 unsigned GetRegForReload(unsigned PhysReg, MachineInstr *MI,
852 AvailableSpills &Spills,
Evan Chengfff3e192007-08-14 09:11:18 +0000853 std::vector<MachineInstr*> &MaybeDeadStores,
Evan Cheng28bb4622007-07-11 19:17:18 +0000854 BitVector &RegKills,
Evan Cheng549f27d32007-08-13 23:45:17 +0000855 std::vector<MachineOperand*> &KillOps,
856 VirtRegMap &VRM) {
Chris Lattner08a4d5a2007-01-23 00:59:48 +0000857 SmallSet<unsigned, 8> Rejected;
Evan Cheng28bb4622007-07-11 19:17:18 +0000858 return GetRegForReload(PhysReg, MI, Spills, MaybeDeadStores, Rejected,
Evan Cheng549f27d32007-08-13 23:45:17 +0000859 RegKills, KillOps, VRM);
Evan Cheng3c82cab2007-01-19 22:40:14 +0000860 }
Chris Lattner540fec62006-02-25 01:51:33 +0000861 };
Chris Lattner7fb64342004-10-01 19:04:51 +0000862}
Chris Lattner8c4d88d2004-09-30 01:54:45 +0000863
Evan Cheng66f71632007-10-19 21:23:22 +0000864/// PrepForUnfoldOpti - Turn a store folding instruction into a load folding
865/// instruction. e.g.
866/// xorl %edi, %eax
867/// movl %eax, -32(%ebp)
868/// movl -36(%ebp), %eax
Bill Wendlingf059deb2008-02-26 10:51:52 +0000869/// orl %eax, -32(%ebp)
Evan Cheng66f71632007-10-19 21:23:22 +0000870/// ==>
871/// xorl %edi, %eax
872/// orl -36(%ebp), %eax
873/// mov %eax, -32(%ebp)
874/// This enables unfolding optimization for a subsequent instruction which will
875/// also eliminate the newly introduced store instruction.
876bool LocalSpiller::PrepForUnfoldOpti(MachineBasicBlock &MBB,
877 MachineBasicBlock::iterator &MII,
878 std::vector<MachineInstr*> &MaybeDeadStores,
879 AvailableSpills &Spills,
880 BitVector &RegKills,
881 std::vector<MachineOperand*> &KillOps,
882 VirtRegMap &VRM) {
883 MachineFunction &MF = *MBB.getParent();
884 MachineInstr &MI = *MII;
885 unsigned UnfoldedOpc = 0;
886 unsigned UnfoldPR = 0;
887 unsigned UnfoldVR = 0;
888 int FoldedSS = VirtRegMap::NO_STACK_SLOT;
889 VirtRegMap::MI2VirtMapTy::const_iterator I, End;
Evan Chengc17ba8a2008-03-14 20:44:01 +0000890 for (tie(I, End) = VRM.getFoldedVirts(&MI); I != End; ) {
Evan Cheng66f71632007-10-19 21:23:22 +0000891 // Only transform a MI that folds a single register.
892 if (UnfoldedOpc)
893 return false;
894 UnfoldVR = I->second.first;
895 VirtRegMap::ModRef MR = I->second.second;
Evan Chengc17ba8a2008-03-14 20:44:01 +0000896 // MI2VirtMap be can updated which invalidate the iterator.
897 // Increment the iterator first.
898 ++I;
Evan Cheng66f71632007-10-19 21:23:22 +0000899 if (VRM.isAssignedReg(UnfoldVR))
900 continue;
901 // If this reference is not a use, any previous store is now dead.
902 // Otherwise, the store to this stack slot is not dead anymore.
903 FoldedSS = VRM.getStackSlot(UnfoldVR);
904 MachineInstr* DeadStore = MaybeDeadStores[FoldedSS];
905 if (DeadStore && (MR & VirtRegMap::isModRef)) {
906 unsigned PhysReg = Spills.getSpillSlotOrReMatPhysReg(FoldedSS);
Evan Cheng6130f662008-03-05 00:59:57 +0000907 if (!PhysReg || !DeadStore->readsRegister(PhysReg))
Evan Cheng66f71632007-10-19 21:23:22 +0000908 continue;
909 UnfoldPR = PhysReg;
Owen Anderson6425f8b2008-01-07 01:35:56 +0000910 UnfoldedOpc = TII->getOpcodeAfterMemoryUnfold(MI.getOpcode(),
Evan Cheng66f71632007-10-19 21:23:22 +0000911 false, true);
912 }
913 }
914
915 if (!UnfoldedOpc)
916 return false;
917
918 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
919 MachineOperand &MO = MI.getOperand(i);
920 if (!MO.isRegister() || MO.getReg() == 0 || !MO.isUse())
921 continue;
922 unsigned VirtReg = MO.getReg();
Dan Gohman6f0d0242008-02-10 18:45:23 +0000923 if (TargetRegisterInfo::isPhysicalRegister(VirtReg) || MO.getSubReg())
Evan Cheng66f71632007-10-19 21:23:22 +0000924 continue;
925 if (VRM.isAssignedReg(VirtReg)) {
926 unsigned PhysReg = VRM.getPhys(VirtReg);
Dan Gohman6f0d0242008-02-10 18:45:23 +0000927 if (PhysReg && TRI->regsOverlap(PhysReg, UnfoldPR))
Evan Cheng66f71632007-10-19 21:23:22 +0000928 return false;
929 } else if (VRM.isReMaterialized(VirtReg))
930 continue;
931 int SS = VRM.getStackSlot(VirtReg);
932 unsigned PhysReg = Spills.getSpillSlotOrReMatPhysReg(SS);
933 if (PhysReg) {
Dan Gohman6f0d0242008-02-10 18:45:23 +0000934 if (TRI->regsOverlap(PhysReg, UnfoldPR))
Evan Cheng66f71632007-10-19 21:23:22 +0000935 return false;
936 continue;
937 }
938 PhysReg = VRM.getPhys(VirtReg);
Dan Gohman6f0d0242008-02-10 18:45:23 +0000939 if (!TRI->regsOverlap(PhysReg, UnfoldPR))
Evan Cheng66f71632007-10-19 21:23:22 +0000940 continue;
941
942 // Ok, we'll need to reload the value into a register which makes
943 // it impossible to perform the store unfolding optimization later.
944 // Let's see if it is possible to fold the load if the store is
945 // unfolded. This allows us to perform the store unfolding
946 // optimization.
947 SmallVector<MachineInstr*, 4> NewMIs;
Owen Anderson6425f8b2008-01-07 01:35:56 +0000948 if (TII->unfoldMemoryOperand(MF, &MI, UnfoldVR, false, false, NewMIs)) {
Evan Cheng66f71632007-10-19 21:23:22 +0000949 assert(NewMIs.size() == 1);
950 MachineInstr *NewMI = NewMIs.back();
951 NewMIs.clear();
Evan Cheng6130f662008-03-05 00:59:57 +0000952 int Idx = NewMI->findRegisterUseOperandIdx(VirtReg, false);
Evan Cheng81a03822007-11-17 00:40:40 +0000953 assert(Idx != -1);
Evan Chengaee4af62007-12-02 08:30:39 +0000954 SmallVector<unsigned, 2> Ops;
955 Ops.push_back(Idx);
Evan Chengf2f8c2a2008-02-08 22:05:27 +0000956 MachineInstr *FoldedMI = TII->foldMemoryOperand(MF, NewMI, Ops, SS);
Evan Cheng66f71632007-10-19 21:23:22 +0000957 if (FoldedMI) {
Evan Cheng21b3f312008-02-27 19:57:11 +0000958 VRM.addSpillSlotUse(SS, FoldedMI);
Evan Chengcbfb9b22007-10-22 03:01:44 +0000959 if (!VRM.hasPhys(UnfoldVR))
Evan Cheng66f71632007-10-19 21:23:22 +0000960 VRM.assignVirt2Phys(UnfoldVR, UnfoldPR);
Evan Cheng66f71632007-10-19 21:23:22 +0000961 VRM.virtFolded(VirtReg, FoldedMI, VirtRegMap::isRef);
962 MII = MBB.insert(MII, FoldedMI);
Evan Cheng7a0f1852008-05-20 08:13:21 +0000963 InvalidateKills(MI, RegKills, KillOps);
Evan Chengcada2452007-11-28 01:28:46 +0000964 VRM.RemoveMachineInstrFromMaps(&MI);
Evan Cheng66f71632007-10-19 21:23:22 +0000965 MBB.erase(&MI);
966 return true;
967 }
968 delete NewMI;
969 }
970 }
971 return false;
972}
Chris Lattner7fb64342004-10-01 19:04:51 +0000973
Evan Cheng7277a7d2007-11-02 17:35:08 +0000974/// findSuperReg - Find the SubReg's super-register of given register class
975/// where its SubIdx sub-register is SubReg.
976static unsigned findSuperReg(const TargetRegisterClass *RC, unsigned SubReg,
Dan Gohman6f0d0242008-02-10 18:45:23 +0000977 unsigned SubIdx, const TargetRegisterInfo *TRI) {
Evan Cheng7277a7d2007-11-02 17:35:08 +0000978 for (TargetRegisterClass::iterator I = RC->begin(), E = RC->end();
979 I != E; ++I) {
980 unsigned Reg = *I;
Dan Gohman6f0d0242008-02-10 18:45:23 +0000981 if (TRI->getSubReg(Reg, SubIdx) == SubReg)
Evan Cheng7277a7d2007-11-02 17:35:08 +0000982 return Reg;
983 }
984 return 0;
985}
986
Evan Cheng81a03822007-11-17 00:40:40 +0000987/// SpillRegToStackSlot - Spill a register to a specified stack slot. Check if
988/// the last store to the same slot is now dead. If so, remove the last store.
989void LocalSpiller::SpillRegToStackSlot(MachineBasicBlock &MBB,
990 MachineBasicBlock::iterator &MII,
991 int Idx, unsigned PhysReg, int StackSlot,
992 const TargetRegisterClass *RC,
Evan Cheng35a3e4a2007-12-04 19:19:45 +0000993 bool isAvailable, MachineInstr *&LastStore,
Evan Cheng81a03822007-11-17 00:40:40 +0000994 AvailableSpills &Spills,
995 SmallSet<MachineInstr*, 4> &ReMatDefs,
996 BitVector &RegKills,
997 std::vector<MachineOperand*> &KillOps,
Evan Chenge4b39002007-12-03 21:31:55 +0000998 VirtRegMap &VRM) {
Owen Andersonf6372aa2008-01-01 21:11:32 +0000999 TII->storeRegToStackSlot(MBB, next(MII), PhysReg, true, StackSlot, RC);
Evan Chengd3653122008-02-27 03:04:06 +00001000 MachineInstr *StoreMI = next(MII);
1001 VRM.addSpillSlotUse(StackSlot, StoreMI);
1002 DOUT << "Store:\t" << *StoreMI;
Evan Cheng81a03822007-11-17 00:40:40 +00001003
1004 // If there is a dead store to this stack slot, nuke it now.
1005 if (LastStore) {
1006 DOUT << "Removed dead store:\t" << *LastStore;
1007 ++NumDSE;
1008 SmallVector<unsigned, 2> KillRegs;
1009 InvalidateKills(*LastStore, RegKills, KillOps, &KillRegs);
1010 MachineBasicBlock::iterator PrevMII = LastStore;
1011 bool CheckDef = PrevMII != MBB.begin();
1012 if (CheckDef)
1013 --PrevMII;
Evan Chengcada2452007-11-28 01:28:46 +00001014 VRM.RemoveMachineInstrFromMaps(LastStore);
Evan Chengd3653122008-02-27 03:04:06 +00001015 MBB.erase(LastStore);
Evan Cheng81a03822007-11-17 00:40:40 +00001016 if (CheckDef) {
1017 // Look at defs of killed registers on the store. Mark the defs
1018 // as dead since the store has been deleted and they aren't
1019 // being reused.
1020 for (unsigned j = 0, ee = KillRegs.size(); j != ee; ++j) {
1021 bool HasOtherDef = false;
1022 if (InvalidateRegDef(PrevMII, *MII, KillRegs[j], HasOtherDef)) {
1023 MachineInstr *DeadDef = PrevMII;
1024 if (ReMatDefs.count(DeadDef) && !HasOtherDef) {
1025 // FIXME: This assumes a remat def does not have side
1026 // effects.
Evan Chengcada2452007-11-28 01:28:46 +00001027 VRM.RemoveMachineInstrFromMaps(DeadDef);
Evan Chengd3653122008-02-27 03:04:06 +00001028 MBB.erase(DeadDef);
Evan Cheng81a03822007-11-17 00:40:40 +00001029 ++NumDRM;
1030 }
1031 }
1032 }
1033 }
1034 }
1035
Evan Chenge4b39002007-12-03 21:31:55 +00001036 LastStore = next(MII);
Evan Cheng81a03822007-11-17 00:40:40 +00001037
1038 // If the stack slot value was previously available in some other
1039 // register, change it now. Otherwise, make the register available,
1040 // in PhysReg.
1041 Spills.ModifyStackSlotOrReMat(StackSlot);
1042 Spills.ClobberPhysReg(PhysReg);
Evan Cheng35a3e4a2007-12-04 19:19:45 +00001043 Spills.addAvailable(StackSlot, LastStore, PhysReg, isAvailable);
Evan Cheng81a03822007-11-17 00:40:40 +00001044 ++NumStores;
1045}
1046
Evan Cheng7a0f1852008-05-20 08:13:21 +00001047/// TransferDeadness - A identity copy definition is dead and it's being
1048/// removed. Find the last def or use and mark it as dead / kill.
1049void LocalSpiller::TransferDeadness(MachineBasicBlock *MBB, unsigned CurDist,
1050 unsigned Reg, BitVector &RegKills,
1051 std::vector<MachineOperand*> &KillOps) {
1052 int LastUDDist = -1;
1053 MachineInstr *LastUDMI = NULL;
1054 for (MachineRegisterInfo::reg_iterator RI = RegInfo->reg_begin(Reg),
1055 RE = RegInfo->reg_end(); RI != RE; ++RI) {
1056 MachineInstr *UDMI = &*RI;
1057 if (UDMI->getParent() != MBB)
1058 continue;
1059 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(UDMI);
1060 if (DI == DistanceMap.end() || DI->second > CurDist)
1061 continue;
1062 if ((int)DI->second < LastUDDist)
1063 continue;
1064 LastUDDist = DI->second;
1065 LastUDMI = UDMI;
1066 }
1067
1068 if (LastUDMI) {
1069 const TargetInstrDesc &TID = LastUDMI->getDesc();
1070 MachineOperand *LastUD = NULL;
1071 for (unsigned i = 0, e = LastUDMI->getNumOperands(); i != e; ++i) {
1072 MachineOperand &MO = LastUDMI->getOperand(i);
1073 if (!MO.isRegister() || MO.getReg() != Reg)
1074 continue;
1075 if (!LastUD || (LastUD->isUse() && MO.isDef()))
1076 LastUD = &MO;
1077 if (TID.getOperandConstraint(i, TOI::TIED_TO) != -1)
1078 return;
1079 }
1080 if (LastUD->isDef())
1081 LastUD->setIsDead();
1082 else {
1083 LastUD->setIsKill();
1084 RegKills.set(Reg);
1085 KillOps[Reg] = LastUD;
1086 }
1087 }
1088}
1089
Chris Lattner7fb64342004-10-01 19:04:51 +00001090/// rewriteMBB - Keep track of which spills are available even after the
Evan Cheng81a03822007-11-17 00:40:40 +00001091/// register allocator is done with them. If possible, avid reloading vregs.
Evan Cheng549f27d32007-08-13 23:45:17 +00001092void LocalSpiller::RewriteMBB(MachineBasicBlock &MBB, VirtRegMap &VRM) {
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001093 DOUT << MBB.getBasicBlock()->getName() << ":\n";
Chris Lattner7fb64342004-10-01 19:04:51 +00001094
Evan Chengfff3e192007-08-14 09:11:18 +00001095 MachineFunction &MF = *MBB.getParent();
Owen Andersond10fd972007-12-31 06:32:00 +00001096
Chris Lattner66cf80f2006-02-03 23:13:58 +00001097 // Spills - Keep track of which spilled values are available in physregs so
1098 // that we can choose to reuse the physregs instead of emitting reloads.
Dan Gohman6f0d0242008-02-10 18:45:23 +00001099 AvailableSpills Spills(TRI, TII);
Chris Lattner66cf80f2006-02-03 23:13:58 +00001100
Chris Lattner52b25db2004-10-01 19:47:12 +00001101 // MaybeDeadStores - When we need to write a value back into a stack slot,
1102 // keep track of the inserted store. If the stack slot value is never read
1103 // (because the value was used from some available register, for example), and
1104 // subsequently stored to, the original store is dead. This map keeps track
1105 // of inserted stores that are not used. If we see a subsequent store to the
1106 // same stack slot, the original store is deleted.
Evan Chengfff3e192007-08-14 09:11:18 +00001107 std::vector<MachineInstr*> MaybeDeadStores;
1108 MaybeDeadStores.resize(MF.getFrameInfo()->getObjectIndexEnd(), NULL);
Chris Lattner52b25db2004-10-01 19:47:12 +00001109
Evan Chengb6ca4b32007-08-14 23:25:37 +00001110 // ReMatDefs - These are rematerializable def MIs which are not deleted.
1111 SmallSet<MachineInstr*, 4> ReMatDefs;
1112
Evan Cheng0c40d722007-07-11 05:28:39 +00001113 // Keep track of kill information.
Dan Gohman6f0d0242008-02-10 18:45:23 +00001114 BitVector RegKills(TRI->getNumRegs());
Evan Cheng0c40d722007-07-11 05:28:39 +00001115 std::vector<MachineOperand*> KillOps;
Dan Gohman6f0d0242008-02-10 18:45:23 +00001116 KillOps.resize(TRI->getNumRegs(), NULL);
Evan Cheng0c40d722007-07-11 05:28:39 +00001117
Evan Cheng7a0f1852008-05-20 08:13:21 +00001118 unsigned Dist = 0;
1119 DistanceMap.clear();
Chris Lattner7fb64342004-10-01 19:04:51 +00001120 for (MachineBasicBlock::iterator MII = MBB.begin(), E = MBB.end();
1121 MII != E; ) {
Chris Lattner7fb64342004-10-01 19:04:51 +00001122 MachineBasicBlock::iterator NextMII = MII; ++NextMII;
Evan Cheng0c40d722007-07-11 05:28:39 +00001123
Evan Cheng66f71632007-10-19 21:23:22 +00001124 VirtRegMap::MI2VirtMapTy::const_iterator I, End;
Evan Cheng0c40d722007-07-11 05:28:39 +00001125 bool Erased = false;
1126 bool BackTracked = false;
Evan Cheng66f71632007-10-19 21:23:22 +00001127 if (PrepForUnfoldOpti(MBB, MII,
1128 MaybeDeadStores, Spills, RegKills, KillOps, VRM))
1129 NextMII = next(MII);
Chris Lattner7fb64342004-10-01 19:04:51 +00001130
Evan Cheng66f71632007-10-19 21:23:22 +00001131 MachineInstr &MI = *MII;
Chris Lattner749c6f62008-01-07 07:27:27 +00001132 const TargetInstrDesc &TID = MI.getDesc();
Evan Chenge077ef62006-11-04 00:21:55 +00001133
Evan Cheng676dd7c2008-03-11 07:19:34 +00001134 if (VRM.hasEmergencySpills(&MI)) {
1135 // Spill physical register(s) in the rare case the allocator has run out
1136 // of registers to allocate.
1137 SmallSet<int, 4> UsedSS;
1138 std::vector<unsigned> &EmSpills = VRM.getEmergencySpills(&MI);
1139 for (unsigned i = 0, e = EmSpills.size(); i != e; ++i) {
1140 unsigned PhysReg = EmSpills[i];
1141 const TargetRegisterClass *RC =
1142 TRI->getPhysicalRegisterRegClass(PhysReg);
1143 assert(RC && "Unable to determine register class!");
1144 int SS = VRM.getEmergencySpillSlot(RC);
1145 if (UsedSS.count(SS))
1146 assert(0 && "Need to spill more than one physical registers!");
1147 UsedSS.insert(SS);
1148 TII->storeRegToStackSlot(MBB, MII, PhysReg, true, SS, RC);
1149 MachineInstr *StoreMI = prior(MII);
1150 VRM.addSpillSlotUse(SS, StoreMI);
1151 TII->loadRegFromStackSlot(MBB, next(MII), PhysReg, SS, RC);
1152 MachineInstr *LoadMI = next(MII);
1153 VRM.addSpillSlotUse(SS, LoadMI);
Evan Chengc1f53c72008-03-11 21:34:46 +00001154 ++NumPSpills;
Evan Cheng676dd7c2008-03-11 07:19:34 +00001155 }
Evan Cheng17d5f542008-03-12 00:14:07 +00001156 NextMII = next(MII);
Evan Cheng676dd7c2008-03-11 07:19:34 +00001157 }
1158
Evan Cheng0cbb1162007-11-29 01:06:25 +00001159 // Insert restores here if asked to.
1160 if (VRM.isRestorePt(&MI)) {
1161 std::vector<unsigned> &RestoreRegs = VRM.getRestorePtRestores(&MI);
1162 for (unsigned i = 0, e = RestoreRegs.size(); i != e; ++i) {
Evan Chengd70dbb52008-02-22 09:24:50 +00001163 unsigned VirtReg = RestoreRegs[e-i-1]; // Reverse order.
Evan Cheng0cbb1162007-11-29 01:06:25 +00001164 if (!VRM.getPreSplitReg(VirtReg))
1165 continue; // Split interval spilled again.
1166 unsigned Phys = VRM.getPhys(VirtReg);
Chris Lattner84bc5422007-12-31 04:13:23 +00001167 RegInfo->setPhysRegUsed(Phys);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001168 if (VRM.isReMaterialized(VirtReg)) {
Evan Chengca1267c2008-03-31 20:40:39 +00001169 ReMaterialize(MBB, MII, Phys, VirtReg, TII, TRI, VRM);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001170 } else {
Chris Lattner84bc5422007-12-31 04:13:23 +00001171 const TargetRegisterClass* RC = RegInfo->getRegClass(VirtReg);
Evan Chengd3653122008-02-27 03:04:06 +00001172 int SS = VRM.getStackSlot(VirtReg);
1173 TII->loadRegFromStackSlot(MBB, &MI, Phys, SS, RC);
1174 MachineInstr *LoadMI = prior(MII);
1175 VRM.addSpillSlotUse(SS, LoadMI);
Evan Cheng0cbb1162007-11-29 01:06:25 +00001176 ++NumLoads;
1177 }
1178 // This invalidates Phys.
1179 Spills.ClobberPhysReg(Phys);
1180 UpdateKills(*prior(MII), RegKills, KillOps);
1181 DOUT << '\t' << *prior(MII);
1182 }
1183 }
1184
Evan Cheng81a03822007-11-17 00:40:40 +00001185 // Insert spills here if asked to.
Evan Chengcada2452007-11-28 01:28:46 +00001186 if (VRM.isSpillPt(&MI)) {
Evan Chengb50bb8c2007-12-05 08:16:32 +00001187 std::vector<std::pair<unsigned,bool> > &SpillRegs =
1188 VRM.getSpillPtSpills(&MI);
Evan Chengcada2452007-11-28 01:28:46 +00001189 for (unsigned i = 0, e = SpillRegs.size(); i != e; ++i) {
Evan Chengb50bb8c2007-12-05 08:16:32 +00001190 unsigned VirtReg = SpillRegs[i].first;
1191 bool isKill = SpillRegs[i].second;
Evan Chengcada2452007-11-28 01:28:46 +00001192 if (!VRM.getPreSplitReg(VirtReg))
1193 continue; // Split interval spilled again.
Chris Lattner84bc5422007-12-31 04:13:23 +00001194 const TargetRegisterClass *RC = RegInfo->getRegClass(VirtReg);
Evan Chengcada2452007-11-28 01:28:46 +00001195 unsigned Phys = VRM.getPhys(VirtReg);
1196 int StackSlot = VRM.getStackSlot(VirtReg);
Owen Andersonf6372aa2008-01-01 21:11:32 +00001197 TII->storeRegToStackSlot(MBB, next(MII), Phys, isKill, StackSlot, RC);
Evan Chengd64b5c82007-12-05 03:14:33 +00001198 MachineInstr *StoreMI = next(MII);
Evan Chengd3653122008-02-27 03:04:06 +00001199 VRM.addSpillSlotUse(StackSlot, StoreMI);
Evan Cheng4191b962008-03-12 00:02:46 +00001200 DOUT << "Store:\t" << *StoreMI;
Evan Chengd64b5c82007-12-05 03:14:33 +00001201 VRM.virtFolded(VirtReg, StoreMI, VirtRegMap::isMod);
Evan Chengcada2452007-11-28 01:28:46 +00001202 }
Evan Chenge4b39002007-12-03 21:31:55 +00001203 NextMII = next(MII);
Evan Cheng81a03822007-11-17 00:40:40 +00001204 }
1205
1206 /// ReusedOperands - Keep track of operand reuse in case we need to undo
1207 /// reuse.
Dan Gohman6f0d0242008-02-10 18:45:23 +00001208 ReuseInfo ReusedOperands(MI, TRI);
Evan Chengb2fd65f2008-02-22 19:22:06 +00001209 SmallVector<unsigned, 4> VirtUseOps;
Chris Lattner7fb64342004-10-01 19:04:51 +00001210 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
1211 MachineOperand &MO = MI.getOperand(i);
Chris Lattner50ea01e2005-09-09 20:29:51 +00001212 if (!MO.isRegister() || MO.getReg() == 0)
1213 continue; // Ignore non-register operands.
1214
Evan Cheng32dfbea2007-10-12 08:50:34 +00001215 unsigned VirtReg = MO.getReg();
Dan Gohman6f0d0242008-02-10 18:45:23 +00001216 if (TargetRegisterInfo::isPhysicalRegister(VirtReg)) {
Chris Lattner50ea01e2005-09-09 20:29:51 +00001217 // Ignore physregs for spilling, but remember that it is used by this
1218 // function.
Chris Lattner84bc5422007-12-31 04:13:23 +00001219 RegInfo->setPhysRegUsed(VirtReg);
Chris Lattner50ea01e2005-09-09 20:29:51 +00001220 continue;
1221 }
Evan Chengb2fd65f2008-02-22 19:22:06 +00001222
1223 // We want to process implicit virtual register uses first.
1224 if (MO.isImplicit())
Evan Cheng4cce6b42008-04-11 17:53:36 +00001225 // If the virtual register is implicitly defined, emit a implicit_def
1226 // before so scavenger knows it's "defined".
Evan Chengb2fd65f2008-02-22 19:22:06 +00001227 VirtUseOps.insert(VirtUseOps.begin(), i);
1228 else
1229 VirtUseOps.push_back(i);
1230 }
1231
1232 // Process all of the spilled uses and all non spilled reg references.
1233 for (unsigned j = 0, e = VirtUseOps.size(); j != e; ++j) {
1234 unsigned i = VirtUseOps[j];
1235 MachineOperand &MO = MI.getOperand(i);
1236 unsigned VirtReg = MO.getReg();
Dan Gohman6f0d0242008-02-10 18:45:23 +00001237 assert(TargetRegisterInfo::isVirtualRegister(VirtReg) &&
Evan Chengb2fd65f2008-02-22 19:22:06 +00001238 "Not a virtual register?");
Evan Cheng70306f82007-12-03 09:58:48 +00001239
Evan Chengc498b022007-11-14 07:59:08 +00001240 unsigned SubIdx = MO.getSubReg();
Evan Cheng549f27d32007-08-13 23:45:17 +00001241 if (VRM.isAssignedReg(VirtReg)) {
Chris Lattner50ea01e2005-09-09 20:29:51 +00001242 // This virtual register was assigned a physreg!
1243 unsigned Phys = VRM.getPhys(VirtReg);
Chris Lattner84bc5422007-12-31 04:13:23 +00001244 RegInfo->setPhysRegUsed(Phys);
Evan Chenge077ef62006-11-04 00:21:55 +00001245 if (MO.isDef())
1246 ReusedOperands.markClobbered(Phys);
Dan Gohman6f0d0242008-02-10 18:45:23 +00001247 unsigned RReg = SubIdx ? TRI->getSubReg(Phys, SubIdx) : Phys;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001248 MI.getOperand(i).setReg(RReg);
Evan Cheng4cce6b42008-04-11 17:53:36 +00001249 if (VRM.isImplicitlyDefined(VirtReg))
1250 BuildMI(MBB, MI, TII->get(TargetInstrInfo::IMPLICIT_DEF), RReg);
Chris Lattner50ea01e2005-09-09 20:29:51 +00001251 continue;
1252 }
1253
1254 // This virtual register is now known to be a spilled value.
1255 if (!MO.isUse())
1256 continue; // Handle defs in the loop below (handle use&def here though)
Chris Lattner7fb64342004-10-01 19:04:51 +00001257
Evan Cheng549f27d32007-08-13 23:45:17 +00001258 bool DoReMat = VRM.isReMaterialized(VirtReg);
1259 int SSorRMId = DoReMat
1260 ? VRM.getReMatId(VirtReg) : VRM.getStackSlot(VirtReg);
Evan Chengdc6be192007-08-14 05:42:54 +00001261 int ReuseSlot = SSorRMId;
Chris Lattner7fb64342004-10-01 19:04:51 +00001262
Chris Lattner50ea01e2005-09-09 20:29:51 +00001263 // Check to see if this stack slot is available.
Evan Chengdc6be192007-08-14 05:42:54 +00001264 unsigned PhysReg = Spills.getSpillSlotOrReMatPhysReg(SSorRMId);
Evan Cheng32dfbea2007-10-12 08:50:34 +00001265
1266 // If this is a sub-register use, make sure the reuse register is in the
1267 // right register class. For example, for x86 not all of the 32-bit
1268 // registers have accessible sub-registers.
1269 // Similarly so for EXTRACT_SUBREG. Consider this:
1270 // EDI = op
1271 // MOV32_mr fi#1, EDI
1272 // ...
1273 // = EXTRACT_SUBREG fi#1
1274 // fi#1 is available in EDI, but it cannot be reused because it's not in
1275 // the right register file.
1276 if (PhysReg &&
Evan Chengc498b022007-11-14 07:59:08 +00001277 (SubIdx || MI.getOpcode() == TargetInstrInfo::EXTRACT_SUBREG)) {
Chris Lattner84bc5422007-12-31 04:13:23 +00001278 const TargetRegisterClass* RC = RegInfo->getRegClass(VirtReg);
Evan Cheng32dfbea2007-10-12 08:50:34 +00001279 if (!RC->contains(PhysReg))
1280 PhysReg = 0;
1281 }
1282
Evan Chengdc6be192007-08-14 05:42:54 +00001283 if (PhysReg) {
Chris Lattner29268692006-09-05 02:12:02 +00001284 // This spilled operand might be part of a two-address operand. If this
1285 // is the case, then changing it will necessarily require changing the
1286 // def part of the instruction as well. However, in some cases, we
1287 // aren't allowed to modify the reused register. If none of these cases
1288 // apply, reuse it.
1289 bool CanReuse = true;
Chris Lattner749c6f62008-01-07 07:27:27 +00001290 int ti = TID.getOperandConstraint(i, TOI::TIED_TO);
Evan Cheng360c2dd2006-11-01 23:06:55 +00001291 if (ti != -1 &&
Dan Gohman92dfe202007-09-14 20:33:02 +00001292 MI.getOperand(ti).isRegister() &&
Evan Cheng360c2dd2006-11-01 23:06:55 +00001293 MI.getOperand(ti).getReg() == VirtReg) {
Chris Lattner29268692006-09-05 02:12:02 +00001294 // Okay, we have a two address operand. We can reuse this physreg as
Evan Cheng3c82cab2007-01-19 22:40:14 +00001295 // long as we are allowed to clobber the value and there isn't an
1296 // earlier def that has already clobbered the physreg.
Evan Chengdc6be192007-08-14 05:42:54 +00001297 CanReuse = Spills.canClobberPhysReg(ReuseSlot) &&
Evan Chenge077ef62006-11-04 00:21:55 +00001298 !ReusedOperands.isClobbered(PhysReg);
Chris Lattner29268692006-09-05 02:12:02 +00001299 }
1300
1301 if (CanReuse) {
Chris Lattneraddc55a2006-04-28 01:46:50 +00001302 // If this stack slot value is already available, reuse it!
Evan Chengdc6be192007-08-14 05:42:54 +00001303 if (ReuseSlot > VirtRegMap::MAX_STACK_SLOT)
1304 DOUT << "Reusing RM#" << ReuseSlot-VirtRegMap::MAX_STACK_SLOT-1;
Evan Cheng2638e1a2007-03-20 08:13:50 +00001305 else
Evan Chengdc6be192007-08-14 05:42:54 +00001306 DOUT << "Reusing SS#" << ReuseSlot;
Evan Cheng2638e1a2007-03-20 08:13:50 +00001307 DOUT << " from physreg "
Bill Wendlinge6d088a2008-02-26 21:47:57 +00001308 << TRI->getName(PhysReg) << " for vreg"
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001309 << VirtReg <<" instead of reloading into physreg "
Bill Wendlinge6d088a2008-02-26 21:47:57 +00001310 << TRI->getName(VRM.getPhys(VirtReg)) << "\n";
Dan Gohman6f0d0242008-02-10 18:45:23 +00001311 unsigned RReg = SubIdx ? TRI->getSubReg(PhysReg, SubIdx) : PhysReg;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001312 MI.getOperand(i).setReg(RReg);
Chris Lattneraddc55a2006-04-28 01:46:50 +00001313
1314 // The only technical detail we have is that we don't know that
1315 // PhysReg won't be clobbered by a reloaded stack slot that occurs
1316 // later in the instruction. In particular, consider 'op V1, V2'.
1317 // If V1 is available in physreg R0, we would choose to reuse it
1318 // here, instead of reloading it into the register the allocator
1319 // indicated (say R1). However, V2 might have to be reloaded
1320 // later, and it might indicate that it needs to live in R0. When
1321 // this occurs, we need to have information available that
1322 // indicates it is safe to use R1 for the reload instead of R0.
1323 //
1324 // To further complicate matters, we might conflict with an alias,
1325 // or R0 and R1 might not be compatible with each other. In this
1326 // case, we actually insert a reload for V1 in R1, ensuring that
1327 // we can get at R0 or its alias.
Evan Chengdc6be192007-08-14 05:42:54 +00001328 ReusedOperands.addReuse(i, ReuseSlot, PhysReg,
Chris Lattneraddc55a2006-04-28 01:46:50 +00001329 VRM.getPhys(VirtReg), VirtReg);
Evan Chenge077ef62006-11-04 00:21:55 +00001330 if (ti != -1)
1331 // Only mark it clobbered if this is a use&def operand.
1332 ReusedOperands.markClobbered(PhysReg);
Chris Lattneraddc55a2006-04-28 01:46:50 +00001333 ++NumReused;
Evan Chengfff3e192007-08-14 09:11:18 +00001334
1335 if (MI.getOperand(i).isKill() &&
1336 ReuseSlot <= VirtRegMap::MAX_STACK_SLOT) {
1337 // This was the last use and the spilled value is still available
1338 // for reuse. That means the spill was unnecessary!
1339 MachineInstr* DeadStore = MaybeDeadStores[ReuseSlot];
1340 if (DeadStore) {
1341 DOUT << "Removed dead store:\t" << *DeadStore;
1342 InvalidateKills(*DeadStore, RegKills, KillOps);
Evan Chengcada2452007-11-28 01:28:46 +00001343 VRM.RemoveMachineInstrFromMaps(DeadStore);
Evan Cheng66f71632007-10-19 21:23:22 +00001344 MBB.erase(DeadStore);
Evan Chengfff3e192007-08-14 09:11:18 +00001345 MaybeDeadStores[ReuseSlot] = NULL;
1346 ++NumDSE;
1347 }
1348 }
Chris Lattneraddc55a2006-04-28 01:46:50 +00001349 continue;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001350 } // CanReuse
Chris Lattneraddc55a2006-04-28 01:46:50 +00001351
1352 // Otherwise we have a situation where we have a two-address instruction
1353 // whose mod/ref operand needs to be reloaded. This reload is already
1354 // available in some register "PhysReg", but if we used PhysReg as the
1355 // operand to our 2-addr instruction, the instruction would modify
1356 // PhysReg. This isn't cool if something later uses PhysReg and expects
1357 // to get its initial value.
Chris Lattner50ea01e2005-09-09 20:29:51 +00001358 //
Chris Lattneraddc55a2006-04-28 01:46:50 +00001359 // To avoid this problem, and to avoid doing a load right after a store,
1360 // we emit a copy from PhysReg into the designated register for this
1361 // operand.
1362 unsigned DesignatedReg = VRM.getPhys(VirtReg);
1363 assert(DesignatedReg && "Must map virtreg to physreg!");
1364
1365 // Note that, if we reused a register for a previous operand, the
1366 // register we want to reload into might not actually be
1367 // available. If this occurs, use the register indicated by the
1368 // reuser.
1369 if (ReusedOperands.hasReuses())
1370 DesignatedReg = ReusedOperands.GetRegForReload(DesignatedReg, &MI,
Evan Cheng549f27d32007-08-13 23:45:17 +00001371 Spills, MaybeDeadStores, RegKills, KillOps, VRM);
Chris Lattneraddc55a2006-04-28 01:46:50 +00001372
Chris Lattnerba1fc3d2006-04-28 04:43:18 +00001373 // If the mapped designated register is actually the physreg we have
1374 // incoming, we don't need to inserted a dead copy.
1375 if (DesignatedReg == PhysReg) {
1376 // If this stack slot value is already available, reuse it!
Evan Chengdc6be192007-08-14 05:42:54 +00001377 if (ReuseSlot > VirtRegMap::MAX_STACK_SLOT)
1378 DOUT << "Reusing RM#" << ReuseSlot-VirtRegMap::MAX_STACK_SLOT-1;
Evan Cheng2638e1a2007-03-20 08:13:50 +00001379 else
Evan Chengdc6be192007-08-14 05:42:54 +00001380 DOUT << "Reusing SS#" << ReuseSlot;
Bill Wendlinge6d088a2008-02-26 21:47:57 +00001381 DOUT << " from physreg " << TRI->getName(PhysReg)
Bill Wendling6ef781f2008-02-27 06:33:05 +00001382 << " for vreg" << VirtReg
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001383 << " instead of reloading into same physreg.\n";
Dan Gohman6f0d0242008-02-10 18:45:23 +00001384 unsigned RReg = SubIdx ? TRI->getSubReg(PhysReg, SubIdx) : PhysReg;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001385 MI.getOperand(i).setReg(RReg);
Evan Cheng7277a7d2007-11-02 17:35:08 +00001386 ReusedOperands.markClobbered(RReg);
Chris Lattnerba1fc3d2006-04-28 04:43:18 +00001387 ++NumReused;
1388 continue;
1389 }
1390
Chris Lattner84bc5422007-12-31 04:13:23 +00001391 const TargetRegisterClass* RC = RegInfo->getRegClass(VirtReg);
1392 RegInfo->setPhysRegUsed(DesignatedReg);
Evan Chenge077ef62006-11-04 00:21:55 +00001393 ReusedOperands.markClobbered(DesignatedReg);
Owen Andersond10fd972007-12-31 06:32:00 +00001394 TII->copyRegToReg(MBB, &MI, DesignatedReg, PhysReg, RC, RC);
Evan Chengde4e9422007-02-25 09:51:27 +00001395
Evan Cheng6b448092007-03-02 08:52:00 +00001396 MachineInstr *CopyMI = prior(MII);
Evan Cheng0c40d722007-07-11 05:28:39 +00001397 UpdateKills(*CopyMI, RegKills, KillOps);
Evan Chengde4e9422007-02-25 09:51:27 +00001398
Chris Lattneraddc55a2006-04-28 01:46:50 +00001399 // This invalidates DesignatedReg.
1400 Spills.ClobberPhysReg(DesignatedReg);
1401
Evan Chengdc6be192007-08-14 05:42:54 +00001402 Spills.addAvailable(ReuseSlot, &MI, DesignatedReg);
Evan Cheng32dfbea2007-10-12 08:50:34 +00001403 unsigned RReg =
Dan Gohman6f0d0242008-02-10 18:45:23 +00001404 SubIdx ? TRI->getSubReg(DesignatedReg, SubIdx) : DesignatedReg;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001405 MI.getOperand(i).setReg(RReg);
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001406 DOUT << '\t' << *prior(MII);
Chris Lattner50ea01e2005-09-09 20:29:51 +00001407 ++NumReused;
1408 continue;
Evan Cheng66f71632007-10-19 21:23:22 +00001409 } // if (PhysReg)
Chris Lattner50ea01e2005-09-09 20:29:51 +00001410
1411 // Otherwise, reload it and remember that we have it.
1412 PhysReg = VRM.getPhys(VirtReg);
Chris Lattner172c3622006-01-04 06:47:48 +00001413 assert(PhysReg && "Must map virtreg to physreg!");
Chris Lattner7fb64342004-10-01 19:04:51 +00001414
Chris Lattner50ea01e2005-09-09 20:29:51 +00001415 // Note that, if we reused a register for a previous operand, the
1416 // register we want to reload into might not actually be
1417 // available. If this occurs, use the register indicated by the
1418 // reuser.
Chris Lattner540fec62006-02-25 01:51:33 +00001419 if (ReusedOperands.hasReuses())
1420 PhysReg = ReusedOperands.GetRegForReload(PhysReg, &MI,
Evan Cheng549f27d32007-08-13 23:45:17 +00001421 Spills, MaybeDeadStores, RegKills, KillOps, VRM);
Chris Lattner540fec62006-02-25 01:51:33 +00001422
Chris Lattner84bc5422007-12-31 04:13:23 +00001423 RegInfo->setPhysRegUsed(PhysReg);
Evan Chenge077ef62006-11-04 00:21:55 +00001424 ReusedOperands.markClobbered(PhysReg);
Evan Cheng549f27d32007-08-13 23:45:17 +00001425 if (DoReMat) {
Evan Chengca1267c2008-03-31 20:40:39 +00001426 ReMaterialize(MBB, MII, PhysReg, VirtReg, TII, TRI, VRM);
Evan Cheng91935142007-04-04 07:40:01 +00001427 } else {
Chris Lattner84bc5422007-12-31 04:13:23 +00001428 const TargetRegisterClass* RC = RegInfo->getRegClass(VirtReg);
Owen Andersonf6372aa2008-01-01 21:11:32 +00001429 TII->loadRegFromStackSlot(MBB, &MI, PhysReg, SSorRMId, RC);
Evan Chengd3653122008-02-27 03:04:06 +00001430 MachineInstr *LoadMI = prior(MII);
1431 VRM.addSpillSlotUse(SSorRMId, LoadMI);
Evan Cheng91935142007-04-04 07:40:01 +00001432 ++NumLoads;
1433 }
Chris Lattner50ea01e2005-09-09 20:29:51 +00001434 // This invalidates PhysReg.
Chris Lattner66cf80f2006-02-03 23:13:58 +00001435 Spills.ClobberPhysReg(PhysReg);
Chris Lattner50ea01e2005-09-09 20:29:51 +00001436
1437 // Any stores to this stack slot are not dead anymore.
Evan Cheng549f27d32007-08-13 23:45:17 +00001438 if (!DoReMat)
Evan Chengfff3e192007-08-14 09:11:18 +00001439 MaybeDeadStores[SSorRMId] = NULL;
Evan Cheng549f27d32007-08-13 23:45:17 +00001440 Spills.addAvailable(SSorRMId, &MI, PhysReg);
Evan Chengde4e9422007-02-25 09:51:27 +00001441 // Assumes this is the last use. IsKill will be unset if reg is reused
1442 // unless it's a two-address operand.
Chris Lattner749c6f62008-01-07 07:27:27 +00001443 if (TID.getOperandConstraint(i, TOI::TIED_TO) == -1)
Evan Chengde4e9422007-02-25 09:51:27 +00001444 MI.getOperand(i).setIsKill();
Dan Gohman6f0d0242008-02-10 18:45:23 +00001445 unsigned RReg = SubIdx ? TRI->getSubReg(PhysReg, SubIdx) : PhysReg;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001446 MI.getOperand(i).setReg(RReg);
Evan Cheng0c40d722007-07-11 05:28:39 +00001447 UpdateKills(*prior(MII), RegKills, KillOps);
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001448 DOUT << '\t' << *prior(MII);
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001449 }
1450
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001451 DOUT << '\t' << MI;
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001452
Evan Cheng81a03822007-11-17 00:40:40 +00001453
Chris Lattner7fb64342004-10-01 19:04:51 +00001454 // If we have folded references to memory operands, make sure we clear all
1455 // physical registers that may contain the value of the spilled virtual
1456 // register
Evan Cheng66f71632007-10-19 21:23:22 +00001457 SmallSet<int, 2> FoldedSS;
Evan Chengc17ba8a2008-03-14 20:44:01 +00001458 for (tie(I, End) = VRM.getFoldedVirts(&MI); I != End; ) {
Chris Lattnerbec6a9e2004-10-01 23:15:36 +00001459 unsigned VirtReg = I->second.first;
1460 VirtRegMap::ModRef MR = I->second.second;
Evan Cheng66f71632007-10-19 21:23:22 +00001461 DOUT << "Folded vreg: " << VirtReg << " MR: " << MR;
Evan Cheng81a03822007-11-17 00:40:40 +00001462
Evan Chengc17ba8a2008-03-14 20:44:01 +00001463 // MI2VirtMap be can updated which invalidate the iterator.
1464 // Increment the iterator first.
1465 ++I;
Chris Lattnercea86882005-09-19 06:56:21 +00001466 int SS = VRM.getStackSlot(VirtReg);
Evan Cheng81a03822007-11-17 00:40:40 +00001467 if (SS == VirtRegMap::NO_STACK_SLOT)
1468 continue;
Evan Cheng90a43c32007-08-15 20:20:34 +00001469 FoldedSS.insert(SS);
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001470 DOUT << " - StackSlot: " << SS << "\n";
Chris Lattnercea86882005-09-19 06:56:21 +00001471
1472 // If this folded instruction is just a use, check to see if it's a
1473 // straight load from the virt reg slot.
1474 if ((MR & VirtRegMap::isRef) && !(MR & VirtRegMap::isMod)) {
1475 int FrameIdx;
Evan Cheng32dfbea2007-10-12 08:50:34 +00001476 unsigned DestReg = TII->isLoadFromStackSlot(&MI, FrameIdx);
1477 if (DestReg && FrameIdx == SS) {
1478 // If this spill slot is available, turn it into a copy (or nothing)
1479 // instead of leaving it as a load!
1480 if (unsigned InReg = Spills.getSpillSlotOrReMatPhysReg(SS)) {
1481 DOUT << "Promoted Load To Copy: " << MI;
1482 if (DestReg != InReg) {
Chris Lattner84bc5422007-12-31 04:13:23 +00001483 const TargetRegisterClass *RC = RegInfo->getRegClass(VirtReg);
Owen Andersond10fd972007-12-31 06:32:00 +00001484 TII->copyRegToReg(MBB, &MI, DestReg, InReg, RC, RC);
Evan Cheng32dfbea2007-10-12 08:50:34 +00001485 // Revisit the copy so we make sure to notice the effects of the
1486 // operation on the destreg (either needing to RA it if it's
1487 // virtual or needing to clobber any values if it's physical).
1488 NextMII = &MI;
1489 --NextMII; // backtrack to the copy.
1490 BackTracked = true;
Evan Cheng39c883c2007-12-11 23:36:57 +00001491 } else {
Evan Cheng32dfbea2007-10-12 08:50:34 +00001492 DOUT << "Removing now-noop copy: " << MI;
Evan Cheng39c883c2007-12-11 23:36:57 +00001493 // Unset last kill since it's being reused.
1494 InvalidateKill(InReg, RegKills, KillOps);
1495 }
Evan Chengde4e9422007-02-25 09:51:27 +00001496
Evan Cheng7a0f1852008-05-20 08:13:21 +00001497 InvalidateKills(MI, RegKills, KillOps);
Evan Chengcada2452007-11-28 01:28:46 +00001498 VRM.RemoveMachineInstrFromMaps(&MI);
Evan Cheng32dfbea2007-10-12 08:50:34 +00001499 MBB.erase(&MI);
1500 Erased = true;
1501 goto ProcessNextInst;
Chris Lattnercea86882005-09-19 06:56:21 +00001502 }
Evan Cheng7f566252007-10-13 02:50:24 +00001503 } else {
1504 unsigned PhysReg = Spills.getSpillSlotOrReMatPhysReg(SS);
1505 SmallVector<MachineInstr*, 4> NewMIs;
1506 if (PhysReg &&
Owen Anderson6425f8b2008-01-07 01:35:56 +00001507 TII->unfoldMemoryOperand(MF, &MI, PhysReg, false, false, NewMIs)) {
Evan Cheng7f566252007-10-13 02:50:24 +00001508 MBB.insert(MII, NewMIs[0]);
Evan Cheng7a0f1852008-05-20 08:13:21 +00001509 InvalidateKills(MI, RegKills, KillOps);
Evan Chengcada2452007-11-28 01:28:46 +00001510 VRM.RemoveMachineInstrFromMaps(&MI);
Evan Cheng7f566252007-10-13 02:50:24 +00001511 MBB.erase(&MI);
1512 Erased = true;
1513 --NextMII; // backtrack to the unfolded instruction.
1514 BackTracked = true;
1515 goto ProcessNextInst;
1516 }
Chris Lattnercea86882005-09-19 06:56:21 +00001517 }
1518 }
1519
1520 // If this reference is not a use, any previous store is now dead.
1521 // Otherwise, the store to this stack slot is not dead anymore.
Evan Chengfff3e192007-08-14 09:11:18 +00001522 MachineInstr* DeadStore = MaybeDeadStores[SS];
1523 if (DeadStore) {
Evan Cheng66f71632007-10-19 21:23:22 +00001524 bool isDead = !(MR & VirtRegMap::isRef);
Evan Cheng7f566252007-10-13 02:50:24 +00001525 MachineInstr *NewStore = NULL;
Evan Chengcbfb9b22007-10-22 03:01:44 +00001526 if (MR & VirtRegMap::isModRef) {
Evan Cheng7f566252007-10-13 02:50:24 +00001527 unsigned PhysReg = Spills.getSpillSlotOrReMatPhysReg(SS);
1528 SmallVector<MachineInstr*, 4> NewMIs;
Evan Cheng35a3e4a2007-12-04 19:19:45 +00001529 // We can reuse this physreg as long as we are allowed to clobber
Chris Lattner84bc5422007-12-31 04:13:23 +00001530 // the value and there isn't an earlier def that has already clobbered
1531 // the physreg.
Evan Cheng7f566252007-10-13 02:50:24 +00001532 if (PhysReg &&
Evan Cheng7ebc06b2008-05-07 00:49:28 +00001533 !TII->isStoreToStackSlot(&MI, SS)) { // Not profitable!
1534 MachineOperand *KillOpnd =
1535 DeadStore->findRegisterUseOperand(PhysReg, true);
1536 // Note, if the store is storing a sub-register, it's possible the
1537 // super-register is needed below.
1538 if (KillOpnd && !KillOpnd->getSubReg() &&
1539 TII->unfoldMemoryOperand(MF, &MI, PhysReg, false, true,NewMIs)){
1540 MBB.insert(MII, NewMIs[0]);
1541 NewStore = NewMIs[1];
1542 MBB.insert(MII, NewStore);
1543 VRM.addSpillSlotUse(SS, NewStore);
Evan Cheng7a0f1852008-05-20 08:13:21 +00001544 InvalidateKills(MI, RegKills, KillOps);
Evan Cheng7ebc06b2008-05-07 00:49:28 +00001545 VRM.RemoveMachineInstrFromMaps(&MI);
1546 MBB.erase(&MI);
1547 Erased = true;
1548 --NextMII;
1549 --NextMII; // backtrack to the unfolded instruction.
1550 BackTracked = true;
1551 isDead = true;
1552 }
Evan Cheng66f71632007-10-19 21:23:22 +00001553 }
Evan Cheng7f566252007-10-13 02:50:24 +00001554 }
1555
1556 if (isDead) { // Previous store is dead.
Chris Lattnercea86882005-09-19 06:56:21 +00001557 // If we get here, the store is dead, nuke it now.
Evan Chengfff3e192007-08-14 09:11:18 +00001558 DOUT << "Removed dead store:\t" << *DeadStore;
1559 InvalidateKills(*DeadStore, RegKills, KillOps);
Evan Chengcada2452007-11-28 01:28:46 +00001560 VRM.RemoveMachineInstrFromMaps(DeadStore);
Evan Cheng7f566252007-10-13 02:50:24 +00001561 MBB.erase(DeadStore);
1562 if (!NewStore)
1563 ++NumDSE;
Chris Lattnercea86882005-09-19 06:56:21 +00001564 }
Evan Cheng7f566252007-10-13 02:50:24 +00001565
Evan Chengfff3e192007-08-14 09:11:18 +00001566 MaybeDeadStores[SS] = NULL;
Evan Cheng7f566252007-10-13 02:50:24 +00001567 if (NewStore) {
1568 // Treat this store as a spill merged into a copy. That makes the
1569 // stack slot value available.
1570 VRM.virtFolded(VirtReg, NewStore, VirtRegMap::isMod);
1571 goto ProcessNextInst;
1572 }
Chris Lattnercea86882005-09-19 06:56:21 +00001573 }
1574
1575 // If the spill slot value is available, and this is a new definition of
1576 // the value, the value is not available anymore.
1577 if (MR & VirtRegMap::isMod) {
Chris Lattner07cf1412006-02-03 00:36:31 +00001578 // Notice that the value in this stack slot has been modified.
Evan Cheng549f27d32007-08-13 23:45:17 +00001579 Spills.ModifyStackSlotOrReMat(SS);
Chris Lattnercd816392006-02-02 23:29:36 +00001580
1581 // If this is *just* a mod of the value, check to see if this is just a
1582 // store to the spill slot (i.e. the spill got merged into the copy). If
1583 // so, realize that the vreg is available now, and add the store to the
1584 // MaybeDeadStore info.
1585 int StackSlot;
1586 if (!(MR & VirtRegMap::isRef)) {
1587 if (unsigned SrcReg = TII->isStoreToStackSlot(&MI, StackSlot)) {
Dan Gohman6f0d0242008-02-10 18:45:23 +00001588 assert(TargetRegisterInfo::isPhysicalRegister(SrcReg) &&
Chris Lattnercd816392006-02-02 23:29:36 +00001589 "Src hasn't been allocated yet?");
Chris Lattner07cf1412006-02-03 00:36:31 +00001590 // Okay, this is certainly a store of SrcReg to [StackSlot]. Mark
Chris Lattnercd816392006-02-02 23:29:36 +00001591 // this as a potentially dead store in case there is a subsequent
1592 // store into the stack slot without a read from it.
1593 MaybeDeadStores[StackSlot] = &MI;
1594
Chris Lattnercd816392006-02-02 23:29:36 +00001595 // If the stack slot value was previously available in some other
1596 // register, change it now. Otherwise, make the register available,
1597 // in PhysReg.
Evan Cheng91e23902007-02-23 01:13:26 +00001598 Spills.addAvailable(StackSlot, &MI, SrcReg, false/*don't clobber*/);
Chris Lattnercd816392006-02-02 23:29:36 +00001599 }
1600 }
Chris Lattner7fb64342004-10-01 19:04:51 +00001601 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001602 }
1603
Chris Lattner7fb64342004-10-01 19:04:51 +00001604 // Process all of the spilled defs.
1605 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
1606 MachineOperand &MO = MI.getOperand(i);
Evan Cheng66f71632007-10-19 21:23:22 +00001607 if (!(MO.isRegister() && MO.getReg() && MO.isDef()))
1608 continue;
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001609
Evan Cheng66f71632007-10-19 21:23:22 +00001610 unsigned VirtReg = MO.getReg();
Dan Gohman6f0d0242008-02-10 18:45:23 +00001611 if (!TargetRegisterInfo::isVirtualRegister(VirtReg)) {
Evan Cheng66f71632007-10-19 21:23:22 +00001612 // Check to see if this is a noop copy. If so, eliminate the
1613 // instruction before considering the dest reg to be changed.
1614 unsigned Src, Dst;
1615 if (TII->isMoveInstr(MI, Src, Dst) && Src == Dst) {
1616 ++NumDCE;
1617 DOUT << "Removing now-noop copy: " << MI;
Evan Cheng7a0f1852008-05-20 08:13:21 +00001618 SmallVector<unsigned, 2> KillRegs;
1619 InvalidateKills(MI, RegKills, KillOps, &KillRegs);
1620 if (MO.isDead() && !KillRegs.empty()) {
1621 assert(KillRegs[0] == Dst);
1622 // Last def is now dead.
1623 TransferDeadness(&MBB, Dist, Src, RegKills, KillOps);
1624 }
Evan Chengd3653122008-02-27 03:04:06 +00001625 VRM.RemoveMachineInstrFromMaps(&MI);
Evan Cheng66f71632007-10-19 21:23:22 +00001626 MBB.erase(&MI);
1627 Erased = true;
Evan Cheng66f71632007-10-19 21:23:22 +00001628 Spills.disallowClobberPhysReg(VirtReg);
1629 goto ProcessNextInst;
1630 }
1631
1632 // If it's not a no-op copy, it clobbers the value in the destreg.
1633 Spills.ClobberPhysReg(VirtReg);
1634 ReusedOperands.markClobbered(VirtReg);
1635
1636 // Check to see if this instruction is a load from a stack slot into
1637 // a register. If so, this provides the stack slot value in the reg.
1638 int FrameIdx;
1639 if (unsigned DestReg = TII->isLoadFromStackSlot(&MI, FrameIdx)) {
1640 assert(DestReg == VirtReg && "Unknown load situation!");
1641
1642 // If it is a folded reference, then it's not safe to clobber.
1643 bool Folded = FoldedSS.count(FrameIdx);
1644 // Otherwise, if it wasn't available, remember that it is now!
1645 Spills.addAvailable(FrameIdx, &MI, DestReg, !Folded);
1646 goto ProcessNextInst;
1647 }
1648
1649 continue;
1650 }
1651
Evan Chengc498b022007-11-14 07:59:08 +00001652 unsigned SubIdx = MO.getSubReg();
Evan Cheng66f71632007-10-19 21:23:22 +00001653 bool DoReMat = VRM.isReMaterialized(VirtReg);
1654 if (DoReMat)
1655 ReMatDefs.insert(&MI);
1656
1657 // The only vregs left are stack slot definitions.
1658 int StackSlot = VRM.getStackSlot(VirtReg);
Chris Lattner84bc5422007-12-31 04:13:23 +00001659 const TargetRegisterClass *RC = RegInfo->getRegClass(VirtReg);
Evan Cheng66f71632007-10-19 21:23:22 +00001660
1661 // If this def is part of a two-address operand, make sure to execute
1662 // the store from the correct physical register.
1663 unsigned PhysReg;
Chris Lattner749c6f62008-01-07 07:27:27 +00001664 int TiedOp = MI.getDesc().findTiedToSrcOperand(i);
Evan Cheng7277a7d2007-11-02 17:35:08 +00001665 if (TiedOp != -1) {
Evan Cheng66f71632007-10-19 21:23:22 +00001666 PhysReg = MI.getOperand(TiedOp).getReg();
Evan Chengc498b022007-11-14 07:59:08 +00001667 if (SubIdx) {
Dan Gohman6f0d0242008-02-10 18:45:23 +00001668 unsigned SuperReg = findSuperReg(RC, PhysReg, SubIdx, TRI);
1669 assert(SuperReg && TRI->getSubReg(SuperReg, SubIdx) == PhysReg &&
Evan Cheng7277a7d2007-11-02 17:35:08 +00001670 "Can't find corresponding super-register!");
1671 PhysReg = SuperReg;
1672 }
1673 } else {
Evan Cheng66f71632007-10-19 21:23:22 +00001674 PhysReg = VRM.getPhys(VirtReg);
1675 if (ReusedOperands.isClobbered(PhysReg)) {
1676 // Another def has taken the assigned physreg. It must have been a
1677 // use&def which got it due to reuse. Undo the reuse!
1678 PhysReg = ReusedOperands.GetRegForReload(PhysReg, &MI,
1679 Spills, MaybeDeadStores, RegKills, KillOps, VRM);
1680 }
1681 }
1682
Evan Chenged70cbb32008-03-26 19:03:01 +00001683 assert(PhysReg && "VR not assigned a physical register?");
Chris Lattner84bc5422007-12-31 04:13:23 +00001684 RegInfo->setPhysRegUsed(PhysReg);
Dan Gohman6f0d0242008-02-10 18:45:23 +00001685 unsigned RReg = SubIdx ? TRI->getSubReg(PhysReg, SubIdx) : PhysReg;
Evan Cheng7277a7d2007-11-02 17:35:08 +00001686 ReusedOperands.markClobbered(RReg);
1687 MI.getOperand(i).setReg(RReg);
1688
Evan Cheng66f71632007-10-19 21:23:22 +00001689 if (!MO.isDead()) {
Evan Cheng66f71632007-10-19 21:23:22 +00001690 MachineInstr *&LastStore = MaybeDeadStores[StackSlot];
Evan Cheng35a3e4a2007-12-04 19:19:45 +00001691 SpillRegToStackSlot(MBB, MII, -1, PhysReg, StackSlot, RC, true,
1692 LastStore, Spills, ReMatDefs, RegKills, KillOps, VRM);
Evan Chenge4b39002007-12-03 21:31:55 +00001693 NextMII = next(MII);
Evan Cheng66f71632007-10-19 21:23:22 +00001694
1695 // Check to see if this is a noop copy. If so, eliminate the
1696 // instruction before considering the dest reg to be changed.
1697 {
Chris Lattner29268692006-09-05 02:12:02 +00001698 unsigned Src, Dst;
1699 if (TII->isMoveInstr(MI, Src, Dst) && Src == Dst) {
1700 ++NumDCE;
Bill Wendlingb2b9c202006-11-17 02:09:07 +00001701 DOUT << "Removing now-noop copy: " << MI;
Evan Cheng7a0f1852008-05-20 08:13:21 +00001702 InvalidateKills(MI, RegKills, KillOps);
Evan Chengd3653122008-02-27 03:04:06 +00001703 VRM.RemoveMachineInstrFromMaps(&MI);
Chris Lattner29268692006-09-05 02:12:02 +00001704 MBB.erase(&MI);
Evan Cheng0c40d722007-07-11 05:28:39 +00001705 Erased = true;
Evan Cheng66f71632007-10-19 21:23:22 +00001706 UpdateKills(*LastStore, RegKills, KillOps);
Chris Lattner29268692006-09-05 02:12:02 +00001707 goto ProcessNextInst;
Chris Lattner7fb64342004-10-01 19:04:51 +00001708 }
Misha Brukmanedf128a2005-04-21 22:36:52 +00001709 }
Evan Cheng66f71632007-10-19 21:23:22 +00001710 }
Chris Lattner7fb64342004-10-01 19:04:51 +00001711 }
Chris Lattnercea86882005-09-19 06:56:21 +00001712 ProcessNextInst:
Evan Cheng7a0f1852008-05-20 08:13:21 +00001713 DistanceMap.insert(std::make_pair(&MI, Dist++));
Evan Cheng35a3e4a2007-12-04 19:19:45 +00001714 if (!Erased && !BackTracked) {
Evan Cheng0c40d722007-07-11 05:28:39 +00001715 for (MachineBasicBlock::iterator II = MI; II != NextMII; ++II)
1716 UpdateKills(*II, RegKills, KillOps);
Evan Cheng35a3e4a2007-12-04 19:19:45 +00001717 }
Chris Lattner7fb64342004-10-01 19:04:51 +00001718 MII = NextMII;
1719 }
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001720}
1721
Chris Lattner8c4d88d2004-09-30 01:54:45 +00001722llvm::Spiller* llvm::createSpiller() {
1723 switch (SpillerOpt) {
1724 default: assert(0 && "Unreachable!");
1725 case local:
1726 return new LocalSpiller();
1727 case simple:
1728 return new SimpleSpiller();
1729 }
Alkis Evlogimenos0d6c5b62004-02-24 08:58:30 +00001730}