blob: 81597afe6b02b4f35569639b99c78a09e444d0ba [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"
Cameron Zwarichb47fb382013-02-11 09:24:47 +000016#include "llvm/CodeGen/LiveIntervalAnalysis.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"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000024#include "llvm/IR/BasicBlock.h"
25#include "llvm/IR/DataLayout.h"
Taewook Oh06a21282017-02-13 18:15:31 +000026#include "llvm/IR/DebugInfoMetadata.h"
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +000027#include "llvm/IR/ModuleSlotTracker.h"
Chris Lattnerd051af72010-01-26 04:55:51 +000028#include "llvm/MC/MCAsmInfo.h"
29#include "llvm/MC/MCContext.h"
Cong Hou663dd012015-12-13 09:52:14 +000030#include "llvm/Support/DataTypes.h"
David Greene6c56cef2010-01-04 23:22:07 +000031#include "llvm/Support/Debug.h"
Daniel Dunbar796e43e2009-07-24 10:36:58 +000032#include "llvm/Support/raw_ostream.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000033#include "llvm/Target/TargetInstrInfo.h"
34#include "llvm/Target/TargetMachine.h"
35#include "llvm/Target/TargetRegisterInfo.h"
Eric Christopherd9134482014-08-04 21:25:23 +000036#include "llvm/Target/TargetSubtargetInfo.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;
Tanya Lattner91fa3a92004-05-24 07:14:35 +000045}
Tanya Lattner91fa3a92004-05-24 07:14:35 +000046
Dan Gohman0ece9432008-07-17 23:49:46 +000047MachineBasicBlock::~MachineBasicBlock() {
Dan Gohman0ece9432008-07-17 23:49:46 +000048}
49
Cong Hou2a02c1c2015-08-12 21:18:54 +000050/// Return the MCSymbol for this basic block.
Chris Lattner29bdac42010-03-13 21:04:28 +000051MCSymbol *MachineBasicBlock::getSymbol() const {
Eli Bendersky58b04b72013-04-22 21:21:08 +000052 if (!CachedMCSymbol) {
53 const MachineFunction *MF = getParent();
54 MCContext &Ctx = MF->getContext();
Mehdi Amini36d33fc2016-10-01 06:46:33 +000055 auto Prefix = Ctx.getAsmInfo()->getPrivateLabelPrefix();
Reid Klecknerb9204a52015-11-11 23:09:31 +000056 assert(getNumber() >= 0 && "cannot get label for unreachable MBB");
Jim Grosbach6f482002015-05-18 18:43:14 +000057 CachedMCSymbol = Ctx.getOrCreateSymbol(Twine(Prefix) + "BB" +
Eli Bendersky58b04b72013-04-22 21:21:08 +000058 Twine(MF->getFunctionNumber()) +
59 "_" + Twine(getNumber()));
60 }
61
62 return CachedMCSymbol;
Chris Lattnerd051af72010-01-26 04:55:51 +000063}
64
65
Chris Lattneraf119ca2009-08-23 00:35:30 +000066raw_ostream &llvm::operator<<(raw_ostream &OS, const MachineBasicBlock &MBB) {
Daniel Dunbar796e43e2009-07-24 10:36:58 +000067 MBB.print(OS);
68 return OS;
69}
Tanya Lattner91fa3a92004-05-24 07:14:35 +000070
Cong Hou2a02c1c2015-08-12 21:18:54 +000071/// When an MBB is added to an MF, we need to update the parent pointer of the
72/// MBB, the MBB numbering, and any instructions in the MBB to be on the right
73/// operand list for registers.
Chris Lattner961e7422008-01-01 01:12:31 +000074///
75/// MBBs start out as #-1. When a MBB is added to a MachineFunction, it
76/// gets the next available unique MBB number. If it is removed from a
77/// MachineFunction, it goes back to being #-1.
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +000078void ilist_callback_traits<MachineBasicBlock>::addNodeToList(
79 MachineBasicBlock *N) {
Dan Gohman3b460302008-07-07 23:14:23 +000080 MachineFunction &MF = *N->getParent();
81 N->Number = MF.addToMBBNumbering(N);
Chris Lattner961e7422008-01-01 01:12:31 +000082
83 // Make sure the instructions have their operands in the reginfo lists.
Dan Gohman3b460302008-07-07 23:14:23 +000084 MachineRegisterInfo &RegInfo = MF.getRegInfo();
Evan Cheng7fae11b2011-12-14 02:11:42 +000085 for (MachineBasicBlock::instr_iterator
86 I = N->instr_begin(), E = N->instr_end(); I != E; ++I)
Chris Lattner961e7422008-01-01 01:12:31 +000087 I->AddRegOperandsToUseLists(RegInfo);
Brian Gaekecb5d22a2004-05-12 21:35:22 +000088}
89
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +000090void ilist_callback_traits<MachineBasicBlock>::removeNodeFromList(
91 MachineBasicBlock *N) {
Chris Lattner574e7162007-12-31 04:56:33 +000092 N->getParent()->removeFromMBBNumbering(N->Number);
Brian Gaekecb5d22a2004-05-12 21:35:22 +000093 N->Number = -1;
94}
95
Cong Hou2a02c1c2015-08-12 21:18:54 +000096/// When we add an instruction to a basic block list, we update its parent
97/// pointer and add its operands from reg use/def lists if appropriate.
Chris Lattneraf119ca2009-08-23 00:35:30 +000098void ilist_traits<MachineInstr>::addNodeToList(MachineInstr *N) {
Craig Topperc0196b12014-04-14 00:51:57 +000099 assert(!N->getParent() && "machine instruction already in a basic block");
Dan Gohman3b460302008-07-07 23:14:23 +0000100 N->setParent(Parent);
Jakub Staszakfeadd432011-06-16 18:01:17 +0000101
Dan Gohman3b460302008-07-07 23:14:23 +0000102 // Add the instruction's register operands to their corresponding
103 // use/def lists.
104 MachineFunction *MF = Parent->getParent();
105 N->AddRegOperandsToUseLists(MF->getRegInfo());
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000106}
107
Cong Hou2a02c1c2015-08-12 21:18:54 +0000108/// When we remove an instruction from a basic block list, we update its parent
109/// pointer and remove its operands from reg use/def lists if appropriate.
Chris Lattneraf119ca2009-08-23 00:35:30 +0000110void ilist_traits<MachineInstr>::removeNodeFromList(MachineInstr *N) {
Craig Topperc0196b12014-04-14 00:51:57 +0000111 assert(N->getParent() && "machine instruction not in a basic block");
Dan Gohman3b460302008-07-07 23:14:23 +0000112
113 // Remove from the use/def lists.
Jakob Stoklund Olesenc4102d42012-08-09 22:49:37 +0000114 if (MachineFunction *MF = N->getParent()->getParent())
115 N->RemoveRegOperandsFromUseLists(MF->getRegInfo());
Jakub Staszakfeadd432011-06-16 18:01:17 +0000116
Craig Topperc0196b12014-04-14 00:51:57 +0000117 N->setParent(nullptr);
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000118}
119
Cong Hou2a02c1c2015-08-12 21:18:54 +0000120/// When moving a range of instructions from one MBB list to another, we need to
121/// update the parent pointers and the use/def lists.
Duncan P. N. Exon Smithcc9edac2016-09-11 16:38:18 +0000122void ilist_traits<MachineInstr>::transferNodesFromList(ilist_traits &FromList,
123 instr_iterator First,
124 instr_iterator Last) {
Cong Hou166e0852015-09-29 19:46:09 +0000125 assert(Parent->getParent() == FromList.Parent->getParent() &&
Dan Gohman804c95d2008-07-28 21:51:04 +0000126 "MachineInstr parent mismatch!");
Duncan P. N. Exon Smithb7668d52016-08-30 18:00:45 +0000127 assert(this != &FromList && "Called without a real transfer...");
128 assert(Parent != FromList.Parent && "Two lists have the same parent?");
Chris Lattner961e7422008-01-01 01:12:31 +0000129
130 // If splicing between two blocks within the same function, just update the
131 // parent pointers.
Cong Hou166e0852015-09-29 19:46:09 +0000132 for (; First != Last; ++First)
133 First->setParent(Parent);
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000134}
135
Duncan P. N. Exon Smithf947c3a2016-08-30 18:40:47 +0000136void ilist_traits<MachineInstr>::deleteNode(MachineInstr *MI) {
Dan Gohman3b460302008-07-07 23:14:23 +0000137 assert(!MI->getParent() && "MI is still in a block!");
138 Parent->getParent()->DeleteMachineInstr(MI);
139}
140
Dan Gohman88c547e2010-07-07 14:33:51 +0000141MachineBasicBlock::iterator MachineBasicBlock::getFirstNonPHI() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000142 instr_iterator I = instr_begin(), E = instr_end();
143 while (I != E && I->isPHI())
Dan Gohman88c547e2010-07-07 14:33:51 +0000144 ++I;
Akira Hatanaka6fe7aca2012-10-26 17:11:42 +0000145 assert((I == E || !I->isInsideBundle()) &&
146 "First non-phi MI cannot be inside a bundle!");
Dan Gohman88c547e2010-07-07 14:33:51 +0000147 return I;
148}
149
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000150MachineBasicBlock::iterator
151MachineBasicBlock::SkipPHIsAndLabels(MachineBasicBlock::iterator I) {
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000152 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
153
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000154 iterator E = end();
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000155 while (I != E && (I->isPHI() || I->isPosition() ||
156 TII->isBasicBlockPrologue(*I)))
Keith Walker830a8c12016-09-16 14:07:29 +0000157 ++I;
158 // FIXME: This needs to change if we wish to bundle labels
159 // inside the bundle.
160 assert((I == E || !I->isInsideBundle()) &&
161 "First non-phi / non-label instruction is inside a bundle!");
162 return I;
163}
164
165MachineBasicBlock::iterator
166MachineBasicBlock::SkipPHIsLabelsAndDebug(MachineBasicBlock::iterator I) {
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000167 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
168
Keith Walker830a8c12016-09-16 14:07:29 +0000169 iterator E = end();
Stanislav Mekhanoshin6ec3e3a2017-01-20 00:44:31 +0000170 while (I != E && (I->isPHI() || I->isPosition() || I->isDebugValue() ||
171 TII->isBasicBlockPrologue(*I)))
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000172 ++I;
Evan Cheng2a81dd42011-12-06 22:12:01 +0000173 // FIXME: This needs to change if we wish to bundle labels / dbg_values
174 // inside the bundle.
Akira Hatanaka6fe7aca2012-10-26 17:11:42 +0000175 assert((I == E || !I->isInsideBundle()) &&
Keith Walker830a8c12016-09-16 14:07:29 +0000176 "First non-phi / non-label / non-debug "
177 "instruction is inside a bundle!");
Jakob Stoklund Olesenef541852010-10-30 01:26:14 +0000178 return I;
179}
180
Chris Lattner4336b872004-10-26 15:43:42 +0000181MachineBasicBlock::iterator MachineBasicBlock::getFirstTerminator() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000182 iterator B = begin(), E = end(), I = E;
183 while (I != B && ((--I)->isTerminator() || I->isDebugValue()))
Jakob Stoklund Olesenc3810282011-01-14 02:12:54 +0000184 ; /*noop */
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000185 while (I != E && !I->isTerminator())
Evan Cheng2a81dd42011-12-06 22:12:01 +0000186 ++I;
187 return I;
188}
189
Evan Cheng7fae11b2011-12-14 02:11:42 +0000190MachineBasicBlock::instr_iterator MachineBasicBlock::getFirstInstrTerminator() {
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000191 instr_iterator B = instr_begin(), E = instr_end(), I = E;
192 while (I != B && ((--I)->isTerminator() || I->isDebugValue()))
Evan Cheng2a81dd42011-12-06 22:12:01 +0000193 ; /*noop */
Benjamin Kramerbaa41d42012-02-10 00:28:31 +0000194 while (I != E && !I->isTerminator())
Jakob Stoklund Olesenab3d6ec2011-01-14 06:33:45 +0000195 ++I;
Jakob Stoklund Olesenc3810282011-01-14 02:12:54 +0000196 return I;
Alkis Evlogimenosaf2de482004-02-23 18:14:48 +0000197}
198
Benjamin Kramer6b568962015-06-23 14:47:29 +0000199MachineBasicBlock::iterator MachineBasicBlock::getFirstNonDebugInstr() {
200 // Skip over begin-of-block dbg_value instructions.
Florian Hahn3c8b8c92016-12-16 11:10:26 +0000201 return skipDebugInstructionsForward(begin(), end());
Benjamin Kramer6b568962015-06-23 14:47:29 +0000202}
203
Jakob Stoklund Olesen4bc5e382011-01-13 21:28:52 +0000204MachineBasicBlock::iterator MachineBasicBlock::getLastNonDebugInstr() {
Evan Cheng2a81dd42011-12-06 22:12:01 +0000205 // Skip over end-of-block dbg_value instructions.
Evan Cheng7fae11b2011-12-14 02:11:42 +0000206 instr_iterator B = instr_begin(), I = instr_end();
Jakob Stoklund Olesen4bc5e382011-01-13 21:28:52 +0000207 while (I != B) {
208 --I;
Evan Cheng2a81dd42011-12-06 22:12:01 +0000209 // Return instruction that starts a bundle.
210 if (I->isDebugValue() || I->isInsideBundle())
211 continue;
212 return I;
213 }
214 // The block is all debug values.
215 return end();
216}
217
Reid Klecknered170792015-09-17 17:19:40 +0000218bool MachineBasicBlock::hasEHPadSuccessor() const {
219 for (const_succ_iterator I = succ_begin(), E = succ_end(); I != E; ++I)
220 if ((*I)->isEHPad())
221 return true;
222 return false;
223}
224
Manman Ren19f49ac2012-09-11 22:23:19 +0000225#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
Yaron Kereneb2a2542016-01-29 20:50:44 +0000226LLVM_DUMP_METHOD void MachineBasicBlock::dump() const {
David Greene6c56cef2010-01-04 23:22:07 +0000227 print(dbgs());
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000228}
Manman Ren742534c2012-09-06 19:06:06 +0000229#endif
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000230
Andrew Kaylord4971192017-06-22 23:27:16 +0000231bool MachineBasicBlock::isLegalToHoistInto() const {
232 if (isReturnBlock() || hasEHPadSuccessor())
233 return false;
234 return true;
235}
236
Jakob Stoklund Olesen2bbeaa82009-11-20 01:17:03 +0000237StringRef MachineBasicBlock::getName() const {
238 if (const BasicBlock *LBB = getBasicBlock())
239 return LBB->getName();
240 else
Matthias Braun43130592017-02-18 00:41:16 +0000241 return StringRef("", 0);
Jakob Stoklund Olesen2bbeaa82009-11-20 01:17:03 +0000242}
243
Andrew Trick320c7032012-03-07 00:18:18 +0000244/// Return a hopefully unique identifier for this block.
245std::string MachineBasicBlock::getFullName() const {
246 std::string Name;
247 if (getParent())
Craig Toppera538d832012-08-22 06:07:19 +0000248 Name = (getParent()->getName() + ":").str();
Andrew Trick320c7032012-03-07 00:18:18 +0000249 if (getBasicBlock())
250 Name += getBasicBlock()->getName();
251 else
Yaron Keren75e0c4b2015-03-27 17:51:30 +0000252 Name += ("BB" + Twine(getNumber())).str();
Andrew Trick320c7032012-03-07 00:18:18 +0000253 return Name;
254}
255
Matthias Braun0e881d62016-05-05 18:14:43 +0000256void MachineBasicBlock::print(raw_ostream &OS, const SlotIndexes *Indexes)
257 const {
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000258 const MachineFunction *MF = getParent();
Chris Lattneraf119ca2009-08-23 00:35:30 +0000259 if (!MF) {
Chris Lattner4336b872004-10-26 15:43:42 +0000260 OS << "Can't print out MachineBasicBlock because parent MachineFunction"
261 << " is null\n";
Tanya Lattnera578cb72004-05-24 06:11:51 +0000262 return;
263 }
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000264 const Function *F = MF->getFunction();
265 const Module *M = F ? F->getParent() : nullptr;
266 ModuleSlotTracker MST(M);
267 print(OS, MST, Indexes);
268}
269
270void MachineBasicBlock::print(raw_ostream &OS, ModuleSlotTracker &MST,
Matthias Braun0e881d62016-05-05 18:14:43 +0000271 const SlotIndexes *Indexes) const {
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000272 const MachineFunction *MF = getParent();
273 if (!MF) {
274 OS << "Can't print out MachineBasicBlock because parent MachineFunction"
275 << " is null\n";
276 return;
277 }
Alkis Evlogimenosfcb3f512004-09-05 18:39:20 +0000278
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000279 if (Indexes)
280 OS << Indexes->getMBBStartIdx(this) << '\t';
281
Dan Gohman34341e62009-10-31 20:19:03 +0000282 OS << "BB#" << getNumber() << ": ";
283
284 const char *Comma = "";
285 if (const BasicBlock *LBB = getBasicBlock()) {
286 OS << Comma << "derived from LLVM BB ";
Duncan P. N. Exon Smith326921542015-06-26 22:04:20 +0000287 LBB->printAsOperand(OS, /*PrintType=*/false, MST);
Dan Gohman34341e62009-10-31 20:19:03 +0000288 Comma = ", ";
289 }
Reid Kleckner0e288232015-08-27 23:27:47 +0000290 if (isEHPad()) { OS << Comma << "EH LANDING PAD"; Comma = ", "; }
Dan Gohman34341e62009-10-31 20:19:03 +0000291 if (hasAddressTaken()) { OS << Comma << "ADDRESS TAKEN"; Comma = ", "; }
Bill Wendling4fd96632012-06-15 19:30:42 +0000292 if (Alignment)
Jakob Stoklund Olesen2a2b37e2011-12-06 21:08:39 +0000293 OS << Comma << "Align " << Alignment << " (" << (1u << Alignment)
294 << " bytes)";
Jakob Stoklund Olesen2a2b37e2011-12-06 21:08:39 +0000295
Chris Lattneraf119ca2009-08-23 00:35:30 +0000296 OS << '\n';
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000297
Eric Christopherfc6de422014-08-05 02:39:49 +0000298 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
Dan Gohmanc731c972007-10-03 19:26:29 +0000299 if (!livein_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000300 if (Indexes) OS << '\t';
Dan Gohman34341e62009-10-31 20:19:03 +0000301 OS << " Live Ins:";
Matthias Braun11723322017-01-05 20:01:19 +0000302 for (const auto &LI : LiveIns) {
Matthias Braund9da1622015-09-09 18:08:03 +0000303 OS << ' ' << PrintReg(LI.PhysReg, TRI);
Krzysztof Parzyszek91b5cf82016-12-15 14:36:06 +0000304 if (!LI.LaneMask.all())
Matthias Braunc804cdb2015-09-25 21:51:24 +0000305 OS << ':' << PrintLaneMask(LI.LaneMask);
Matthias Braunb2b7ef12015-08-24 22:59:52 +0000306 }
Chris Lattneraf119ca2009-08-23 00:35:30 +0000307 OS << '\n';
Evan Chengbcf1d7f2007-02-10 02:38:19 +0000308 }
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000309 // Print the preds of this block according to the CFG.
310 if (!pred_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000311 if (Indexes) OS << '\t';
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000312 OS << " Predecessors according to CFG:";
313 for (const_pred_iterator PI = pred_begin(), E = pred_end(); PI != E; ++PI)
Dan Gohman34341e62009-10-31 20:19:03 +0000314 OS << " BB#" << (*PI)->getNumber();
Chris Lattneraf119ca2009-08-23 00:35:30 +0000315 OS << '\n';
Chris Lattner9a1e91b2006-09-26 03:41:59 +0000316 }
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000317
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000318 for (auto &I : instrs()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000319 if (Indexes) {
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000320 if (Indexes->hasIndex(I))
321 OS << Indexes->getInstructionIndex(I);
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000322 OS << '\t';
323 }
Chris Lattner1216f542009-08-23 00:47:04 +0000324 OS << '\t';
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000325 if (I.isInsideBundle())
Evan Cheng7fae11b2011-12-14 02:11:42 +0000326 OS << " * ";
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000327 I.print(OS, MST);
Alkis Evlogimenosfcb3f512004-09-05 18:39:20 +0000328 }
Chris Lattner329c14a2005-04-01 06:48:38 +0000329
330 // Print the successors of this block according to the CFG.
331 if (!succ_empty()) {
Jakob Stoklund Olesenb7050232010-10-26 20:21:46 +0000332 if (Indexes) OS << '\t';
Chris Lattner329c14a2005-04-01 06:48:38 +0000333 OS << " Successors according to CFG:";
Jakob Stoklund Olesen1dc107a2012-08-13 23:13:23 +0000334 for (const_succ_iterator SI = succ_begin(), E = succ_end(); SI != E; ++SI) {
Dan Gohman34341e62009-10-31 20:19:03 +0000335 OS << " BB#" << (*SI)->getNumber();
Cong Houd97c1002015-12-01 05:29:22 +0000336 if (!Probs.empty())
337 OS << '(' << *getProbabilityIterator(SI) << ')';
Jakob Stoklund Olesen1dc107a2012-08-13 23:13:23 +0000338 }
Chris Lattneraf119ca2009-08-23 00:35:30 +0000339 OS << '\n';
Chris Lattner329c14a2005-04-01 06:48:38 +0000340 }
Alkis Evlogimenos14f3fe82004-02-16 07:17:43 +0000341}
Chris Lattner4336b872004-10-26 15:43:42 +0000342
Cong Hou2793e722015-08-10 22:27:10 +0000343void MachineBasicBlock::printAsOperand(raw_ostream &OS,
344 bool /*PrintType*/) const {
Chandler Carruthd48cdbf2014-01-09 02:29:41 +0000345 OS << "BB#" << getNumber();
346}
347
Matthias Braune6a24852015-09-25 21:51:14 +0000348void MachineBasicBlock::removeLiveIn(MCPhysReg Reg, LaneBitmask LaneMask) {
David Majnemer42531262016-08-12 03:55:06 +0000349 LiveInVector::iterator I = find_if(
350 LiveIns, [Reg](const RegisterMaskPair &LI) { return LI.PhysReg == Reg; });
Matthias Braund9da1622015-09-09 18:08:03 +0000351 if (I == LiveIns.end())
352 return;
353
354 I->LaneMask &= ~LaneMask;
Krzysztof Parzyszek91b5cf82016-12-15 14:36:06 +0000355 if (I->LaneMask.none())
Jakob Stoklund Olesen8e58c902012-03-28 20:11:42 +0000356 LiveIns.erase(I);
Evan Chengf7ed82d2007-02-19 21:49:54 +0000357}
358
Matthias Braunac4becc2017-05-31 20:30:22 +0000359MachineBasicBlock::livein_iterator
360MachineBasicBlock::removeLiveIn(MachineBasicBlock::livein_iterator I) {
Matthias Braune2e65912017-05-31 21:25:03 +0000361 // Get non-const version of iterator.
362 LiveInVector::iterator LI = LiveIns.begin() + (I - LiveIns.begin());
363 return LiveIns.erase(LI);
Matthias Braunac4becc2017-05-31 20:30:22 +0000364}
365
Matthias Braune6a24852015-09-25 21:51:14 +0000366bool MachineBasicBlock::isLiveIn(MCPhysReg Reg, LaneBitmask LaneMask) const {
David Majnemer42531262016-08-12 03:55:06 +0000367 livein_iterator I = find_if(
368 LiveIns, [Reg](const RegisterMaskPair &LI) { return LI.PhysReg == Reg; });
Krzysztof Parzyszekea9f8ce2016-12-16 19:11:56 +0000369 return I != livein_end() && (I->LaneMask & LaneMask).any();
Matthias Braund9da1622015-09-09 18:08:03 +0000370}
371
372void MachineBasicBlock::sortUniqueLiveIns() {
373 std::sort(LiveIns.begin(), LiveIns.end(),
374 [](const RegisterMaskPair &LI0, const RegisterMaskPair &LI1) {
375 return LI0.PhysReg < LI1.PhysReg;
376 });
377 // Liveins are sorted by physreg now we can merge their lanemasks.
378 LiveInVector::const_iterator I = LiveIns.begin();
379 LiveInVector::const_iterator J;
380 LiveInVector::iterator Out = LiveIns.begin();
381 for (; I != LiveIns.end(); ++Out, I = J) {
382 unsigned PhysReg = I->PhysReg;
Matthias Braune6a24852015-09-25 21:51:14 +0000383 LaneBitmask LaneMask = I->LaneMask;
Matthias Braund9da1622015-09-09 18:08:03 +0000384 for (J = std::next(I); J != LiveIns.end() && J->PhysReg == PhysReg; ++J)
385 LaneMask |= J->LaneMask;
386 Out->PhysReg = PhysReg;
387 Out->LaneMask = LaneMask;
388 }
389 LiveIns.erase(Out, LiveIns.end());
Aaron Ballman9f154f62015-07-29 15:57:49 +0000390}
391
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000392unsigned
Matthias Braun130bd902015-08-25 22:05:55 +0000393MachineBasicBlock::addLiveIn(MCPhysReg PhysReg, const TargetRegisterClass *RC) {
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000394 assert(getParent() && "MBB must be inserted in function");
395 assert(TargetRegisterInfo::isPhysicalRegister(PhysReg) && "Expected physreg");
396 assert(RC && "Register class is required");
Reid Kleckner0e288232015-08-27 23:27:47 +0000397 assert((isEHPad() || this == &getParent()->front()) &&
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000398 "Only the entry block and landing pads can have physreg live ins");
399
400 bool LiveIn = isLiveIn(PhysReg);
Jakob Stoklund Olesenbbbb5322013-07-04 04:32:35 +0000401 iterator I = SkipPHIsAndLabels(begin()), E = end();
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000402 MachineRegisterInfo &MRI = getParent()->getRegInfo();
Eric Christopherfc6de422014-08-05 02:39:49 +0000403 const TargetInstrInfo &TII = *getParent()->getSubtarget().getInstrInfo();
Jakob Stoklund Olesen533c3bf2013-07-03 23:56:20 +0000404
405 // Look for an existing copy.
406 if (LiveIn)
407 for (;I != E && I->isCopy(); ++I)
408 if (I->getOperand(1).getReg() == PhysReg) {
409 unsigned VirtReg = I->getOperand(0).getReg();
410 if (!MRI.constrainRegClass(VirtReg, RC))
411 llvm_unreachable("Incompatible live-in register class.");
412 return VirtReg;
413 }
414
415 // No luck, create a virtual register.
416 unsigned VirtReg = MRI.createVirtualRegister(RC);
417 BuildMI(*this, I, DebugLoc(), TII.get(TargetOpcode::COPY), VirtReg)
418 .addReg(PhysReg, RegState::Kill);
419 if (!LiveIn)
420 addLiveIn(PhysReg);
421 return VirtReg;
422}
423
Chris Lattner94866be2006-10-24 00:02:26 +0000424void MachineBasicBlock::moveBefore(MachineBasicBlock *NewAfter) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000425 getParent()->splice(NewAfter->getIterator(), getIterator());
Chris Lattner94866be2006-10-24 00:02:26 +0000426}
427
428void MachineBasicBlock::moveAfter(MachineBasicBlock *NewBefore) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000429 getParent()->splice(++NewBefore->getIterator(), getIterator());
Chris Lattner94866be2006-10-24 00:02:26 +0000430}
431
Jim Grosbach801b33b2009-11-12 03:55:33 +0000432void MachineBasicBlock::updateTerminator() {
Eric Christopherfc6de422014-08-05 02:39:49 +0000433 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Jim Grosbach801b33b2009-11-12 03:55:33 +0000434 // A block with no successors has no concerns with fall-through edges.
Chad Rosierdca76512016-05-25 21:53:46 +0000435 if (this->succ_empty())
436 return;
Jim Grosbach801b33b2009-11-12 03:55:33 +0000437
Craig Topperc0196b12014-04-14 00:51:57 +0000438 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
Jim Grosbach801b33b2009-11-12 03:55:33 +0000439 SmallVector<MachineOperand, 4> Cond;
Taewook Oh06a21282017-02-13 18:15:31 +0000440 DebugLoc DL = findBranchDebugLoc();
Jacques Pienaar71c30a12016-07-15 14:41:04 +0000441 bool B = TII->analyzeBranch(*this, TBB, FBB, Cond);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000442 (void) B;
443 assert(!B && "UpdateTerminators requires analyzable predecessors!");
444 if (Cond.empty()) {
445 if (TBB) {
Chad Rosierdca76512016-05-25 21:53:46 +0000446 // The block has an unconditional branch. If its successor is now its
447 // layout successor, delete the branch.
Jim Grosbach801b33b2009-11-12 03:55:33 +0000448 if (isLayoutSuccessor(TBB))
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000449 TII->removeBranch(*this);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000450 } else {
Chad Rosierdca76512016-05-25 21:53:46 +0000451 // The block has an unconditional fallthrough. If its successor is not its
452 // layout successor, insert a branch. First we have to locate the only
453 // non-landing-pad successor, as that is the fallthrough block.
Chandler Carruthee54feb2011-11-22 13:13:16 +0000454 for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) {
Reid Kleckner0e288232015-08-27 23:27:47 +0000455 if ((*SI)->isEHPad())
Chandler Carruthee54feb2011-11-22 13:13:16 +0000456 continue;
457 assert(!TBB && "Found more than one non-landing-pad successor!");
458 TBB = *SI;
459 }
Chandler Carruth8c68f1f2011-11-23 08:23:54 +0000460
Chad Rosierdca76512016-05-25 21:53:46 +0000461 // If there is no non-landing-pad successor, the block has no fall-through
462 // edges to be concerned with.
Chandler Carruth8c68f1f2011-11-23 08:23:54 +0000463 if (!TBB)
464 return;
465
466 // Finally update the unconditional successor to be reached via a branch
467 // if it would not be reached by fallthrough.
Jim Grosbach801b33b2009-11-12 03:55:33 +0000468 if (!isLayoutSuccessor(TBB))
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000469 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000470 }
Chad Rosierdca76512016-05-25 21:53:46 +0000471 return;
472 }
Chandler Carruth1f5580b2012-04-16 22:03:00 +0000473
Chad Rosierdca76512016-05-25 21:53:46 +0000474 if (FBB) {
475 // The block has a non-fallthrough conditional branch. If one of its
476 // successors is its layout successor, rewrite it to a fallthrough
477 // conditional branch.
478 if (isLayoutSuccessor(TBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000479 if (TII->reverseBranchCondition(Cond))
Chandler Carruth1f5580b2012-04-16 22:03:00 +0000480 return;
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000481 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000482 TII->insertBranch(*this, FBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000483 } else if (isLayoutSuccessor(FBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000484 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000485 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000486 }
Chad Rosierdca76512016-05-25 21:53:46 +0000487 return;
488 }
489
490 // Walk through the successors and find the successor which is not a landing
491 // pad and is not the conditional branch destination (in TBB) as the
492 // fallthrough successor.
493 MachineBasicBlock *FallthroughBB = nullptr;
494 for (succ_iterator SI = succ_begin(), SE = succ_end(); SI != SE; ++SI) {
495 if ((*SI)->isEHPad() || *SI == TBB)
496 continue;
497 assert(!FallthroughBB && "Found more than one fallthrough successor.");
498 FallthroughBB = *SI;
499 }
500
Haicheng Wub71b2f62016-07-03 19:14:17 +0000501 if (!FallthroughBB) {
502 if (canFallThrough()) {
503 // We fallthrough to the same basic block as the conditional jump targets.
504 // Remove the conditional jump, leaving unconditional fallthrough.
505 // FIXME: This does not seem like a reasonable pattern to support, but it
506 // has been seen in the wild coming out of degenerate ARM test cases.
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000507 TII->removeBranch(*this);
Taewook Oh06a21282017-02-13 18:15:31 +0000508
Haicheng Wub71b2f62016-07-03 19:14:17 +0000509 // Finally update the unconditional successor to be reached via a branch if
510 // it would not be reached by fallthrough.
511 if (!isLayoutSuccessor(TBB))
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000512 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Haicheng Wub71b2f62016-07-03 19:14:17 +0000513 return;
514 }
Chad Rosierdca76512016-05-25 21:53:46 +0000515
Haicheng Wub71b2f62016-07-03 19:14:17 +0000516 // We enter here iff exactly one successor is TBB which cannot fallthrough
517 // and the rest successors if any are EHPads. In this case, we need to
518 // change the conditional branch into unconditional branch.
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000519 TII->removeBranch(*this);
Haicheng Wub71b2f62016-07-03 19:14:17 +0000520 Cond.clear();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000521 TII->insertBranch(*this, TBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000522 return;
523 }
524
525 // The block has a fallthrough conditional branch.
526 if (isLayoutSuccessor(TBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000527 if (TII->reverseBranchCondition(Cond)) {
Chad Rosierdca76512016-05-25 21:53:46 +0000528 // We can't reverse the condition, add an unconditional branch.
529 Cond.clear();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000530 TII->insertBranch(*this, FallthroughBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000531 return;
532 }
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000533 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000534 TII->insertBranch(*this, FallthroughBB, nullptr, Cond, DL);
Chad Rosierdca76512016-05-25 21:53:46 +0000535 } else if (!isLayoutSuccessor(FallthroughBB)) {
Matt Arsenault1b9fc8e2016-09-14 20:43:16 +0000536 TII->removeBranch(*this);
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000537 TII->insertBranch(*this, TBB, FallthroughBB, Cond, DL);
Jim Grosbach801b33b2009-11-12 03:55:33 +0000538 }
539}
Chris Lattner94866be2006-10-24 00:02:26 +0000540
Cong Houc1069892015-12-13 09:26:17 +0000541void MachineBasicBlock::validateSuccProbs() const {
542#ifndef NDEBUG
543 int64_t Sum = 0;
544 for (auto Prob : Probs)
545 Sum += Prob.getNumerator();
546 // Due to precision issue, we assume that the sum of probabilities is one if
547 // the difference between the sum of their numerators and the denominator is
548 // no greater than the number of successors.
Cong Houc00e65a2015-12-13 17:00:25 +0000549 assert((uint64_t)std::abs(Sum - BranchProbability::getDenominator()) <=
Cong Houc1069892015-12-13 09:26:17 +0000550 Probs.size() &&
551 "The sum of successors's probabilities exceeds one.");
552#endif // NDEBUG
553}
554
Cong Hou23a3bf02015-11-04 21:37:58 +0000555void MachineBasicBlock::addSuccessor(MachineBasicBlock *Succ,
556 BranchProbability Prob) {
557 // Probability list is either empty (if successor list isn't empty, this means
558 // disabled optimization) or has the same size as successor list.
Cong Hou4aef7ef2015-12-01 11:05:39 +0000559 if (!(Probs.empty() && !Successors.empty()))
Cong Hou23a3bf02015-11-04 21:37:58 +0000560 Probs.push_back(Prob);
561 Successors.push_back(Succ);
562 Succ->addPredecessor(this);
563}
564
565void MachineBasicBlock::addSuccessorWithoutProb(MachineBasicBlock *Succ) {
566 // We need to make sure probability list is either empty or has the same size
567 // of successor list. When this function is called, we can safely delete all
568 // probability in the list.
569 Probs.clear();
570 Successors.push_back(Succ);
571 Succ->addPredecessor(this);
572}
573
Cong Houc1069892015-12-13 09:26:17 +0000574void MachineBasicBlock::removeSuccessor(MachineBasicBlock *Succ,
575 bool NormalizeSuccProbs) {
David Majnemer0d955d02016-08-11 22:21:41 +0000576 succ_iterator I = find(Successors, Succ);
Cong Houc1069892015-12-13 09:26:17 +0000577 removeSuccessor(I, NormalizeSuccProbs);
Chris Lattner4336b872004-10-26 15:43:42 +0000578}
579
Jakub Staszakfeadd432011-06-16 18:01:17 +0000580MachineBasicBlock::succ_iterator
Cong Houc1069892015-12-13 09:26:17 +0000581MachineBasicBlock::removeSuccessor(succ_iterator I, bool NormalizeSuccProbs) {
Chris Lattner4336b872004-10-26 15:43:42 +0000582 assert(I != Successors.end() && "Not a current successor!");
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000583
Cong Hou23a3bf02015-11-04 21:37:58 +0000584 // If probability list is empty it means we don't use it (disabled
585 // optimization).
586 if (!Probs.empty()) {
587 probability_iterator WI = getProbabilityIterator(I);
588 Probs.erase(WI);
Cong Houc1069892015-12-13 09:26:17 +0000589 if (NormalizeSuccProbs)
590 normalizeSuccProbs();
Cong Hou23a3bf02015-11-04 21:37:58 +0000591 }
592
Chris Lattner4336b872004-10-26 15:43:42 +0000593 (*I)->removePredecessor(this);
Dan Gohmanf87dc922009-01-08 22:19:34 +0000594 return Successors.erase(I);
Chris Lattner4336b872004-10-26 15:43:42 +0000595}
596
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000597void MachineBasicBlock::replaceSuccessor(MachineBasicBlock *Old,
598 MachineBasicBlock *New) {
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000599 if (Old == New)
600 return;
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000601
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000602 succ_iterator E = succ_end();
603 succ_iterator NewI = E;
604 succ_iterator OldI = E;
605 for (succ_iterator I = succ_begin(); I != E; ++I) {
606 if (*I == Old) {
607 OldI = I;
608 if (NewI != E)
609 break;
610 }
611 if (*I == New) {
612 NewI = I;
613 if (OldI != E)
614 break;
615 }
616 }
617 assert(OldI != E && "Old is not a successor of this block");
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000618
619 // If New isn't already a successor, let it take Old's place.
620 if (NewI == E) {
Cong Hou11c14202015-11-18 01:45:10 +0000621 Old->removePredecessor(this);
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000622 New->addPredecessor(this);
623 *OldI = New;
624 return;
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000625 }
626
Jakob Stoklund Olesen8c28ac92012-08-10 03:23:27 +0000627 // New is already a successor.
Cong Hou23a3bf02015-11-04 21:37:58 +0000628 // Update its probability instead of adding a duplicate edge.
Cong Houd97c1002015-12-01 05:29:22 +0000629 if (!Probs.empty()) {
630 auto ProbIter = getProbabilityIterator(NewI);
631 if (!ProbIter->isUnknown())
632 *ProbIter += *getProbabilityIterator(OldI);
633 }
Cong Hou11c14202015-11-18 01:45:10 +0000634 removeSuccessor(OldI);
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000635}
636
Cong Hou166e0852015-09-29 19:46:09 +0000637void MachineBasicBlock::addPredecessor(MachineBasicBlock *Pred) {
638 Predecessors.push_back(Pred);
Chris Lattner4336b872004-10-26 15:43:42 +0000639}
640
Cong Hou166e0852015-09-29 19:46:09 +0000641void MachineBasicBlock::removePredecessor(MachineBasicBlock *Pred) {
David Majnemer0d955d02016-08-11 22:21:41 +0000642 pred_iterator I = find(Predecessors, Pred);
Chris Lattner4336b872004-10-26 15:43:42 +0000643 assert(I != Predecessors.end() && "Pred is not a predecessor of this block!");
644 Predecessors.erase(I);
645}
Evan Chenga92b2b32007-05-17 23:58:53 +0000646
Cong Hou166e0852015-09-29 19:46:09 +0000647void MachineBasicBlock::transferSuccessors(MachineBasicBlock *FromMBB) {
648 if (this == FromMBB)
Mon P Wang3e583932008-05-05 19:05:59 +0000649 return;
Jakub Staszakfeadd432011-06-16 18:01:17 +0000650
Cong Hou166e0852015-09-29 19:46:09 +0000651 while (!FromMBB->succ_empty()) {
652 MachineBasicBlock *Succ = *FromMBB->succ_begin();
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000653
Cong Houd97c1002015-12-01 05:29:22 +0000654 // If probability list is empty it means we don't use it (disabled optimization).
655 if (!FromMBB->Probs.empty()) {
656 auto Prob = *FromMBB->Probs.begin();
657 addSuccessor(Succ, Prob);
658 } else
659 addSuccessorWithoutProb(Succ);
Jakub Staszak12a43bd2011-06-16 20:22:37 +0000660
Cong Hou166e0852015-09-29 19:46:09 +0000661 FromMBB->removeSuccessor(Succ);
Dan Gohman34396292010-07-06 20:24:04 +0000662 }
663}
664
665void
Cong Hou166e0852015-09-29 19:46:09 +0000666MachineBasicBlock::transferSuccessorsAndUpdatePHIs(MachineBasicBlock *FromMBB) {
667 if (this == FromMBB)
Dan Gohman34396292010-07-06 20:24:04 +0000668 return;
Jakub Staszakfeadd432011-06-16 18:01:17 +0000669
Cong Hou166e0852015-09-29 19:46:09 +0000670 while (!FromMBB->succ_empty()) {
671 MachineBasicBlock *Succ = *FromMBB->succ_begin();
Cong Houd97c1002015-12-01 05:29:22 +0000672 if (!FromMBB->Probs.empty()) {
673 auto Prob = *FromMBB->Probs.begin();
674 addSuccessor(Succ, Prob);
675 } else
676 addSuccessorWithoutProb(Succ);
Cong Hou166e0852015-09-29 19:46:09 +0000677 FromMBB->removeSuccessor(Succ);
Dan Gohman34396292010-07-06 20:24:04 +0000678
679 // Fix up any PHI nodes in the successor.
Evan Cheng7fae11b2011-12-14 02:11:42 +0000680 for (MachineBasicBlock::instr_iterator MI = Succ->instr_begin(),
681 ME = Succ->instr_end(); MI != ME && MI->isPHI(); ++MI)
Dan Gohman34396292010-07-06 20:24:04 +0000682 for (unsigned i = 2, e = MI->getNumOperands()+1; i != e; i += 2) {
683 MachineOperand &MO = MI->getOperand(i);
Cong Hou166e0852015-09-29 19:46:09 +0000684 if (MO.getMBB() == FromMBB)
Dan Gohman34396292010-07-06 20:24:04 +0000685 MO.setMBB(this);
686 }
687 }
Cong Houc1069892015-12-13 09:26:17 +0000688 normalizeSuccProbs();
Mon P Wang3e583932008-05-05 19:05:59 +0000689}
690
Jakob Stoklund Olesenfee94ca2012-07-30 17:36:47 +0000691bool MachineBasicBlock::isPredecessor(const MachineBasicBlock *MBB) const {
David Majnemer42531262016-08-12 03:55:06 +0000692 return is_contained(predecessors(), MBB);
Jakob Stoklund Olesenfee94ca2012-07-30 17:36:47 +0000693}
694
Dan Gohmanff62c622009-03-30 20:06:29 +0000695bool MachineBasicBlock::isSuccessor(const MachineBasicBlock *MBB) const {
David Majnemer42531262016-08-12 03:55:06 +0000696 return is_contained(successors(), MBB);
Evan Chenga92b2b32007-05-17 23:58:53 +0000697}
Evan Chengdf757852007-06-04 06:44:01 +0000698
Dan Gohmanff62c622009-03-30 20:06:29 +0000699bool MachineBasicBlock::isLayoutSuccessor(const MachineBasicBlock *MBB) const {
Dan Gohmana78bae32008-10-02 22:09:09 +0000700 MachineFunction::const_iterator I(this);
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000701 return std::next(I) == MachineFunction::const_iterator(MBB);
Dan Gohmana78bae32008-10-02 22:09:09 +0000702}
703
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000704MachineBasicBlock *MachineBasicBlock::getFallThrough() {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000705 MachineFunction::iterator Fallthrough = getIterator();
Bob Wilson2d4ff122009-11-26 00:32:21 +0000706 ++Fallthrough;
707 // If FallthroughBlock is off the end of the function, it can't fall through.
708 if (Fallthrough == getParent()->end())
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000709 return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000710
711 // If FallthroughBlock isn't a successor, no fallthrough is possible.
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000712 if (!isSuccessor(&*Fallthrough))
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000713 return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000714
Dan Gohman0b44cb02009-12-05 00:32:59 +0000715 // Analyze the branches, if any, at the end of the block.
Craig Topperc0196b12014-04-14 00:51:57 +0000716 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
Dan Gohman0b44cb02009-12-05 00:32:59 +0000717 SmallVector<MachineOperand, 4> Cond;
Eric Christopherfc6de422014-08-05 02:39:49 +0000718 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Jacques Pienaar71c30a12016-07-15 14:41:04 +0000719 if (TII->analyzeBranch(*this, TBB, FBB, Cond)) {
Dan Gohman0b44cb02009-12-05 00:32:59 +0000720 // If we couldn't analyze the branch, examine the last instruction.
721 // If the block doesn't end in a known control barrier, assume fallthrough
Chad Rosier1a1531d2012-01-26 18:24:25 +0000722 // is possible. The isPredicated check is needed because this code can be
Dan Gohman0b44cb02009-12-05 00:32:59 +0000723 // called during IfConversion, where an instruction which is normally a
Chad Rosier9b61cf32012-01-26 20:19:05 +0000724 // Barrier is predicated and thus no longer an actual control barrier.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000725 return (empty() || !back().isBarrier() || TII->isPredicated(back()))
726 ? &*Fallthrough
727 : nullptr;
Dan Gohman0b44cb02009-12-05 00:32:59 +0000728 }
Bob Wilson2d4ff122009-11-26 00:32:21 +0000729
730 // If there is no branch, control always falls through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000731 if (!TBB) return &*Fallthrough;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000732
733 // If there is some explicit branch to the fallthrough block, it can obviously
734 // reach, even though the branch should get folded to fall through implicitly.
735 if (MachineFunction::iterator(TBB) == Fallthrough ||
736 MachineFunction::iterator(FBB) == Fallthrough)
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000737 return &*Fallthrough;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000738
739 // If it's an unconditional branch to some block not the fall through, it
740 // doesn't fall through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000741 if (Cond.empty()) return nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000742
743 // Otherwise, if it is conditional and has no explicit false block, it falls
744 // through.
Jan Sjodinf1a30f12017-03-31 15:55:37 +0000745 return (FBB == nullptr) ? &*Fallthrough : nullptr;
746}
747
748bool MachineBasicBlock::canFallThrough() {
749 return getFallThrough() != nullptr;
Bob Wilson2d4ff122009-11-26 00:32:21 +0000750}
751
Quentin Colombet23341a82016-04-21 21:01:13 +0000752MachineBasicBlock *MachineBasicBlock::SplitCriticalEdge(MachineBasicBlock *Succ,
753 Pass &P) {
Quentin Colombet77e18782016-04-21 20:46:27 +0000754 if (!canSplitCriticalEdge(Succ))
Craig Topperc0196b12014-04-14 00:51:57 +0000755 return nullptr;
Evan Cheng2d14d8a2012-04-24 19:06:55 +0000756
Dan Gohman3570f812010-06-22 17:25:57 +0000757 MachineFunction *MF = getParent();
Cong Hou166e0852015-09-29 19:46:09 +0000758 DebugLoc DL; // FIXME: this is nowhere
Dan Gohman3570f812010-06-22 17:25:57 +0000759
Dan Gohman3570f812010-06-22 17:25:57 +0000760 MachineBasicBlock *NMBB = MF->CreateMachineBasicBlock();
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000761 MF->insert(std::next(MachineFunction::iterator(this)), NMBB);
Evan Chenga6848242010-08-17 17:15:14 +0000762 DEBUG(dbgs() << "Splitting critical edge:"
Dan Gohman3570f812010-06-22 17:25:57 +0000763 " BB#" << getNumber()
764 << " -- BB#" << NMBB->getNumber()
765 << " -- BB#" << Succ->getNumber() << '\n');
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000766
Quentin Colombet23341a82016-04-21 21:01:13 +0000767 LiveIntervals *LIS = P.getAnalysisIfAvailable<LiveIntervals>();
768 SlotIndexes *Indexes = P.getAnalysisIfAvailable<SlotIndexes>();
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000769 if (LIS)
770 LIS->insertMBBInMaps(NMBB);
771 else if (Indexes)
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000772 Indexes->insertMBBInMaps(NMBB);
Dan Gohman3570f812010-06-22 17:25:57 +0000773
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000774 // On some targets like Mips, branches may kill virtual registers. Make sure
775 // that LiveVariables is properly updated after updateTerminator replaces the
776 // terminators.
Quentin Colombet23341a82016-04-21 21:01:13 +0000777 LiveVariables *LV = P.getAnalysisIfAvailable<LiveVariables>();
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000778
779 // Collect a list of virtual registers killed by the terminators.
780 SmallVector<unsigned, 4> KilledRegs;
781 if (LV)
Evan Cheng7fae11b2011-12-14 02:11:42 +0000782 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
Evan Cheng2a81dd42011-12-06 22:12:01 +0000783 I != E; ++I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000784 MachineInstr *MI = &*I;
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000785 for (MachineInstr::mop_iterator OI = MI->operands_begin(),
786 OE = MI->operands_end(); OI != OE; ++OI) {
Lang Hamesedeea172012-02-09 05:59:36 +0000787 if (!OI->isReg() || OI->getReg() == 0 ||
788 !OI->isUse() || !OI->isKill() || OI->isUndef())
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000789 continue;
790 unsigned Reg = OI->getReg();
Lang Hamesedeea172012-02-09 05:59:36 +0000791 if (TargetRegisterInfo::isPhysicalRegister(Reg) ||
Duncan P. N. Exon Smithd26fdc82016-07-01 01:51:32 +0000792 LV->getVarInfo(Reg).removeKill(*MI)) {
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000793 KilledRegs.push_back(Reg);
794 DEBUG(dbgs() << "Removing terminator kill: " << *MI);
795 OI->setIsKill(false);
796 }
797 }
798 }
799
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000800 SmallVector<unsigned, 4> UsedRegs;
801 if (LIS) {
802 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
803 I != E; ++I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000804 MachineInstr *MI = &*I;
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000805
806 for (MachineInstr::mop_iterator OI = MI->operands_begin(),
807 OE = MI->operands_end(); OI != OE; ++OI) {
808 if (!OI->isReg() || OI->getReg() == 0)
809 continue;
810
811 unsigned Reg = OI->getReg();
David Majnemer0d955d02016-08-11 22:21:41 +0000812 if (!is_contained(UsedRegs, Reg))
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000813 UsedRegs.push_back(Reg);
814 }
815 }
816 }
817
Dan Gohman3570f812010-06-22 17:25:57 +0000818 ReplaceUsesOfBlockWith(Succ, NMBB);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000819
820 // If updateTerminator() removes instructions, we need to remove them from
821 // SlotIndexes.
822 SmallVector<MachineInstr*, 4> Terminators;
823 if (Indexes) {
824 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
825 I != E; ++I)
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000826 Terminators.push_back(&*I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000827 }
828
Dan Gohman3570f812010-06-22 17:25:57 +0000829 updateTerminator();
830
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000831 if (Indexes) {
832 SmallVector<MachineInstr*, 4> NewTerminators;
833 for (instr_iterator I = getFirstInstrTerminator(), E = instr_end();
834 I != E; ++I)
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000835 NewTerminators.push_back(&*I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000836
837 for (SmallVectorImpl<MachineInstr*>::iterator I = Terminators.begin(),
838 E = Terminators.end(); I != E; ++I) {
David Majnemer0d955d02016-08-11 22:21:41 +0000839 if (!is_contained(NewTerminators, *I))
840 Indexes->removeMachineInstrFromMaps(**I);
Cameron Zwarichba378ce2013-02-11 09:24:45 +0000841 }
842 }
843
Dan Gohman3570f812010-06-22 17:25:57 +0000844 // Insert unconditional "jump Succ" instruction in NMBB if necessary.
845 NMBB->addSuccessor(Succ);
846 if (!NMBB->isLayoutSuccessor(Succ)) {
Quentin Colombet77e18782016-04-21 20:46:27 +0000847 SmallVector<MachineOperand, 4> Cond;
848 const TargetInstrInfo *TII = getParent()->getSubtarget().getInstrInfo();
Matt Arsenaulte8e0f5c2016-09-14 17:24:15 +0000849 TII->insertBranch(*NMBB, Succ, nullptr, Cond, DL);
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000850
851 if (Indexes) {
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000852 for (MachineInstr &MI : NMBB->instrs()) {
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000853 // Some instructions may have been moved to NMBB by updateTerminator(),
854 // so we first remove any instruction that already has an index.
Duncan P. N. Exon Smith3ac9cc62016-02-27 06:40:41 +0000855 if (Indexes->hasIndex(MI))
856 Indexes->removeMachineInstrFromMaps(MI);
857 Indexes->insertMachineInstrInMaps(MI);
Cameron Zwarich21beaf62013-02-10 23:29:54 +0000858 }
859 }
Dan Gohman3570f812010-06-22 17:25:57 +0000860 }
861
862 // Fix PHI nodes in Succ so they refer to NMBB instead of this
Evan Cheng7fae11b2011-12-14 02:11:42 +0000863 for (MachineBasicBlock::instr_iterator
864 i = Succ->instr_begin(),e = Succ->instr_end();
865 i != e && i->isPHI(); ++i)
Dan Gohman3570f812010-06-22 17:25:57 +0000866 for (unsigned ni = 1, ne = i->getNumOperands(); ni != ne; ni += 2)
867 if (i->getOperand(ni+1).getMBB() == this)
868 i->getOperand(ni+1).setMBB(NMBB);
869
Jakob Stoklund Olesen06b6ccf2011-10-14 17:25:46 +0000870 // Inherit live-ins from the successor
Matthias Braund9da1622015-09-09 18:08:03 +0000871 for (const auto &LI : Succ->liveins())
Matthias Braunb2b7ef12015-08-24 22:59:52 +0000872 NMBB->addLiveIn(LI);
Jakob Stoklund Olesen06b6ccf2011-10-14 17:25:46 +0000873
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000874 // Update LiveVariables.
Eric Christopherfc6de422014-08-05 02:39:49 +0000875 const TargetRegisterInfo *TRI = MF->getSubtarget().getRegisterInfo();
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000876 if (LV) {
877 // Restore kills of virtual registers that were killed by the terminators.
878 while (!KilledRegs.empty()) {
879 unsigned Reg = KilledRegs.pop_back_val();
Evan Cheng7fae11b2011-12-14 02:11:42 +0000880 for (instr_iterator I = instr_end(), E = instr_begin(); I != E;) {
Lang Hamesedeea172012-02-09 05:59:36 +0000881 if (!(--I)->addRegisterKilled(Reg, TRI, /* addIfNotFound= */ false))
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000882 continue;
Lang Hamesedeea172012-02-09 05:59:36 +0000883 if (TargetRegisterInfo::isVirtualRegister(Reg))
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +0000884 LV->getVarInfo(Reg).Kills.push_back(&*I);
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000885 DEBUG(dbgs() << "Restored terminator kill: " << *I);
886 break;
887 }
888 }
889 // Update relevant live-through information.
Dan Gohman3570f812010-06-22 17:25:57 +0000890 LV->addNewBlock(NMBB, this, Succ);
Jakob Stoklund Olesendd6fcc42011-05-29 20:10:28 +0000891 }
Dan Gohman3570f812010-06-22 17:25:57 +0000892
Cameron Zwarichcdcab382013-02-12 03:49:20 +0000893 if (LIS) {
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000894 // After splitting the edge and updating SlotIndexes, live intervals may be
895 // in one of two situations, depending on whether this block was the last in
Cong Hou2793e722015-08-10 22:27:10 +0000896 // the function. If the original block was the last in the function, all
897 // live intervals will end prior to the beginning of the new split block. If
898 // the original block was not at the end of the function, all live intervals
899 // will extend to the end of the new split block.
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000900
901 bool isLastMBB =
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000902 std::next(MachineFunction::iterator(NMBB)) == getParent()->end();
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000903
904 SlotIndex StartIndex = Indexes->getMBBEndIdx(this);
905 SlotIndex PrevIndex = StartIndex.getPrevSlot();
906 SlotIndex EndIndex = Indexes->getMBBEndIdx(NMBB);
907
908 // Find the registers used from NMBB in PHIs in Succ.
909 SmallSet<unsigned, 8> PHISrcRegs;
910 for (MachineBasicBlock::instr_iterator
911 I = Succ->instr_begin(), E = Succ->instr_end();
912 I != E && I->isPHI(); ++I) {
913 for (unsigned ni = 1, ne = I->getNumOperands(); ni != ne; ni += 2) {
914 if (I->getOperand(ni+1).getMBB() == NMBB) {
915 MachineOperand &MO = I->getOperand(ni);
916 unsigned Reg = MO.getReg();
917 PHISrcRegs.insert(Reg);
Cameron Zwarichaf349312013-02-12 03:49:17 +0000918 if (MO.isUndef())
919 continue;
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000920
921 LiveInterval &LI = LIS->getInterval(Reg);
922 VNInfo *VNI = LI.getVNInfoAt(PrevIndex);
Cong Hou2793e722015-08-10 22:27:10 +0000923 assert(VNI &&
924 "PHI sources should be live out of their predecessors.");
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000925 LI.addSegment(LiveInterval::Segment(StartIndex, EndIndex, VNI));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000926 }
927 }
928 }
929
930 MachineRegisterInfo *MRI = &getParent()->getRegInfo();
931 for (unsigned i = 0, e = MRI->getNumVirtRegs(); i != e; ++i) {
932 unsigned Reg = TargetRegisterInfo::index2VirtReg(i);
933 if (PHISrcRegs.count(Reg) || !LIS->hasInterval(Reg))
934 continue;
935
936 LiveInterval &LI = LIS->getInterval(Reg);
937 if (!LI.liveAt(PrevIndex))
938 continue;
939
Cameron Zwarichaf349312013-02-12 03:49:17 +0000940 bool isLiveOut = LI.liveAt(LIS->getMBBStartIdx(Succ));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000941 if (isLiveOut && isLastMBB) {
942 VNInfo *VNI = LI.getVNInfoAt(PrevIndex);
943 assert(VNI && "LiveInterval should have VNInfo where it is live.");
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000944 LI.addSegment(LiveInterval::Segment(StartIndex, EndIndex, VNI));
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000945 } else if (!isLiveOut && !isLastMBB) {
Matthias Braun13ddb7c2013-10-10 21:28:43 +0000946 LI.removeSegment(StartIndex, EndIndex);
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000947 }
948 }
Cameron Zwarichbfebb412013-02-17 00:10:44 +0000949
950 // Update all intervals for registers whose uses may have been modified by
951 // updateTerminator().
Cameron Zwarich24955962013-02-17 11:09:00 +0000952 LIS->repairIntervalsInRange(this, getFirstTerminator(), end(), UsedRegs);
Cameron Zwarichb47fb382013-02-11 09:24:47 +0000953 }
954
Dan Gohman3570f812010-06-22 17:25:57 +0000955 if (MachineDominatorTree *MDT =
Quentin Colombet23341a82016-04-21 21:01:13 +0000956 P.getAnalysisIfAvailable<MachineDominatorTree>())
Quentin Colombetabea99f2014-08-13 21:00:07 +0000957 MDT->recordSplitCriticalEdge(this, Succ, NMBB);
Dan Gohman3570f812010-06-22 17:25:57 +0000958
Quentin Colombet23341a82016-04-21 21:01:13 +0000959 if (MachineLoopInfo *MLI = P.getAnalysisIfAvailable<MachineLoopInfo>())
Dan Gohman3570f812010-06-22 17:25:57 +0000960 if (MachineLoop *TIL = MLI->getLoopFor(this)) {
961 // If one or the other blocks were not in a loop, the new block is not
962 // either, and thus LI doesn't need to be updated.
963 if (MachineLoop *DestLoop = MLI->getLoopFor(Succ)) {
964 if (TIL == DestLoop) {
965 // Both in the same loop, the NMBB joins loop.
966 DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase());
967 } else if (TIL->contains(DestLoop)) {
968 // Edge from an outer loop to an inner loop. Add to the outer loop.
969 TIL->addBasicBlockToLoop(NMBB, MLI->getBase());
970 } else if (DestLoop->contains(TIL)) {
971 // Edge from an inner loop to an outer loop. Add to the outer loop.
972 DestLoop->addBasicBlockToLoop(NMBB, MLI->getBase());
973 } else {
974 // Edge from two loops with no containment relation. Because these
975 // are natural loops, we know that the destination block must be the
976 // header of its loop (adding a branch into a loop elsewhere would
977 // create an irreducible loop).
978 assert(DestLoop->getHeader() == Succ &&
979 "Should not create irreducible loops!");
980 if (MachineLoop *P = DestLoop->getParentLoop())
981 P->addBasicBlockToLoop(NMBB, MLI->getBase());
982 }
983 }
984 }
985
986 return NMBB;
987}
988
Quentin Colombet77e18782016-04-21 20:46:27 +0000989bool MachineBasicBlock::canSplitCriticalEdge(
990 const MachineBasicBlock *Succ) const {
991 // Splitting the critical edge to a landing pad block is non-trivial. Don't do
992 // it in this generic function.
993 if (Succ->isEHPad())
994 return false;
995
996 const MachineFunction *MF = getParent();
997
998 // Performance might be harmed on HW that implements branching using exec mask
999 // where both sides of the branches are always executed.
1000 if (MF->getTarget().requiresStructuredCFG())
1001 return false;
1002
1003 // We may need to update this's terminator, but we can't do that if
1004 // AnalyzeBranch fails. If this uses a jump table, we won't touch it.
1005 const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
1006 MachineBasicBlock *TBB = nullptr, *FBB = nullptr;
1007 SmallVector<MachineOperand, 4> Cond;
1008 // AnalyzeBanch should modify this, since we did not allow modification.
Jacques Pienaar71c30a12016-07-15 14:41:04 +00001009 if (TII->analyzeBranch(*const_cast<MachineBasicBlock *>(this), TBB, FBB, Cond,
Quentin Colombet77e18782016-04-21 20:46:27 +00001010 /*AllowModify*/ false))
1011 return false;
1012
1013 // Avoid bugpoint weirdness: A block may end with a conditional branch but
1014 // jumps to the same MBB is either case. We have duplicate CFG edges in that
1015 // case that we can't handle. Since this never happens in properly optimized
1016 // code, just skip those edges.
1017 if (TBB && TBB == FBB) {
1018 DEBUG(dbgs() << "Won't split critical edge after degenerate BB#"
1019 << getNumber() << '\n');
1020 return false;
1021 }
1022 return true;
1023}
1024
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001025/// Prepare MI to be removed from its bundle. This fixes bundle flags on MI's
1026/// neighboring instructions so the bundle won't be broken by removing MI.
1027static void unbundleSingleMI(MachineInstr *MI) {
1028 // Removing the first instruction in a bundle.
1029 if (MI->isBundledWithSucc() && !MI->isBundledWithPred())
1030 MI->unbundleFromSucc();
1031 // Removing the last instruction in a bundle.
1032 if (MI->isBundledWithPred() && !MI->isBundledWithSucc())
1033 MI->unbundleFromPred();
1034 // If MI is not bundled, or if it is internal to a bundle, the neighbor flags
1035 // are already fine.
Evan Cheng7fae11b2011-12-14 02:11:42 +00001036}
1037
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001038MachineBasicBlock::instr_iterator
1039MachineBasicBlock::erase(MachineBasicBlock::instr_iterator I) {
Duncan P. N. Exon Smith0ac8eb92015-10-09 19:23:20 +00001040 unbundleSingleMI(&*I);
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001041 return Insts.erase(I);
1042}
Evan Cheng7fae11b2011-12-14 02:11:42 +00001043
Jakob Stoklund Olesenccfb5fb2012-12-17 23:55:38 +00001044MachineInstr *MachineBasicBlock::remove_instr(MachineInstr *MI) {
1045 unbundleSingleMI(MI);
1046 MI->clearFlag(MachineInstr::BundledPred);
1047 MI->clearFlag(MachineInstr::BundledSucc);
1048 return Insts.remove(MI);
Evan Cheng7fae11b2011-12-14 02:11:42 +00001049}
1050
Jakob Stoklund Olesen422e07b2012-12-18 17:54:53 +00001051MachineBasicBlock::instr_iterator
1052MachineBasicBlock::insert(instr_iterator I, MachineInstr *MI) {
1053 assert(!MI->isBundledWithPred() && !MI->isBundledWithSucc() &&
1054 "Cannot insert instruction with bundle flags");
1055 // Set the bundle flags when inserting inside a bundle.
1056 if (I != instr_end() && I->isBundledWithPred()) {
1057 MI->setFlag(MachineInstr::BundledPred);
1058 MI->setFlag(MachineInstr::BundledSucc);
1059 }
1060 return Insts.insert(I, MI);
1061}
1062
Cong Hou2a02c1c2015-08-12 21:18:54 +00001063/// This method unlinks 'this' from the containing function, and returns it, but
1064/// does not delete it.
Dan Gohman3b460302008-07-07 23:14:23 +00001065MachineBasicBlock *MachineBasicBlock::removeFromParent() {
1066 assert(getParent() && "Not embedded in a function!");
1067 getParent()->remove(this);
1068 return this;
1069}
1070
Cong Hou2a02c1c2015-08-12 21:18:54 +00001071/// This method unlinks 'this' from the containing function, and deletes it.
Dan Gohman3b460302008-07-07 23:14:23 +00001072void MachineBasicBlock::eraseFromParent() {
1073 assert(getParent() && "Not embedded in a function!");
1074 getParent()->erase(this);
1075}
1076
Cong Hou2a02c1c2015-08-12 21:18:54 +00001077/// Given a machine basic block that branched to 'Old', change the code and CFG
1078/// so that it branches to 'New' instead.
Evan Chengdf757852007-06-04 06:44:01 +00001079void MachineBasicBlock::ReplaceUsesOfBlockWith(MachineBasicBlock *Old,
1080 MachineBasicBlock *New) {
1081 assert(Old != New && "Cannot replace self with self!");
1082
Evan Cheng7fae11b2011-12-14 02:11:42 +00001083 MachineBasicBlock::instr_iterator I = instr_end();
1084 while (I != instr_begin()) {
Evan Chengdf757852007-06-04 06:44:01 +00001085 --I;
Evan Cheng7f8e5632011-12-07 07:15:52 +00001086 if (!I->isTerminator()) break;
Evan Chengdf757852007-06-04 06:44:01 +00001087
1088 // Scan the operands of this machine instruction, replacing any uses of Old
1089 // with New.
1090 for (unsigned i = 0, e = I->getNumOperands(); i != e; ++i)
Dan Gohman0d1e9a82008-10-03 15:45:36 +00001091 if (I->getOperand(i).isMBB() &&
Dan Gohman38453ee2008-09-13 17:58:21 +00001092 I->getOperand(i).getMBB() == Old)
Chris Lattnera5bb3702007-12-30 23:10:15 +00001093 I->getOperand(i).setMBB(New);
Evan Chengdf757852007-06-04 06:44:01 +00001094 }
1095
Dan Gohmanbb2f1072009-05-05 21:10:19 +00001096 // Update the successor information.
Jakub Staszak12a43bd2011-06-16 20:22:37 +00001097 replaceSuccessor(Old, New);
Evan Chengdf757852007-06-04 06:44:01 +00001098}
1099
Cong Hou2a02c1c2015-08-12 21:18:54 +00001100/// Various pieces of code can cause excess edges in the CFG to be inserted. If
1101/// we have proven that MBB can only branch to DestA and DestB, remove any other
1102/// MBB successors from the CFG. DestA and DestB can be null.
Jakub Staszakfeadd432011-06-16 18:01:17 +00001103///
Chris Lattner574e7162007-12-31 04:56:33 +00001104/// Besides DestA and DestB, retain other edges leading to LandingPads
1105/// (currently there can be only one; we don't check or require that here).
Evan Cheng2afd7022007-06-18 22:43:58 +00001106/// Note it is possible that DestA and/or DestB are LandingPads.
1107bool MachineBasicBlock::CorrectExtraCFGEdges(MachineBasicBlock *DestA,
1108 MachineBasicBlock *DestB,
Cong Hou166e0852015-09-29 19:46:09 +00001109 bool IsCond) {
Bill Wendling2d5967d2009-12-16 00:08:36 +00001110 // The values of DestA and DestB frequently come from a call to the
1111 // 'TargetInstrInfo::AnalyzeBranch' method. We take our meaning of the initial
1112 // values from there.
1113 //
1114 // 1. If both DestA and DestB are null, then the block ends with no branches
1115 // (it falls through to its successor).
Cong Hou166e0852015-09-29 19:46:09 +00001116 // 2. If DestA is set, DestB is null, and IsCond is false, then the block ends
Bill Wendling2d5967d2009-12-16 00:08:36 +00001117 // with only an unconditional branch.
Cong Hou166e0852015-09-29 19:46:09 +00001118 // 3. If DestA is set, DestB is null, and IsCond is true, then the block ends
Bill Wendling2d5967d2009-12-16 00:08:36 +00001119 // with a conditional branch that falls through to a successor (DestB).
Cong Hou166e0852015-09-29 19:46:09 +00001120 // 4. If DestA and DestB is set and IsCond is true, then the block ends with a
Bill Wendling2d5967d2009-12-16 00:08:36 +00001121 // conditional branch followed by an unconditional branch. DestA is the
1122 // 'true' destination and DestB is the 'false' destination.
1123
Bill Wendling95643402010-04-01 00:00:43 +00001124 bool Changed = false;
Evan Cheng2afd7022007-06-18 22:43:58 +00001125
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001126 MachineBasicBlock *FallThru = getNextNode();
Bill Wendling95643402010-04-01 00:00:43 +00001127
Craig Topperc0196b12014-04-14 00:51:57 +00001128 if (!DestA && !DestB) {
Bill Wendling95643402010-04-01 00:00:43 +00001129 // Block falls through to successor.
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001130 DestA = FallThru;
1131 DestB = FallThru;
Craig Topperc0196b12014-04-14 00:51:57 +00001132 } else if (DestA && !DestB) {
Cong Hou166e0852015-09-29 19:46:09 +00001133 if (IsCond)
Bill Wendling95643402010-04-01 00:00:43 +00001134 // Block ends in conditional jump that falls through to successor.
Duncan P. N. Exon Smith41cf73c2016-08-16 21:46:03 +00001135 DestB = FallThru;
Evan Cheng2afd7022007-06-18 22:43:58 +00001136 } else {
Cong Hou166e0852015-09-29 19:46:09 +00001137 assert(DestA && DestB && IsCond &&
Bill Wendling95643402010-04-01 00:00:43 +00001138 "CFG in a bad state. Cannot correct CFG edges");
Evan Cheng2afd7022007-06-18 22:43:58 +00001139 }
Bill Wendling95643402010-04-01 00:00:43 +00001140
1141 // Remove superfluous edges. I.e., those which aren't destinations of this
1142 // basic block, duplicate edges, or landing pads.
1143 SmallPtrSet<const MachineBasicBlock*, 8> SeenMBBs;
Evan Cheng2afd7022007-06-18 22:43:58 +00001144 MachineBasicBlock::succ_iterator SI = succ_begin();
Evan Cheng2afd7022007-06-18 22:43:58 +00001145 while (SI != succ_end()) {
Bill Wendling2d5967d2009-12-16 00:08:36 +00001146 const MachineBasicBlock *MBB = *SI;
David Blaikie70573dc2014-11-19 07:49:26 +00001147 if (!SeenMBBs.insert(MBB).second ||
Reid Kleckner0e288232015-08-27 23:27:47 +00001148 (MBB != DestA && MBB != DestB && !MBB->isEHPad())) {
Bill Wendling95643402010-04-01 00:00:43 +00001149 // This is a superfluous edge, remove it.
Bill Wendlingfa9810d2010-03-31 23:26:26 +00001150 SI = removeSuccessor(SI);
Bill Wendling95643402010-04-01 00:00:43 +00001151 Changed = true;
1152 } else {
1153 ++SI;
Evan Cheng2afd7022007-06-18 22:43:58 +00001154 }
1155 }
Bill Wendling2d5967d2009-12-16 00:08:36 +00001156
Cong Houc1069892015-12-13 09:26:17 +00001157 if (Changed)
1158 normalizeSuccProbs();
Bill Wendling95643402010-04-01 00:00:43 +00001159 return Changed;
Evan Cheng2afd7022007-06-18 22:43:58 +00001160}
Evan Cheng57be2f22009-11-17 19:19:59 +00001161
Cong Hou2a02c1c2015-08-12 21:18:54 +00001162/// Find the next valid DebugLoc starting at MBBI, skipping any DBG_VALUE
1163/// instructions. Return UnknownLoc if there is none.
Dale Johannesenc5db5992010-01-20 21:36:02 +00001164DebugLoc
Evan Cheng7fae11b2011-12-14 02:11:42 +00001165MachineBasicBlock::findDebugLoc(instr_iterator MBBI) {
Evan Cheng2a81dd42011-12-06 22:12:01 +00001166 // Skip debug declarations, we don't want a DebugLoc from them.
Florian Hahn3c8b8c92016-12-16 11:10:26 +00001167 MBBI = skipDebugInstructionsForward(MBBI, instr_end());
1168 if (MBBI != instr_end())
1169 return MBBI->getDebugLoc();
1170 return {};
Dale Johannesenc5db5992010-01-20 21:36:02 +00001171}
1172
Taewook Oh06a21282017-02-13 18:15:31 +00001173/// Find and return the merged DebugLoc of the branch instructions of the block.
1174/// Return UnknownLoc if there is none.
1175DebugLoc
1176MachineBasicBlock::findBranchDebugLoc() {
Taewook Oh4d35f9e2017-02-13 21:12:27 +00001177 DebugLoc DL;
Taewook Oh06a21282017-02-13 18:15:31 +00001178 auto TI = getFirstTerminator();
1179 while (TI != end() && !TI->isBranch())
1180 ++TI;
1181
1182 if (TI != end()) {
1183 DL = TI->getDebugLoc();
1184 for (++TI ; TI != end() ; ++TI)
1185 if (TI->isBranch())
1186 DL = DILocation::getMergedLocation(DL, TI->getDebugLoc());
1187 }
1188 return DL;
1189}
1190
Cong Houd97c1002015-12-01 05:29:22 +00001191/// Return probability of the edge from this block to MBB.
Cong Hou23a3bf02015-11-04 21:37:58 +00001192BranchProbability
1193MachineBasicBlock::getSuccProbability(const_succ_iterator Succ) const {
Cong Hou4aef7ef2015-12-01 11:05:39 +00001194 if (Probs.empty())
Cong Hou23a3bf02015-11-04 21:37:58 +00001195 return BranchProbability(1, succ_size());
1196
Cong Hou4aef7ef2015-12-01 11:05:39 +00001197 const auto &Prob = *getProbabilityIterator(Succ);
1198 if (Prob.isUnknown()) {
1199 // For unknown probabilities, collect the sum of all known ones, and evenly
1200 // ditribute the complemental of the sum to each unknown probability.
1201 unsigned KnownProbNum = 0;
1202 auto Sum = BranchProbability::getZero();
1203 for (auto &P : Probs) {
1204 if (!P.isUnknown()) {
1205 Sum += P;
1206 KnownProbNum++;
1207 }
1208 }
1209 return Sum.getCompl() / (Probs.size() - KnownProbNum);
1210 } else
1211 return Prob;
Manman Renb6819182014-01-29 23:18:47 +00001212}
1213
Cong Hou23a3bf02015-11-04 21:37:58 +00001214/// Set successor probability of a given iterator.
1215void MachineBasicBlock::setSuccProbability(succ_iterator I,
1216 BranchProbability Prob) {
1217 assert(!Prob.isUnknown());
Cong Houd97c1002015-12-01 05:29:22 +00001218 if (Probs.empty())
Cong Hou23a3bf02015-11-04 21:37:58 +00001219 return;
1220 *getProbabilityIterator(I) = Prob;
Cong Hou23a3bf02015-11-04 21:37:58 +00001221}
1222
Hans Wennborg1dbaf672015-12-01 03:49:42 +00001223/// Return probability iterator corresonding to the I successor iterator
1224MachineBasicBlock::const_probability_iterator
1225MachineBasicBlock::getProbabilityIterator(
1226 MachineBasicBlock::const_succ_iterator I) const {
Cong Houfa1917c2015-12-01 00:02:51 +00001227 assert(Probs.size() == Successors.size() && "Async probability list!");
1228 const size_t index = std::distance(Successors.begin(), I);
1229 assert(index < Probs.size() && "Not a current successor!");
1230 return Probs.begin() + index;
1231}
1232
Cong Houd97c1002015-12-01 05:29:22 +00001233/// Return probability iterator corresonding to the I successor iterator.
1234MachineBasicBlock::probability_iterator
1235MachineBasicBlock::getProbabilityIterator(MachineBasicBlock::succ_iterator I) {
1236 assert(Probs.size() == Successors.size() && "Async probability list!");
1237 const size_t index = std::distance(Successors.begin(), I);
1238 assert(index < Probs.size() && "Not a current successor!");
1239 return Probs.begin() + index;
1240}
1241
James Molloyc747cda2012-09-12 10:18:23 +00001242/// Return whether (physical) register "Reg" has been <def>ined and not <kill>ed
1243/// as of just before "MI".
Junmo Park12386102016-01-07 10:26:32 +00001244///
James Molloyc747cda2012-09-12 10:18:23 +00001245/// Search is localised to a neighborhood of
1246/// Neighborhood instructions before (searching for defs or kills) and N
1247/// instructions after (searching just for defs) MI.
1248MachineBasicBlock::LivenessQueryResult
1249MachineBasicBlock::computeRegisterLiveness(const TargetRegisterInfo *TRI,
Matthias Braun07a07ba2015-05-27 05:12:39 +00001250 unsigned Reg, const_iterator Before,
1251 unsigned Neighborhood) const {
James Molloyc747cda2012-09-12 10:18:23 +00001252 unsigned N = Neighborhood;
James Molloyc747cda2012-09-12 10:18:23 +00001253
Matthias Braun07a07ba2015-05-27 05:12:39 +00001254 // Start by searching backwards from Before, looking for kills, reads or defs.
1255 const_iterator I(Before);
James Molloyc747cda2012-09-12 10:18:23 +00001256 // If this is the first insn in the block, don't search backwards.
Matthias Braun07a07ba2015-05-27 05:12:39 +00001257 if (I != begin()) {
James Molloyc747cda2012-09-12 10:18:23 +00001258 do {
1259 --I;
1260
Matthias Braun60d69e22015-12-11 19:42:09 +00001261 MachineOperandIteratorBase::PhysRegInfo Info =
Duncan P. N. Exon Smithf9ab4162016-02-27 17:05:33 +00001262 ConstMIOperands(*I).analyzePhysReg(Reg, TRI);
James Molloyc747cda2012-09-12 10:18:23 +00001263
Matthias Braun60d69e22015-12-11 19:42:09 +00001264 // Defs happen after uses so they take precedence if both are present.
Tim Northoverdd219d02012-11-20 09:56:11 +00001265
Matthias Braun60d69e22015-12-11 19:42:09 +00001266 // Register is dead after a dead def of the full register.
1267 if (Info.DeadDef)
James Molloyc747cda2012-09-12 10:18:23 +00001268 return LQR_Dead;
Matthias Braun60d69e22015-12-11 19:42:09 +00001269 // Register is (at least partially) live after a def.
Quentin Colombet08e79992016-04-26 23:14:29 +00001270 if (Info.Defined) {
1271 if (!Info.PartialDeadDef)
1272 return LQR_Live;
1273 // As soon as we saw a partial definition (dead or not),
1274 // we cannot tell if the value is partial live without
1275 // tracking the lanemasks. We are not going to do this,
1276 // so fall back on the remaining of the analysis.
1277 break;
1278 }
Matthias Braun60d69e22015-12-11 19:42:09 +00001279 // Register is dead after a full kill or clobber and no def.
1280 if (Info.Killed || Info.Clobbered)
1281 return LQR_Dead;
1282 // Register must be live if we read it.
1283 if (Info.Read)
1284 return LQR_Live;
Matthias Braun07a07ba2015-05-27 05:12:39 +00001285 } while (I != begin() && --N > 0);
James Molloyc747cda2012-09-12 10:18:23 +00001286 }
1287
1288 // Did we get to the start of the block?
Matthias Braun07a07ba2015-05-27 05:12:39 +00001289 if (I == begin()) {
James Molloyc747cda2012-09-12 10:18:23 +00001290 // If so, the register's state is definitely defined by the live-in state.
Matthias Braun60d69e22015-12-11 19:42:09 +00001291 for (MCRegAliasIterator RAI(Reg, TRI, /*IncludeSelf=*/true); RAI.isValid();
1292 ++RAI)
Matthias Braun07a07ba2015-05-27 05:12:39 +00001293 if (isLiveIn(*RAI))
Matthias Braun60d69e22015-12-11 19:42:09 +00001294 return LQR_Live;
James Molloyc747cda2012-09-12 10:18:23 +00001295
1296 return LQR_Dead;
1297 }
1298
1299 N = Neighborhood;
1300
Matthias Braun07a07ba2015-05-27 05:12:39 +00001301 // Try searching forwards from Before, looking for reads or defs.
1302 I = const_iterator(Before);
James Molloyc747cda2012-09-12 10:18:23 +00001303 // If this is the last insn in the block, don't search forwards.
Matthias Braun07a07ba2015-05-27 05:12:39 +00001304 if (I != end()) {
1305 for (++I; I != end() && N > 0; ++I, --N) {
Matthias Braun60d69e22015-12-11 19:42:09 +00001306 MachineOperandIteratorBase::PhysRegInfo Info =
Duncan P. N. Exon Smithf9ab4162016-02-27 17:05:33 +00001307 ConstMIOperands(*I).analyzePhysReg(Reg, TRI);
James Molloyc747cda2012-09-12 10:18:23 +00001308
Matthias Braun60d69e22015-12-11 19:42:09 +00001309 // Register is live when we read it here.
1310 if (Info.Read)
1311 return LQR_Live;
1312 // Register is dead if we can fully overwrite or clobber it here.
1313 if (Info.FullyDefined || Info.Clobbered)
James Molloyc747cda2012-09-12 10:18:23 +00001314 return LQR_Dead;
1315 }
1316 }
1317
1318 // At this point we have no idea of the liveness of the register.
1319 return LQR_Unknown;
1320}
Reid Klecknerb8fd1622015-11-06 17:06:38 +00001321
1322const uint32_t *
1323MachineBasicBlock::getBeginClobberMask(const TargetRegisterInfo *TRI) const {
1324 // EH funclet entry does not preserve any registers.
1325 return isEHFuncletEntry() ? TRI->getNoPreservedMask() : nullptr;
1326}
1327
1328const uint32_t *
1329MachineBasicBlock::getEndClobberMask(const TargetRegisterInfo *TRI) const {
1330 // If we see a return block with successors, this must be a funclet return,
1331 // which does not preserve any registers. If there are no successors, we don't
1332 // care what kind of return it is, putting a mask after it is a no-op.
1333 return isReturnBlock() && !succ_empty() ? TRI->getNoPreservedMask() : nullptr;
1334}
Matthias Braun18198302016-12-16 23:55:37 +00001335
1336void MachineBasicBlock::clearLiveIns() {
1337 LiveIns.clear();
1338}
Matthias Braun11723322017-01-05 20:01:19 +00001339
1340MachineBasicBlock::livein_iterator MachineBasicBlock::livein_begin() const {
1341 assert(getParent()->getProperties().hasProperty(
1342 MachineFunctionProperties::Property::TracksLiveness) &&
1343 "Liveness information is accurate");
1344 return LiveIns.begin();
1345}