blob: 9c0bec0f2d1901a2369a38fd93b672e6261eb267 [file] [log] [blame]
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +00001//===-- llvm/CodeGen/MachineBasicBlock.cpp ----------------------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-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 Evlogimenos14f3fe82004-02-16 07:17:43 +00007//
8//===----------------------------------------------------------------------===//
9//
10// Collect the sequence of machine instructions for a basic block.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/CodeGen/MachineBasicBlock.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000015#include "llvm/ADT/SmallPtrSet.h"
Matthias Braunf8422972017-12-13 02:51:04 +000016#include "llvm/CodeGen/LiveIntervals.h"
Dan Gohman3570f812010-06-22 17:25:57 +000017#include "llvm/CodeGen/LiveVariables.h"
18#include "llvm/CodeGen/MachineDominators.h"
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +000019#include "llvm/CodeGen/MachineFunction.h"
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +000020#include "llvm/CodeGen/MachineInstrBuilder.h"
Dan Gohman3570f812010-06-22 17:25:57 +000021#include "llvm/CodeGen/MachineLoopInfo.h"
Cameron Zwarichb47fb382013-02-11 09:24:47 +000022#include "llvm/CodeGen/MachineRegisterInfo.h"
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +000023#include "llvm/CodeGen/SlotIndexes.h"
David Blaikie3f833ed2017-11-08 01:01:31 +000024#include "llvm/CodeGen/TargetInstrInfo.h"
David Blaikieb3bde2e2017-11-17 01:07:10 +000025#include "llvm/CodeGen/TargetRegisterInfo.h"
26#include "llvm/CodeGen/TargetSubtargetInfo.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000027#include "llvm/IR/BasicBlock.h"
28#include "llvm/IR/DataLayout.h"
Taewook Oh06a21282017-02-13 18:15:31 +000029#include "llvm/IR/DebugInfoMetadata.h"
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +000030#include "llvm/IR/ModuleSlotTracker.h"
Chris Lattnerd051af72010-01-26 04:55:51 +000031#include "llvm/MC/MCAsmInfo.h"
32#include "llvm/MC/MCContext.h"
Cong Hou663dd012015-12-13 09:52:14 +000033#include "llvm/Support/DataTypes.h"
David Greene6c56cef2010-01-04 23:22:07 +000034#include "llvm/Support/Debug.h"
Daniel Dunbar796e43e2009-07-24 10:36:58 +000035#include "llvm/Support/raw_ostream.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000036#include "llvm/Target/TargetMachine.h"
Chris Lattner4336b872004-10-26 15:43:42 +000037#include <algorithm>
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +000038using namespace llvm;
39
Chandler Carruthe96dd892014-04-21 22:55:11 +000040#define DEBUG_TYPE "codegen"
41
Cong Hou166e0852015-09-29 19:46:09 +000042MachineBasicBlock::MachineBasicBlock(MachineFunction &MF, const BasicBlock *B)
43 : BB(B), Number(-1), xParent(&MF) {
Dan Gohman804c95d2008-07-28 21:51:04 +000044 Insts.Parent = this;
Hiroshi Yamauchidce9def2017-11-02 22:26:51 +000045 if (B)
46 IrrLoopHeaderWeight = B->getIrrLoopHeaderWeight();
Tanya Lattner91fa3a92004-05-24 07:14:35 +000047}
Tanya Lattner91fa3a92004-05-24 07:14:35 +000048
Dan Gohman0ece9432008-07-17 23:49:46 +000049MachineBasicBlock::~MachineBasicBlock() {
Dan Gohman0ece9432008-07-17 23:49:46 +000050}
51
Cong Hou2a02c1c2015-08-12 21:18:54 +000052/// Return the MCSymbol for this basic block.
Chris Lattner29bdac42010-03-13 21:04:28 +000053MCSymbol *MachineBasicBlock::getSymbol() const {
Eli Bendersky58b04b72013-04-22 21:21:08 +000054 if (!CachedMCSymbol) {
55 const MachineFunction *MF = getParent();
56 MCContext &Ctx = MF->getContext();
Mehdi Amini36d33fc2016-10-01 06:46:33 +000057 auto Prefix = Ctx.getAsmInfo()->getPrivateLabelPrefix();
Reid Klecknerb9204a52015-11-11 23:09:31 +000058 assert(getNumber() >= 0 && "cannot get label for unreachable MBB");
Jim Grosbach6f482002015-05-18 18:43:14 +000059 CachedMCSymbol = Ctx.getOrCreateSymbol(Twine(Prefix) + "BB" +
Eli Bendersky58b04b72013-04-22 21:21:08 +000060 Twine(MF->getFunctionNumber()) +
61 "_" + Twine(getNumber()));
62 }
63
64 return CachedMCSymbol;
Chris Lattnerd051af72010-01-26 04:55:51 +000065}
66
67
Chris Lattneraf119ca2009-08-23 00:35:30 +000068raw_ostream &llvm::operator<<(raw_ostream &OS, const MachineBasicBlock &MBB) {
Daniel Dunbar796e43e2009-07-24 10:36:58 +000069 MBB.print(OS);
70 return OS;
71}
Tanya Lattner91fa3a92004-05-24 07:14:35 +000072
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +000073Printable llvm::printMBBReference(const MachineBasicBlock &MBB) {
74 return Printable([&MBB](raw_ostream &OS) { return MBB.printAsOperand(OS); });
75}
76
Cong Hou2a02c1c2015-08-12 21:18:54 +000077/// When an MBB is added to an MF, we need to update the parent pointer of the
78/// MBB, the MBB numbering, and any instructions in the MBB to be on the right
79/// operand list for registers.
Chris Lattner961e7422008-01-01 01:12:31 +000080///
81/// MBBs start out as #-1. When a MBB is added to a MachineFunction, it
82/// gets the next available unique MBB number. If it is removed from a
83/// MachineFunction, it goes back to being #-1.
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +000084void ilist_callback_traits<MachineBasicBlock>::addNodeToList(
85 MachineBasicBlock *N) {
Dan Gohman3b460302008-07-07 23:14:23 +000086 MachineFunction &MF = *N->getParent();
87 N->Number = MF.addToMBBNumbering(N);
Chris Lattner961e7422008-01-01 01:12:31 +000088
89 // Make sure the instructions have their operands in the reginfo lists.
Dan Gohman3b460302008-07-07 23:14:23 +000090 MachineRegisterInfo &RegInfo = MF.getRegInfo();
Evan Cheng7fae11b2011-12-14 02:11:42 +000091 for (MachineBasicBlock::instr_iterator
92 I = N->instr_begin(), E = N->instr_end(); I != E; ++I)
Chris Lattner961e7422008-01-01 01:12:31 +000093 I->AddRegOperandsToUseLists(RegInfo);
Brian Gaekecb5d22a2004-05-12 21:35:22 +000094}
95
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +000096void ilist_callback_traits<MachineBasicBlock>::removeNodeFromList(
97 MachineBasicBlock *N) {
Chris Lattner574e7162007-12-31 04:56:33 +000098 N->getParent()->removeFromMBBNumbering(N->Number);
Brian Gaekecb5d22a2004-05-12 21:35:22 +000099 N->Number = -1;
100}
101
Cong Hou2a02c1c2015-08-12 21:18:54 +0000102/// When we add an instruction to a basic block list, we update its parent
103/// pointer and add its operands from reg use/def lists if appropriate.
Chris Lattneraf119ca2009-08-23 00:35:30 +0000104void ilist_traits<MachineInstr>::addNodeToList(MachineInstr *N) {
Craig Topperc0196b12014-04-14 00:51:57 +0000105 assert(!N->getParent() && "machine instruction already in a basic block");
Dan Gohman3b460302008-07-07 23:14:23 +0000106 N->setParent(Parent);
Jakub Staszakfeadd432011-06-16 18:01:17 +0000107
Dan Gohman3b460302008-07-07 23:14:23 +0000108 // Add the instruction's register operands to their corresponding
109 // use/def lists.
110 MachineFunction *MF = Parent->getParent();
111 N->AddRegOperandsToUseLists(MF->getRegInfo());
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000112}
113
Cong Hou2a02c1c2015-08-12 21:18:54 +0000114/// When we remove an instruction from a basic block list, we update its parent
115/// pointer and remove its operands from reg use/def lists if appropriate.
Chris Lattneraf119ca2009-08-23 00:35:30 +0000116void ilist_traits<MachineInstr>::removeNodeFromList(MachineInstr *N) {
Craig Topperc0196b12014-04-14 00:51:57 +0000117 assert(N->getParent() && "machine instruction not in a basic block");
Dan Gohman3b460302008-07-07 23:14:23 +0000118
119 // Remove from the use/def lists.
Justin Bognerfdf9bf42017-10-10 23:50:49 +0000120 if (MachineFunction *MF = N->getMF())
Jakob Stoklund Olesenc4102d42012-08-09 22:49:37 +0000121 N->RemoveRegOperandsFromUseLists(MF->getRegInfo());
Jakub Staszakfeadd432011-06-16 18:01:17 +0000122
Craig Topperc0196b12014-04-14 00:51:57 +0000123 N->setParent(nullptr);
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000124}
125
Cong Hou2a02c1c2015-08-12 21:18:54 +0000126/// When moving a range of instructions from one MBB list to another, we need to
127/// update the parent pointers and the use/def lists.
Duncan P. N. Exon Smithcc9edac2016-09-11 16:38:18 +0000128void ilist_traits<MachineInstr>::transferNodesFromList(ilist_traits &FromList,
129 instr_iterator First,
130 instr_iterator Last) {
Cong Hou166e0852015-09-29 19:46:09 +0000131 assert(Parent->getParent() == FromList.Parent->getParent() &&
Dan Gohman804c95d2008-07-28 21:51:04 +0000132 "MachineInstr parent mismatch!");
Duncan P. N. Exon Smithb7668d52016-08-30 18:00:45 +0000133 assert(this != &FromList && "Called without a real transfer...");
134 assert(Parent != FromList.Parent && "Two lists have the same parent?");
Chris Lattner961e7422008-01-01 01:12:31 +0000135
136 // If splicing between two blocks within the same function, just update the
137 // parent pointers.
Cong Hou166e0852015-09-29 19:46:09 +0000138 for (; First != Last; ++First)
139 First->setParent(Parent);
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000140}
141
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +0000142void ilist_traits<MachineInstr>::deleteNode(MachineInstr *MI) {
Dan Gohman3b460302008-07-07 23:14:23 +0000143 assert(!MI->getParent() && "MI is still in a block!");
144 Parent->getParent()->DeleteMachineInstr(MI);
145}
146
Dan Gohman88c547e2010-07-07 14:33:51 +0000147MachineBasicBlock::iterator MachineBasicBlock::getFirstNonPHI() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000148 instr_iterator I = instr_begin(), E = instr_end();
149 while (I != E && I->isPHI())
Dan Gohman88c547e2010-07-07 14:33:51 +0000150 ++I;
Akira Hatanaka6fe7aca2012-10-26 17:11:42 +0000151 assert((I == E || !I->isInsideBundle()) &&
152 "First non-phi MI cannot be inside a bundle!");
Dan Gohman88c547e2010-07-07 14:33:51 +0000153 return I;
154}
155
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000156MachineBasicBlock::iterator
157MachineBasicBlock::SkipPHIsAndLabels(MachineBasicBlock::iterator I) {
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000158 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
159
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000160 iterator E = end();
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000161 while (I != E && (I->isPHI() || I->isPosition() ||
162 TII->isBasicBlockPrologue(*I)))
Keith Walker830a8c12016-09-16 14:07:29 +0000163 ++I;
164 // FIXME: This needs to change if we wish to bundle labels
165 // inside the bundle.
166 assert((I == E || !I->isInsideBundle()) &&
167 "First non-phi / non-label instruction is inside a bundle!");
168 return I;
169}
170
171MachineBasicBlock::iterator
172MachineBasicBlock::SkipPHIsLabelsAndDebug(MachineBasicBlock::iterator I) {
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000173 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
174
Keith Walker830a8c12016-09-16 14:07:29 +0000175 iterator E = end();
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000176 while (I != E && (I->isPHI() || I->isPosition() || I->isDebugValue() ||
177 TII->isBasicBlockPrologue(*I)))
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000178 ++I;
Evan Cheng2a81dd42011-12-06 22:12:01 +0000179 // FIXME: This needs to change if we wish to bundle labels / dbg_values
180 // inside the bundle.
Akira Hatanaka6fe7aca2012-10-26 17:11:42 +0000181 assert((I == E || !I->isInsideBundle()) &&
Keith Walker830a8c12016-09-16 14:07:29 +0000182 "First non-phi / non-label / non-debug "
183 "instruction is inside a bundle!");
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000184 return I;
185}
186
Chris Lattner4336b872004-10-26 15:43:42 +0000187MachineBasicBlock::iterator MachineBasicBlock::getFirstTerminator() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000188 iterator B = begin(), E = end(), I = E;
189 while (I != B && ((--I)->isTerminator() || I->isDebugValue()))
Jakob Stoklund Olesenc3810282011-01-14 02:12:54 +0000190 ; /*noop */
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000191 while (I != E && !I->isTerminator())
Evan Cheng2a81dd42011-12-06 22:12:01 +0000192 ++I;
193 return I;
194}
195
Evan Cheng7fae11b2011-12-14 02:11:42 +0000196MachineBasicBlock::instr_iterator MachineBasicBlock::getFirstInstrTerminator() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000197 instr_iterator B = instr_begin(), E = instr_end(), I = E;
198 while (I != B && ((--I)->isTerminator() || I->isDebugValue()))
Evan Cheng2a81dd42011-12-06 22:12:01 +0000199 ; /*noop */
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000200 while (I != E && !I->isTerminator())
Jakob Stoklund Olesenab3d6ec2011-01-14 06:33:45 +0000201 ++I;
Jakob Stoklund Olesenc3810282011-01-14 02:12:54 +0000202 return I;
Alkis Evlogimenosaf2de482004-02-23 18:14:48 +0000203}
204
Benjamin Kramer6b568962015-06-23 14:47:29 +0000205MachineBasicBlock::iterator MachineBasicBlock::getFirstNonDebugInstr() {
206 // Skip over begin-of-block dbg_value instructions.
Florian Hahn3c8b8c92016-12-16 11:10:26 +0000207 return skipDebugInstructionsForward(begin(), end());
Benjamin Kramer6b568962015-06-23 14:47:29 +0000208}
209
Jakob Stoklund Olesen4bc5e382011-01-13 21:28:52 +0000210MachineBasicBlock::iterator MachineBasicBlock::getLastNonDebugInstr() {
Evan Cheng2a81dd42011-12-06 22:12:01 +0000211 // Skip over end-of-block dbg_value instructions.
Evan Cheng7fae11b2011-12-14 02:11:42 +0000212 instr_iterator B = instr_begin(), I = instr_end();
Jakob Stoklund Olesen4bc5e382011-01-13 21:28:52 +0000213 while (I != B) {
214 --I;
Evan Cheng2a81dd42011-12-06 22:12:01 +0000215 // Return instruction that starts a bundle.
216 if (I->isDebugValue() || I->isInsideBundle())
217 continue;
218 return I;
219 }
220 // The block is all debug values.
221 return end();
222}
223
Reid Klecknered170792015-09-17 17:19:40 +0000224bool MachineBasicBlock::hasEHPadSuccessor() const {
225 for (const_succ_iterator I = succ_begin(), E = succ_end(); I != E; ++I)
226 if ((*I)->isEHPad())
227 return true;
228 return false;
229}
230
Aaron Ballman615eb472017-10-15 14:32:27 +0000231#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
Yaron Kereneb2a2542016-01-29 20:50:44 +0000232LLVM_DUMP_METHOD void MachineBasicBlock::dump() const {
David Greene6c56cef2010-01-04 23:22:07 +0000233 print(dbgs());
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000234}
Manman Ren742534c2012-09-06 19:06:06 +0000235#endif
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000236
Andrew Kaylord4971192017-06-22 23:27:16 +0000237bool MachineBasicBlock::isLegalToHoistInto() const {
238 if (isReturnBlock() || hasEHPadSuccessor())
239 return false;
240 return true;
241}
242
Jakob Stoklund Olesen2bbeaa82009-11-20 01:17:03 +0000243StringRef MachineBasicBlock::getName() const {
244 if (const BasicBlock *LBB = getBasicBlock())
245 return LBB->getName();
246 else
Matthias Braun43130592017-02-18 00:41:16 +0000247 return StringRef("", 0);
Jakob Stoklund Olesen2bbeaa82009-11-20 01:17:03 +0000248}
249
Andrew Trick320c7032012-03-07 00:18:18 +0000250/// Return a hopefully unique identifier for this block.
251std::string MachineBasicBlock::getFullName() const {
252 std::string Name;
253 if (getParent())
Craig Toppera538d832012-08-22 06:07:19 +0000254 Name = (getParent()->getName() + ":").str();
Andrew Trick320c7032012-03-07 00:18:18 +0000255 if (getBasicBlock())
256 Name += getBasicBlock()->getName();
257 else
Yaron Keren75e0c4b2015-03-27 17:51:30 +0000258 Name += ("BB" + Twine(getNumber())).str();
Andrew Trick320c7032012-03-07 00:18:18 +0000259 return Name;
260}
261
Francis Visoiu Mistrih378b5f32018-01-18 17:59:06 +0000262void MachineBasicBlock::print(raw_ostream &OS, const SlotIndexes *Indexes,
263 bool IsVerbose) const {
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000264 const MachineFunction *MF = getParent();
Chris Lattneraf119ca2009-08-23 00:35:30 +0000265 if (!MF) {
Chris Lattner4336b872004-10-26 15:43:42 +0000266 OS << "Can't print out MachineBasicBlock because parent MachineFunction"
267 << " is null\n";
Tanya Lattnera578cb72004-05-24 06:11:51 +0000268 return;
269 }
Matthias Braunf1caa282017-12-15 22:22:58 +0000270 const Function &F = MF->getFunction();
271 const Module *M = F.getParent();
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000272 ModuleSlotTracker MST(M);
Francis Visoiu Mistrih378b5f32018-01-18 17:59:06 +0000273 print(OS, MST, Indexes, IsVerbose);
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000274}
275
276void MachineBasicBlock::print(raw_ostream &OS, ModuleSlotTracker &MST,
Francis Visoiu Mistrih378b5f32018-01-18 17:59:06 +0000277 const SlotIndexes *Indexes,
278 bool IsVerbose) const {
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000279 const MachineFunction *MF = getParent();
280 if (!MF) {
281 OS << "Can't print out MachineBasicBlock because parent MachineFunction"
282 << " is null\n";
283 return;
284 }
Alkis Evlogimenosfcb3f512004-09-05 18:39:20 +0000285
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000286 if (Indexes)
287 OS << Indexes->getMBBStartIdx(this) << '\t';
288
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000289 OS << printMBBReference(*this) << ": ";
Dan Gohman34341e62009-10-31 20:19:03 +0000290
291 const char *Comma = "";
292 if (const BasicBlock *LBB = getBasicBlock()) {
293 OS << Comma << "derived from LLVM BB ";
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000294 LBB->printAsOperand(OS, /*PrintType=*/false, MST);
Dan Gohman34341e62009-10-31 20:19:03 +0000295 Comma = ", ";
296 }
Reid Kleckner0e288232015-08-27 23:27:47 +0000297 if (isEHPad()) { OS << Comma << "EH LANDING PAD"; Comma = ", "; }
Dan Gohman34341e62009-10-31 20:19:03 +0000298 if (hasAddressTaken()) { OS << Comma << "ADDRESS TAKEN"; Comma = ", "; }
Bill Wendling4fd96632012-06-15 19:30:42 +0000299 if (Alignment)
Jakob Stoklund Olesen2a2b37e2011-12-06 21:08:39 +0000300 OS << Comma << "Align " << Alignment << " (" << (1u << Alignment)
301 << " bytes)";
Jakob Stoklund Olesen2a2b37e2011-12-06 21:08:39 +0000302
Chris Lattneraf119ca2009-08-23 00:35:30 +0000303 OS << '\n';
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000304
Eric Christopherfc6de422014-08-05 02:39:49 +0000305 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
Dan Gohmanc731c972007-10-03 19:26:29 +0000306 if (!livein_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000307 if (Indexes) OS << '\t';
Dan Gohman34341e62009-10-31 20:19:03 +0000308 OS << " Live Ins:";
Matthias Braun11723322017-01-05 20:01:19 +0000309 for (const auto &LI : LiveIns) {
Francis Visoiu Mistrih9d419d32017-11-28 12:42:37 +0000310 OS << ' ' << printReg(LI.PhysReg, TRI);
Krzysztof Parzyszek91b5cf82016-12-15 14:36:06 +0000311 if (!LI.LaneMask.all())
Matthias Braunc804cdb2015-09-25 21:51:24 +0000312 OS << ':' << PrintLaneMask(LI.LaneMask);
Matthias Braunb2b7ef12015-08-24 22:59:52 +0000313 }
Chris Lattneraf119ca2009-08-23 00:35:30 +0000314 OS << '\n';
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000315 }
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000316 // Print the preds of this block according to the CFG.
317 if (!pred_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000318 if (Indexes) OS << '\t';
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000319 OS << " Predecessors according to CFG:";
320 for (const_pred_iterator PI = pred_begin(), E = pred_end(); PI != E; ++PI)
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000321 OS << " " << printMBBReference(*(*PI));
Chris Lattneraf119ca2009-08-23 00:35:30 +0000322 OS << '\n';
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000323 }
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000324
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000325 for (auto &I : instrs()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000326 if (Indexes) {
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000327 if (Indexes->hasIndex(I))
328 OS << Indexes->getInstructionIndex(I);
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000329 OS << '\t';
330 }
Chris Lattner1216f542009-08-23 00:47:04 +0000331 OS << '\t';
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000332 if (I.isInsideBundle())
Evan Cheng7fae11b2011-12-14 02:11:42 +0000333 OS << " * ";
Francis Visoiu Mistrih378b5f32018-01-18 17:59:06 +0000334 I.print(OS, MST, IsVerbose);
Alkis Evlogimenosfcb3f512004-09-05 18:39:20 +0000335 }
Chris Lattner329c14a2005-04-01 06:48:38 +0000336
337 // Print the successors of this block according to the CFG.
338 if (!succ_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000339 if (Indexes) OS << '\t';
Chris Lattner329c14a2005-04-01 06:48:38 +0000340 OS << " Successors according to CFG:";
Jakob Stoklund Olesen1dc107a2012-08-13 23:13:23 +0000341 for (const_succ_iterator SI = succ_begin(), E = succ_end(); SI != E; ++SI) {
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000342 OS << " " << printMBBReference(*(*SI));
Cong Houd97c1002015-12-01 05:29:22 +0000343 if (!Probs.empty())
344 OS << '(' << *getProbabilityIterator(SI) << ')';
Jakob Stoklund Olesen1dc107a2012-08-13 23:13:23 +0000345 }
Chris Lattneraf119ca2009-08-23 00:35:30 +0000346 OS << '\n';
Chris Lattner329c14a2005-04-01 06:48:38 +0000347 }
Hiroshi Yamauchidce9def2017-11-02 22:26:51 +0000348 if (IrrLoopHeaderWeight) {
349 if (Indexes) OS << '\t';
350 OS << " Irreducible loop header weight: "
351 << IrrLoopHeaderWeight.getValue();
352 OS << '\n';
353 }
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000354}
Chris Lattner4336b872004-10-26 15:43:42 +0000355
Cong Hou2793e722015-08-10 22:27:10 +0000356void MachineBasicBlock::printAsOperand(raw_ostream &OS,
357 bool /*PrintType*/) const {
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000358 OS << "%bb." << getNumber();
Chandler Carruthd48cdbf2014-01-09 02:29:41 +0000359}
360
Matthias Braune6a24852015-09-25 21:51:14 +0000361void MachineBasicBlock::removeLiveIn(MCPhysReg Reg, LaneBitmask LaneMask) {
David Majnemer42531262016-08-12 03:55:06 +0000362 LiveInVector::iterator I = find_if(
363 LiveIns, [Reg](const RegisterMaskPair &LI) { return LI.PhysReg == Reg; });
Matthias Braund9da1622015-09-09 18:08:03 +0000364 if (I == LiveIns.end())
365 return;
366
367 I->LaneMask &= ~LaneMask;
Krzysztof Parzyszek91b5cf82016-12-15 14:36:06 +0000368 if (I->LaneMask.none())
Jakob Stoklund Olesen8e58c902012-03-28 20:11:42 +0000369 LiveIns.erase(I);
Evan Chengf7ed82d2007-02-19 21:49:54 +0000370}
371
Matthias Braunac4becc2017-05-31 20:30:22 +0000372MachineBasicBlock::livein_iterator
373MachineBasicBlock::removeLiveIn(MachineBasicBlock::livein_iterator I) {
Matthias Braune2e65912017-05-31 21:25:03 +0000374 // Get non-const version of iterator.
375 LiveInVector::iterator LI = LiveIns.begin() + (I - LiveIns.begin());
376 return LiveIns.erase(LI);
Matthias Braunac4becc2017-05-31 20:30:22 +0000377}
378
Matthias Braune6a24852015-09-25 21:51:14 +0000379bool MachineBasicBlock::isLiveIn(MCPhysReg Reg, LaneBitmask LaneMask) const {
David Majnemer42531262016-08-12 03:55:06 +0000380 livein_iterator I = find_if(
381 LiveIns, [Reg](const RegisterMaskPair &LI) { return LI.PhysReg == Reg; });
Krzysztof Parzyszekea9f8ce2016-12-16 19:11:56 +0000382 return I != livein_end() && (I->LaneMask & LaneMask).any();
Matthias Braund9da1622015-09-09 18:08:03 +0000383}
384
385void MachineBasicBlock::sortUniqueLiveIns() {
386 std::sort(LiveIns.begin(), LiveIns.end(),
387 [](const RegisterMaskPair &LI0, const RegisterMaskPair &LI1) {
388 return LI0.PhysReg < LI1.PhysReg;
389 });
390 // Liveins are sorted by physreg now we can merge their lanemasks.
391 LiveInVector::const_iterator I = LiveIns.begin();
392 LiveInVector::const_iterator J;
393 LiveInVector::iterator Out = LiveIns.begin();
394 for (; I != LiveIns.end(); ++Out, I = J) {
395 unsigned PhysReg = I->PhysReg;
Matthias Braune6a24852015-09-25 21:51:14 +0000396 LaneBitmask LaneMask = I->LaneMask;
Matthias Braund9da1622015-09-09 18:08:03 +0000397 for (J = std::next(I); J != LiveIns.end() && J->PhysReg == PhysReg; ++J)
398 LaneMask |= J->LaneMask;
399 Out->PhysReg = PhysReg;
400 Out->LaneMask = LaneMask;
401 }
402 LiveIns.erase(Out, LiveIns.end());
Aaron Ballman9f154f62015-07-29 15:57:49 +0000403}
404
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000405unsigned
Matthias Braun130bd902015-08-25 22:05:55 +0000406MachineBasicBlock::addLiveIn(MCPhysReg PhysReg, const TargetRegisterClass *RC) {
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000407 assert(getParent() && "MBB must be inserted in function");
408 assert(TargetRegisterInfo::isPhysicalRegister(PhysReg) && "Expected physreg");
409 assert(RC && "Register class is required");
Reid Kleckner0e288232015-08-27 23:27:47 +0000410 assert((isEHPad() || this == &getParent()->front()) &&
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000411 "Only the entry block and landing pads can have physreg live ins");
412
413 bool LiveIn = isLiveIn(PhysReg);
Jakob Stoklund Olesenbbbb5322013-07-04 04:32:35 +0000414 iterator I = SkipPHIsAndLabels(begin()), E = end();
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000415 MachineRegisterInfo &MRI = getParent()->getRegInfo();
Eric Christopherfc6de422014-08-05 02:39:49 +0000416 const TargetInstrInfo &TII = *getParent()->getSubtarget().getInstrInfo();
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000417
418 // Look for an existing copy.
419 if (LiveIn)
420 for (;I != E && I->isCopy(); ++I)
421 if (I->getOperand(1).getReg() == PhysReg) {
422 unsigned VirtReg = I->getOperand(0).getReg();
423 if (!MRI.constrainRegClass(VirtReg, RC))
424 llvm_unreachable("Incompatible live-in register class.");
425 return VirtReg;
426 }
427
428 // No luck, create a virtual register.
429 unsigned VirtReg = MRI.createVirtualRegister(RC);
430 BuildMI(*this, I, DebugLoc(), TII.get(TargetOpcode::COPY), VirtReg)
431 .addReg(PhysReg, RegState::Kill);
432 if (!LiveIn)
433 addLiveIn(PhysReg);
434 return VirtReg;
435}
436
Chris Lattner94866be2006-10-24 00:02:26 +0000437void MachineBasicBlock::moveBefore(MachineBasicBlock *NewAfter) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000438 getParent()->splice(NewAfter->getIterator(), getIterator());
Chris Lattner94866be2006-10-24 00:02:26 +0000439}
440
441void MachineBasicBlock::moveAfter(MachineBasicBlock *NewBefore) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000442 getParent()->splice(++NewBefore->getIterator(), getIterator());
Chris Lattner94866be2006-10-24 00:02:26 +0000443}
444
Jim Grosbach801b33b2009-11-12 03:55:33 +0000445void MachineBasicBlock::updateTerminator() {
Eric Christopherfc6de422014-08-05 02:39:49 +0000446 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Jim Grosbach801b33b2009-11-12 03:55:33 +0000447 // A block with no successors has no concerns with fall-through edges.
Chad Rosierdca76512016-05-25 21:53:46 +0000448 if (this->succ_empty())
449 return;
Jim Grosbach801b33b2009-11-12 03:55:33 +0000450
Craig Topperc0196b12014-04-14 00:51:57 +0000451 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
Jim Grosbach801b33b2009-11-12 03:55:33 +0000452 SmallVector<MachineOperand, 4> Cond;
Taewook Oh06a21282017-02-13 18:15:31 +0000453 DebugLoc DL = findBranchDebugLoc();
Jacques Pienaar71c30a12016-07-15 14:41:04 +0000454 bool B = TII->analyzeBranch(*this, TBB, FBB, Cond);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000455 (void) B;
456 assert(!B && "UpdateTerminators requires analyzable predecessors!");
457 if (Cond.empty()) {
458 if (TBB) {
Chad Rosierdca76512016-05-25 21:53:46 +0000459 // The block has an unconditional branch. If its successor is now its
460 // layout successor, delete the branch.
Jim Grosbach801b33b2009-11-12 03:55:33 +0000461 if (isLayoutSuccessor(TBB))
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000462 TII->removeBranch(*this);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000463 } else {
Chad Rosierdca76512016-05-25 21:53:46 +0000464 // The block has an unconditional fallthrough. If its successor is not its
465 // layout successor, insert a branch. First we have to locate the only
466 // non-landing-pad successor, as that is the fallthrough block.
Chandler Carruthee54feb2011-11-22 13:13:16 +0000467 for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) {
Reid Kleckner0e288232015-08-27 23:27:47 +0000468 if ((*SI)->isEHPad())
Chandler Carruthee54feb2011-11-22 13:13:16 +0000469 continue;
470 assert(!TBB && "Found more than one non-landing-pad successor!");
471 TBB = *SI;
472 }
Chandler Carruth8c68f1f2011-11-23 08:23:54 +0000473
Chad Rosierdca76512016-05-25 21:53:46 +0000474 // If there is no non-landing-pad successor, the block has no fall-through
475 // edges to be concerned with.
Chandler Carruth8c68f1f2011-11-23 08:23:54 +0000476 if (!TBB)
477 return;
478
479 // Finally update the unconditional successor to be reached via a branch
480 // if it would not be reached by fallthrough.
Jim Grosbach801b33b2009-11-12 03:55:33 +0000481 if (!isLayoutSuccessor(TBB))
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000482 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000483 }
Chad Rosierdca76512016-05-25 21:53:46 +0000484 return;
485 }
Chandler Carruth1f5580b2012-04-16 22:03:00 +0000486
Chad Rosierdca76512016-05-25 21:53:46 +0000487 if (FBB) {
488 // The block has a non-fallthrough conditional branch. If one of its
489 // successors is its layout successor, rewrite it to a fallthrough
490 // conditional branch.
491 if (isLayoutSuccessor(TBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000492 if (TII->reverseBranchCondition(Cond))
Chandler Carruth1f5580b2012-04-16 22:03:00 +0000493 return;
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000494 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000495 TII->insertBranch(*this, FBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000496 } else if (isLayoutSuccessor(FBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000497 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000498 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000499 }
Chad Rosierdca76512016-05-25 21:53:46 +0000500 return;
501 }
502
503 // Walk through the successors and find the successor which is not a landing
504 // pad and is not the conditional branch destination (in TBB) as the
505 // fallthrough successor.
506 MachineBasicBlock *FallthroughBB = nullptr;
507 for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) {
508 if ((*SI)->isEHPad() || *SI == TBB)
509 continue;
510 assert(!FallthroughBB && "Found more than one fallthrough successor.");
511 FallthroughBB = *SI;
512 }
513
Haicheng Wub71b2f62016-07-03 19:14:17 +0000514 if (!FallthroughBB) {
515 if (canFallThrough()) {
516 // We fallthrough to the same basic block as the conditional jump targets.
517 // Remove the conditional jump, leaving unconditional fallthrough.
518 // FIXME: This does not seem like a reasonable pattern to support, but it
519 // has been seen in the wild coming out of degenerate ARM test cases.
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000520 TII->removeBranch(*this);
Taewook Oh06a21282017-02-13 18:15:31 +0000521
Haicheng Wub71b2f62016-07-03 19:14:17 +0000522 // Finally update the unconditional successor to be reached via a branch if
523 // it would not be reached by fallthrough.
524 if (!isLayoutSuccessor(TBB))
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000525 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Haicheng Wub71b2f62016-07-03 19:14:17 +0000526 return;
527 }
Chad Rosierdca76512016-05-25 21:53:46 +0000528
Haicheng Wub71b2f62016-07-03 19:14:17 +0000529 // We enter here iff exactly one successor is TBB which cannot fallthrough
530 // and the rest successors if any are EHPads. In this case, we need to
531 // change the conditional branch into unconditional branch.
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000532 TII->removeBranch(*this);
Haicheng Wub71b2f62016-07-03 19:14:17 +0000533 Cond.clear();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000534 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000535 return;
536 }
537
538 // The block has a fallthrough conditional branch.
539 if (isLayoutSuccessor(TBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000540 if (TII->reverseBranchCondition(Cond)) {
Chad Rosierdca76512016-05-25 21:53:46 +0000541 // We can't reverse the condition, add an unconditional branch.
542 Cond.clear();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000543 TII->insertBranch(*this, FallthroughBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000544 return;
545 }
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000546 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000547 TII->insertBranch(*this, FallthroughBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000548 } else if (!isLayoutSuccessor(FallthroughBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000549 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000550 TII->insertBranch(*this, TBB, FallthroughBB, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000551 }
552}
Chris Lattner94866be2006-10-24 00:02:26 +0000553
Cong Houc1069892015-12-13 09:26:17 +0000554void MachineBasicBlock::validateSuccProbs() const {
555#ifndef NDEBUG
556 int64_t Sum = 0;
557 for (auto Prob : Probs)
558 Sum += Prob.getNumerator();
559 // Due to precision issue, we assume that the sum of probabilities is one if
560 // the difference between the sum of their numerators and the denominator is
561 // no greater than the number of successors.
Cong Houc00e65a2015-12-13 17:00:25 +0000562 assert((uint64_t)std::abs(Sum - BranchProbability::getDenominator()) <=
Cong Houc1069892015-12-13 09:26:17 +0000563 Probs.size() &&
564 "The sum of successors's probabilities exceeds one.");
565#endif // NDEBUG
566}
567
Cong Hou23a3bf02015-11-04 21:37:58 +0000568void MachineBasicBlock::addSuccessor(MachineBasicBlock *Succ,
569 BranchProbability Prob) {
570 // Probability list is either empty (if successor list isn't empty, this means
571 // disabled optimization) or has the same size as successor list.
Cong Hou4aef7ef2015-12-01 11:05:39 +0000572 if (!(Probs.empty() && !Successors.empty()))
Cong Hou23a3bf02015-11-04 21:37:58 +0000573 Probs.push_back(Prob);
574 Successors.push_back(Succ);
575 Succ->addPredecessor(this);
576}
577
578void MachineBasicBlock::addSuccessorWithoutProb(MachineBasicBlock *Succ) {
579 // We need to make sure probability list is either empty or has the same size
580 // of successor list. When this function is called, we can safely delete all
581 // probability in the list.
582 Probs.clear();
583 Successors.push_back(Succ);
584 Succ->addPredecessor(this);
585}
586
Cong Houc1069892015-12-13 09:26:17 +0000587void MachineBasicBlock::removeSuccessor(MachineBasicBlock *Succ,
588 bool NormalizeSuccProbs) {
David Majnemer0d955d02016-08-11 22:21:41 +0000589 succ_iterator I = find(Successors, Succ);
Cong Houc1069892015-12-13 09:26:17 +0000590 removeSuccessor(I, NormalizeSuccProbs);
Chris Lattner4336b872004-10-26 15:43:42 +0000591}
592
Jakub Staszakfeadd432011-06-16 18:01:17 +0000593MachineBasicBlock::succ_iterator
Cong Houc1069892015-12-13 09:26:17 +0000594MachineBasicBlock::removeSuccessor(succ_iterator I, bool NormalizeSuccProbs) {
Chris Lattner4336b872004-10-26 15:43:42 +0000595 assert(I != Successors.end() && "Not a current successor!");
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000596
Cong Hou23a3bf02015-11-04 21:37:58 +0000597 // If probability list is empty it means we don't use it (disabled
598 // optimization).
599 if (!Probs.empty()) {
600 probability_iterator WI = getProbabilityIterator(I);
601 Probs.erase(WI);
Cong Houc1069892015-12-13 09:26:17 +0000602 if (NormalizeSuccProbs)
603 normalizeSuccProbs();
Cong Hou23a3bf02015-11-04 21:37:58 +0000604 }
605
Chris Lattner4336b872004-10-26 15:43:42 +0000606 (*I)->removePredecessor(this);
Dan Gohmanf87dc922009-01-08 22:19:34 +0000607 return Successors.erase(I);
Chris Lattner4336b872004-10-26 15:43:42 +0000608}
609
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000610void MachineBasicBlock::replaceSuccessor(MachineBasicBlock *Old,
611 MachineBasicBlock *New) {
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000612 if (Old == New)
613 return;
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000614
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000615 succ_iterator E = succ_end();
616 succ_iterator NewI = E;
617 succ_iterator OldI = E;
618 for (succ_iterator I = succ_begin(); I != E; ++I) {
619 if (*I == Old) {
620 OldI = I;
621 if (NewI != E)
622 break;
623 }
624 if (*I == New) {
625 NewI = I;
626 if (OldI != E)
627 break;
628 }
629 }
630 assert(OldI != E && "Old is not a successor of this block");
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000631
632 // If New isn't already a successor, let it take Old's place.
633 if (NewI == E) {
Cong Hou11c14202015-11-18 01:45:10 +0000634 Old->removePredecessor(this);
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000635 New->addPredecessor(this);
636 *OldI = New;
637 return;
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000638 }
639
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000640 // New is already a successor.
Cong Hou23a3bf02015-11-04 21:37:58 +0000641 // Update its probability instead of adding a duplicate edge.
Cong Houd97c1002015-12-01 05:29:22 +0000642 if (!Probs.empty()) {
643 auto ProbIter = getProbabilityIterator(NewI);
644 if (!ProbIter->isUnknown())
645 *ProbIter += *getProbabilityIterator(OldI);
646 }
Cong Hou11c14202015-11-18 01:45:10 +0000647 removeSuccessor(OldI);
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000648}
649
Cong Hou166e0852015-09-29 19:46:09 +0000650void MachineBasicBlock::addPredecessor(MachineBasicBlock *Pred) {
651 Predecessors.push_back(Pred);
Chris Lattner4336b872004-10-26 15:43:42 +0000652}
653
Cong Hou166e0852015-09-29 19:46:09 +0000654void MachineBasicBlock::removePredecessor(MachineBasicBlock *Pred) {
David Majnemer0d955d02016-08-11 22:21:41 +0000655 pred_iterator I = find(Predecessors, Pred);
Chris Lattner4336b872004-10-26 15:43:42 +0000656 assert(I != Predecessors.end() && "Pred is not a predecessor of this block!");
657 Predecessors.erase(I);
658}
Evan Chenga92b2b32007-05-17 23:58:53 +0000659
Cong Hou166e0852015-09-29 19:46:09 +0000660void MachineBasicBlock::transferSuccessors(MachineBasicBlock *FromMBB) {
661 if (this == FromMBB)
Mon P Wang3e583932008-05-05 19:05:59 +0000662 return;
Jakub Staszakfeadd432011-06-16 18:01:17 +0000663
Cong Hou166e0852015-09-29 19:46:09 +0000664 while (!FromMBB->succ_empty()) {
665 MachineBasicBlock *Succ = *FromMBB->succ_begin();
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000666
Cong Houd97c1002015-12-01 05:29:22 +0000667 // If probability list is empty it means we don't use it (disabled optimization).
668 if (!FromMBB->Probs.empty()) {
669 auto Prob = *FromMBB->Probs.begin();
670 addSuccessor(Succ, Prob);
671 } else
672 addSuccessorWithoutProb(Succ);
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000673
Cong Hou166e0852015-09-29 19:46:09 +0000674 FromMBB->removeSuccessor(Succ);
Dan Gohman34396292010-07-06 20:24:04 +0000675 }
676}
677
678void
Cong Hou166e0852015-09-29 19:46:09 +0000679MachineBasicBlock::transferSuccessorsAndUpdatePHIs(MachineBasicBlock *FromMBB) {
680 if (this == FromMBB)
Dan Gohman34396292010-07-06 20:24:04 +0000681 return;
Jakub Staszakfeadd432011-06-16 18:01:17 +0000682
Cong Hou166e0852015-09-29 19:46:09 +0000683 while (!FromMBB->succ_empty()) {
684 MachineBasicBlock *Succ = *FromMBB->succ_begin();
Cong Houd97c1002015-12-01 05:29:22 +0000685 if (!FromMBB->Probs.empty()) {
686 auto Prob = *FromMBB->Probs.begin();
687 addSuccessor(Succ, Prob);
688 } else
689 addSuccessorWithoutProb(Succ);
Cong Hou166e0852015-09-29 19:46:09 +0000690 FromMBB->removeSuccessor(Succ);
Dan Gohman34396292010-07-06 20:24:04 +0000691
692 // Fix up any PHI nodes in the successor.
Evan Cheng7fae11b2011-12-14 02:11:42 +0000693 for (MachineBasicBlock::instr_iterator MI = Succ->instr_begin(),
694 ME = Succ->instr_end(); MI != ME && MI->isPHI(); ++MI)
Dan Gohman34396292010-07-06 20:24:04 +0000695 for (unsigned i = 2, e = MI->getNumOperands()+1; i != e; i += 2) {
696 MachineOperand &MO = MI->getOperand(i);
Cong Hou166e0852015-09-29 19:46:09 +0000697 if (MO.getMBB() == FromMBB)
Dan Gohman34396292010-07-06 20:24:04 +0000698 MO.setMBB(this);
699 }
700 }
Cong Houc1069892015-12-13 09:26:17 +0000701 normalizeSuccProbs();
Mon P Wang3e583932008-05-05 19:05:59 +0000702}
703
Jakob Stoklund Olesenfee94ca2012-07-30 17:36:47 +0000704bool MachineBasicBlock::isPredecessor(const MachineBasicBlock *MBB) const {
David Majnemer42531262016-08-12 03:55:06 +0000705 return is_contained(predecessors(), MBB);
Jakob Stoklund Olesenfee94ca2012-07-30 17:36:47 +0000706}
707
Dan Gohmanff62c622009-03-30 20:06:29 +0000708bool MachineBasicBlock::isSuccessor(const MachineBasicBlock *MBB) const {
David Majnemer42531262016-08-12 03:55:06 +0000709 return is_contained(successors(), MBB);
Evan Chenga92b2b32007-05-17 23:58:53 +0000710}
Evan Chengdf757852007-06-04 06:44:01 +0000711
Dan Gohmanff62c622009-03-30 20:06:29 +0000712bool MachineBasicBlock::isLayoutSuccessor(const MachineBasicBlock *MBB) const {
Dan Gohmana78bae32008-10-02 22:09:09 +0000713 MachineFunction::const_iterator I(this);
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000714 return std::next(I) == MachineFunction::const_iterator(MBB);
Dan Gohmana78bae32008-10-02 22:09:09 +0000715}
716
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000717MachineBasicBlock *MachineBasicBlock::getFallThrough() {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000718 MachineFunction::iterator Fallthrough = getIterator();
Bob Wilson2d4ff122009-11-26 00:32:21 +0000719 ++Fallthrough;
720 // If FallthroughBlock is off the end of the function, it can't fall through.
721 if (Fallthrough == getParent()->end())
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000722 return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000723
724 // If FallthroughBlock isn't a successor, no fallthrough is possible.
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000725 if (!isSuccessor(&*Fallthrough))
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000726 return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000727
Dan Gohman0b44cb02009-12-05 00:32:59 +0000728 // Analyze the branches, if any, at the end of the block.
Craig Topperc0196b12014-04-14 00:51:57 +0000729 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
Dan Gohman0b44cb02009-12-05 00:32:59 +0000730 SmallVector<MachineOperand, 4> Cond;
Eric Christopherfc6de422014-08-05 02:39:49 +0000731 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Jacques Pienaar71c30a12016-07-15 14:41:04 +0000732 if (TII->analyzeBranch(*this, TBB, FBB, Cond)) {
Dan Gohman0b44cb02009-12-05 00:32:59 +0000733 // If we couldn't analyze the branch, examine the last instruction.
734 // If the block doesn't end in a known control barrier, assume fallthrough
Chad Rosier1a1531d2012-01-26 18:24:25 +0000735 // is possible. The isPredicated check is needed because this code can be
Dan Gohman0b44cb02009-12-05 00:32:59 +0000736 // called during IfConversion, where an instruction which is normally a
Chad Rosier9b61cf32012-01-26 20:19:05 +0000737 // Barrier is predicated and thus no longer an actual control barrier.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000738 return (empty() || !back().isBarrier() || TII->isPredicated(back()))
739 ? &*Fallthrough
740 : nullptr;
Dan Gohman0b44cb02009-12-05 00:32:59 +0000741 }
Bob Wilson2d4ff122009-11-26 00:32:21 +0000742
743 // If there is no branch, control always falls through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000744 if (!TBB) return &*Fallthrough;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000745
746 // If there is some explicit branch to the fallthrough block, it can obviously
747 // reach, even though the branch should get folded to fall through implicitly.
748 if (MachineFunction::iterator(TBB) == Fallthrough ||
749 MachineFunction::iterator(FBB) == Fallthrough)
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000750 return &*Fallthrough;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000751
752 // If it's an unconditional branch to some block not the fall through, it
753 // doesn't fall through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000754 if (Cond.empty()) return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000755
756 // Otherwise, if it is conditional and has no explicit false block, it falls
757 // through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000758 return (FBB == nullptr) ? &*Fallthrough : nullptr;
759}
760
761bool MachineBasicBlock::canFallThrough() {
762 return getFallThrough() != nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000763}
764
Quentin Colombet23341a82016-04-21 21:01:13 +0000765MachineBasicBlock *MachineBasicBlock::SplitCriticalEdge(MachineBasicBlock *Succ,
766 Pass &P) {
Quentin Colombet77e18782016-04-21 20:46:27 +0000767 if (!canSplitCriticalEdge(Succ))
Craig Topperc0196b12014-04-14 00:51:57 +0000768 return nullptr;
Evan Cheng2d14d8a2012-04-24 19:06:55 +0000769
Dan Gohman3570f812010-06-22 17:25:57 +0000770 MachineFunction *MF = getParent();
Cong Hou166e0852015-09-29 19:46:09 +0000771 DebugLoc DL; // FIXME: this is nowhere
Dan Gohman3570f812010-06-22 17:25:57 +0000772
Dan Gohman3570f812010-06-22 17:25:57 +0000773 MachineBasicBlock *NMBB = MF->CreateMachineBasicBlock();
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000774 MF->insert(std::next(MachineFunction::iterator(this)), NMBB);
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +0000775 DEBUG(dbgs() << "Splitting critical edge: " << printMBBReference(*this)
776 << " -- " << printMBBReference(*NMBB) << " -- "
777 << printMBBReference(*Succ) << '\n');
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000778
Quentin Colombet23341a82016-04-21 21:01:13 +0000779 LiveIntervals *LIS = P.getAnalysisIfAvailable<LiveIntervals>();
780 SlotIndexes *Indexes = P.getAnalysisIfAvailable<SlotIndexes>();
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000781 if (LIS)
782 LIS->insertMBBInMaps(NMBB);
783 else if (Indexes)
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000784 Indexes->insertMBBInMaps(NMBB);
Dan Gohman3570f812010-06-22 17:25:57 +0000785
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000786 // On some targets like Mips, branches may kill virtual registers. Make sure
787 // that LiveVariables is properly updated after updateTerminator replaces the
788 // terminators.
Quentin Colombet23341a82016-04-21 21:01:13 +0000789 LiveVariables *LV = P.getAnalysisIfAvailable<LiveVariables>();
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000790
791 // Collect a list of virtual registers killed by the terminators.
792 SmallVector<unsigned, 4> KilledRegs;
793 if (LV)
Evan Cheng7fae11b2011-12-14 02:11:42 +0000794 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
Evan Cheng2a81dd42011-12-06 22:12:01 +0000795 I != E; ++I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000796 MachineInstr *MI = &*I;
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000797 for (MachineInstr::mop_iterator OI = MI->operands_begin(),
798 OE = MI->operands_end(); OI != OE; ++OI) {
Lang Hamesedeea172012-02-09 05:59:36 +0000799 if (!OI->isReg() || OI->getReg() == 0 ||
800 !OI->isUse() || !OI->isKill() || OI->isUndef())
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000801 continue;
802 unsigned Reg = OI->getReg();
Lang Hamesedeea172012-02-09 05:59:36 +0000803 if (TargetRegisterInfo::isPhysicalRegister(Reg) ||
Duncan P. N. Exon Smithd26fdc82016-07-01 01:51:32 +0000804 LV->getVarInfo(Reg).removeKill(*MI)) {
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000805 KilledRegs.push_back(Reg);
806 DEBUG(dbgs() << "Removing terminator kill: " << *MI);
807 OI->setIsKill(false);
808 }
809 }
810 }
811
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000812 SmallVector<unsigned, 4> UsedRegs;
813 if (LIS) {
814 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
815 I != E; ++I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000816 MachineInstr *MI = &*I;
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000817
818 for (MachineInstr::mop_iterator OI = MI->operands_begin(),
819 OE = MI->operands_end(); OI != OE; ++OI) {
820 if (!OI->isReg() || OI->getReg() == 0)
821 continue;
822
823 unsigned Reg = OI->getReg();
David Majnemer0d955d02016-08-11 22:21:41 +0000824 if (!is_contained(UsedRegs, Reg))
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000825 UsedRegs.push_back(Reg);
826 }
827 }
828 }
829
Dan Gohman3570f812010-06-22 17:25:57 +0000830 ReplaceUsesOfBlockWith(Succ, NMBB);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000831
832 // If updateTerminator() removes instructions, we need to remove them from
833 // SlotIndexes.
834 SmallVector<MachineInstr*, 4> Terminators;
835 if (Indexes) {
836 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
837 I != E; ++I)
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000838 Terminators.push_back(&*I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000839 }
840
Dan Gohman3570f812010-06-22 17:25:57 +0000841 updateTerminator();
842
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000843 if (Indexes) {
844 SmallVector<MachineInstr*, 4> NewTerminators;
845 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
846 I != E; ++I)
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000847 NewTerminators.push_back(&*I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000848
849 for (SmallVectorImpl<MachineInstr*>::iterator I = Terminators.begin(),
850 E = Terminators.end(); I != E; ++I) {
David Majnemer0d955d02016-08-11 22:21:41 +0000851 if (!is_contained(NewTerminators, *I))
852 Indexes->removeMachineInstrFromMaps(**I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000853 }
854 }
855
Dan Gohman3570f812010-06-22 17:25:57 +0000856 // Insert unconditional "jump Succ" instruction in NMBB if necessary.
857 NMBB->addSuccessor(Succ);
858 if (!NMBB->isLayoutSuccessor(Succ)) {
Quentin Colombet77e18782016-04-21 20:46:27 +0000859 SmallVector<MachineOperand, 4> Cond;
860 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000861 TII->insertBranch(*NMBB, Succ, nullptr, Cond, DL);
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000862
863 if (Indexes) {
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000864 for (MachineInstr &MI : NMBB->instrs()) {
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000865 // Some instructions may have been moved to NMBB by updateTerminator(),
866 // so we first remove any instruction that already has an index.
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000867 if (Indexes->hasIndex(MI))
868 Indexes->removeMachineInstrFromMaps(MI);
869 Indexes->insertMachineInstrInMaps(MI);
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000870 }
871 }
Dan Gohman3570f812010-06-22 17:25:57 +0000872 }
873
874 // Fix PHI nodes in Succ so they refer to NMBB instead of this
Evan Cheng7fae11b2011-12-14 02:11:42 +0000875 for (MachineBasicBlock::instr_iterator
876 i = Succ->instr_begin(),e = Succ->instr_end();
877 i != e && i->isPHI(); ++i)
Dan Gohman3570f812010-06-22 17:25:57 +0000878 for (unsigned ni = 1, ne = i->getNumOperands(); ni != ne; ni += 2)
879 if (i->getOperand(ni+1).getMBB() == this)
880 i->getOperand(ni+1).setMBB(NMBB);
881
Jakob Stoklund Olesen06b6ccf2011-10-14 17:25:46 +0000882 // Inherit live-ins from the successor
Matthias Braund9da1622015-09-09 18:08:03 +0000883 for (const auto &LI : Succ->liveins())
Matthias Braunb2b7ef12015-08-24 22:59:52 +0000884 NMBB->addLiveIn(LI);
Jakob Stoklund Olesen06b6ccf2011-10-14 17:25:46 +0000885
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000886 // Update LiveVariables.
Eric Christopherfc6de422014-08-05 02:39:49 +0000887 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000888 if (LV) {
889 // Restore kills of virtual registers that were killed by the terminators.
890 while (!KilledRegs.empty()) {
891 unsigned Reg = KilledRegs.pop_back_val();
Evan Cheng7fae11b2011-12-14 02:11:42 +0000892 for (instr_iterator I = instr_end(), E = instr_begin(); I != E;) {
Lang Hamesedeea172012-02-09 05:59:36 +0000893 if (!(--I)->addRegisterKilled(Reg, TRI, /* addIfNotFound= */ false))
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000894 continue;
Lang Hamesedeea172012-02-09 05:59:36 +0000895 if (TargetRegisterInfo::isVirtualRegister(Reg))
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000896 LV->getVarInfo(Reg).Kills.push_back(&*I);
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000897 DEBUG(dbgs() << "Restored terminator kill: " << *I);
898 break;
899 }
900 }
901 // Update relevant live-through information.
Dan Gohman3570f812010-06-22 17:25:57 +0000902 LV->addNewBlock(NMBB, this, Succ);
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000903 }
Dan Gohman3570f812010-06-22 17:25:57 +0000904
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000905 if (LIS) {
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000906 // After splitting the edge and updating SlotIndexes, live intervals may be
907 // in one of two situations, depending on whether this block was the last in
Cong Hou2793e722015-08-10 22:27:10 +0000908 // the function. If the original block was the last in the function, all
909 // live intervals will end prior to the beginning of the new split block. If
910 // the original block was not at the end of the function, all live intervals
911 // will extend to the end of the new split block.
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000912
913 bool isLastMBB =
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000914 std::next(MachineFunction::iterator(NMBB)) == getParent()->end();
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000915
916 SlotIndex StartIndex = Indexes->getMBBEndIdx(this);
917 SlotIndex PrevIndex = StartIndex.getPrevSlot();
918 SlotIndex EndIndex = Indexes->getMBBEndIdx(NMBB);
919
920 // Find the registers used from NMBB in PHIs in Succ.
921 SmallSet<unsigned, 8> PHISrcRegs;
922 for (MachineBasicBlock::instr_iterator
923 I = Succ->instr_begin(), E = Succ->instr_end();
924 I != E && I->isPHI(); ++I) {
925 for (unsigned ni = 1, ne = I->getNumOperands(); ni != ne; ni += 2) {
926 if (I->getOperand(ni+1).getMBB() == NMBB) {
927 MachineOperand &MO = I->getOperand(ni);
928 unsigned Reg = MO.getReg();
929 PHISrcRegs.insert(Reg);
Cameron Zwarichaf349312013-02-12 03:49:17 +0000930 if (MO.isUndef())
931 continue;
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000932
933 LiveInterval &LI = LIS->getInterval(Reg);
934 VNInfo *VNI = LI.getVNInfoAt(PrevIndex);
Cong Hou2793e722015-08-10 22:27:10 +0000935 assert(VNI &&
936 "PHI sources should be live out of their predecessors.");
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000937 LI.addSegment(LiveInterval::Segment(StartIndex, EndIndex, VNI));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000938 }
939 }
940 }
941
942 MachineRegisterInfo *MRI = &getParent()->getRegInfo();
943 for (unsigned i = 0, e = MRI->getNumVirtRegs(); i != e; ++i) {
944 unsigned Reg = TargetRegisterInfo::index2VirtReg(i);
945 if (PHISrcRegs.count(Reg) || !LIS->hasInterval(Reg))
946 continue;
947
948 LiveInterval &LI = LIS->getInterval(Reg);
949 if (!LI.liveAt(PrevIndex))
950 continue;
951
Cameron Zwarichaf349312013-02-12 03:49:17 +0000952 bool isLiveOut = LI.liveAt(LIS->getMBBStartIdx(Succ));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000953 if (isLiveOut && isLastMBB) {
954 VNInfo *VNI = LI.getVNInfoAt(PrevIndex);
955 assert(VNI && "LiveInterval should have VNInfo where it is live.");
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000956 LI.addSegment(LiveInterval::Segment(StartIndex, EndIndex, VNI));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000957 } else if (!isLiveOut && !isLastMBB) {
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000958 LI.removeSegment(StartIndex, EndIndex);
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000959 }
960 }
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000961
962 // Update all intervals for registers whose uses may have been modified by
963 // updateTerminator().
Cameron Zwarich24955962013-02-17 11:09:00 +0000964 LIS->repairIntervalsInRange(this, getFirstTerminator(), end(), UsedRegs);
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000965 }
966
Dan Gohman3570f812010-06-22 17:25:57 +0000967 if (MachineDominatorTree *MDT =
Quentin Colombet23341a82016-04-21 21:01:13 +0000968 P.getAnalysisIfAvailable<MachineDominatorTree>())
Quentin Colombetabea99f2014-08-13 21:00:07 +0000969 MDT->recordSplitCriticalEdge(this, Succ, NMBB);
Dan Gohman3570f812010-06-22 17:25:57 +0000970
Quentin Colombet23341a82016-04-21 21:01:13 +0000971 if (MachineLoopInfo *MLI = P.getAnalysisIfAvailable<MachineLoopInfo>())
Dan Gohman3570f812010-06-22 17:25:57 +0000972 if (MachineLoop *TIL = MLI->getLoopFor(this)) {
973 // If one or the other blocks were not in a loop, the new block is not
974 // either, and thus LI doesn't need to be updated.
975 if (MachineLoop *DestLoop = MLI->getLoopFor(Succ)) {
976 if (TIL == DestLoop) {
977 // Both in the same loop, the NMBB joins loop.
978 DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase());
979 } else if (TIL->contains(DestLoop)) {
980 // Edge from an outer loop to an inner loop. Add to the outer loop.
981 TIL->addBasicBlockToLoop(NMBB, MLI->getBase());
982 } else if (DestLoop->contains(TIL)) {
983 // Edge from an inner loop to an outer loop. Add to the outer loop.
984 DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase());
985 } else {
986 // Edge from two loops with no containment relation. Because these
987 // are natural loops, we know that the destination block must be the
988 // header of its loop (adding a branch into a loop elsewhere would
989 // create an irreducible loop).
990 assert(DestLoop->getHeader() == Succ &&
991 "Should not create irreducible loops!");
992 if (MachineLoop *P = DestLoop->getParentLoop())
993 P->addBasicBlockToLoop(NMBB, MLI->getBase());
994 }
995 }
996 }
997
998 return NMBB;
999}
1000
Quentin Colombet77e18782016-04-21 20:46:27 +00001001bool MachineBasicBlock::canSplitCriticalEdge(
1002 const MachineBasicBlock *Succ) const {
1003 // Splitting the critical edge to a landing pad block is non-trivial. Don't do
1004 // it in this generic function.
1005 if (Succ->isEHPad())
1006 return false;
1007
1008 const MachineFunction *MF = getParent();
1009
1010 // Performance might be harmed on HW that implements branching using exec mask
1011 // where both sides of the branches are always executed.
1012 if (MF->getTarget().requiresStructuredCFG())
1013 return false;
1014
1015 // We may need to update this's terminator, but we can't do that if
1016 // AnalyzeBranch fails. If this uses a jump table, we won't touch it.
1017 const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
1018 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
1019 SmallVector<MachineOperand, 4> Cond;
1020 // AnalyzeBanch should modify this, since we did not allow modification.
Jacques Pienaar71c30a12016-07-15 14:41:04 +00001021 if (TII->analyzeBranch(*const_cast<MachineBasicBlock *>(this), TBB, FBB, Cond,
Quentin Colombet77e18782016-04-21 20:46:27 +00001022 /*AllowModify*/ false))
1023 return false;
1024
1025 // Avoid bugpoint weirdness: A block may end with a conditional branch but
1026 // jumps to the same MBB is either case. We have duplicate CFG edges in that
1027 // case that we can't handle. Since this never happens in properly optimized
1028 // code, just skip those edges.
1029 if (TBB && TBB == FBB) {
Francis Visoiu Mistrih25528d62017-12-04 17:18:51 +00001030 DEBUG(dbgs() << "Won't split critical edge after degenerate "
1031 << printMBBReference(*this) << '\n');
Quentin Colombet77e18782016-04-21 20:46:27 +00001032 return false;
1033 }
1034 return true;
1035}
1036
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001037/// Prepare MI to be removed from its bundle. This fixes bundle flags on MI's
1038/// neighboring instructions so the bundle won't be broken by removing MI.
1039static void unbundleSingleMI(MachineInstr *MI) {
1040 // Removing the first instruction in a bundle.
1041 if (MI->isBundledWithSucc() && !MI->isBundledWithPred())
1042 MI->unbundleFromSucc();
1043 // Removing the last instruction in a bundle.
1044 if (MI->isBundledWithPred() && !MI->isBundledWithSucc())
1045 MI->unbundleFromPred();
1046 // If MI is not bundled, or if it is internal to a bundle, the neighbor flags
1047 // are already fine.
Evan Cheng7fae11b2011-12-14 02:11:42 +00001048}
1049
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001050MachineBasicBlock::instr_iterator
1051MachineBasicBlock::erase(MachineBasicBlock::instr_iterator I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +00001052 unbundleSingleMI(&*I);
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001053 return Insts.erase(I);
1054}
Evan Cheng7fae11b2011-12-14 02:11:42 +00001055
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001056MachineInstr *MachineBasicBlock::remove_instr(MachineInstr *MI) {
1057 unbundleSingleMI(MI);
1058 MI->clearFlag(MachineInstr::BundledPred);
1059 MI->clearFlag(MachineInstr::BundledSucc);
1060 return Insts.remove(MI);
Evan Cheng7fae11b2011-12-14 02:11:42 +00001061}
1062
Jakob Stoklund Olesen422e07b2012-12-18 17:54:53 +00001063MachineBasicBlock::instr_iterator
1064MachineBasicBlock::insert(instr_iterator I, MachineInstr *MI) {
1065 assert(!MI->isBundledWithPred() && !MI->isBundledWithSucc() &&
1066 "Cannot insert instruction with bundle flags");
1067 // Set the bundle flags when inserting inside a bundle.
1068 if (I != instr_end() && I->isBundledWithPred()) {
1069 MI->setFlag(MachineInstr::BundledPred);
1070 MI->setFlag(MachineInstr::BundledSucc);
1071 }
1072 return Insts.insert(I, MI);
1073}
1074
Cong Hou2a02c1c2015-08-12 21:18:54 +00001075/// This method unlinks 'this' from the containing function, and returns it, but
1076/// does not delete it.
Dan Gohman3b460302008-07-07 23:14:23 +00001077MachineBasicBlock *MachineBasicBlock::removeFromParent() {
1078 assert(getParent() && "Not embedded in a function!");
1079 getParent()->remove(this);
1080 return this;
1081}
1082
Cong Hou2a02c1c2015-08-12 21:18:54 +00001083/// This method unlinks 'this' from the containing function, and deletes it.
Dan Gohman3b460302008-07-07 23:14:23 +00001084void MachineBasicBlock::eraseFromParent() {
1085 assert(getParent() && "Not embedded in a function!");
1086 getParent()->erase(this);
1087}
1088
Cong Hou2a02c1c2015-08-12 21:18:54 +00001089/// Given a machine basic block that branched to 'Old', change the code and CFG
1090/// so that it branches to 'New' instead.
Evan Chengdf757852007-06-04 06:44:01 +00001091void MachineBasicBlock::ReplaceUsesOfBlockWith(MachineBasicBlock *Old,
1092 MachineBasicBlock *New) {
1093 assert(Old != New && "Cannot replace self with self!");
1094
Evan Cheng7fae11b2011-12-14 02:11:42 +00001095 MachineBasicBlock::instr_iterator I = instr_end();
1096 while (I != instr_begin()) {
Evan Chengdf757852007-06-04 06:44:01 +00001097 --I;
Evan Cheng7f8e5632011-12-07 07:15:52 +00001098 if (!I->isTerminator()) break;
Evan Chengdf757852007-06-04 06:44:01 +00001099
1100 // Scan the operands of this machine instruction, replacing any uses of Old
1101 // with New.
1102 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
Dan Gohman0d1e9a82008-10-03 15:45:36 +00001103 if (I->getOperand(i).isMBB() &&
Dan Gohman38453ee2008-09-13 17:58:21 +00001104 I->getOperand(i).getMBB() == Old)
Chris Lattnera5bb3702007-12-30 23:10:15 +00001105 I->getOperand(i).setMBB(New);
Evan Chengdf757852007-06-04 06:44:01 +00001106 }
1107
Dan Gohmanbb2f1072009-05-05 21:10:19 +00001108 // Update the successor information.
Jakub Staszak12a43bd2011-06-16 20:22:37 +00001109 replaceSuccessor(Old, New);
Evan Chengdf757852007-06-04 06:44:01 +00001110}
1111
Cong Hou2a02c1c2015-08-12 21:18:54 +00001112/// Various pieces of code can cause excess edges in the CFG to be inserted. If
1113/// we have proven that MBB can only branch to DestA and DestB, remove any other
1114/// MBB successors from the CFG. DestA and DestB can be null.
Jakub Staszakfeadd432011-06-16 18:01:17 +00001115///
Chris Lattner574e7162007-12-31 04:56:33 +00001116/// Besides DestA and DestB, retain other edges leading to LandingPads
1117/// (currently there can be only one; we don't check or require that here).
Evan Cheng2afd7022007-06-18 22:43:58 +00001118/// Note it is possible that DestA and/or DestB are LandingPads.
1119bool MachineBasicBlock::CorrectExtraCFGEdges(MachineBasicBlock *DestA,
1120 MachineBasicBlock *DestB,
Cong Hou166e0852015-09-29 19:46:09 +00001121 bool IsCond) {
Bill Wendling2d5967d2009-12-16 00:08:36 +00001122 // The values of DestA and DestB frequently come from a call to the
1123 // 'TargetInstrInfo::AnalyzeBranch' method. We take our meaning of the initial
1124 // values from there.
1125 //
1126 // 1. If both DestA and DestB are null, then the block ends with no branches
1127 // (it falls through to its successor).
Cong Hou166e0852015-09-29 19:46:09 +00001128 // 2. If DestA is set, DestB is null, and IsCond is false, then the block ends
Bill Wendling2d5967d2009-12-16 00:08:36 +00001129 // with only an unconditional branch.
Cong Hou166e0852015-09-29 19:46:09 +00001130 // 3. If DestA is set, DestB is null, and IsCond is true, then the block ends
Bill Wendling2d5967d2009-12-16 00:08:36 +00001131 // with a conditional branch that falls through to a successor (DestB).
Cong Hou166e0852015-09-29 19:46:09 +00001132 // 4. If DestA and DestB is set and IsCond is true, then the block ends with a
Bill Wendling2d5967d2009-12-16 00:08:36 +00001133 // conditional branch followed by an unconditional branch. DestA is the
1134 // 'true' destination and DestB is the 'false' destination.
1135
Bill Wendling95643402010-04-01 00:00:43 +00001136 bool Changed = false;
Evan Cheng2afd7022007-06-18 22:43:58 +00001137
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001138 MachineBasicBlock *FallThru = getNextNode();
Bill Wendling95643402010-04-01 00:00:43 +00001139
Craig Topperc0196b12014-04-14 00:51:57 +00001140 if (!DestA && !DestB) {
Bill Wendling95643402010-04-01 00:00:43 +00001141 // Block falls through to successor.
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001142 DestA = FallThru;
1143 DestB = FallThru;
Craig Topperc0196b12014-04-14 00:51:57 +00001144 } else if (DestA && !DestB) {
Cong Hou166e0852015-09-29 19:46:09 +00001145 if (IsCond)
Bill Wendling95643402010-04-01 00:00:43 +00001146 // Block ends in conditional jump that falls through to successor.
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001147 DestB = FallThru;
Evan Cheng2afd7022007-06-18 22:43:58 +00001148 } else {
Cong Hou166e0852015-09-29 19:46:09 +00001149 assert(DestA && DestB && IsCond &&
Bill Wendling95643402010-04-01 00:00:43 +00001150 "CFG in a bad state. Cannot correct CFG edges");
Evan Cheng2afd7022007-06-18 22:43:58 +00001151 }
Bill Wendling95643402010-04-01 00:00:43 +00001152
1153 // Remove superfluous edges. I.e., those which aren't destinations of this
1154 // basic block, duplicate edges, or landing pads.
1155 SmallPtrSet<const MachineBasicBlock*, 8> SeenMBBs;
Evan Cheng2afd7022007-06-18 22:43:58 +00001156 MachineBasicBlock::succ_iterator SI = succ_begin();
Evan Cheng2afd7022007-06-18 22:43:58 +00001157 while (SI != succ_end()) {
Bill Wendling2d5967d2009-12-16 00:08:36 +00001158 const MachineBasicBlock *MBB = *SI;
David Blaikie70573dc2014-11-19 07:49:26 +00001159 if (!SeenMBBs.insert(MBB).second ||
Reid Kleckner0e288232015-08-27 23:27:47 +00001160 (MBB != DestA && MBB != DestB && !MBB->isEHPad())) {
Bill Wendling95643402010-04-01 00:00:43 +00001161 // This is a superfluous edge, remove it.
Bill Wendlingfa9810d2010-03-31 23:26:26 +00001162 SI = removeSuccessor(SI);
Bill Wendling95643402010-04-01 00:00:43 +00001163 Changed = true;
1164 } else {
1165 ++SI;
Evan Cheng2afd7022007-06-18 22:43:58 +00001166 }
1167 }
Bill Wendling2d5967d2009-12-16 00:08:36 +00001168
Cong Houc1069892015-12-13 09:26:17 +00001169 if (Changed)
1170 normalizeSuccProbs();
Bill Wendling95643402010-04-01 00:00:43 +00001171 return Changed;
Evan Cheng2afd7022007-06-18 22:43:58 +00001172}
Evan Cheng57be2f22009-11-17 19:19:59 +00001173
Cong Hou2a02c1c2015-08-12 21:18:54 +00001174/// Find the next valid DebugLoc starting at MBBI, skipping any DBG_VALUE
1175/// instructions. Return UnknownLoc if there is none.
Dale Johannesenc5db5992010-01-20 21:36:02 +00001176DebugLoc
Evan Cheng7fae11b2011-12-14 02:11:42 +00001177MachineBasicBlock::findDebugLoc(instr_iterator MBBI) {
Evan Cheng2a81dd42011-12-06 22:12:01 +00001178 // Skip debug declarations, we don't want a DebugLoc from them.
Florian Hahn3c8b8c92016-12-16 11:10:26 +00001179 MBBI = skipDebugInstructionsForward(MBBI, instr_end());
1180 if (MBBI != instr_end())
1181 return MBBI->getDebugLoc();
1182 return {};
Dale Johannesenc5db5992010-01-20 21:36:02 +00001183}
1184
Taewook Oh06a21282017-02-13 18:15:31 +00001185/// Find and return the merged DebugLoc of the branch instructions of the block.
1186/// Return UnknownLoc if there is none.
1187DebugLoc
1188MachineBasicBlock::findBranchDebugLoc() {
Taewook Oh4d35f9e2017-02-13 21:12:27 +00001189 DebugLoc DL;
Taewook Oh06a21282017-02-13 18:15:31 +00001190 auto TI = getFirstTerminator();
1191 while (TI != end() && !TI->isBranch())
1192 ++TI;
1193
1194 if (TI != end()) {
1195 DL = TI->getDebugLoc();
1196 for (++TI ; TI != end() ; ++TI)
1197 if (TI->isBranch())
1198 DL = DILocation::getMergedLocation(DL, TI->getDebugLoc());
1199 }
1200 return DL;
1201}
1202
Cong Houd97c1002015-12-01 05:29:22 +00001203/// Return probability of the edge from this block to MBB.
Cong Hou23a3bf02015-11-04 21:37:58 +00001204BranchProbability
1205MachineBasicBlock::getSuccProbability(const_succ_iterator Succ) const {
Cong Hou4aef7ef2015-12-01 11:05:39 +00001206 if (Probs.empty())
Cong Hou23a3bf02015-11-04 21:37:58 +00001207 return BranchProbability(1, succ_size());
1208
Cong Hou4aef7ef2015-12-01 11:05:39 +00001209 const auto &Prob = *getProbabilityIterator(Succ);
1210 if (Prob.isUnknown()) {
1211 // For unknown probabilities, collect the sum of all known ones, and evenly
1212 // ditribute the complemental of the sum to each unknown probability.
1213 unsigned KnownProbNum = 0;
1214 auto Sum = BranchProbability::getZero();
1215 for (auto &P : Probs) {
1216 if (!P.isUnknown()) {
1217 Sum += P;
1218 KnownProbNum++;
1219 }
1220 }
1221 return Sum.getCompl() / (Probs.size() - KnownProbNum);
1222 } else
1223 return Prob;
Manman Renb6819182014-01-29 23:18:47 +00001224}
1225
Cong Hou23a3bf02015-11-04 21:37:58 +00001226/// Set successor probability of a given iterator.
1227void MachineBasicBlock::setSuccProbability(succ_iterator I,
1228 BranchProbability Prob) {
1229 assert(!Prob.isUnknown());
Cong Houd97c1002015-12-01 05:29:22 +00001230 if (Probs.empty())
Cong Hou23a3bf02015-11-04 21:37:58 +00001231 return;
1232 *getProbabilityIterator(I) = Prob;
Cong Hou23a3bf02015-11-04 21:37:58 +00001233}
1234
Hans Wennborg1dbaf672015-12-01 03:49:42 +00001235/// Return probability iterator corresonding to the I successor iterator
1236MachineBasicBlock::const_probability_iterator
1237MachineBasicBlock::getProbabilityIterator(
1238 MachineBasicBlock::const_succ_iterator I) const {
Cong Houfa1917c2015-12-01 00:02:51 +00001239 assert(Probs.size() == Successors.size() && "Async probability list!");
1240 const size_t index = std::distance(Successors.begin(), I);
1241 assert(index < Probs.size() && "Not a current successor!");
1242 return Probs.begin() + index;
1243}
1244
Cong Houd97c1002015-12-01 05:29:22 +00001245/// Return probability iterator corresonding to the I successor iterator.
1246MachineBasicBlock::probability_iterator
1247MachineBasicBlock::getProbabilityIterator(MachineBasicBlock::succ_iterator I) {
1248 assert(Probs.size() == Successors.size() && "Async probability list!");
1249 const size_t index = std::distance(Successors.begin(), I);
1250 assert(index < Probs.size() && "Not a current successor!");
1251 return Probs.begin() + index;
1252}
1253
James Molloyc747cda2012-09-12 10:18:23 +00001254/// Return whether (physical) register "Reg" has been <def>ined and not <kill>ed
1255/// as of just before "MI".
Junmo Park12386102016-01-07 10:26:32 +00001256///
James Molloyc747cda2012-09-12 10:18:23 +00001257/// Search is localised to a neighborhood of
1258/// Neighborhood instructions before (searching for defs or kills) and N
1259/// instructions after (searching just for defs) MI.
1260MachineBasicBlock::LivenessQueryResult
1261MachineBasicBlock::computeRegisterLiveness(const TargetRegisterInfo *TRI,
Matthias Braun07a07ba2015-05-27 05:12:39 +00001262 unsigned Reg, const_iterator Before,
1263 unsigned Neighborhood) const {
James Molloyc747cda2012-09-12 10:18:23 +00001264 unsigned N = Neighborhood;
James Molloyc747cda2012-09-12 10:18:23 +00001265
Matthias Braun07a07ba2015-05-27 05:12:39 +00001266 // Start by searching backwards from Before, looking for kills, reads or defs.
1267 const_iterator I(Before);
James Molloyc747cda2012-09-12 10:18:23 +00001268 // If this is the first insn in the block, don't search backwards.
Matthias Braun07a07ba2015-05-27 05:12:39 +00001269 if (I != begin()) {
James Molloyc747cda2012-09-12 10:18:23 +00001270 do {
1271 --I;
1272
Matthias Braun60d69e22015-12-11 19:42:09 +00001273 MachineOperandIteratorBase::PhysRegInfo Info =
Duncan P. N. Exon Smithf9ab4162016-02-27 17:05:33 +00001274 ConstMIOperands(*I).analyzePhysReg(Reg, TRI);
James Molloyc747cda2012-09-12 10:18:23 +00001275
Matthias Braun60d69e22015-12-11 19:42:09 +00001276 // Defs happen after uses so they take precedence if both are present.
Tim Northoverdd219d02012-11-20 09:56:11 +00001277
Matthias Braun60d69e22015-12-11 19:42:09 +00001278 // Register is dead after a dead def of the full register.
1279 if (Info.DeadDef)
James Molloyc747cda2012-09-12 10:18:23 +00001280 return LQR_Dead;
Matthias Braun60d69e22015-12-11 19:42:09 +00001281 // Register is (at least partially) live after a def.
Quentin Colombet08e79992016-04-26 23:14:29 +00001282 if (Info.Defined) {
1283 if (!Info.PartialDeadDef)
1284 return LQR_Live;
1285 // As soon as we saw a partial definition (dead or not),
1286 // we cannot tell if the value is partial live without
1287 // tracking the lanemasks. We are not going to do this,
1288 // so fall back on the remaining of the analysis.
1289 break;
1290 }
Matthias Braun60d69e22015-12-11 19:42:09 +00001291 // Register is dead after a full kill or clobber and no def.
1292 if (Info.Killed || Info.Clobbered)
1293 return LQR_Dead;
1294 // Register must be live if we read it.
1295 if (Info.Read)
1296 return LQR_Live;
Matthias Braun07a07ba2015-05-27 05:12:39 +00001297 } while (I != begin() && --N > 0);
James Molloyc747cda2012-09-12 10:18:23 +00001298 }
1299
1300 // Did we get to the start of the block?
Matthias Braun07a07ba2015-05-27 05:12:39 +00001301 if (I == begin()) {
James Molloyc747cda2012-09-12 10:18:23 +00001302 // If so, the register's state is definitely defined by the live-in state.
Matthias Braun60d69e22015-12-11 19:42:09 +00001303 for (MCRegAliasIterator RAI(Reg, TRI, /*IncludeSelf=*/true); RAI.isValid();
1304 ++RAI)
Matthias Braun07a07ba2015-05-27 05:12:39 +00001305 if (isLiveIn(*RAI))
Matthias Braun60d69e22015-12-11 19:42:09 +00001306 return LQR_Live;
James Molloyc747cda2012-09-12 10:18:23 +00001307
1308 return LQR_Dead;
1309 }
1310
1311 N = Neighborhood;
1312
Matthias Braun07a07ba2015-05-27 05:12:39 +00001313 // Try searching forwards from Before, looking for reads or defs.
1314 I = const_iterator(Before);
James Molloyc747cda2012-09-12 10:18:23 +00001315 // If this is the last insn in the block, don't search forwards.
Matthias Braun07a07ba2015-05-27 05:12:39 +00001316 if (I != end()) {
1317 for (++I; I != end() && N > 0; ++I, --N) {
Matthias Braun60d69e22015-12-11 19:42:09 +00001318 MachineOperandIteratorBase::PhysRegInfo Info =
Duncan P. N. Exon Smithf9ab4162016-02-27 17:05:33 +00001319 ConstMIOperands(*I).analyzePhysReg(Reg, TRI);
James Molloyc747cda2012-09-12 10:18:23 +00001320
Matthias Braun60d69e22015-12-11 19:42:09 +00001321 // Register is live when we read it here.
1322 if (Info.Read)
1323 return LQR_Live;
1324 // Register is dead if we can fully overwrite or clobber it here.
1325 if (Info.FullyDefined || Info.Clobbered)
James Molloyc747cda2012-09-12 10:18:23 +00001326 return LQR_Dead;
1327 }
1328 }
1329
1330 // At this point we have no idea of the liveness of the register.
1331 return LQR_Unknown;
1332}
Reid Klecknerb8fd1622015-11-06 17:06:38 +00001333
1334const uint32_t *
1335MachineBasicBlock::getBeginClobberMask(const TargetRegisterInfo *TRI) const {
1336 // EH funclet entry does not preserve any registers.
1337 return isEHFuncletEntry() ? TRI->getNoPreservedMask() : nullptr;
1338}
1339
1340const uint32_t *
1341MachineBasicBlock::getEndClobberMask(const TargetRegisterInfo *TRI) const {
1342 // If we see a return block with successors, this must be a funclet return,
1343 // which does not preserve any registers. If there are no successors, we don't
1344 // care what kind of return it is, putting a mask after it is a no-op.
1345 return isReturnBlock() && !succ_empty() ? TRI->getNoPreservedMask() : nullptr;
1346}
Matthias Braun18198302016-12-16 23:55:37 +00001347
1348void MachineBasicBlock::clearLiveIns() {
1349 LiveIns.clear();
1350}
Matthias Braun11723322017-01-05 20:01:19 +00001351
1352MachineBasicBlock::livein_iterator MachineBasicBlock::livein_begin() const {
1353 assert(getParent()->getProperties().hasProperty(
1354 MachineFunctionProperties::Property::TracksLiveness) &&
1355 "Liveness information is accurate");
1356 return LiveIns.begin();
1357}