blob: 2e5111dee5629d0837fa746bff407a7591647911 [file] [log] [blame]
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001//===-- TwoAddressInstructionPass.cpp - Two-Address instruction pass ------===//
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 Evlogimenos71499de2003-12-18 13:06:04 +00007//
8//===----------------------------------------------------------------------===//
9//
Alkis Evlogimenos50c047d2004-01-04 23:09:24 +000010// This file implements the TwoAddress instruction pass which is used
11// by most register allocators. Two-Address instructions are rewritten
12// from:
13//
14// A = B op C
15//
16// to:
17//
18// A = B
Alkis Evlogimenos14be6402004-02-04 22:17:40 +000019// A op= C
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000020//
Alkis Evlogimenos14be6402004-02-04 22:17:40 +000021// Note that if a register allocator chooses to use this pass, that it
22// has to be capable of handling the non-SSA nature of these rewritten
23// virtual registers.
24//
25// It is also worth noting that the duplicate operand of the two
26// address instruction is removed.
Chris Lattnerbd91c1c2004-01-31 21:07:15 +000027//
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000028//===----------------------------------------------------------------------===//
29
30#define DEBUG_TYPE "twoaddrinstr"
Chris Lattnerbd91c1c2004-01-31 21:07:15 +000031#include "llvm/CodeGen/Passes.h"
Chris Lattner1e313632004-07-21 23:17:57 +000032#include "llvm/Function.h"
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000033#include "llvm/CodeGen/LiveVariables.h"
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000034#include "llvm/CodeGen/MachineFunctionPass.h"
35#include "llvm/CodeGen/MachineInstr.h"
Bob Wilson852a7e32010-06-15 05:56:31 +000036#include "llvm/CodeGen/MachineInstrBuilder.h"
Chris Lattner84bc5422007-12-31 04:13:23 +000037#include "llvm/CodeGen/MachineRegisterInfo.h"
Dan Gohmana70dca12009-10-09 23:27:56 +000038#include "llvm/Analysis/AliasAnalysis.h"
Evan Cheng2a4410d2011-11-14 19:48:55 +000039#include "llvm/MC/MCInstrItineraries.h"
Dan Gohman6f0d0242008-02-10 18:45:23 +000040#include "llvm/Target/TargetRegisterInfo.h"
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000041#include "llvm/Target/TargetInstrInfo.h"
42#include "llvm/Target/TargetMachine.h"
Owen Anderson95dad832008-10-07 20:22:28 +000043#include "llvm/Target/TargetOptions.h"
Evan Cheng875357d2008-03-13 06:37:55 +000044#include "llvm/Support/Debug.h"
Evan Cheng3d720fb2010-05-05 18:45:40 +000045#include "llvm/Support/ErrorHandling.h"
Evan Cheng7543e582008-06-18 07:49:14 +000046#include "llvm/ADT/BitVector.h"
47#include "llvm/ADT/DenseMap.h"
Dan Gohmand68a0762009-01-05 17:59:02 +000048#include "llvm/ADT/SmallSet.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000049#include "llvm/ADT/Statistic.h"
50#include "llvm/ADT/STLExtras.h"
Alkis Evlogimenos71499de2003-12-18 13:06:04 +000051using namespace llvm;
52
Chris Lattnercd3245a2006-12-19 22:41:21 +000053STATISTIC(NumTwoAddressInstrs, "Number of two-address instructions");
54STATISTIC(NumCommuted , "Number of instructions commuted to coalesce");
Evan Chengd498c8f2009-01-25 03:53:59 +000055STATISTIC(NumAggrCommuted , "Number of instructions aggressively commuted");
Chris Lattnercd3245a2006-12-19 22:41:21 +000056STATISTIC(NumConvertedTo3Addr, "Number of instructions promoted to 3-address");
Evan Cheng875357d2008-03-13 06:37:55 +000057STATISTIC(Num3AddrSunk, "Number of 3-address instructions sunk");
Evan Cheng7543e582008-06-18 07:49:14 +000058STATISTIC(NumReMats, "Number of instructions re-materialized");
Evan Cheng28c7ce32009-02-21 03:14:25 +000059STATISTIC(NumDeletes, "Number of dead instructions deleted");
Evan Cheng2a4410d2011-11-14 19:48:55 +000060STATISTIC(NumReSchedUps, "Number of instructions re-scheduled up");
61STATISTIC(NumReSchedDowns, "Number of instructions re-scheduled down");
Evan Cheng875357d2008-03-13 06:37:55 +000062
63namespace {
Nick Lewycky6726b6d2009-10-25 06:33:48 +000064 class TwoAddressInstructionPass : public MachineFunctionPass {
Evan Cheng875357d2008-03-13 06:37:55 +000065 const TargetInstrInfo *TII;
66 const TargetRegisterInfo *TRI;
Evan Cheng2a4410d2011-11-14 19:48:55 +000067 const InstrItineraryData *InstrItins;
Evan Cheng875357d2008-03-13 06:37:55 +000068 MachineRegisterInfo *MRI;
69 LiveVariables *LV;
Dan Gohmana70dca12009-10-09 23:27:56 +000070 AliasAnalysis *AA;
Evan Cheng875357d2008-03-13 06:37:55 +000071
Evan Cheng870b8072009-03-01 02:03:43 +000072 // DistanceMap - Keep track the distance of a MI from the start of the
73 // current basic block.
74 DenseMap<MachineInstr*, unsigned> DistanceMap;
75
76 // SrcRegMap - A map from virtual registers to physical registers which
77 // are likely targets to be coalesced to due to copies from physical
78 // registers to virtual registers. e.g. v1024 = move r0.
79 DenseMap<unsigned, unsigned> SrcRegMap;
80
81 // DstRegMap - A map from virtual registers to physical registers which
82 // are likely targets to be coalesced to due to copies to physical
83 // registers from virtual registers. e.g. r1 = move v1024.
84 DenseMap<unsigned, unsigned> DstRegMap;
85
Evan Cheng3d720fb2010-05-05 18:45:40 +000086 /// RegSequences - Keep track the list of REG_SEQUENCE instructions seen
87 /// during the initial walk of the machine function.
88 SmallVector<MachineInstr*, 16> RegSequences;
89
Bill Wendling637980e2008-05-10 00:12:52 +000090 bool Sink3AddrInstruction(MachineBasicBlock *MBB, MachineInstr *MI,
91 unsigned Reg,
92 MachineBasicBlock::iterator OldPos);
Evan Cheng7543e582008-06-18 07:49:14 +000093
Evan Cheng7543e582008-06-18 07:49:14 +000094 bool isProfitableToReMat(unsigned Reg, const TargetRegisterClass *RC,
Evan Cheng601ca4b2008-06-25 01:16:38 +000095 MachineInstr *MI, MachineInstr *DefMI,
Evan Cheng870b8072009-03-01 02:03:43 +000096 MachineBasicBlock *MBB, unsigned Loc);
Evan Cheng81913712009-01-23 23:27:33 +000097
Evan Chengd498c8f2009-01-25 03:53:59 +000098 bool NoUseAfterLastDef(unsigned Reg, MachineBasicBlock *MBB, unsigned Dist,
Evan Chengd498c8f2009-01-25 03:53:59 +000099 unsigned &LastDef);
100
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000101 MachineInstr *FindLastUseInMBB(unsigned Reg, MachineBasicBlock *MBB,
102 unsigned Dist);
103
Evan Chengd498c8f2009-01-25 03:53:59 +0000104 bool isProfitableToCommute(unsigned regB, unsigned regC,
105 MachineInstr *MI, MachineBasicBlock *MBB,
Evan Cheng870b8072009-03-01 02:03:43 +0000106 unsigned Dist);
Evan Chengd498c8f2009-01-25 03:53:59 +0000107
Evan Cheng81913712009-01-23 23:27:33 +0000108 bool CommuteInstruction(MachineBasicBlock::iterator &mi,
109 MachineFunction::iterator &mbbi,
Evan Cheng870b8072009-03-01 02:03:43 +0000110 unsigned RegB, unsigned RegC, unsigned Dist);
111
Evan Chengf06e6c22011-03-02 01:08:17 +0000112 bool isProfitableToConv3Addr(unsigned RegA, unsigned RegB);
Evan Chenge6f350d2009-03-30 21:34:07 +0000113
114 bool ConvertInstTo3Addr(MachineBasicBlock::iterator &mi,
115 MachineBasicBlock::iterator &nmi,
116 MachineFunction::iterator &mbbi,
Evan Cheng4d96c632011-02-10 02:20:55 +0000117 unsigned RegA, unsigned RegB, unsigned Dist);
Evan Chenge6f350d2009-03-30 21:34:07 +0000118
Bob Wilson326f4382009-09-01 22:51:08 +0000119 typedef std::pair<std::pair<unsigned, bool>, MachineInstr*> NewKill;
120 bool canUpdateDeletedKills(SmallVector<unsigned, 4> &Kills,
121 SmallVector<NewKill, 4> &NewKills,
122 MachineBasicBlock *MBB, unsigned Dist);
123 bool DeleteUnusedInstr(MachineBasicBlock::iterator &mi,
124 MachineBasicBlock::iterator &nmi,
Jakob Stoklund Olesen0b25ae12009-11-18 21:33:35 +0000125 MachineFunction::iterator &mbbi, unsigned Dist);
Bob Wilson326f4382009-09-01 22:51:08 +0000126
Evan Cheng2a4410d2011-11-14 19:48:55 +0000127 bool isDefTooClose(unsigned Reg, unsigned Dist,
128 MachineInstr *MI, MachineBasicBlock *MBB);
129
130 bool RescheduleMIBelowKill(MachineBasicBlock *MBB,
131 MachineBasicBlock::iterator &mi,
132 MachineBasicBlock::iterator &nmi,
133 unsigned Reg);
134 bool RescheduleKillAboveMI(MachineBasicBlock *MBB,
135 MachineBasicBlock::iterator &mi,
136 MachineBasicBlock::iterator &nmi,
137 unsigned Reg);
138
Bob Wilsoncc80df92009-09-03 20:58:42 +0000139 bool TryInstructionTransform(MachineBasicBlock::iterator &mi,
140 MachineBasicBlock::iterator &nmi,
141 MachineFunction::iterator &mbbi,
142 unsigned SrcIdx, unsigned DstIdx,
Evan Chengf06e6c22011-03-02 01:08:17 +0000143 unsigned Dist,
144 SmallPtrSet<MachineInstr*, 8> &Processed);
145
146 void ScanUses(unsigned DstReg, MachineBasicBlock *MBB,
147 SmallPtrSet<MachineInstr*, 8> &Processed);
Bob Wilsoncc80df92009-09-03 20:58:42 +0000148
Evan Cheng870b8072009-03-01 02:03:43 +0000149 void ProcessCopy(MachineInstr *MI, MachineBasicBlock *MBB,
150 SmallPtrSet<MachineInstr*, 8> &Processed);
Evan Cheng3a3cce52009-08-07 00:28:58 +0000151
Evan Cheng53c779b2010-05-17 20:57:12 +0000152 void CoalesceExtSubRegs(SmallVector<unsigned,4> &Srcs, unsigned DstReg);
153
Evan Cheng3d720fb2010-05-05 18:45:40 +0000154 /// EliminateRegSequences - Eliminate REG_SEQUENCE instructions as part
155 /// of the de-ssa process. This replaces sources of REG_SEQUENCE as
156 /// sub-register references of the register defined by REG_SEQUENCE.
157 bool EliminateRegSequences();
Evan Chengc6dcce32010-05-17 23:24:12 +0000158
Evan Cheng875357d2008-03-13 06:37:55 +0000159 public:
Nick Lewyckyecd94c82007-05-06 13:37:16 +0000160 static char ID; // Pass identification, replacement for typeid
Owen Anderson081c34b2010-10-19 17:21:58 +0000161 TwoAddressInstructionPass() : MachineFunctionPass(ID) {
162 initializeTwoAddressInstructionPassPass(*PassRegistry::getPassRegistry());
163 }
Devang Patel794fd752007-05-01 21:15:47 +0000164
Bill Wendling637980e2008-05-10 00:12:52 +0000165 virtual void getAnalysisUsage(AnalysisUsage &AU) const {
Dan Gohman845012e2009-07-31 23:37:33 +0000166 AU.setPreservesCFG();
Dan Gohmana70dca12009-10-09 23:27:56 +0000167 AU.addRequired<AliasAnalysis>();
Bill Wendling637980e2008-05-10 00:12:52 +0000168 AU.addPreserved<LiveVariables>();
169 AU.addPreservedID(MachineLoopInfoID);
170 AU.addPreservedID(MachineDominatorsID);
Cameron Zwarichd959da92010-12-19 18:03:27 +0000171 AU.addPreservedID(PHIEliminationID);
Bill Wendling637980e2008-05-10 00:12:52 +0000172 MachineFunctionPass::getAnalysisUsage(AU);
173 }
Alkis Evlogimenos4c080862003-12-18 22:40:24 +0000174
Bill Wendling637980e2008-05-10 00:12:52 +0000175 /// runOnMachineFunction - Pass entry point.
Misha Brukman75fa4e42004-07-22 15:26:23 +0000176 bool runOnMachineFunction(MachineFunction&);
177 };
Chris Lattnerd74ea2b2006-05-24 17:04:05 +0000178}
Alkis Evlogimenos71499de2003-12-18 13:06:04 +0000179
Dan Gohman844731a2008-05-13 00:00:25 +0000180char TwoAddressInstructionPass::ID = 0;
Owen Anderson2ab36d32010-10-12 19:48:12 +0000181INITIALIZE_PASS_BEGIN(TwoAddressInstructionPass, "twoaddressinstruction",
182 "Two-Address instruction pass", false, false)
183INITIALIZE_AG_DEPENDENCY(AliasAnalysis)
184INITIALIZE_PASS_END(TwoAddressInstructionPass, "twoaddressinstruction",
Owen Andersonce665bd2010-10-07 22:25:06 +0000185 "Two-Address instruction pass", false, false)
Dan Gohman844731a2008-05-13 00:00:25 +0000186
Owen Anderson90c579d2010-08-06 18:33:48 +0000187char &llvm::TwoAddressInstructionPassID = TwoAddressInstructionPass::ID;
Alkis Evlogimenos4c080862003-12-18 22:40:24 +0000188
Evan Cheng875357d2008-03-13 06:37:55 +0000189/// Sink3AddrInstruction - A two-address instruction has been converted to a
190/// three-address instruction to avoid clobbering a register. Try to sink it
Bill Wendling637980e2008-05-10 00:12:52 +0000191/// past the instruction that would kill the above mentioned register to reduce
192/// register pressure.
Evan Cheng875357d2008-03-13 06:37:55 +0000193bool TwoAddressInstructionPass::Sink3AddrInstruction(MachineBasicBlock *MBB,
194 MachineInstr *MI, unsigned SavedReg,
195 MachineBasicBlock::iterator OldPos) {
Eli Friedmanbde81d52011-09-23 22:41:57 +0000196 // FIXME: Shouldn't we be trying to do this before we three-addressify the
197 // instruction? After this transformation is done, we no longer need
198 // the instruction to be in three-address form.
199
Evan Cheng875357d2008-03-13 06:37:55 +0000200 // Check if it's safe to move this instruction.
201 bool SeenStore = true; // Be conservative.
Evan Chengac1abde2010-03-02 19:03:01 +0000202 if (!MI->isSafeToMove(TII, AA, SeenStore))
Evan Cheng875357d2008-03-13 06:37:55 +0000203 return false;
204
205 unsigned DefReg = 0;
206 SmallSet<unsigned, 4> UseRegs;
Bill Wendling637980e2008-05-10 00:12:52 +0000207
Evan Cheng875357d2008-03-13 06:37:55 +0000208 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
209 const MachineOperand &MO = MI->getOperand(i);
Dan Gohmand735b802008-10-03 15:45:36 +0000210 if (!MO.isReg())
Evan Cheng875357d2008-03-13 06:37:55 +0000211 continue;
212 unsigned MOReg = MO.getReg();
213 if (!MOReg)
214 continue;
215 if (MO.isUse() && MOReg != SavedReg)
216 UseRegs.insert(MO.getReg());
217 if (!MO.isDef())
218 continue;
219 if (MO.isImplicit())
220 // Don't try to move it if it implicitly defines a register.
221 return false;
222 if (DefReg)
223 // For now, don't move any instructions that define multiple registers.
224 return false;
225 DefReg = MO.getReg();
226 }
227
228 // Find the instruction that kills SavedReg.
229 MachineInstr *KillMI = NULL;
Evan Chengf1250ee2010-03-23 20:36:12 +0000230 for (MachineRegisterInfo::use_nodbg_iterator
231 UI = MRI->use_nodbg_begin(SavedReg),
232 UE = MRI->use_nodbg_end(); UI != UE; ++UI) {
Evan Cheng875357d2008-03-13 06:37:55 +0000233 MachineOperand &UseMO = UI.getOperand();
234 if (!UseMO.isKill())
235 continue;
236 KillMI = UseMO.getParent();
237 break;
238 }
Bill Wendling637980e2008-05-10 00:12:52 +0000239
Eli Friedmanbde81d52011-09-23 22:41:57 +0000240 // If we find the instruction that kills SavedReg, and it is in an
241 // appropriate location, we can try to sink the current instruction
242 // past it.
243 if (!KillMI || KillMI->getParent() != MBB || KillMI == MI ||
244 KillMI->getDesc().isTerminator())
Evan Cheng875357d2008-03-13 06:37:55 +0000245 return false;
246
Bill Wendling637980e2008-05-10 00:12:52 +0000247 // If any of the definitions are used by another instruction between the
248 // position and the kill use, then it's not safe to sink it.
249 //
250 // FIXME: This can be sped up if there is an easy way to query whether an
Evan Cheng7543e582008-06-18 07:49:14 +0000251 // instruction is before or after another instruction. Then we can use
Bill Wendling637980e2008-05-10 00:12:52 +0000252 // MachineRegisterInfo def / use instead.
Evan Cheng875357d2008-03-13 06:37:55 +0000253 MachineOperand *KillMO = NULL;
254 MachineBasicBlock::iterator KillPos = KillMI;
255 ++KillPos;
Bill Wendling637980e2008-05-10 00:12:52 +0000256
Evan Cheng7543e582008-06-18 07:49:14 +0000257 unsigned NumVisited = 0;
Chris Lattner7896c9f2009-12-03 00:50:42 +0000258 for (MachineBasicBlock::iterator I = llvm::next(OldPos); I != KillPos; ++I) {
Evan Cheng875357d2008-03-13 06:37:55 +0000259 MachineInstr *OtherMI = I;
Dale Johannesen3bfef032010-02-11 18:22:31 +0000260 // DBG_VALUE cannot be counted against the limit.
261 if (OtherMI->isDebugValue())
262 continue;
Evan Cheng7543e582008-06-18 07:49:14 +0000263 if (NumVisited > 30) // FIXME: Arbitrary limit to reduce compile time cost.
264 return false;
265 ++NumVisited;
Evan Cheng875357d2008-03-13 06:37:55 +0000266 for (unsigned i = 0, e = OtherMI->getNumOperands(); i != e; ++i) {
267 MachineOperand &MO = OtherMI->getOperand(i);
Dan Gohmand735b802008-10-03 15:45:36 +0000268 if (!MO.isReg())
Evan Cheng875357d2008-03-13 06:37:55 +0000269 continue;
270 unsigned MOReg = MO.getReg();
271 if (!MOReg)
272 continue;
273 if (DefReg == MOReg)
274 return false;
Bill Wendling637980e2008-05-10 00:12:52 +0000275
Evan Cheng875357d2008-03-13 06:37:55 +0000276 if (MO.isKill()) {
277 if (OtherMI == KillMI && MOReg == SavedReg)
Evan Cheng7543e582008-06-18 07:49:14 +0000278 // Save the operand that kills the register. We want to unset the kill
279 // marker if we can sink MI past it.
Evan Cheng875357d2008-03-13 06:37:55 +0000280 KillMO = &MO;
281 else if (UseRegs.count(MOReg))
282 // One of the uses is killed before the destination.
283 return false;
284 }
285 }
286 }
287
Evan Cheng875357d2008-03-13 06:37:55 +0000288 // Update kill and LV information.
289 KillMO->setIsKill(false);
290 KillMO = MI->findRegisterUseOperand(SavedReg, false, TRI);
291 KillMO->setIsKill(true);
Owen Anderson802af112008-07-02 21:28:58 +0000292
Evan Cheng9f1c8312008-07-03 09:09:37 +0000293 if (LV)
294 LV->replaceKillInstruction(SavedReg, KillMI, MI);
Evan Cheng875357d2008-03-13 06:37:55 +0000295
296 // Move instruction to its destination.
297 MBB->remove(MI);
298 MBB->insert(KillPos, MI);
299
300 ++Num3AddrSunk;
301 return true;
302}
303
Evan Cheng7543e582008-06-18 07:49:14 +0000304/// isTwoAddrUse - Return true if the specified MI is using the specified
305/// register as a two-address operand.
306static bool isTwoAddrUse(MachineInstr *UseMI, unsigned Reg) {
Evan Chenge837dea2011-06-28 19:10:37 +0000307 const MCInstrDesc &MCID = UseMI->getDesc();
308 for (unsigned i = 0, e = MCID.getNumOperands(); i != e; ++i) {
Evan Cheng7543e582008-06-18 07:49:14 +0000309 MachineOperand &MO = UseMI->getOperand(i);
Dan Gohmand735b802008-10-03 15:45:36 +0000310 if (MO.isReg() && MO.getReg() == Reg &&
Evan Chenga24752f2009-03-19 20:30:06 +0000311 (MO.isDef() || UseMI->isRegTiedToDefOperand(i)))
Evan Cheng7543e582008-06-18 07:49:14 +0000312 // Earlier use is a two-address one.
313 return true;
314 }
315 return false;
316}
317
318/// isProfitableToReMat - Return true if the heuristics determines it is likely
319/// to be profitable to re-materialize the definition of Reg rather than copy
320/// the register.
321bool
322TwoAddressInstructionPass::isProfitableToReMat(unsigned Reg,
Evan Cheng870b8072009-03-01 02:03:43 +0000323 const TargetRegisterClass *RC,
324 MachineInstr *MI, MachineInstr *DefMI,
325 MachineBasicBlock *MBB, unsigned Loc) {
Evan Cheng7543e582008-06-18 07:49:14 +0000326 bool OtherUse = false;
Evan Chengf1250ee2010-03-23 20:36:12 +0000327 for (MachineRegisterInfo::use_nodbg_iterator UI = MRI->use_nodbg_begin(Reg),
328 UE = MRI->use_nodbg_end(); UI != UE; ++UI) {
Evan Cheng7543e582008-06-18 07:49:14 +0000329 MachineOperand &UseMO = UI.getOperand();
Evan Cheng7543e582008-06-18 07:49:14 +0000330 MachineInstr *UseMI = UseMO.getParent();
Evan Cheng601ca4b2008-06-25 01:16:38 +0000331 MachineBasicBlock *UseMBB = UseMI->getParent();
332 if (UseMBB == MBB) {
333 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(UseMI);
334 if (DI != DistanceMap.end() && DI->second == Loc)
335 continue; // Current use.
336 OtherUse = true;
337 // There is at least one other use in the MBB that will clobber the
338 // register.
339 if (isTwoAddrUse(UseMI, Reg))
340 return true;
341 }
Evan Cheng7543e582008-06-18 07:49:14 +0000342 }
Evan Cheng601ca4b2008-06-25 01:16:38 +0000343
344 // If other uses in MBB are not two-address uses, then don't remat.
345 if (OtherUse)
346 return false;
347
348 // No other uses in the same block, remat if it's defined in the same
349 // block so it does not unnecessarily extend the live range.
350 return MBB == DefMI->getParent();
Evan Cheng7543e582008-06-18 07:49:14 +0000351}
352
Evan Chengd498c8f2009-01-25 03:53:59 +0000353/// NoUseAfterLastDef - Return true if there are no intervening uses between the
354/// last instruction in the MBB that defines the specified register and the
355/// two-address instruction which is being processed. It also returns the last
356/// def location by reference
357bool TwoAddressInstructionPass::NoUseAfterLastDef(unsigned Reg,
Evan Cheng870b8072009-03-01 02:03:43 +0000358 MachineBasicBlock *MBB, unsigned Dist,
359 unsigned &LastDef) {
Evan Chengd498c8f2009-01-25 03:53:59 +0000360 LastDef = 0;
361 unsigned LastUse = Dist;
362 for (MachineRegisterInfo::reg_iterator I = MRI->reg_begin(Reg),
363 E = MRI->reg_end(); I != E; ++I) {
364 MachineOperand &MO = I.getOperand();
365 MachineInstr *MI = MO.getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000366 if (MI->getParent() != MBB || MI->isDebugValue())
Dale Johannesend94998f2010-02-09 02:01:46 +0000367 continue;
Evan Chengd498c8f2009-01-25 03:53:59 +0000368 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(MI);
369 if (DI == DistanceMap.end())
370 continue;
371 if (MO.isUse() && DI->second < LastUse)
372 LastUse = DI->second;
373 if (MO.isDef() && DI->second > LastDef)
374 LastDef = DI->second;
375 }
376
377 return !(LastUse > LastDef && LastUse < Dist);
378}
379
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000380MachineInstr *TwoAddressInstructionPass::FindLastUseInMBB(unsigned Reg,
381 MachineBasicBlock *MBB,
382 unsigned Dist) {
Lang Hamesa7c9dea2009-05-14 04:26:30 +0000383 unsigned LastUseDist = 0;
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000384 MachineInstr *LastUse = 0;
385 for (MachineRegisterInfo::reg_iterator I = MRI->reg_begin(Reg),
386 E = MRI->reg_end(); I != E; ++I) {
387 MachineOperand &MO = I.getOperand();
388 MachineInstr *MI = MO.getParent();
Chris Lattner518bb532010-02-09 19:54:29 +0000389 if (MI->getParent() != MBB || MI->isDebugValue())
Dale Johannesend94998f2010-02-09 02:01:46 +0000390 continue;
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000391 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(MI);
392 if (DI == DistanceMap.end())
393 continue;
Lang Hamesa7c9dea2009-05-14 04:26:30 +0000394 if (DI->second >= Dist)
395 continue;
396
397 if (MO.isUse() && DI->second > LastUseDist) {
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000398 LastUse = DI->first;
399 LastUseDist = DI->second;
400 }
401 }
402 return LastUse;
403}
404
Evan Cheng870b8072009-03-01 02:03:43 +0000405/// isCopyToReg - Return true if the specified MI is a copy instruction or
406/// a extract_subreg instruction. It also returns the source and destination
407/// registers and whether they are physical registers by reference.
408static bool isCopyToReg(MachineInstr &MI, const TargetInstrInfo *TII,
409 unsigned &SrcReg, unsigned &DstReg,
410 bool &IsSrcPhys, bool &IsDstPhys) {
411 SrcReg = 0;
412 DstReg = 0;
Jakob Stoklund Olesen04c528a2010-07-16 04:45:42 +0000413 if (MI.isCopy()) {
414 DstReg = MI.getOperand(0).getReg();
415 SrcReg = MI.getOperand(1).getReg();
416 } else if (MI.isInsertSubreg() || MI.isSubregToReg()) {
417 DstReg = MI.getOperand(0).getReg();
418 SrcReg = MI.getOperand(2).getReg();
419 } else
420 return false;
Evan Cheng870b8072009-03-01 02:03:43 +0000421
Jakob Stoklund Olesen04c528a2010-07-16 04:45:42 +0000422 IsSrcPhys = TargetRegisterInfo::isPhysicalRegister(SrcReg);
423 IsDstPhys = TargetRegisterInfo::isPhysicalRegister(DstReg);
424 return true;
Evan Cheng870b8072009-03-01 02:03:43 +0000425}
426
Dan Gohman97121ba2009-04-08 00:15:30 +0000427/// isKilled - Test if the given register value, which is used by the given
428/// instruction, is killed by the given instruction. This looks through
429/// coalescable copies to see if the original value is potentially not killed.
430///
431/// For example, in this code:
432///
433/// %reg1034 = copy %reg1024
434/// %reg1035 = copy %reg1025<kill>
435/// %reg1036 = add %reg1034<kill>, %reg1035<kill>
436///
437/// %reg1034 is not considered to be killed, since it is copied from a
438/// register which is not killed. Treating it as not killed lets the
439/// normal heuristics commute the (two-address) add, which lets
440/// coalescing eliminate the extra copy.
441///
442static bool isKilled(MachineInstr &MI, unsigned Reg,
443 const MachineRegisterInfo *MRI,
444 const TargetInstrInfo *TII) {
445 MachineInstr *DefMI = &MI;
446 for (;;) {
447 if (!DefMI->killsRegister(Reg))
448 return false;
449 if (TargetRegisterInfo::isPhysicalRegister(Reg))
450 return true;
451 MachineRegisterInfo::def_iterator Begin = MRI->def_begin(Reg);
452 // If there are multiple defs, we can't do a simple analysis, so just
453 // go with what the kill flag says.
Chris Lattner7896c9f2009-12-03 00:50:42 +0000454 if (llvm::next(Begin) != MRI->def_end())
Dan Gohman97121ba2009-04-08 00:15:30 +0000455 return true;
456 DefMI = &*Begin;
457 bool IsSrcPhys, IsDstPhys;
458 unsigned SrcReg, DstReg;
459 // If the def is something other than a copy, then it isn't going to
460 // be coalesced, so follow the kill flag.
461 if (!isCopyToReg(*DefMI, TII, SrcReg, DstReg, IsSrcPhys, IsDstPhys))
462 return true;
463 Reg = SrcReg;
464 }
465}
466
Evan Cheng870b8072009-03-01 02:03:43 +0000467/// isTwoAddrUse - Return true if the specified MI uses the specified register
468/// as a two-address use. If so, return the destination register by reference.
469static bool isTwoAddrUse(MachineInstr &MI, unsigned Reg, unsigned &DstReg) {
Evan Chenge837dea2011-06-28 19:10:37 +0000470 const MCInstrDesc &MCID = MI.getDesc();
471 unsigned NumOps = MI.isInlineAsm()
472 ? MI.getNumOperands() : MCID.getNumOperands();
Evan Chenge6f350d2009-03-30 21:34:07 +0000473 for (unsigned i = 0; i != NumOps; ++i) {
Evan Cheng870b8072009-03-01 02:03:43 +0000474 const MachineOperand &MO = MI.getOperand(i);
475 if (!MO.isReg() || !MO.isUse() || MO.getReg() != Reg)
476 continue;
Evan Chenga24752f2009-03-19 20:30:06 +0000477 unsigned ti;
478 if (MI.isRegTiedToDefOperand(i, &ti)) {
Evan Cheng870b8072009-03-01 02:03:43 +0000479 DstReg = MI.getOperand(ti).getReg();
480 return true;
481 }
482 }
483 return false;
484}
485
Evan Cheng2a4410d2011-11-14 19:48:55 +0000486/// findLocalKill - Look for an instruction below MI in the MBB that kills the
487/// specified register. Returns null if there are any other Reg use between the
488/// instructions.
489static
490MachineInstr *findLocalKill(unsigned Reg, MachineBasicBlock *MBB,
491 MachineInstr *MI, MachineRegisterInfo *MRI,
492 DenseMap<MachineInstr*, unsigned> &DistanceMap) {
493 MachineInstr *KillMI = 0;
494 for (MachineRegisterInfo::use_nodbg_iterator
495 UI = MRI->use_nodbg_begin(Reg),
496 UE = MRI->use_nodbg_end(); UI != UE; ++UI) {
497 MachineInstr *UseMI = &*UI;
498 if (UseMI == MI || UseMI->getParent() != MBB)
499 continue;
500 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(UseMI);
501 if (DI != DistanceMap.end())
502 continue;
503 if (!UI.getOperand().isKill())
504 return 0;
Evan Cheng41e00172011-11-14 21:02:09 +0000505 if (KillMI)
506 return 0; // -O0 kill markers cannot be trusted?
Evan Cheng2a4410d2011-11-14 19:48:55 +0000507 KillMI = UseMI;
508 }
509
510 return KillMI;
511}
512
Evan Cheng870b8072009-03-01 02:03:43 +0000513/// findOnlyInterestingUse - Given a register, if has a single in-basic block
514/// use, return the use instruction if it's a copy or a two-address use.
515static
516MachineInstr *findOnlyInterestingUse(unsigned Reg, MachineBasicBlock *MBB,
517 MachineRegisterInfo *MRI,
518 const TargetInstrInfo *TII,
Evan Cheng87d696a2009-04-14 00:32:25 +0000519 bool &IsCopy,
Evan Cheng870b8072009-03-01 02:03:43 +0000520 unsigned &DstReg, bool &IsDstPhys) {
Evan Cheng1423c702010-03-03 21:18:38 +0000521 if (!MRI->hasOneNonDBGUse(Reg))
522 // None or more than one use.
Evan Cheng870b8072009-03-01 02:03:43 +0000523 return 0;
Evan Cheng1423c702010-03-03 21:18:38 +0000524 MachineInstr &UseMI = *MRI->use_nodbg_begin(Reg);
Evan Cheng870b8072009-03-01 02:03:43 +0000525 if (UseMI.getParent() != MBB)
526 return 0;
527 unsigned SrcReg;
528 bool IsSrcPhys;
Evan Cheng87d696a2009-04-14 00:32:25 +0000529 if (isCopyToReg(UseMI, TII, SrcReg, DstReg, IsSrcPhys, IsDstPhys)) {
530 IsCopy = true;
Evan Cheng870b8072009-03-01 02:03:43 +0000531 return &UseMI;
Evan Cheng87d696a2009-04-14 00:32:25 +0000532 }
Evan Cheng870b8072009-03-01 02:03:43 +0000533 IsDstPhys = false;
Evan Cheng87d696a2009-04-14 00:32:25 +0000534 if (isTwoAddrUse(UseMI, Reg, DstReg)) {
535 IsDstPhys = TargetRegisterInfo::isPhysicalRegister(DstReg);
Evan Cheng870b8072009-03-01 02:03:43 +0000536 return &UseMI;
Evan Cheng87d696a2009-04-14 00:32:25 +0000537 }
Evan Cheng870b8072009-03-01 02:03:43 +0000538 return 0;
539}
540
541/// getMappedReg - Return the physical register the specified virtual register
542/// might be mapped to.
543static unsigned
544getMappedReg(unsigned Reg, DenseMap<unsigned, unsigned> &RegMap) {
545 while (TargetRegisterInfo::isVirtualRegister(Reg)) {
546 DenseMap<unsigned, unsigned>::iterator SI = RegMap.find(Reg);
547 if (SI == RegMap.end())
548 return 0;
549 Reg = SI->second;
550 }
551 if (TargetRegisterInfo::isPhysicalRegister(Reg))
552 return Reg;
553 return 0;
554}
555
556/// regsAreCompatible - Return true if the two registers are equal or aliased.
557///
558static bool
559regsAreCompatible(unsigned RegA, unsigned RegB, const TargetRegisterInfo *TRI) {
560 if (RegA == RegB)
561 return true;
562 if (!RegA || !RegB)
563 return false;
564 return TRI->regsOverlap(RegA, RegB);
565}
566
567
Evan Chengd498c8f2009-01-25 03:53:59 +0000568/// isProfitableToReMat - Return true if it's potentially profitable to commute
569/// the two-address instruction that's being processed.
570bool
571TwoAddressInstructionPass::isProfitableToCommute(unsigned regB, unsigned regC,
Evan Cheng870b8072009-03-01 02:03:43 +0000572 MachineInstr *MI, MachineBasicBlock *MBB,
573 unsigned Dist) {
Evan Chengd498c8f2009-01-25 03:53:59 +0000574 // Determine if it's profitable to commute this two address instruction. In
575 // general, we want no uses between this instruction and the definition of
576 // the two-address register.
577 // e.g.
578 // %reg1028<def> = EXTRACT_SUBREG %reg1027<kill>, 1
579 // %reg1029<def> = MOV8rr %reg1028
580 // %reg1029<def> = SHR8ri %reg1029, 7, %EFLAGS<imp-def,dead>
581 // insert => %reg1030<def> = MOV8rr %reg1028
582 // %reg1030<def> = ADD8rr %reg1028<kill>, %reg1029<kill>, %EFLAGS<imp-def,dead>
583 // In this case, it might not be possible to coalesce the second MOV8rr
584 // instruction if the first one is coalesced. So it would be profitable to
585 // commute it:
586 // %reg1028<def> = EXTRACT_SUBREG %reg1027<kill>, 1
587 // %reg1029<def> = MOV8rr %reg1028
588 // %reg1029<def> = SHR8ri %reg1029, 7, %EFLAGS<imp-def,dead>
589 // insert => %reg1030<def> = MOV8rr %reg1029
590 // %reg1030<def> = ADD8rr %reg1029<kill>, %reg1028<kill>, %EFLAGS<imp-def,dead>
591
592 if (!MI->killsRegister(regC))
593 return false;
594
595 // Ok, we have something like:
596 // %reg1030<def> = ADD8rr %reg1028<kill>, %reg1029<kill>, %EFLAGS<imp-def,dead>
597 // let's see if it's worth commuting it.
598
Evan Cheng870b8072009-03-01 02:03:43 +0000599 // Look for situations like this:
600 // %reg1024<def> = MOV r1
601 // %reg1025<def> = MOV r0
602 // %reg1026<def> = ADD %reg1024, %reg1025
603 // r0 = MOV %reg1026
604 // Commute the ADD to hopefully eliminate an otherwise unavoidable copy.
605 unsigned FromRegB = getMappedReg(regB, SrcRegMap);
606 unsigned FromRegC = getMappedReg(regC, SrcRegMap);
607 unsigned ToRegB = getMappedReg(regB, DstRegMap);
608 unsigned ToRegC = getMappedReg(regC, DstRegMap);
Evan Cheng4d96c632011-02-10 02:20:55 +0000609 if ((FromRegB && ToRegB && !regsAreCompatible(FromRegB, ToRegB, TRI)) &&
Evan Chengbbc726d2010-12-14 21:34:53 +0000610 ((!FromRegC && !ToRegC) ||
611 regsAreCompatible(FromRegB, ToRegC, TRI) ||
Evan Cheng870b8072009-03-01 02:03:43 +0000612 regsAreCompatible(FromRegC, ToRegB, TRI)))
613 return true;
614
Evan Chengd498c8f2009-01-25 03:53:59 +0000615 // If there is a use of regC between its last def (could be livein) and this
616 // instruction, then bail.
617 unsigned LastDefC = 0;
Evan Cheng870b8072009-03-01 02:03:43 +0000618 if (!NoUseAfterLastDef(regC, MBB, Dist, LastDefC))
Evan Chengd498c8f2009-01-25 03:53:59 +0000619 return false;
620
621 // If there is a use of regB between its last def (could be livein) and this
622 // instruction, then go ahead and make this transformation.
623 unsigned LastDefB = 0;
Evan Cheng870b8072009-03-01 02:03:43 +0000624 if (!NoUseAfterLastDef(regB, MBB, Dist, LastDefB))
Evan Chengd498c8f2009-01-25 03:53:59 +0000625 return true;
626
627 // Since there are no intervening uses for both registers, then commute
628 // if the def of regC is closer. Its live interval is shorter.
629 return LastDefB && LastDefC && LastDefC > LastDefB;
630}
631
Evan Cheng81913712009-01-23 23:27:33 +0000632/// CommuteInstruction - Commute a two-address instruction and update the basic
633/// block, distance map, and live variables if needed. Return true if it is
634/// successful.
635bool
636TwoAddressInstructionPass::CommuteInstruction(MachineBasicBlock::iterator &mi,
Evan Cheng870b8072009-03-01 02:03:43 +0000637 MachineFunction::iterator &mbbi,
638 unsigned RegB, unsigned RegC, unsigned Dist) {
Evan Cheng81913712009-01-23 23:27:33 +0000639 MachineInstr *MI = mi;
David Greeneeb00b182010-01-05 01:24:21 +0000640 DEBUG(dbgs() << "2addr: COMMUTING : " << *MI);
Evan Cheng81913712009-01-23 23:27:33 +0000641 MachineInstr *NewMI = TII->commuteInstruction(MI);
642
643 if (NewMI == 0) {
David Greeneeb00b182010-01-05 01:24:21 +0000644 DEBUG(dbgs() << "2addr: COMMUTING FAILED!\n");
Evan Cheng81913712009-01-23 23:27:33 +0000645 return false;
646 }
647
David Greeneeb00b182010-01-05 01:24:21 +0000648 DEBUG(dbgs() << "2addr: COMMUTED TO: " << *NewMI);
Evan Cheng81913712009-01-23 23:27:33 +0000649 // If the instruction changed to commute it, update livevar.
650 if (NewMI != MI) {
651 if (LV)
652 // Update live variables
653 LV->replaceKillInstruction(RegC, MI, NewMI);
654
655 mbbi->insert(mi, NewMI); // Insert the new inst
656 mbbi->erase(mi); // Nuke the old inst.
657 mi = NewMI;
658 DistanceMap.insert(std::make_pair(NewMI, Dist));
659 }
Evan Cheng870b8072009-03-01 02:03:43 +0000660
661 // Update source register map.
662 unsigned FromRegC = getMappedReg(RegC, SrcRegMap);
663 if (FromRegC) {
664 unsigned RegA = MI->getOperand(0).getReg();
665 SrcRegMap[RegA] = FromRegC;
666 }
667
Evan Cheng81913712009-01-23 23:27:33 +0000668 return true;
669}
670
Evan Chenge6f350d2009-03-30 21:34:07 +0000671/// isProfitableToConv3Addr - Return true if it is profitable to convert the
672/// given 2-address instruction to a 3-address one.
673bool
Evan Chengf06e6c22011-03-02 01:08:17 +0000674TwoAddressInstructionPass::isProfitableToConv3Addr(unsigned RegA,unsigned RegB){
Evan Chenge6f350d2009-03-30 21:34:07 +0000675 // Look for situations like this:
676 // %reg1024<def> = MOV r1
677 // %reg1025<def> = MOV r0
678 // %reg1026<def> = ADD %reg1024, %reg1025
679 // r2 = MOV %reg1026
680 // Turn ADD into a 3-address instruction to avoid a copy.
Evan Chengf06e6c22011-03-02 01:08:17 +0000681 unsigned FromRegB = getMappedReg(RegB, SrcRegMap);
682 if (!FromRegB)
683 return false;
Evan Chenge6f350d2009-03-30 21:34:07 +0000684 unsigned ToRegA = getMappedReg(RegA, DstRegMap);
Evan Chengf06e6c22011-03-02 01:08:17 +0000685 return (ToRegA && !regsAreCompatible(FromRegB, ToRegA, TRI));
Evan Chenge6f350d2009-03-30 21:34:07 +0000686}
687
688/// ConvertInstTo3Addr - Convert the specified two-address instruction into a
689/// three address one. Return true if this transformation was successful.
690bool
691TwoAddressInstructionPass::ConvertInstTo3Addr(MachineBasicBlock::iterator &mi,
692 MachineBasicBlock::iterator &nmi,
693 MachineFunction::iterator &mbbi,
Evan Cheng4d96c632011-02-10 02:20:55 +0000694 unsigned RegA, unsigned RegB,
695 unsigned Dist) {
Evan Chenge6f350d2009-03-30 21:34:07 +0000696 MachineInstr *NewMI = TII->convertToThreeAddress(mbbi, mi, LV);
697 if (NewMI) {
David Greeneeb00b182010-01-05 01:24:21 +0000698 DEBUG(dbgs() << "2addr: CONVERTING 2-ADDR: " << *mi);
699 DEBUG(dbgs() << "2addr: TO 3-ADDR: " << *NewMI);
Evan Chenge6f350d2009-03-30 21:34:07 +0000700 bool Sunk = false;
701
702 if (NewMI->findRegisterUseOperand(RegB, false, TRI))
703 // FIXME: Temporary workaround. If the new instruction doesn't
704 // uses RegB, convertToThreeAddress must have created more
705 // then one instruction.
706 Sunk = Sink3AddrInstruction(mbbi, NewMI, RegB, mi);
707
708 mbbi->erase(mi); // Nuke the old inst.
709
710 if (!Sunk) {
711 DistanceMap.insert(std::make_pair(NewMI, Dist));
712 mi = NewMI;
Chris Lattner7896c9f2009-12-03 00:50:42 +0000713 nmi = llvm::next(mi);
Evan Chenge6f350d2009-03-30 21:34:07 +0000714 }
Evan Cheng4d96c632011-02-10 02:20:55 +0000715
716 // Update source and destination register maps.
717 SrcRegMap.erase(RegA);
718 DstRegMap.erase(RegB);
Evan Chenge6f350d2009-03-30 21:34:07 +0000719 return true;
720 }
721
722 return false;
723}
724
Evan Chengf06e6c22011-03-02 01:08:17 +0000725/// ScanUses - Scan forward recursively for only uses, update maps if the use
726/// is a copy or a two-address instruction.
727void
728TwoAddressInstructionPass::ScanUses(unsigned DstReg, MachineBasicBlock *MBB,
729 SmallPtrSet<MachineInstr*, 8> &Processed) {
730 SmallVector<unsigned, 4> VirtRegPairs;
731 bool IsDstPhys;
732 bool IsCopy = false;
733 unsigned NewReg = 0;
734 unsigned Reg = DstReg;
735 while (MachineInstr *UseMI = findOnlyInterestingUse(Reg, MBB, MRI, TII,IsCopy,
736 NewReg, IsDstPhys)) {
737 if (IsCopy && !Processed.insert(UseMI))
738 break;
739
740 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(UseMI);
741 if (DI != DistanceMap.end())
742 // Earlier in the same MBB.Reached via a back edge.
743 break;
744
745 if (IsDstPhys) {
746 VirtRegPairs.push_back(NewReg);
747 break;
748 }
749 bool isNew = SrcRegMap.insert(std::make_pair(NewReg, Reg)).second;
750 if (!isNew)
751 assert(SrcRegMap[NewReg] == Reg && "Can't map to two src registers!");
752 VirtRegPairs.push_back(NewReg);
753 Reg = NewReg;
754 }
755
756 if (!VirtRegPairs.empty()) {
757 unsigned ToReg = VirtRegPairs.back();
758 VirtRegPairs.pop_back();
759 while (!VirtRegPairs.empty()) {
760 unsigned FromReg = VirtRegPairs.back();
761 VirtRegPairs.pop_back();
762 bool isNew = DstRegMap.insert(std::make_pair(FromReg, ToReg)).second;
763 if (!isNew)
764 assert(DstRegMap[FromReg] == ToReg &&"Can't map to two dst registers!");
765 ToReg = FromReg;
766 }
767 bool isNew = DstRegMap.insert(std::make_pair(DstReg, ToReg)).second;
768 if (!isNew)
769 assert(DstRegMap[DstReg] == ToReg && "Can't map to two dst registers!");
770 }
771}
772
Evan Cheng870b8072009-03-01 02:03:43 +0000773/// ProcessCopy - If the specified instruction is not yet processed, process it
774/// if it's a copy. For a copy instruction, we find the physical registers the
775/// source and destination registers might be mapped to. These are kept in
776/// point-to maps used to determine future optimizations. e.g.
777/// v1024 = mov r0
778/// v1025 = mov r1
779/// v1026 = add v1024, v1025
780/// r1 = mov r1026
781/// If 'add' is a two-address instruction, v1024, v1026 are both potentially
782/// coalesced to r0 (from the input side). v1025 is mapped to r1. v1026 is
783/// potentially joined with r1 on the output side. It's worthwhile to commute
784/// 'add' to eliminate a copy.
785void TwoAddressInstructionPass::ProcessCopy(MachineInstr *MI,
786 MachineBasicBlock *MBB,
787 SmallPtrSet<MachineInstr*, 8> &Processed) {
788 if (Processed.count(MI))
789 return;
790
791 bool IsSrcPhys, IsDstPhys;
792 unsigned SrcReg, DstReg;
793 if (!isCopyToReg(*MI, TII, SrcReg, DstReg, IsSrcPhys, IsDstPhys))
794 return;
795
796 if (IsDstPhys && !IsSrcPhys)
797 DstRegMap.insert(std::make_pair(SrcReg, DstReg));
798 else if (!IsDstPhys && IsSrcPhys) {
Evan Cheng3005ed62009-04-13 20:04:24 +0000799 bool isNew = SrcRegMap.insert(std::make_pair(DstReg, SrcReg)).second;
800 if (!isNew)
801 assert(SrcRegMap[DstReg] == SrcReg &&
802 "Can't map to two src physical registers!");
Evan Cheng870b8072009-03-01 02:03:43 +0000803
Evan Chengf06e6c22011-03-02 01:08:17 +0000804 ScanUses(DstReg, MBB, Processed);
Evan Cheng870b8072009-03-01 02:03:43 +0000805 }
806
807 Processed.insert(MI);
Evan Chengf06e6c22011-03-02 01:08:17 +0000808 return;
Evan Cheng870b8072009-03-01 02:03:43 +0000809}
810
Evan Cheng28c7ce32009-02-21 03:14:25 +0000811/// isSafeToDelete - If the specified instruction does not produce any side
812/// effects and all of its defs are dead, then it's safe to delete.
Jakob Stoklund Olesen0b25ae12009-11-18 21:33:35 +0000813static bool isSafeToDelete(MachineInstr *MI,
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000814 const TargetInstrInfo *TII,
815 SmallVector<unsigned, 4> &Kills) {
Evan Chenge837dea2011-06-28 19:10:37 +0000816 const MCInstrDesc &MCID = MI->getDesc();
817 if (MCID.mayStore() || MCID.isCall())
Evan Cheng28c7ce32009-02-21 03:14:25 +0000818 return false;
Evan Chenge837dea2011-06-28 19:10:37 +0000819 if (MCID.isTerminator() || MI->hasUnmodeledSideEffects())
Evan Cheng28c7ce32009-02-21 03:14:25 +0000820 return false;
821
822 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
823 MachineOperand &MO = MI->getOperand(i);
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000824 if (!MO.isReg())
Evan Cheng28c7ce32009-02-21 03:14:25 +0000825 continue;
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000826 if (MO.isDef() && !MO.isDead())
Evan Cheng28c7ce32009-02-21 03:14:25 +0000827 return false;
Jakob Stoklund Olesen0b25ae12009-11-18 21:33:35 +0000828 if (MO.isUse() && MO.isKill())
Evan Chenge9ccb3a2009-04-28 02:12:36 +0000829 Kills.push_back(MO.getReg());
Evan Cheng28c7ce32009-02-21 03:14:25 +0000830 }
Evan Cheng28c7ce32009-02-21 03:14:25 +0000831 return true;
832}
833
Bob Wilson326f4382009-09-01 22:51:08 +0000834/// canUpdateDeletedKills - Check if all the registers listed in Kills are
835/// killed by instructions in MBB preceding the current instruction at
836/// position Dist. If so, return true and record information about the
837/// preceding kills in NewKills.
838bool TwoAddressInstructionPass::
839canUpdateDeletedKills(SmallVector<unsigned, 4> &Kills,
840 SmallVector<NewKill, 4> &NewKills,
841 MachineBasicBlock *MBB, unsigned Dist) {
842 while (!Kills.empty()) {
843 unsigned Kill = Kills.back();
844 Kills.pop_back();
845 if (TargetRegisterInfo::isPhysicalRegister(Kill))
846 return false;
847
848 MachineInstr *LastKill = FindLastUseInMBB(Kill, MBB, Dist);
849 if (!LastKill)
850 return false;
851
Evan Cheng1015ba72010-05-21 20:53:24 +0000852 bool isModRef = LastKill->definesRegister(Kill);
Bob Wilson326f4382009-09-01 22:51:08 +0000853 NewKills.push_back(std::make_pair(std::make_pair(Kill, isModRef),
854 LastKill));
855 }
856 return true;
857}
858
859/// DeleteUnusedInstr - If an instruction with a tied register operand can
860/// be safely deleted, just delete it.
861bool
862TwoAddressInstructionPass::DeleteUnusedInstr(MachineBasicBlock::iterator &mi,
863 MachineBasicBlock::iterator &nmi,
864 MachineFunction::iterator &mbbi,
Bob Wilson326f4382009-09-01 22:51:08 +0000865 unsigned Dist) {
866 // Check if the instruction has no side effects and if all its defs are dead.
867 SmallVector<unsigned, 4> Kills;
Jakob Stoklund Olesen0b25ae12009-11-18 21:33:35 +0000868 if (!isSafeToDelete(mi, TII, Kills))
Bob Wilson326f4382009-09-01 22:51:08 +0000869 return false;
870
871 // If this instruction kills some virtual registers, we need to
872 // update the kill information. If it's not possible to do so,
873 // then bail out.
874 SmallVector<NewKill, 4> NewKills;
875 if (!canUpdateDeletedKills(Kills, NewKills, &*mbbi, Dist))
876 return false;
877
878 if (LV) {
879 while (!NewKills.empty()) {
880 MachineInstr *NewKill = NewKills.back().second;
881 unsigned Kill = NewKills.back().first.first;
882 bool isDead = NewKills.back().first.second;
883 NewKills.pop_back();
884 if (LV->removeVirtualRegisterKilled(Kill, mi)) {
885 if (isDead)
886 LV->addVirtualRegisterDead(Kill, NewKill);
887 else
888 LV->addVirtualRegisterKilled(Kill, NewKill);
889 }
890 }
Bob Wilson326f4382009-09-01 22:51:08 +0000891 }
892
893 mbbi->erase(mi); // Nuke the old inst.
894 mi = nmi;
895 return true;
896}
897
Evan Cheng2a4410d2011-11-14 19:48:55 +0000898/// RescheduleMIBelowKill - If there is one more local instruction that reads
899/// 'Reg' and it kills 'Reg, consider moving the instruction below the kill
900/// instruction in order to eliminate the need for the copy.
901bool
902TwoAddressInstructionPass::RescheduleMIBelowKill(MachineBasicBlock *MBB,
903 MachineBasicBlock::iterator &mi,
904 MachineBasicBlock::iterator &nmi,
905 unsigned Reg) {
906 MachineInstr *MI = &*mi;
907 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(MI);
908 if (DI == DistanceMap.end())
909 // Must be created from unfolded load. Don't waste time trying this.
910 return false;
911
912 MachineInstr *KillMI = findLocalKill(Reg, MBB, mi, MRI, DistanceMap);
913 if (!KillMI || KillMI->isCopy() || KillMI->isCopyLike())
914 // Don't mess with copies, they may be coalesced later.
915 return false;
916
917 const MCInstrDesc &MCID = KillMI->getDesc();
918 if (MCID.hasUnmodeledSideEffects() || MCID.isCall() || MCID.isBranch() ||
919 MCID.isTerminator())
920 // Don't move pass calls, etc.
921 return false;
922
923 unsigned DstReg;
924 if (isTwoAddrUse(*KillMI, Reg, DstReg))
925 return false;
926
Evan Chengf1784182011-11-15 06:26:51 +0000927 bool SeenStore = true;
Evan Cheng2a4410d2011-11-14 19:48:55 +0000928 if (!MI->isSafeToMove(TII, AA, SeenStore))
929 return false;
930
931 if (TII->getInstrLatency(InstrItins, MI) > 1)
932 // FIXME: Needs more sophisticated heuristics.
933 return false;
934
935 SmallSet<unsigned, 2> Uses;
936 SmallSet<unsigned, 2> Defs;
937 for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
938 const MachineOperand &MO = MI->getOperand(i);
939 if (!MO.isReg())
940 continue;
941 unsigned MOReg = MO.getReg();
942 if (!MOReg)
943 continue;
944 if (MO.isDef())
945 Defs.insert(MOReg);
946 else
947 Uses.insert(MOReg);
948 }
949
950 // Move the copies connected to MI down as well.
951 MachineBasicBlock::iterator From = MI;
952 MachineBasicBlock::iterator To = llvm::next(From);
953 while (To->isCopy() && Defs.count(To->getOperand(1).getReg())) {
954 Defs.insert(To->getOperand(0).getReg());
955 ++To;
956 }
957
958 // Check if the reschedule will not break depedencies.
959 unsigned NumVisited = 0;
960 MachineBasicBlock::iterator KillPos = KillMI;
961 ++KillPos;
962 for (MachineBasicBlock::iterator I = To; I != KillPos; ++I) {
963 MachineInstr *OtherMI = I;
964 // DBG_VALUE cannot be counted against the limit.
965 if (OtherMI->isDebugValue())
966 continue;
967 if (NumVisited > 10) // FIXME: Arbitrary limit to reduce compile time cost.
968 return false;
969 ++NumVisited;
970 const MCInstrDesc &OMCID = OtherMI->getDesc();
971 if (OMCID.hasUnmodeledSideEffects() || OMCID.isCall() || OMCID.isBranch() ||
972 OMCID.isTerminator())
973 // Don't move pass calls, etc.
974 return false;
975 for (unsigned i = 0, e = OtherMI->getNumOperands(); i != e; ++i) {
976 const MachineOperand &MO = OtherMI->getOperand(i);
977 if (!MO.isReg())
978 continue;
979 unsigned MOReg = MO.getReg();
980 if (!MOReg)
981 continue;
982 if (MO.isDef()) {
983 if (Uses.count(MOReg))
984 // Physical register use would be clobbered.
985 return false;
986 if (!MO.isDead() && Defs.count(MOReg))
987 // May clobber a physical register def.
988 // FIXME: This may be too conservative. It's ok if the instruction
989 // is sunken completely below the use.
990 return false;
991 } else {
992 if (Defs.count(MOReg))
993 return false;
994 if (MOReg != Reg && MO.isKill() && Uses.count(MOReg))
995 // Don't want to extend other live ranges and update kills.
996 return false;
997 }
998 }
999 }
1000
1001 // Move debug info as well.
Evan Cheng8aee7d82011-11-14 21:11:15 +00001002 while (From != MBB->begin() && llvm::prior(From)->isDebugValue())
1003 --From;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001004
1005 // Copies following MI may have been moved as well.
1006 nmi = To;
1007 MBB->splice(KillPos, MBB, From, To);
1008 DistanceMap.erase(DI);
1009
1010 if (LV) {
1011 // Update live variables
1012 LV->removeVirtualRegisterKilled(Reg, KillMI);
1013 LV->addVirtualRegisterKilled(Reg, MI);
1014 } else {
1015 for (unsigned i = 0, e = KillMI->getNumOperands(); i != e; ++i) {
1016 MachineOperand &MO = KillMI->getOperand(i);
1017 if (!MO.isReg() || !MO.isUse() || MO.getReg() != Reg)
1018 continue;
1019 MO.setIsKill(false);
1020 }
1021 MI->addRegisterKilled(Reg, 0);
1022 }
1023
1024 return true;
1025}
1026
1027/// isDefTooClose - Return true if the re-scheduling will put the given
1028/// instruction too close to the defs of its register dependencies.
1029bool TwoAddressInstructionPass::isDefTooClose(unsigned Reg, unsigned Dist,
1030 MachineInstr *MI,
1031 MachineBasicBlock *MBB) {
1032 for (MachineRegisterInfo::def_iterator DI = MRI->def_begin(Reg),
1033 DE = MRI->def_end(); DI != DE; ++DI) {
1034 MachineInstr *DefMI = &*DI;
1035 if (DefMI->getParent() != MBB || DefMI->isCopy() || DefMI->isCopyLike())
1036 continue;
1037 if (DefMI == MI)
1038 return true; // MI is defining something KillMI uses
1039 DenseMap<MachineInstr*, unsigned>::iterator DDI = DistanceMap.find(DefMI);
1040 if (DDI == DistanceMap.end())
1041 return true; // Below MI
1042 unsigned DefDist = DDI->second;
1043 assert(Dist > DefDist && "Visited def already?");
1044 if (TII->getInstrLatency(InstrItins, DefMI) > (int)(Dist - DefDist))
1045 return true;
1046 }
1047 return false;
1048}
1049
1050/// RescheduleKillAboveMI - If there is one more local instruction that reads
1051/// 'Reg' and it kills 'Reg, consider moving the kill instruction above the
1052/// current two-address instruction in order to eliminate the need for the
1053/// copy.
1054bool
1055TwoAddressInstructionPass::RescheduleKillAboveMI(MachineBasicBlock *MBB,
1056 MachineBasicBlock::iterator &mi,
1057 MachineBasicBlock::iterator &nmi,
1058 unsigned Reg) {
1059 MachineInstr *MI = &*mi;
1060 DenseMap<MachineInstr*, unsigned>::iterator DI = DistanceMap.find(MI);
1061 if (DI == DistanceMap.end())
1062 // Must be created from unfolded load. Don't waste time trying this.
1063 return false;
1064
1065 MachineInstr *KillMI = findLocalKill(Reg, MBB, mi, MRI, DistanceMap);
1066 if (!KillMI || KillMI->isCopy() || KillMI->isCopyLike())
1067 // Don't mess with copies, they may be coalesced later.
1068 return false;
1069
1070 unsigned DstReg;
1071 if (isTwoAddrUse(*KillMI, Reg, DstReg))
1072 return false;
1073
Evan Chengf1784182011-11-15 06:26:51 +00001074 bool SeenStore = true;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001075 if (!KillMI->isSafeToMove(TII, AA, SeenStore))
1076 return false;
1077
1078 SmallSet<unsigned, 2> Uses;
1079 SmallSet<unsigned, 2> Kills;
1080 SmallSet<unsigned, 2> Defs;
1081 SmallSet<unsigned, 2> LiveDefs;
1082 for (unsigned i = 0, e = KillMI->getNumOperands(); i != e; ++i) {
1083 const MachineOperand &MO = KillMI->getOperand(i);
1084 if (!MO.isReg())
1085 continue;
1086 unsigned MOReg = MO.getReg();
1087 if (MO.isUse()) {
1088 if (!MOReg)
1089 continue;
1090 if (isDefTooClose(MOReg, DI->second, MI, MBB))
1091 return false;
1092 Uses.insert(MOReg);
1093 if (MO.isKill() && MOReg != Reg)
1094 Kills.insert(MOReg);
1095 } else if (TargetRegisterInfo::isPhysicalRegister(MOReg)) {
1096 Defs.insert(MOReg);
1097 if (!MO.isDead())
1098 LiveDefs.insert(MOReg);
1099 }
1100 }
1101
1102 // Check if the reschedule will not break depedencies.
1103 unsigned NumVisited = 0;
1104 MachineBasicBlock::iterator KillPos = KillMI;
1105 for (MachineBasicBlock::iterator I = mi; I != KillPos; ++I) {
1106 MachineInstr *OtherMI = I;
1107 // DBG_VALUE cannot be counted against the limit.
1108 if (OtherMI->isDebugValue())
1109 continue;
1110 if (NumVisited > 10) // FIXME: Arbitrary limit to reduce compile time cost.
1111 return false;
1112 ++NumVisited;
1113 const MCInstrDesc &MCID = OtherMI->getDesc();
1114 if (MCID.hasUnmodeledSideEffects() || MCID.isCall() || MCID.isBranch() ||
1115 MCID.isTerminator())
1116 // Don't move pass calls, etc.
1117 return false;
Evan Chengae7db7a2011-11-16 03:05:12 +00001118 SmallVector<unsigned, 2> OtherDefs;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001119 for (unsigned i = 0, e = OtherMI->getNumOperands(); i != e; ++i) {
1120 const MachineOperand &MO = OtherMI->getOperand(i);
1121 if (!MO.isReg())
1122 continue;
1123 unsigned MOReg = MO.getReg();
1124 if (!MOReg)
1125 continue;
1126 if (MO.isUse()) {
1127 if (Defs.count(MOReg))
1128 // Moving KillMI can clobber the physical register if the def has
1129 // not been seen.
1130 return false;
1131 if (Kills.count(MOReg))
1132 // Don't want to extend other live ranges and update kills.
1133 return false;
1134 } else {
Evan Chengae7db7a2011-11-16 03:05:12 +00001135 OtherDefs.push_back(MOReg);
Evan Cheng2a4410d2011-11-14 19:48:55 +00001136 }
1137 }
Evan Chengae7db7a2011-11-16 03:05:12 +00001138
1139 for (unsigned i = 0, e = OtherDefs.size(); i != e; ++i) {
1140 unsigned MOReg = OtherDefs[i];
1141 if (Uses.count(MOReg))
1142 return false;
1143 if (TargetRegisterInfo::isPhysicalRegister(MOReg) &&
1144 LiveDefs.count(MOReg))
1145 return false;
1146 // Physical register def is seen.
1147 Defs.erase(MOReg);
1148 }
Evan Cheng2a4410d2011-11-14 19:48:55 +00001149 }
1150
1151 // Move the old kill above MI, don't forget to move debug info as well.
1152 MachineBasicBlock::iterator InsertPos = mi;
Evan Cheng8aee7d82011-11-14 21:11:15 +00001153 while (InsertPos != MBB->begin() && llvm::prior(InsertPos)->isDebugValue())
1154 --InsertPos;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001155 MachineBasicBlock::iterator From = KillMI;
1156 MachineBasicBlock::iterator To = llvm::next(From);
1157 while (llvm::prior(From)->isDebugValue())
1158 --From;
1159 MBB->splice(InsertPos, MBB, From, To);
1160
1161 nmi = llvm::prior(mi); // Backtrack so we process the moved instruction.
1162 DistanceMap.erase(DI);
1163
1164 if (LV) {
1165 // Update live variables
1166 LV->removeVirtualRegisterKilled(Reg, KillMI);
1167 LV->addVirtualRegisterKilled(Reg, MI);
1168 } else {
1169 for (unsigned i = 0, e = KillMI->getNumOperands(); i != e; ++i) {
1170 MachineOperand &MO = KillMI->getOperand(i);
1171 if (!MO.isReg() || !MO.isUse() || MO.getReg() != Reg)
1172 continue;
1173 MO.setIsKill(false);
1174 }
1175 MI->addRegisterKilled(Reg, 0);
1176 }
1177 return true;
1178}
1179
Bob Wilsoncc80df92009-09-03 20:58:42 +00001180/// TryInstructionTransform - For the case where an instruction has a single
1181/// pair of tied register operands, attempt some transformations that may
1182/// either eliminate the tied operands or improve the opportunities for
1183/// coalescing away the register copy. Returns true if the tied operands
1184/// are eliminated altogether.
1185bool TwoAddressInstructionPass::
1186TryInstructionTransform(MachineBasicBlock::iterator &mi,
1187 MachineBasicBlock::iterator &nmi,
1188 MachineFunction::iterator &mbbi,
Evan Chengf06e6c22011-03-02 01:08:17 +00001189 unsigned SrcIdx, unsigned DstIdx, unsigned Dist,
1190 SmallPtrSet<MachineInstr*, 8> &Processed) {
Evan Cheng2a4410d2011-11-14 19:48:55 +00001191 MachineInstr &MI = *mi;
1192 const MCInstrDesc &MCID = MI.getDesc();
1193 unsigned regA = MI.getOperand(DstIdx).getReg();
1194 unsigned regB = MI.getOperand(SrcIdx).getReg();
Bob Wilsoncc80df92009-09-03 20:58:42 +00001195
1196 assert(TargetRegisterInfo::isVirtualRegister(regB) &&
1197 "cannot make instruction into two-address form");
1198
1199 // If regA is dead and the instruction can be deleted, just delete
1200 // it so it doesn't clobber regB.
Evan Cheng2a4410d2011-11-14 19:48:55 +00001201 bool regBKilled = isKilled(MI, regB, MRI, TII);
1202 if (!regBKilled && MI.getOperand(DstIdx).isDead() &&
Jakob Stoklund Olesen0b25ae12009-11-18 21:33:35 +00001203 DeleteUnusedInstr(mi, nmi, mbbi, Dist)) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001204 ++NumDeletes;
1205 return true; // Done with this instruction.
1206 }
1207
1208 // Check if it is profitable to commute the operands.
1209 unsigned SrcOp1, SrcOp2;
1210 unsigned regC = 0;
1211 unsigned regCIdx = ~0U;
1212 bool TryCommute = false;
1213 bool AggressiveCommute = false;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001214 if (MCID.isCommutable() && MI.getNumOperands() >= 3 &&
1215 TII->findCommutedOpIndices(&MI, SrcOp1, SrcOp2)) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001216 if (SrcIdx == SrcOp1)
1217 regCIdx = SrcOp2;
1218 else if (SrcIdx == SrcOp2)
1219 regCIdx = SrcOp1;
1220
1221 if (regCIdx != ~0U) {
Evan Cheng2a4410d2011-11-14 19:48:55 +00001222 regC = MI.getOperand(regCIdx).getReg();
1223 if (!regBKilled && isKilled(MI, regC, MRI, TII))
Bob Wilsoncc80df92009-09-03 20:58:42 +00001224 // If C dies but B does not, swap the B and C operands.
1225 // This makes the live ranges of A and C joinable.
1226 TryCommute = true;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001227 else if (isProfitableToCommute(regB, regC, &MI, mbbi, Dist)) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001228 TryCommute = true;
1229 AggressiveCommute = true;
1230 }
1231 }
1232 }
1233
1234 // If it's profitable to commute, try to do so.
1235 if (TryCommute && CommuteInstruction(mi, mbbi, regB, regC, Dist)) {
1236 ++NumCommuted;
1237 if (AggressiveCommute)
1238 ++NumAggrCommuted;
1239 return false;
1240 }
1241
Evan Cheng2a4410d2011-11-14 19:48:55 +00001242 // If there is one more use of regB later in the same MBB, consider
1243 // re-schedule this MI below it.
1244 if (RescheduleMIBelowKill(mbbi, mi, nmi, regB)) {
1245 ++NumReSchedDowns;
1246 return true;
1247 }
1248
Evan Chengf06e6c22011-03-02 01:08:17 +00001249 if (TargetRegisterInfo::isVirtualRegister(regA))
1250 ScanUses(regA, &*mbbi, Processed);
1251
Evan Chenge837dea2011-06-28 19:10:37 +00001252 if (MCID.isConvertibleTo3Addr()) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001253 // This instruction is potentially convertible to a true
1254 // three-address instruction. Check if it is profitable.
Evan Chengf06e6c22011-03-02 01:08:17 +00001255 if (!regBKilled || isProfitableToConv3Addr(regA, regB)) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001256 // Try to convert it.
Evan Cheng4d96c632011-02-10 02:20:55 +00001257 if (ConvertInstTo3Addr(mi, nmi, mbbi, regA, regB, Dist)) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001258 ++NumConvertedTo3Addr;
1259 return true; // Done with this instruction.
1260 }
1261 }
1262 }
Dan Gohman584fedf2010-06-21 22:17:20 +00001263
Evan Cheng2a4410d2011-11-14 19:48:55 +00001264 // If there is one more use of regB later in the same MBB, consider
1265 // re-schedule it before this MI if it's legal.
1266 if (RescheduleKillAboveMI(mbbi, mi, nmi, regB)) {
1267 ++NumReSchedUps;
1268 return true;
1269 }
1270
Dan Gohman584fedf2010-06-21 22:17:20 +00001271 // If this is an instruction with a load folded into it, try unfolding
1272 // the load, e.g. avoid this:
1273 // movq %rdx, %rcx
1274 // addq (%rax), %rcx
1275 // in favor of this:
1276 // movq (%rax), %rcx
1277 // addq %rdx, %rcx
1278 // because it's preferable to schedule a load than a register copy.
Evan Chenge837dea2011-06-28 19:10:37 +00001279 if (MCID.mayLoad() && !regBKilled) {
Dan Gohman584fedf2010-06-21 22:17:20 +00001280 // Determine if a load can be unfolded.
1281 unsigned LoadRegIndex;
1282 unsigned NewOpc =
Evan Cheng2a4410d2011-11-14 19:48:55 +00001283 TII->getOpcodeAfterMemoryUnfold(MI.getOpcode(),
Dan Gohman584fedf2010-06-21 22:17:20 +00001284 /*UnfoldLoad=*/true,
1285 /*UnfoldStore=*/false,
1286 &LoadRegIndex);
1287 if (NewOpc != 0) {
Evan Chenge837dea2011-06-28 19:10:37 +00001288 const MCInstrDesc &UnfoldMCID = TII->get(NewOpc);
1289 if (UnfoldMCID.getNumDefs() == 1) {
Dan Gohman584fedf2010-06-21 22:17:20 +00001290 MachineFunction &MF = *mbbi->getParent();
1291
1292 // Unfold the load.
Evan Cheng2a4410d2011-11-14 19:48:55 +00001293 DEBUG(dbgs() << "2addr: UNFOLDING: " << MI);
Dan Gohman584fedf2010-06-21 22:17:20 +00001294 const TargetRegisterClass *RC =
Evan Chenge837dea2011-06-28 19:10:37 +00001295 TII->getRegClass(UnfoldMCID, LoadRegIndex, TRI);
Dan Gohman584fedf2010-06-21 22:17:20 +00001296 unsigned Reg = MRI->createVirtualRegister(RC);
1297 SmallVector<MachineInstr *, 2> NewMIs;
Evan Cheng2a4410d2011-11-14 19:48:55 +00001298 if (!TII->unfoldMemoryOperand(MF, &MI, Reg,
Evan Cheng98ec91e2010-07-02 20:36:18 +00001299 /*UnfoldLoad=*/true,/*UnfoldStore=*/false,
1300 NewMIs)) {
1301 DEBUG(dbgs() << "2addr: ABANDONING UNFOLD\n");
1302 return false;
1303 }
Dan Gohman584fedf2010-06-21 22:17:20 +00001304 assert(NewMIs.size() == 2 &&
1305 "Unfolded a load into multiple instructions!");
1306 // The load was previously folded, so this is the only use.
1307 NewMIs[1]->addRegisterKilled(Reg, TRI);
1308
1309 // Tentatively insert the instructions into the block so that they
1310 // look "normal" to the transformation logic.
1311 mbbi->insert(mi, NewMIs[0]);
1312 mbbi->insert(mi, NewMIs[1]);
1313
1314 DEBUG(dbgs() << "2addr: NEW LOAD: " << *NewMIs[0]
1315 << "2addr: NEW INST: " << *NewMIs[1]);
1316
1317 // Transform the instruction, now that it no longer has a load.
1318 unsigned NewDstIdx = NewMIs[1]->findRegisterDefOperandIdx(regA);
1319 unsigned NewSrcIdx = NewMIs[1]->findRegisterUseOperandIdx(regB);
1320 MachineBasicBlock::iterator NewMI = NewMIs[1];
1321 bool TransformSuccess =
1322 TryInstructionTransform(NewMI, mi, mbbi,
Evan Chengf06e6c22011-03-02 01:08:17 +00001323 NewSrcIdx, NewDstIdx, Dist, Processed);
Dan Gohman584fedf2010-06-21 22:17:20 +00001324 if (TransformSuccess ||
1325 NewMIs[1]->getOperand(NewSrcIdx).isKill()) {
1326 // Success, or at least we made an improvement. Keep the unfolded
1327 // instructions and discard the original.
1328 if (LV) {
Evan Cheng2a4410d2011-11-14 19:48:55 +00001329 for (unsigned i = 0, e = MI.getNumOperands(); i != e; ++i) {
1330 MachineOperand &MO = MI.getOperand(i);
Jakob Stoklund Olesenc9df0252011-01-10 02:58:51 +00001331 if (MO.isReg() &&
Dan Gohman7aa7bc72010-06-22 00:32:04 +00001332 TargetRegisterInfo::isVirtualRegister(MO.getReg())) {
1333 if (MO.isUse()) {
Dan Gohmancc1ca982010-06-22 02:07:21 +00001334 if (MO.isKill()) {
1335 if (NewMIs[0]->killsRegister(MO.getReg()))
Evan Cheng2a4410d2011-11-14 19:48:55 +00001336 LV->replaceKillInstruction(MO.getReg(), &MI, NewMIs[0]);
Dan Gohmancc1ca982010-06-22 02:07:21 +00001337 else {
1338 assert(NewMIs[1]->killsRegister(MO.getReg()) &&
1339 "Kill missing after load unfold!");
Evan Cheng2a4410d2011-11-14 19:48:55 +00001340 LV->replaceKillInstruction(MO.getReg(), &MI, NewMIs[1]);
Dan Gohmancc1ca982010-06-22 02:07:21 +00001341 }
1342 }
Evan Cheng2a4410d2011-11-14 19:48:55 +00001343 } else if (LV->removeVirtualRegisterDead(MO.getReg(), &MI)) {
Dan Gohmancc1ca982010-06-22 02:07:21 +00001344 if (NewMIs[1]->registerDefIsDead(MO.getReg()))
1345 LV->addVirtualRegisterDead(MO.getReg(), NewMIs[1]);
1346 else {
1347 assert(NewMIs[0]->registerDefIsDead(MO.getReg()) &&
1348 "Dead flag missing after load unfold!");
1349 LV->addVirtualRegisterDead(MO.getReg(), NewMIs[0]);
1350 }
1351 }
Dan Gohman7aa7bc72010-06-22 00:32:04 +00001352 }
Dan Gohman584fedf2010-06-21 22:17:20 +00001353 }
1354 LV->addVirtualRegisterKilled(Reg, NewMIs[1]);
1355 }
Evan Cheng2a4410d2011-11-14 19:48:55 +00001356 MI.eraseFromParent();
Dan Gohman584fedf2010-06-21 22:17:20 +00001357 mi = NewMIs[1];
1358 if (TransformSuccess)
1359 return true;
1360 } else {
1361 // Transforming didn't eliminate the tie and didn't lead to an
1362 // improvement. Clean up the unfolded instructions and keep the
1363 // original.
1364 DEBUG(dbgs() << "2addr: ABANDONING UNFOLD\n");
1365 NewMIs[0]->eraseFromParent();
1366 NewMIs[1]->eraseFromParent();
1367 }
1368 }
1369 }
1370 }
1371
Bob Wilsoncc80df92009-09-03 20:58:42 +00001372 return false;
1373}
1374
Bill Wendling637980e2008-05-10 00:12:52 +00001375/// runOnMachineFunction - Reduce two-address instructions to two operands.
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001376///
Chris Lattner163c1e72004-01-31 21:14:04 +00001377bool TwoAddressInstructionPass::runOnMachineFunction(MachineFunction &MF) {
David Greeneeb00b182010-01-05 01:24:21 +00001378 DEBUG(dbgs() << "Machine Function\n");
Misha Brukman75fa4e42004-07-22 15:26:23 +00001379 const TargetMachine &TM = MF.getTarget();
Evan Cheng875357d2008-03-13 06:37:55 +00001380 MRI = &MF.getRegInfo();
1381 TII = TM.getInstrInfo();
1382 TRI = TM.getRegisterInfo();
Evan Cheng2a4410d2011-11-14 19:48:55 +00001383 InstrItins = TM.getInstrItineraryData();
Duncan Sands1465d612009-01-28 13:14:17 +00001384 LV = getAnalysisIfAvailable<LiveVariables>();
Dan Gohmana70dca12009-10-09 23:27:56 +00001385 AA = &getAnalysis<AliasAnalysis>();
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001386
Misha Brukman75fa4e42004-07-22 15:26:23 +00001387 bool MadeChange = false;
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001388
David Greeneeb00b182010-01-05 01:24:21 +00001389 DEBUG(dbgs() << "********** REWRITING TWO-ADDR INSTRS **********\n");
1390 DEBUG(dbgs() << "********** Function: "
Daniel Dunbarce63ffb2009-07-25 00:23:56 +00001391 << MF.getFunction()->getName() << '\n');
Alkis Evlogimenos3a9986f2004-02-18 00:35:06 +00001392
Jakob Stoklund Olesen73e7dce2011-07-29 22:51:22 +00001393 // This pass takes the function out of SSA form.
1394 MRI->leaveSSA();
1395
Evan Cheng7543e582008-06-18 07:49:14 +00001396 // ReMatRegs - Keep track of the registers whose def's are remat'ed.
Jakob Stoklund Olesen00f93fc2011-01-09 03:45:44 +00001397 BitVector ReMatRegs(MRI->getNumVirtRegs());
Evan Cheng7543e582008-06-18 07:49:14 +00001398
Bob Wilsoncc80df92009-09-03 20:58:42 +00001399 typedef DenseMap<unsigned, SmallVector<std::pair<unsigned, unsigned>, 4> >
1400 TiedOperandMap;
1401 TiedOperandMap TiedOperands(4);
1402
Evan Cheng870b8072009-03-01 02:03:43 +00001403 SmallPtrSet<MachineInstr*, 8> Processed;
Misha Brukman75fa4e42004-07-22 15:26:23 +00001404 for (MachineFunction::iterator mbbi = MF.begin(), mbbe = MF.end();
1405 mbbi != mbbe; ++mbbi) {
Evan Cheng7543e582008-06-18 07:49:14 +00001406 unsigned Dist = 0;
1407 DistanceMap.clear();
Evan Cheng870b8072009-03-01 02:03:43 +00001408 SrcRegMap.clear();
1409 DstRegMap.clear();
1410 Processed.clear();
Misha Brukman75fa4e42004-07-22 15:26:23 +00001411 for (MachineBasicBlock::iterator mi = mbbi->begin(), me = mbbi->end();
Evan Cheng7a963fa2008-03-27 01:27:25 +00001412 mi != me; ) {
Chris Lattner7896c9f2009-12-03 00:50:42 +00001413 MachineBasicBlock::iterator nmi = llvm::next(mi);
Dale Johannesenb8ff9342010-02-10 21:47:48 +00001414 if (mi->isDebugValue()) {
1415 mi = nmi;
1416 continue;
1417 }
Evan Chengf1250ee2010-03-23 20:36:12 +00001418
Evan Cheng3d720fb2010-05-05 18:45:40 +00001419 // Remember REG_SEQUENCE instructions, we'll deal with them later.
1420 if (mi->isRegSequence())
1421 RegSequences.push_back(&*mi);
1422
Evan Chenge837dea2011-06-28 19:10:37 +00001423 const MCInstrDesc &MCID = mi->getDesc();
Evan Cheng360c2dd2006-11-01 23:06:55 +00001424 bool FirstTied = true;
Bill Wendling637980e2008-05-10 00:12:52 +00001425
Evan Cheng7543e582008-06-18 07:49:14 +00001426 DistanceMap.insert(std::make_pair(mi, ++Dist));
Evan Cheng870b8072009-03-01 02:03:43 +00001427
1428 ProcessCopy(&*mi, &*mbbi, Processed);
1429
Bob Wilsoncc80df92009-09-03 20:58:42 +00001430 // First scan through all the tied register uses in this instruction
1431 // and record a list of pairs of tied operands for each register.
Chris Lattner518bb532010-02-09 19:54:29 +00001432 unsigned NumOps = mi->isInlineAsm()
Evan Chenge837dea2011-06-28 19:10:37 +00001433 ? mi->getNumOperands() : MCID.getNumOperands();
Bob Wilsoncc80df92009-09-03 20:58:42 +00001434 for (unsigned SrcIdx = 0; SrcIdx < NumOps; ++SrcIdx) {
1435 unsigned DstIdx = 0;
1436 if (!mi->isRegTiedToDefOperand(SrcIdx, &DstIdx))
Evan Cheng360c2dd2006-11-01 23:06:55 +00001437 continue;
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001438
Evan Cheng360c2dd2006-11-01 23:06:55 +00001439 if (FirstTied) {
Bob Wilsoncc80df92009-09-03 20:58:42 +00001440 FirstTied = false;
Evan Cheng360c2dd2006-11-01 23:06:55 +00001441 ++NumTwoAddressInstrs;
David Greeneeb00b182010-01-05 01:24:21 +00001442 DEBUG(dbgs() << '\t' << *mi);
Evan Cheng360c2dd2006-11-01 23:06:55 +00001443 }
Bill Wendling637980e2008-05-10 00:12:52 +00001444
Bob Wilsoncc80df92009-09-03 20:58:42 +00001445 assert(mi->getOperand(SrcIdx).isReg() &&
1446 mi->getOperand(SrcIdx).getReg() &&
1447 mi->getOperand(SrcIdx).isUse() &&
1448 "two address instruction invalid");
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001449
Bob Wilsoncc80df92009-09-03 20:58:42 +00001450 unsigned regB = mi->getOperand(SrcIdx).getReg();
Benjamin Kramer4e39f8f2011-06-18 13:53:47 +00001451 TiedOperands[regB].push_back(std::make_pair(SrcIdx, DstIdx));
Bob Wilsoncc80df92009-09-03 20:58:42 +00001452 }
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001453
Bob Wilsoncc80df92009-09-03 20:58:42 +00001454 // Now iterate over the information collected above.
1455 for (TiedOperandMap::iterator OI = TiedOperands.begin(),
1456 OE = TiedOperands.end(); OI != OE; ++OI) {
1457 SmallVector<std::pair<unsigned, unsigned>, 4> &TiedPairs = OI->second;
Evan Cheng360c2dd2006-11-01 23:06:55 +00001458
Bob Wilsoncc80df92009-09-03 20:58:42 +00001459 // If the instruction has a single pair of tied operands, try some
1460 // transformations that may either eliminate the tied operands or
1461 // improve the opportunities for coalescing away the register copy.
1462 if (TiedOperands.size() == 1 && TiedPairs.size() == 1) {
1463 unsigned SrcIdx = TiedPairs[0].first;
1464 unsigned DstIdx = TiedPairs[0].second;
Bob Wilson43449792009-08-31 21:54:55 +00001465
Bob Wilsoncc80df92009-09-03 20:58:42 +00001466 // If the registers are already equal, nothing needs to be done.
1467 if (mi->getOperand(SrcIdx).getReg() ==
1468 mi->getOperand(DstIdx).getReg())
1469 break; // Done with this instruction.
1470
Evan Chengf06e6c22011-03-02 01:08:17 +00001471 if (TryInstructionTransform(mi, nmi, mbbi, SrcIdx, DstIdx, Dist,
1472 Processed))
Bob Wilsoncc80df92009-09-03 20:58:42 +00001473 break; // The tied operands have been eliminated.
1474 }
1475
Cameron Zwarichaaa5f142011-06-07 23:54:00 +00001476 bool IsEarlyClobber = false;
Bob Wilsoncc80df92009-09-03 20:58:42 +00001477 bool RemovedKillFlag = false;
1478 bool AllUsesCopied = true;
1479 unsigned LastCopiedReg = 0;
1480 unsigned regB = OI->first;
1481 for (unsigned tpi = 0, tpe = TiedPairs.size(); tpi != tpe; ++tpi) {
1482 unsigned SrcIdx = TiedPairs[tpi].first;
1483 unsigned DstIdx = TiedPairs[tpi].second;
Cameron Zwarichaaa5f142011-06-07 23:54:00 +00001484
1485 const MachineOperand &DstMO = mi->getOperand(DstIdx);
1486 unsigned regA = DstMO.getReg();
1487 IsEarlyClobber |= DstMO.isEarlyClobber();
1488
Bob Wilsoncc80df92009-09-03 20:58:42 +00001489 // Grab regB from the instruction because it may have changed if the
1490 // instruction was commuted.
1491 regB = mi->getOperand(SrcIdx).getReg();
1492
1493 if (regA == regB) {
1494 // The register is tied to multiple destinations (or else we would
1495 // not have continued this far), but this use of the register
1496 // already matches the tied destination. Leave it.
1497 AllUsesCopied = false;
1498 continue;
1499 }
1500 LastCopiedReg = regA;
1501
1502 assert(TargetRegisterInfo::isVirtualRegister(regB) &&
1503 "cannot make instruction into two-address form");
Chris Lattner6b507672004-01-31 21:21:43 +00001504
Chris Lattner1e313632004-07-21 23:17:57 +00001505#ifndef NDEBUG
Bob Wilsoncc80df92009-09-03 20:58:42 +00001506 // First, verify that we don't have a use of "a" in the instruction
1507 // (a = b + a for example) because our transformation will not
1508 // work. This should never occur because we are in SSA form.
1509 for (unsigned i = 0; i != mi->getNumOperands(); ++i)
1510 assert(i == DstIdx ||
1511 !mi->getOperand(i).isReg() ||
1512 mi->getOperand(i).getReg() != regA);
Chris Lattner1e313632004-07-21 23:17:57 +00001513#endif
Alkis Evlogimenos14be6402004-02-04 22:17:40 +00001514
Bob Wilsoncc80df92009-09-03 20:58:42 +00001515 // Emit a copy or rematerialize the definition.
1516 const TargetRegisterClass *rc = MRI->getRegClass(regB);
1517 MachineInstr *DefMI = MRI->getVRegDef(regB);
1518 // If it's safe and profitable, remat the definition instead of
1519 // copying it.
1520 if (DefMI &&
1521 DefMI->getDesc().isAsCheapAsAMove() &&
Evan Chengac1abde2010-03-02 19:03:01 +00001522 DefMI->isSafeToReMat(TII, AA, regB) &&
Bob Wilsoncc80df92009-09-03 20:58:42 +00001523 isProfitableToReMat(regB, rc, mi, DefMI, mbbi, Dist)){
David Greeneeb00b182010-01-05 01:24:21 +00001524 DEBUG(dbgs() << "2addr: REMATTING : " << *DefMI << "\n");
Bob Wilsoncc80df92009-09-03 20:58:42 +00001525 unsigned regASubIdx = mi->getOperand(DstIdx).getSubReg();
Jakob Stoklund Olesen9edf7de2010-06-02 22:47:25 +00001526 TII->reMaterialize(*mbbi, mi, regA, regASubIdx, DefMI, *TRI);
Jakob Stoklund Olesen00f93fc2011-01-09 03:45:44 +00001527 ReMatRegs.set(TargetRegisterInfo::virtReg2Index(regB));
Bob Wilsoncc80df92009-09-03 20:58:42 +00001528 ++NumReMats;
Bob Wilson71124f62009-09-01 04:18:40 +00001529 } else {
Jakob Stoklund Olesen92c1f722010-07-10 19:08:25 +00001530 BuildMI(*mbbi, mi, mi->getDebugLoc(), TII->get(TargetOpcode::COPY),
1531 regA).addReg(regB);
Bob Wilsoncc80df92009-09-03 20:58:42 +00001532 }
1533
1534 MachineBasicBlock::iterator prevMI = prior(mi);
1535 // Update DistanceMap.
1536 DistanceMap.insert(std::make_pair(prevMI, Dist));
1537 DistanceMap[mi] = ++Dist;
1538
David Greeneeb00b182010-01-05 01:24:21 +00001539 DEBUG(dbgs() << "\t\tprepend:\t" << *prevMI);
Bob Wilsoncc80df92009-09-03 20:58:42 +00001540
1541 MachineOperand &MO = mi->getOperand(SrcIdx);
1542 assert(MO.isReg() && MO.getReg() == regB && MO.isUse() &&
1543 "inconsistent operand info for 2-reg pass");
1544 if (MO.isKill()) {
1545 MO.setIsKill(false);
1546 RemovedKillFlag = true;
1547 }
1548 MO.setReg(regA);
1549 }
1550
1551 if (AllUsesCopied) {
Cameron Zwarichaaa5f142011-06-07 23:54:00 +00001552 if (!IsEarlyClobber) {
1553 // Replace other (un-tied) uses of regB with LastCopiedReg.
1554 for (unsigned i = 0, e = mi->getNumOperands(); i != e; ++i) {
1555 MachineOperand &MO = mi->getOperand(i);
1556 if (MO.isReg() && MO.getReg() == regB && MO.isUse()) {
1557 if (MO.isKill()) {
1558 MO.setIsKill(false);
1559 RemovedKillFlag = true;
1560 }
1561 MO.setReg(LastCopiedReg);
Bob Wilsoncc80df92009-09-03 20:58:42 +00001562 }
Bob Wilsoncc80df92009-09-03 20:58:42 +00001563 }
1564 }
1565
1566 // Update live variables for regB.
1567 if (RemovedKillFlag && LV && LV->getVarInfo(regB).removeKill(mi))
1568 LV->addVirtualRegisterKilled(regB, prior(mi));
1569
1570 } else if (RemovedKillFlag) {
1571 // Some tied uses of regB matched their destination registers, so
1572 // regB is still used in this instruction, but a kill flag was
1573 // removed from a different tied use of regB, so now we need to add
1574 // a kill flag to one of the remaining uses of regB.
1575 for (unsigned i = 0, e = mi->getNumOperands(); i != e; ++i) {
1576 MachineOperand &MO = mi->getOperand(i);
1577 if (MO.isReg() && MO.getReg() == regB && MO.isUse()) {
1578 MO.setIsKill(true);
1579 break;
Bob Wilson71124f62009-09-01 04:18:40 +00001580 }
1581 }
Bob Wilson43449792009-08-31 21:54:55 +00001582 }
Evan Cheng68fc2da2010-06-09 19:26:01 +00001583
1584 // Schedule the source copy / remat inserted to form two-address
1585 // instruction. FIXME: Does it matter the distance map may not be
1586 // accurate after it's scheduled?
1587 TII->scheduleTwoAddrSource(prior(mi), mi, *TRI);
1588
Bob Wilson43449792009-08-31 21:54:55 +00001589 MadeChange = true;
1590
David Greeneeb00b182010-01-05 01:24:21 +00001591 DEBUG(dbgs() << "\t\trewrite to:\t" << *mi);
Misha Brukman75fa4e42004-07-22 15:26:23 +00001592 }
Bill Wendling637980e2008-05-10 00:12:52 +00001593
Jakob Stoklund Olesened2185e2010-07-06 23:26:25 +00001594 // Rewrite INSERT_SUBREG as COPY now that we no longer need SSA form.
1595 if (mi->isInsertSubreg()) {
1596 // From %reg = INSERT_SUBREG %reg, %subreg, subidx
1597 // To %reg:subidx = COPY %subreg
1598 unsigned SubIdx = mi->getOperand(3).getImm();
1599 mi->RemoveOperand(3);
1600 assert(mi->getOperand(0).getSubReg() == 0 && "Unexpected subreg idx");
1601 mi->getOperand(0).setSubReg(SubIdx);
1602 mi->RemoveOperand(1);
1603 mi->setDesc(TII->get(TargetOpcode::COPY));
1604 DEBUG(dbgs() << "\t\tconvert to:\t" << *mi);
1605 }
1606
Bob Wilsoncc80df92009-09-03 20:58:42 +00001607 // Clear TiedOperands here instead of at the top of the loop
1608 // since most instructions do not have tied operands.
1609 TiedOperands.clear();
Evan Cheng7a963fa2008-03-27 01:27:25 +00001610 mi = nmi;
Misha Brukman75fa4e42004-07-22 15:26:23 +00001611 }
1612 }
1613
Evan Cheng601ca4b2008-06-25 01:16:38 +00001614 // Some remat'ed instructions are dead.
Jakob Stoklund Olesen00f93fc2011-01-09 03:45:44 +00001615 for (int i = ReMatRegs.find_first(); i != -1; i = ReMatRegs.find_next(i)) {
1616 unsigned VReg = TargetRegisterInfo::index2VirtReg(i);
Evan Chengf1250ee2010-03-23 20:36:12 +00001617 if (MRI->use_nodbg_empty(VReg)) {
Evan Cheng601ca4b2008-06-25 01:16:38 +00001618 MachineInstr *DefMI = MRI->getVRegDef(VReg);
1619 DefMI->eraseFromParent();
Bill Wendlinga16157a2008-05-26 05:49:49 +00001620 }
Bill Wendling48f7f232008-05-26 05:18:34 +00001621 }
1622
Evan Cheng3d720fb2010-05-05 18:45:40 +00001623 // Eliminate REG_SEQUENCE instructions. Their whole purpose was to preseve
1624 // SSA form. It's now safe to de-SSA.
1625 MadeChange |= EliminateRegSequences();
1626
Misha Brukman75fa4e42004-07-22 15:26:23 +00001627 return MadeChange;
Alkis Evlogimenos71499de2003-12-18 13:06:04 +00001628}
Evan Cheng3d720fb2010-05-05 18:45:40 +00001629
1630static void UpdateRegSequenceSrcs(unsigned SrcReg,
Evan Cheng53c779b2010-05-17 20:57:12 +00001631 unsigned DstReg, unsigned SubIdx,
Jakob Stoklund Olesen5a0d4fc2010-05-29 00:14:14 +00001632 MachineRegisterInfo *MRI,
1633 const TargetRegisterInfo &TRI) {
Evan Cheng3d720fb2010-05-05 18:45:40 +00001634 for (MachineRegisterInfo::reg_iterator RI = MRI->reg_begin(SrcReg),
Evan Cheng3ae56bc2010-05-12 01:27:49 +00001635 RE = MRI->reg_end(); RI != RE; ) {
Evan Cheng3d720fb2010-05-05 18:45:40 +00001636 MachineOperand &MO = RI.getOperand();
1637 ++RI;
Jakob Stoklund Olesen5a0d4fc2010-05-29 00:14:14 +00001638 MO.substVirtReg(DstReg, SubIdx, TRI);
Evan Cheng53c779b2010-05-17 20:57:12 +00001639 }
1640}
1641
1642/// CoalesceExtSubRegs - If a number of sources of the REG_SEQUENCE are
1643/// EXTRACT_SUBREG from the same register and to the same virtual register
1644/// with different sub-register indices, attempt to combine the
1645/// EXTRACT_SUBREGs and pre-coalesce them. e.g.
1646/// %reg1026<def> = VLDMQ %reg1025<kill>, 260, pred:14, pred:%reg0
1647/// %reg1029:6<def> = EXTRACT_SUBREG %reg1026, 6
1648/// %reg1029:5<def> = EXTRACT_SUBREG %reg1026<kill>, 5
1649/// Since D subregs 5, 6 can combine to a Q register, we can coalesce
1650/// reg1026 to reg1029.
1651void
1652TwoAddressInstructionPass::CoalesceExtSubRegs(SmallVector<unsigned,4> &Srcs,
1653 unsigned DstReg) {
1654 SmallSet<unsigned, 4> Seen;
1655 for (unsigned i = 0, e = Srcs.size(); i != e; ++i) {
1656 unsigned SrcReg = Srcs[i];
1657 if (!Seen.insert(SrcReg))
1658 continue;
1659
Bob Wilson26bf8f92010-06-03 23:53:58 +00001660 // Check that the instructions are all in the same basic block.
1661 MachineInstr *SrcDefMI = MRI->getVRegDef(SrcReg);
1662 MachineInstr *DstDefMI = MRI->getVRegDef(DstReg);
1663 if (SrcDefMI->getParent() != DstDefMI->getParent())
1664 continue;
1665
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001666 // If there are no other uses than copies which feed into
Evan Cheng53c779b2010-05-17 20:57:12 +00001667 // the reg_sequence, then we might be able to coalesce them.
1668 bool CanCoalesce = true;
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001669 SmallVector<unsigned, 4> SrcSubIndices, DstSubIndices;
Evan Cheng53c779b2010-05-17 20:57:12 +00001670 for (MachineRegisterInfo::use_nodbg_iterator
1671 UI = MRI->use_nodbg_begin(SrcReg),
1672 UE = MRI->use_nodbg_end(); UI != UE; ++UI) {
1673 MachineInstr *UseMI = &*UI;
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001674 if (!UseMI->isCopy() || UseMI->getOperand(0).getReg() != DstReg) {
Evan Cheng53c779b2010-05-17 20:57:12 +00001675 CanCoalesce = false;
1676 break;
1677 }
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001678 SrcSubIndices.push_back(UseMI->getOperand(1).getSubReg());
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001679 DstSubIndices.push_back(UseMI->getOperand(0).getSubReg());
Evan Cheng53c779b2010-05-17 20:57:12 +00001680 }
1681
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001682 if (!CanCoalesce || SrcSubIndices.size() < 2)
Evan Cheng53c779b2010-05-17 20:57:12 +00001683 continue;
1684
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001685 // Check that the source subregisters can be combined.
1686 std::sort(SrcSubIndices.begin(), SrcSubIndices.end());
Bob Wilson852a7e32010-06-15 05:56:31 +00001687 unsigned NewSrcSubIdx = 0;
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001688 if (!TRI->canCombineSubRegIndices(MRI->getRegClass(SrcReg), SrcSubIndices,
Bob Wilson852a7e32010-06-15 05:56:31 +00001689 NewSrcSubIdx))
Bob Wilson26bf8f92010-06-03 23:53:58 +00001690 continue;
1691
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001692 // Check that the destination subregisters can also be combined.
1693 std::sort(DstSubIndices.begin(), DstSubIndices.end());
1694 unsigned NewDstSubIdx = 0;
1695 if (!TRI->canCombineSubRegIndices(MRI->getRegClass(DstReg), DstSubIndices,
1696 NewDstSubIdx))
1697 continue;
1698
1699 // If neither source nor destination can be combined to the full register,
1700 // just give up. This could be improved if it ever matters.
1701 if (NewSrcSubIdx != 0 && NewDstSubIdx != 0)
1702 continue;
1703
Bob Wilson852a7e32010-06-15 05:56:31 +00001704 // Now that we know that all the uses are extract_subregs and that those
1705 // subregs can somehow be combined, scan all the extract_subregs again to
1706 // make sure the subregs are in the right order and can be composed.
Bob Wilson852a7e32010-06-15 05:56:31 +00001707 MachineInstr *SomeMI = 0;
1708 CanCoalesce = true;
1709 for (MachineRegisterInfo::use_nodbg_iterator
1710 UI = MRI->use_nodbg_begin(SrcReg),
1711 UE = MRI->use_nodbg_end(); UI != UE; ++UI) {
1712 MachineInstr *UseMI = &*UI;
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001713 assert(UseMI->isCopy());
Bob Wilson852a7e32010-06-15 05:56:31 +00001714 unsigned DstSubIdx = UseMI->getOperand(0).getSubReg();
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001715 unsigned SrcSubIdx = UseMI->getOperand(1).getSubReg();
Bob Wilson852a7e32010-06-15 05:56:31 +00001716 assert(DstSubIdx != 0 && "missing subreg from RegSequence elimination");
Bob Wilson4ffd22d2010-06-15 17:27:54 +00001717 if ((NewDstSubIdx == 0 &&
1718 TRI->composeSubRegIndices(NewSrcSubIdx, DstSubIdx) != SrcSubIdx) ||
1719 (NewSrcSubIdx == 0 &&
1720 TRI->composeSubRegIndices(NewDstSubIdx, SrcSubIdx) != DstSubIdx)) {
Bob Wilson852a7e32010-06-15 05:56:31 +00001721 CanCoalesce = false;
1722 break;
Evan Cheng53c779b2010-05-17 20:57:12 +00001723 }
Bob Wilson852a7e32010-06-15 05:56:31 +00001724 // Keep track of one of the uses.
1725 SomeMI = UseMI;
1726 }
1727 if (!CanCoalesce)
1728 continue;
1729
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001730 // Insert a copy to replace the original.
Jakob Stoklund Olesen5c00e072010-07-08 16:40:15 +00001731 MachineInstr *CopyMI = BuildMI(*SomeMI->getParent(), SomeMI,
1732 SomeMI->getDebugLoc(),
1733 TII->get(TargetOpcode::COPY))
1734 .addReg(DstReg, RegState::Define, NewDstSubIdx)
1735 .addReg(SrcReg, 0, NewSrcSubIdx);
Bob Wilson852a7e32010-06-15 05:56:31 +00001736
1737 // Remove all the old extract instructions.
1738 for (MachineRegisterInfo::use_nodbg_iterator
1739 UI = MRI->use_nodbg_begin(SrcReg),
1740 UE = MRI->use_nodbg_end(); UI != UE; ) {
1741 MachineInstr *UseMI = &*UI;
1742 ++UI;
1743 if (UseMI == CopyMI)
1744 continue;
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001745 assert(UseMI->isCopy());
Bob Wilson852a7e32010-06-15 05:56:31 +00001746 // Move any kills to the new copy or extract instruction.
1747 if (UseMI->getOperand(1).isKill()) {
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001748 CopyMI->getOperand(1).setIsKill();
Bob Wilson852a7e32010-06-15 05:56:31 +00001749 if (LV)
1750 // Update live variables
1751 LV->replaceKillInstruction(SrcReg, UseMI, &*CopyMI);
1752 }
1753 UseMI->eraseFromParent();
1754 }
Evan Cheng3d720fb2010-05-05 18:45:40 +00001755 }
1756}
1757
Evan Chengc6dcce32010-05-17 23:24:12 +00001758static bool HasOtherRegSequenceUses(unsigned Reg, MachineInstr *RegSeq,
1759 MachineRegisterInfo *MRI) {
1760 for (MachineRegisterInfo::use_iterator UI = MRI->use_begin(Reg),
1761 UE = MRI->use_end(); UI != UE; ++UI) {
1762 MachineInstr *UseMI = &*UI;
1763 if (UseMI != RegSeq && UseMI->isRegSequence())
1764 return true;
1765 }
1766 return false;
1767}
1768
Evan Cheng3d720fb2010-05-05 18:45:40 +00001769/// EliminateRegSequences - Eliminate REG_SEQUENCE instructions as part
1770/// of the de-ssa process. This replaces sources of REG_SEQUENCE as
1771/// sub-register references of the register defined by REG_SEQUENCE. e.g.
1772///
1773/// %reg1029<def>, %reg1030<def> = VLD1q16 %reg1024<kill>, ...
1774/// %reg1031<def> = REG_SEQUENCE %reg1029<kill>, 5, %reg1030<kill>, 6
1775/// =>
1776/// %reg1031:5<def>, %reg1031:6<def> = VLD1q16 %reg1024<kill>, ...
1777bool TwoAddressInstructionPass::EliminateRegSequences() {
1778 if (RegSequences.empty())
1779 return false;
1780
1781 for (unsigned i = 0, e = RegSequences.size(); i != e; ++i) {
1782 MachineInstr *MI = RegSequences[i];
1783 unsigned DstReg = MI->getOperand(0).getReg();
1784 if (MI->getOperand(0).getSubReg() ||
1785 TargetRegisterInfo::isPhysicalRegister(DstReg) ||
1786 !(MI->getNumOperands() & 1)) {
1787 DEBUG(dbgs() << "Illegal REG_SEQUENCE instruction:" << *MI);
1788 llvm_unreachable(0);
1789 }
Evan Cheng0bcccac2010-05-11 00:04:31 +00001790
Evan Cheng44bfdd32010-05-17 22:09:49 +00001791 bool IsImpDef = true;
Evan Chengb990a2f2010-05-14 23:21:14 +00001792 SmallVector<unsigned, 4> RealSrcs;
Evan Cheng0bcccac2010-05-11 00:04:31 +00001793 SmallSet<unsigned, 4> Seen;
Evan Cheng3d720fb2010-05-05 18:45:40 +00001794 for (unsigned i = 1, e = MI->getNumOperands(); i < e; i += 2) {
1795 unsigned SrcReg = MI->getOperand(i).getReg();
Bob Wilson495de3b2010-12-17 01:21:12 +00001796 unsigned SubIdx = MI->getOperand(i+1).getImm();
Evan Cheng3d720fb2010-05-05 18:45:40 +00001797 if (MI->getOperand(i).getSubReg() ||
1798 TargetRegisterInfo::isPhysicalRegister(SrcReg)) {
1799 DEBUG(dbgs() << "Illegal REG_SEQUENCE instruction:" << *MI);
1800 llvm_unreachable(0);
1801 }
Evan Cheng0bcccac2010-05-11 00:04:31 +00001802
Evan Cheng054dbb82010-05-13 00:00:35 +00001803 MachineInstr *DefMI = MRI->getVRegDef(SrcReg);
Evan Chengb990a2f2010-05-14 23:21:14 +00001804 if (DefMI->isImplicitDef()) {
1805 DefMI->eraseFromParent();
1806 continue;
1807 }
Evan Cheng44bfdd32010-05-17 22:09:49 +00001808 IsImpDef = false;
Evan Chengb990a2f2010-05-14 23:21:14 +00001809
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001810 // Remember COPY sources. These might be candidate for coalescing.
Jakob Stoklund Olesenc0075cc2010-07-10 22:42:53 +00001811 if (DefMI->isCopy() && DefMI->getOperand(1).getSubReg())
Evan Chengb990a2f2010-05-14 23:21:14 +00001812 RealSrcs.push_back(DefMI->getOperand(1).getReg());
1813
Jakob Stoklund Olesen1e1098c2010-07-10 22:42:59 +00001814 bool isKill = MI->getOperand(i).isKill();
1815 if (!Seen.insert(SrcReg) || MI->getParent() != DefMI->getParent() ||
Bob Wilson495de3b2010-12-17 01:21:12 +00001816 !isKill || HasOtherRegSequenceUses(SrcReg, MI, MRI) ||
1817 !TRI->getMatchingSuperRegClass(MRI->getRegClass(DstReg),
1818 MRI->getRegClass(SrcReg), SubIdx)) {
Evan Cheng054dbb82010-05-13 00:00:35 +00001819 // REG_SEQUENCE cannot have duplicated operands, add a copy.
Jakob Stoklund Olesen34373522010-05-19 20:08:00 +00001820 // Also add an copy if the source is live-in the block. We don't want
Evan Cheng054dbb82010-05-13 00:00:35 +00001821 // to end up with a partial-redef of a livein, e.g.
1822 // BB0:
1823 // reg1051:10<def> =
1824 // ...
1825 // BB1:
1826 // ... = reg1051:10
1827 // BB2:
1828 // reg1051:9<def> =
1829 // LiveIntervalAnalysis won't like it.
Jakob Stoklund Olesen34373522010-05-19 20:08:00 +00001830 //
1831 // If the REG_SEQUENCE doesn't kill its source, keeping live variables
1832 // correctly up to date becomes very difficult. Insert a copy.
Jakob Stoklund Olesene4b9c4f2010-08-09 20:19:16 +00001833
1834 // Defer any kill flag to the last operand using SrcReg. Otherwise, we
1835 // might insert a COPY that uses SrcReg after is was killed.
1836 if (isKill)
1837 for (unsigned j = i + 2; j < e; j += 2)
1838 if (MI->getOperand(j).getReg() == SrcReg) {
1839 MI->getOperand(j).setIsKill();
1840 isKill = false;
1841 break;
1842 }
1843
Evan Cheng054dbb82010-05-13 00:00:35 +00001844 MachineBasicBlock::iterator InsertLoc = MI;
Jakob Stoklund Olesen1e1098c2010-07-10 22:42:59 +00001845 MachineInstr *CopyMI = BuildMI(*MI->getParent(), InsertLoc,
1846 MI->getDebugLoc(), TII->get(TargetOpcode::COPY))
Bob Wilson495de3b2010-12-17 01:21:12 +00001847 .addReg(DstReg, RegState::Define, SubIdx)
Jakob Stoklund Olesen1e1098c2010-07-10 22:42:59 +00001848 .addReg(SrcReg, getKillRegState(isKill));
1849 MI->getOperand(i).setReg(0);
1850 if (LV && isKill)
1851 LV->replaceKillInstruction(SrcReg, MI, CopyMI);
1852 DEBUG(dbgs() << "Inserted: " << *CopyMI);
Evan Cheng0bcccac2010-05-11 00:04:31 +00001853 }
1854 }
1855
1856 for (unsigned i = 1, e = MI->getNumOperands(); i < e; i += 2) {
1857 unsigned SrcReg = MI->getOperand(i).getReg();
Jakob Stoklund Olesen1e1098c2010-07-10 22:42:59 +00001858 if (!SrcReg) continue;
Evan Cheng53c779b2010-05-17 20:57:12 +00001859 unsigned SubIdx = MI->getOperand(i+1).getImm();
Jakob Stoklund Olesen5a0d4fc2010-05-29 00:14:14 +00001860 UpdateRegSequenceSrcs(SrcReg, DstReg, SubIdx, MRI, *TRI);
Evan Cheng3d720fb2010-05-05 18:45:40 +00001861 }
1862
Evan Cheng44bfdd32010-05-17 22:09:49 +00001863 if (IsImpDef) {
1864 DEBUG(dbgs() << "Turned: " << *MI << " into an IMPLICIT_DEF");
1865 MI->setDesc(TII->get(TargetOpcode::IMPLICIT_DEF));
1866 for (int j = MI->getNumOperands() - 1, ee = 0; j > ee; --j)
1867 MI->RemoveOperand(j);
1868 } else {
1869 DEBUG(dbgs() << "Eliminated: " << *MI);
1870 MI->eraseFromParent();
1871 }
Evan Chengb990a2f2010-05-14 23:21:14 +00001872
Jakob Stoklund Olesenfe181f42010-06-18 23:10:20 +00001873 // Try coalescing some EXTRACT_SUBREG instructions. This can create
1874 // INSERT_SUBREG instructions that must have <undef> flags added by
1875 // LiveIntervalAnalysis, so only run it when LiveVariables is available.
1876 if (LV)
1877 CoalesceExtSubRegs(RealSrcs, DstReg);
Evan Cheng3d720fb2010-05-05 18:45:40 +00001878 }
1879
Evan Chengfc6e6a92010-05-10 21:24:55 +00001880 RegSequences.clear();
Evan Cheng3d720fb2010-05-05 18:45:40 +00001881 return true;
1882}