blob: bd939fdf1045a17cdb1cf7cb2058b02377134997 [file] [log] [blame]
Puyan Lotfia521c4ac52017-11-02 23:37:32 +00001//===-------------- MIRCanonicalizer.cpp - MIR Canonicalizer --------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// The purpose of this pass is to employ a canonical code transformation so
11// that code compiled with slightly different IR passes can be diffed more
12// effectively than otherwise. This is done by renaming vregs in a given
13// LiveRange in a canonical way. This pass also does a pseudo-scheduling to
14// move defs closer to their use inorder to reduce diffs caused by slightly
15// different schedules.
16//
17// Basic Usage:
18//
19// llc -o - -run-pass mir-canonicalizer example.mir
20//
21// Reorders instructions canonically.
22// Renames virtual register operands canonically.
23// Strips certain MIR artifacts (optionally).
24//
25//===----------------------------------------------------------------------===//
26
27#include "llvm/ADT/PostOrderIterator.h"
28#include "llvm/ADT/STLExtras.h"
Puyan Lotfia521c4ac52017-11-02 23:37:32 +000029#include "llvm/CodeGen/MachineFunctionPass.h"
30#include "llvm/CodeGen/MachineInstrBuilder.h"
31#include "llvm/CodeGen/MachineRegisterInfo.h"
David Blaikie3f833ed2017-11-08 01:01:31 +000032#include "llvm/CodeGen/Passes.h"
Puyan Lotfia521c4ac52017-11-02 23:37:32 +000033#include "llvm/Support/raw_ostream.h"
Puyan Lotfia521c4ac52017-11-02 23:37:32 +000034
35#include <queue>
36
37using namespace llvm;
38
39namespace llvm {
40extern char &MIRCanonicalizerID;
41} // namespace llvm
42
43#define DEBUG_TYPE "mir-canonicalizer"
44
45static cl::opt<unsigned>
46CanonicalizeFunctionNumber("canon-nth-function", cl::Hidden, cl::init(~0u),
47 cl::value_desc("N"),
48 cl::desc("Function number to canonicalize."));
49
50static cl::opt<unsigned>
51CanonicalizeBasicBlockNumber("canon-nth-basicblock", cl::Hidden, cl::init(~0u),
52 cl::value_desc("N"),
53 cl::desc("BasicBlock number to canonicalize."));
54
55namespace {
56
57class MIRCanonicalizer : public MachineFunctionPass {
58public:
59 static char ID;
60 MIRCanonicalizer() : MachineFunctionPass(ID) {}
61
62 StringRef getPassName() const override {
63 return "Rename register operands in a canonical ordering.";
64 }
65
66 void getAnalysisUsage(AnalysisUsage &AU) const override {
67 AU.setPreservesCFG();
68 MachineFunctionPass::getAnalysisUsage(AU);
69 }
70
71 bool runOnMachineFunction(MachineFunction &MF) override;
72};
73
74} // end anonymous namespace
75
76enum VRType { RSE_Reg = 0, RSE_FrameIndex, RSE_NewCandidate };
77class TypedVReg {
78 VRType type;
79 unsigned reg;
80
81public:
82 TypedVReg(unsigned reg) : type(RSE_Reg), reg(reg) {}
83 TypedVReg(VRType type) : type(type), reg(~0U) {
84 assert(type != RSE_Reg && "Expected a non-register type.");
85 }
86
87 bool isReg() const { return type == RSE_Reg; }
88 bool isFrameIndex() const { return type == RSE_FrameIndex; }
89 bool isCandidate() const { return type == RSE_NewCandidate; }
90
91 VRType getType() const { return type; }
92 unsigned getReg() const {
93 assert(this->isReg() && "Expected a virtual or physical register.");
94 return reg;
95 }
96};
97
98char MIRCanonicalizer::ID;
99
100char &llvm::MIRCanonicalizerID = MIRCanonicalizer::ID;
101
102INITIALIZE_PASS_BEGIN(MIRCanonicalizer, "mir-canonicalizer",
Craig Topper666e23b2017-11-03 18:02:46 +0000103 "Rename Register Operands Canonically", false, false)
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000104
105INITIALIZE_PASS_END(MIRCanonicalizer, "mir-canonicalizer",
Craig Topper666e23b2017-11-03 18:02:46 +0000106 "Rename Register Operands Canonically", false, false)
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000107
108static std::vector<MachineBasicBlock *> GetRPOList(MachineFunction &MF) {
109 ReversePostOrderTraversal<MachineBasicBlock *> RPOT(&*MF.begin());
110 std::vector<MachineBasicBlock *> RPOList;
111 for (auto MBB : RPOT) {
112 RPOList.push_back(MBB);
113 }
114
115 return RPOList;
116}
117
118// Set a dummy vreg. We use this vregs register class to generate throw-away
119// vregs that are used to skip vreg numbers so that vreg numbers line up.
120static unsigned GetDummyVReg(const MachineFunction &MF) {
121 for (auto &MBB : MF) {
122 for (auto &MI : MBB) {
123 for (auto &MO : MI.operands()) {
124 if (!MO.isReg() || !TargetRegisterInfo::isVirtualRegister(MO.getReg()))
125 continue;
126 return MO.getReg();
127 }
128 }
129 }
130
131 return ~0U;
132}
133
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000134static bool
135rescheduleLexographically(std::vector<MachineInstr *> instructions,
136 MachineBasicBlock *MBB,
137 std::function<MachineBasicBlock::iterator()> getPos) {
138
139 bool Changed = false;
140 std::map<std::string, MachineInstr*> StringInstrMap;
141
142 for (auto *II : instructions) {
143 std::string S;
144 raw_string_ostream OS(S);
145 II->print(OS);
146 OS.flush();
147
148 // Trim the assignment, or start from the begining in the case of a store.
149 const size_t i = S.find("=");
150 StringInstrMap.insert({(i == std::string::npos) ? S : S.substr(i), II});
151 }
152
153 for (auto &II : StringInstrMap) {
154
155 DEBUG({
156 dbgs() << "Splicing ";
157 II.second->dump();
158 dbgs() << " right before: ";
159 getPos()->dump();
160 });
161
162 Changed = true;
163 MBB->splice(getPos(), MBB, II.second);
164 }
165
166 return Changed;
167}
168
169static bool rescheduleCanonically(unsigned &PseudoIdempotentInstCount,
170 MachineBasicBlock *MBB) {
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000171
172 bool Changed = false;
173
174 // Calculates the distance of MI from the begining of its parent BB.
175 auto getInstrIdx = [](const MachineInstr &MI) {
176 unsigned i = 0;
177 for (auto &CurMI : *MI.getParent()) {
178 if (&CurMI == &MI)
179 return i;
180 i++;
181 }
182 return ~0U;
183 };
184
185 // Pre-Populate vector of instructions to reschedule so that we don't
186 // clobber the iterator.
187 std::vector<MachineInstr *> Instructions;
188 for (auto &MI : *MBB) {
189 Instructions.push_back(&MI);
190 }
191
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000192 std::vector<MachineInstr *> PseudoIdempotentInstructions;
193 std::vector<unsigned> PhysRegDefs;
194 for (auto *II : Instructions) {
195 for (unsigned i = 1; i < II->getNumOperands(); i++) {
196 MachineOperand &MO = II->getOperand(i);
197 if (!MO.isReg())
198 continue;
199
200 if (TargetRegisterInfo::isVirtualRegister(MO.getReg()))
201 continue;
202
203 if (!MO.isDef())
204 continue;
205
206 PhysRegDefs.push_back(MO.getReg());
207 }
208 }
209
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000210 for (auto *II : Instructions) {
211 if (II->getNumOperands() == 0)
212 continue;
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000213 if (II->mayLoadOrStore())
214 continue;
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000215
216 MachineOperand &MO = II->getOperand(0);
217 if (!MO.isReg() || !TargetRegisterInfo::isVirtualRegister(MO.getReg()))
218 continue;
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000219 if (!MO.isDef())
220 continue;
221
222 bool IsPseudoIdempotent = true;
223 for (unsigned i = 1; i < II->getNumOperands(); i++) {
224
225 if (II->getOperand(i).isImm()) {
226 continue;
227 }
228
229 if (II->getOperand(i).isReg()) {
230 if (!TargetRegisterInfo::isVirtualRegister(II->getOperand(i).getReg()))
231 if (llvm::find(PhysRegDefs, II->getOperand(i).getReg()) ==
232 PhysRegDefs.end()) {
233 continue;
234 }
235 }
236
237 IsPseudoIdempotent = false;
238 break;
239 }
240
241 if (IsPseudoIdempotent) {
242 PseudoIdempotentInstructions.push_back(II);
243 continue;
244 }
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000245
246 DEBUG(dbgs() << "Operand " << 0 << " of "; II->dump(); MO.dump(););
247
248 MachineInstr *Def = II;
249 unsigned Distance = ~0U;
250 MachineInstr *UseToBringDefCloserTo = nullptr;
251 MachineRegisterInfo *MRI = &MBB->getParent()->getRegInfo();
252 for (auto &UO : MRI->use_nodbg_operands(MO.getReg())) {
253 MachineInstr *UseInst = UO.getParent();
254
255 const unsigned DefLoc = getInstrIdx(*Def);
256 const unsigned UseLoc = getInstrIdx(*UseInst);
257 const unsigned Delta = (UseLoc - DefLoc);
258
259 if (UseInst->getParent() != Def->getParent())
260 continue;
261 if (DefLoc >= UseLoc)
262 continue;
263
264 if (Delta < Distance) {
265 Distance = Delta;
266 UseToBringDefCloserTo = UseInst;
267 }
268 }
269
270 const auto BBE = MBB->instr_end();
271 MachineBasicBlock::iterator DefI = BBE;
272 MachineBasicBlock::iterator UseI = BBE;
273
274 for (auto BBI = MBB->instr_begin(); BBI != BBE; ++BBI) {
275
276 if (DefI != BBE && UseI != BBE)
277 break;
278
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000279 if (&*BBI == Def) {
280 DefI = BBI;
281 continue;
282 }
283
284 if (&*BBI == UseToBringDefCloserTo) {
285 UseI = BBI;
286 continue;
287 }
288 }
289
290 if (DefI == BBE || UseI == BBE)
291 continue;
292
293 DEBUG({
294 dbgs() << "Splicing ";
295 DefI->dump();
296 dbgs() << " right before: ";
297 UseI->dump();
298 });
299
300 Changed = true;
301 MBB->splice(UseI, MBB, DefI);
302 }
303
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000304 PseudoIdempotentInstCount = PseudoIdempotentInstructions.size();
305 DEBUG(dbgs() << "Rescheduling Idempotent Instructions Lexographically.";);
306 Changed |= rescheduleLexographically(
307 PseudoIdempotentInstructions, MBB,
308 [&]() -> MachineBasicBlock::iterator { return MBB->begin(); });
309
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000310 return Changed;
311}
312
313/// Here we find our candidates. What makes an interesting candidate?
314/// An candidate for a canonicalization tree root is normally any kind of
315/// instruction that causes side effects such as a store to memory or a copy to
316/// a physical register or a return instruction. We use these as an expression
317/// tree root that we walk inorder to build a canonical walk which should result
318/// in canoncal vreg renaming.
319static std::vector<MachineInstr *> populateCandidates(MachineBasicBlock *MBB) {
320 std::vector<MachineInstr *> Candidates;
321 MachineRegisterInfo &MRI = MBB->getParent()->getRegInfo();
322
323 for (auto II = MBB->begin(), IE = MBB->end(); II != IE; ++II) {
324 MachineInstr *MI = &*II;
325
326 bool DoesMISideEffect = false;
327
328 if (MI->getNumOperands() > 0 && MI->getOperand(0).isReg()) {
329 const unsigned Dst = MI->getOperand(0).getReg();
330 DoesMISideEffect |= !TargetRegisterInfo::isVirtualRegister(Dst);
331
332 for (auto UI = MRI.use_begin(Dst); UI != MRI.use_end(); ++UI) {
333 if (DoesMISideEffect) break;
334 DoesMISideEffect |= (UI->getParent()->getParent() != MI->getParent());
335 }
336 }
337
338 if (!MI->mayStore() && !MI->isBranch() && !DoesMISideEffect)
339 continue;
340
341 DEBUG(dbgs() << "Found Candidate: "; MI->dump(););
342 Candidates.push_back(MI);
343 }
344
345 return Candidates;
346}
347
Benjamin Kramer51ebcaa2017-11-24 14:55:41 +0000348static void doCandidateWalk(std::vector<TypedVReg> &VRegs,
349 std::queue<TypedVReg> &RegQueue,
350 std::vector<MachineInstr *> &VisitedMIs,
351 const MachineBasicBlock *MBB) {
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000352
353 const MachineFunction &MF = *MBB->getParent();
354 const MachineRegisterInfo &MRI = MF.getRegInfo();
355
356 while (!RegQueue.empty()) {
357
358 auto TReg = RegQueue.front();
359 RegQueue.pop();
360
361 if (TReg.isFrameIndex()) {
362 DEBUG(dbgs() << "Popping frame index.\n";);
363 VRegs.push_back(TypedVReg(RSE_FrameIndex));
364 continue;
365 }
366
367 assert(TReg.isReg() && "Expected vreg or physreg.");
368 unsigned Reg = TReg.getReg();
369
370 if (TargetRegisterInfo::isVirtualRegister(Reg)) {
371 DEBUG({
372 dbgs() << "Popping vreg ";
373 MRI.def_begin(Reg)->dump();
374 dbgs() << "\n";
375 });
376
377 if (!llvm::any_of(VRegs, [&](const TypedVReg &TR) {
378 return TR.isReg() && TR.getReg() == Reg;
379 })) {
380 VRegs.push_back(TypedVReg(Reg));
381 }
382 } else {
383 DEBUG(dbgs() << "Popping physreg.\n";);
384 VRegs.push_back(TypedVReg(Reg));
385 continue;
386 }
387
388 for (auto RI = MRI.def_begin(Reg), RE = MRI.def_end(); RI != RE; ++RI) {
389 MachineInstr *Def = RI->getParent();
390
391 if (Def->getParent() != MBB)
392 continue;
393
394 if (llvm::any_of(VisitedMIs,
395 [&](const MachineInstr *VMI) { return Def == VMI; })) {
396 break;
397 }
398
399 DEBUG({
400 dbgs() << "\n========================\n";
401 dbgs() << "Visited MI: ";
402 Def->dump();
403 dbgs() << "BB Name: " << Def->getParent()->getName() << "\n";
404 dbgs() << "\n========================\n";
405 });
406 VisitedMIs.push_back(Def);
407 for (unsigned I = 1, E = Def->getNumOperands(); I != E; ++I) {
408
409 MachineOperand &MO = Def->getOperand(I);
410 if (MO.isFI()) {
411 DEBUG(dbgs() << "Pushing frame index.\n";);
412 RegQueue.push(TypedVReg(RSE_FrameIndex));
413 }
414
415 if (!MO.isReg())
416 continue;
417 RegQueue.push(TypedVReg(MO.getReg()));
418 }
419 }
420 }
421}
422
423// TODO: Work to remove this in the future. One day when we have named vregs
424// we should be able to form the canonical name based on some characteristic
425// we see in that point of the expression tree (like if we were to name based
426// on some sort of value numbering scheme).
427static void SkipVRegs(unsigned &VRegGapIndex, MachineRegisterInfo &MRI,
428 const TargetRegisterClass *RC) {
429 const unsigned VR_GAP = (++VRegGapIndex * 1000);
430
431 DEBUG({
432 dbgs() << "Adjusting per-BB VR_GAP for BB" << VRegGapIndex << " to "
433 << VR_GAP << "\n";
434 });
435
436 unsigned I = MRI.createVirtualRegister(RC);
437 const unsigned E = (((I + VR_GAP) / VR_GAP) + 1) * VR_GAP;
438 while (I != E) {
439 I = MRI.createVirtualRegister(RC);
440 }
441}
442
443static std::map<unsigned, unsigned>
444GetVRegRenameMap(const std::vector<TypedVReg> &VRegs,
445 const std::vector<unsigned> &renamedInOtherBB,
446 MachineRegisterInfo &MRI,
447 const TargetRegisterClass *RC) {
448 std::map<unsigned, unsigned> VRegRenameMap;
449 unsigned LastRenameReg = MRI.createVirtualRegister(RC);
450 bool FirstCandidate = true;
451
452 for (auto &vreg : VRegs) {
453 if (vreg.isFrameIndex()) {
454 // We skip one vreg for any frame index because there is a good chance
455 // (especially when comparing SelectionDAG to GlobalISel generated MIR)
456 // that in the other file we are just getting an incoming vreg that comes
457 // from a copy from a frame index. So it's safe to skip by one.
458 LastRenameReg = MRI.createVirtualRegister(RC);
459 DEBUG(dbgs() << "Skipping rename for FI " << LastRenameReg << "\n";);
460 continue;
461 } else if (vreg.isCandidate()) {
462
463 // After the first candidate, for every subsequent candidate, we skip mod
464 // 10 registers so that the candidates are more likely to start at the
465 // same vreg number making it more likely that the canonical walk from the
466 // candidate insruction. We don't need to skip from the first candidate of
467 // the BasicBlock because we already skip ahead several vregs for each BB.
468 while (LastRenameReg % 10) {
469 if (!FirstCandidate) break;
470 LastRenameReg = MRI.createVirtualRegister(RC);
471
472 DEBUG({
473 dbgs() << "Skipping rename for new candidate " << LastRenameReg
474 << "\n";
475 });
476 }
477 FirstCandidate = false;
478 continue;
479 } else if (!TargetRegisterInfo::isVirtualRegister(vreg.getReg())) {
480 LastRenameReg = MRI.createVirtualRegister(RC);
481 DEBUG({
482 dbgs() << "Skipping rename for Phys Reg " << LastRenameReg << "\n";
483 });
484 continue;
485 }
486
487 auto Reg = vreg.getReg();
488 if (llvm::find(renamedInOtherBB, Reg) != renamedInOtherBB.end()) {
489 DEBUG(dbgs() << "Vreg " << Reg << " already renamed in other BB.\n";);
490 continue;
491 }
492
493 auto Rename = MRI.createVirtualRegister(MRI.getRegClass(Reg));
494 LastRenameReg = Rename;
495
496 if (VRegRenameMap.find(Reg) == VRegRenameMap.end()) {
497 DEBUG(dbgs() << "Mapping vreg ";);
498 if (MRI.reg_begin(Reg) != MRI.reg_end()) {
499 DEBUG(auto foo = &*MRI.reg_begin(Reg); foo->dump(););
500 } else {
501 DEBUG(dbgs() << Reg;);
502 }
503 DEBUG(dbgs() << " to ";);
504 if (MRI.reg_begin(Rename) != MRI.reg_end()) {
505 DEBUG(auto foo = &*MRI.reg_begin(Rename); foo->dump(););
506 } else {
507 DEBUG(dbgs() << Rename;);
508 }
509 DEBUG(dbgs() << "\n";);
510
511 VRegRenameMap.insert(std::pair<unsigned, unsigned>(Reg, Rename));
512 }
513 }
514
515 return VRegRenameMap;
516}
517
518static bool doVRegRenaming(std::vector<unsigned> &RenamedInOtherBB,
519 const std::map<unsigned, unsigned> &VRegRenameMap,
520 MachineRegisterInfo &MRI) {
521 bool Changed = false;
522 for (auto I = VRegRenameMap.begin(), E = VRegRenameMap.end(); I != E; ++I) {
523
524 auto VReg = I->first;
525 auto Rename = I->second;
526
527 RenamedInOtherBB.push_back(Rename);
528
529 std::vector<MachineOperand *> RenameMOs;
530 for (auto &MO : MRI.reg_operands(VReg)) {
531 RenameMOs.push_back(&MO);
532 }
533
534 for (auto *MO : RenameMOs) {
535 Changed = true;
536 MO->setReg(Rename);
537
538 if (!MO->isDef())
539 MO->setIsKill(false);
540 }
541 }
542
543 return Changed;
544}
545
546static bool doDefKillClear(MachineBasicBlock *MBB) {
547 bool Changed = false;
548
549 for (auto &MI : *MBB) {
550 for (auto &MO : MI.operands()) {
551 if (!MO.isReg())
552 continue;
553 if (!MO.isDef() && MO.isKill()) {
554 Changed = true;
555 MO.setIsKill(false);
556 }
557
558 if (MO.isDef() && MO.isDead()) {
559 Changed = true;
560 MO.setIsDead(false);
561 }
562 }
563 }
564
565 return Changed;
566}
567
568static bool runOnBasicBlock(MachineBasicBlock *MBB,
569 std::vector<StringRef> &bbNames,
570 std::vector<unsigned> &renamedInOtherBB,
571 unsigned &basicBlockNum, unsigned &VRegGapIndex) {
572
573 if (CanonicalizeBasicBlockNumber != ~0U) {
574 if (CanonicalizeBasicBlockNumber != basicBlockNum++)
575 return false;
576 DEBUG(dbgs() << "\n Canonicalizing BasicBlock " << MBB->getName() << "\n";);
577 }
578
579 if (llvm::find(bbNames, MBB->getName()) != bbNames.end()) {
580 DEBUG({
581 dbgs() << "Found potentially duplicate BasicBlocks: " << MBB->getName()
582 << "\n";
583 });
584 return false;
585 }
586
587 DEBUG({
588 dbgs() << "\n\n NEW BASIC BLOCK: " << MBB->getName() << " \n\n";
589 dbgs() << "\n\n================================================\n\n";
590 });
591
592 bool Changed = false;
593 MachineFunction &MF = *MBB->getParent();
594 MachineRegisterInfo &MRI = MF.getRegInfo();
595
596 const unsigned DummyVReg = GetDummyVReg(MF);
597 const TargetRegisterClass *DummyRC =
598 (DummyVReg == ~0U) ? nullptr : MRI.getRegClass(DummyVReg);
599 if (!DummyRC) return false;
600
601 bbNames.push_back(MBB->getName());
602 DEBUG(dbgs() << "\n\n NEW BASIC BLOCK: " << MBB->getName() << "\n\n";);
603
604 DEBUG(dbgs() << "MBB Before Scheduling:\n"; MBB->dump(););
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000605 unsigned IdempotentInstCount = 0;
606 Changed |= rescheduleCanonically(IdempotentInstCount, MBB);
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000607 DEBUG(dbgs() << "MBB After Scheduling:\n"; MBB->dump(););
608
609 std::vector<MachineInstr *> Candidates = populateCandidates(MBB);
610 std::vector<MachineInstr *> VisitedMIs;
611 std::copy(Candidates.begin(), Candidates.end(),
612 std::back_inserter(VisitedMIs));
613
614 std::vector<TypedVReg> VRegs;
615 for (auto candidate : Candidates) {
616 VRegs.push_back(TypedVReg(RSE_NewCandidate));
617
618 std::queue<TypedVReg> RegQueue;
619
620 // Here we walk the vreg operands of a non-root node along our walk.
621 // The root nodes are the original candidates (stores normally).
622 // These are normally not the root nodes (except for the case of copies to
623 // physical registers).
624 for (unsigned i = 1; i < candidate->getNumOperands(); i++) {
625 if (candidate->mayStore() || candidate->isBranch())
626 break;
627
628 MachineOperand &MO = candidate->getOperand(i);
629 if (!(MO.isReg() && TargetRegisterInfo::isVirtualRegister(MO.getReg())))
630 continue;
631
632 DEBUG(dbgs() << "Enqueue register"; MO.dump(); dbgs() << "\n";);
633 RegQueue.push(TypedVReg(MO.getReg()));
634 }
635
636 // Here we walk the root candidates. We start from the 0th operand because
637 // the root is normally a store to a vreg.
638 for (unsigned i = 0; i < candidate->getNumOperands(); i++) {
639
640 if (!candidate->mayStore() && !candidate->isBranch())
641 break;
642
643 MachineOperand &MO = candidate->getOperand(i);
644
645 // TODO: Do we want to only add vregs here?
646 if (!MO.isReg() && !MO.isFI())
647 continue;
648
649 DEBUG(dbgs() << "Enqueue Reg/FI"; MO.dump(); dbgs() << "\n";);
650
651 RegQueue.push(MO.isReg() ? TypedVReg(MO.getReg()) :
652 TypedVReg(RSE_FrameIndex));
653 }
654
655 doCandidateWalk(VRegs, RegQueue, VisitedMIs, MBB);
656 }
657
658 // If we have populated no vregs to rename then bail.
659 // The rest of this function does the vreg remaping.
660 if (VRegs.size() == 0)
661 return Changed;
662
663 // Skip some vregs, so we can recon where we'll land next.
664 SkipVRegs(VRegGapIndex, MRI, DummyRC);
665
666 auto VRegRenameMap = GetVRegRenameMap(VRegs, renamedInOtherBB, MRI, DummyRC);
667 Changed |= doVRegRenaming(renamedInOtherBB, VRegRenameMap, MRI);
Puyan Lotfi57c4f382018-03-31 05:48:51 +0000668
669 // Here we renumber the def vregs for the idempotent instructions from the top
670 // of the MachineBasicBlock so that they are named in the order that we sorted
671 // them alphabetically. Eventually we wont need SkipVRegs because we will use
672 // named vregs instead.
673 unsigned gap = 1;
674 SkipVRegs(gap, MRI, DummyRC);
675
676 auto MII = MBB->begin();
677 for (unsigned i = 0; i < IdempotentInstCount && MII != MBB->end(); ++i) {
678 MachineInstr &MI = *MII++;
679 Changed = true;
680 unsigned vRegToRename = MI.getOperand(0).getReg();
681 auto Rename = MRI.createVirtualRegister(MRI.getRegClass(vRegToRename));
682
683 std::vector<MachineOperand *> RenameMOs;
684 for (auto &MO : MRI.reg_operands(vRegToRename)) {
685 RenameMOs.push_back(&MO);
686 }
687
688 for (auto *MO : RenameMOs) {
689 MO->setReg(Rename);
690 }
691 }
692
Puyan Lotfia521c4ac52017-11-02 23:37:32 +0000693 Changed |= doDefKillClear(MBB);
694
695 DEBUG(dbgs() << "Updated MachineBasicBlock:\n"; MBB->dump(); dbgs() << "\n";);
696 DEBUG(dbgs() << "\n\n================================================\n\n");
697 return Changed;
698}
699
700bool MIRCanonicalizer::runOnMachineFunction(MachineFunction &MF) {
701
702 static unsigned functionNum = 0;
703 if (CanonicalizeFunctionNumber != ~0U) {
704 if (CanonicalizeFunctionNumber != functionNum++)
705 return false;
706 DEBUG(dbgs() << "\n Canonicalizing Function " << MF.getName() << "\n";);
707 }
708
709 // we need a valid vreg to create a vreg type for skipping all those
710 // stray vreg numbers so reach alignment/canonical vreg values.
711 std::vector<MachineBasicBlock*> RPOList = GetRPOList(MF);
712
713 DEBUG(
714 dbgs() << "\n\n NEW MACHINE FUNCTION: " << MF.getName() << " \n\n";
715 dbgs() << "\n\n================================================\n\n";
716 dbgs() << "Total Basic Blocks: " << RPOList.size() << "\n";
717 for (auto MBB : RPOList) {
718 dbgs() << MBB->getName() << "\n";
719 }
720 dbgs() << "\n\n================================================\n\n";
721 );
722
723 std::vector<StringRef> BBNames;
724 std::vector<unsigned> RenamedInOtherBB;
725
726 unsigned GapIdx = 0;
727 unsigned BBNum = 0;
728
729 bool Changed = false;
730
731 for (auto MBB : RPOList)
732 Changed |= runOnBasicBlock(MBB, BBNames, RenamedInOtherBB, BBNum, GapIdx);
733
734 return Changed;
735}
736