blob: 3846284ec10d6f5b57f25e15d9a3f501eab3e697 [file] [log] [blame]
Evan Cheng4e654852007-05-16 02:00:57 +00001//===-- IfConversion.cpp - Machine code if conversion pass. ---------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by the Evan Cheng and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the machine instruction level if-conversion pass.
11//
12//===----------------------------------------------------------------------===//
13
Evan Cheng5f702182007-06-01 00:12:12 +000014#define DEBUG_TYPE "ifcvt"
15#include "llvm/Function.h"
Evan Cheng4e654852007-05-16 02:00:57 +000016#include "llvm/CodeGen/Passes.h"
17#include "llvm/CodeGen/MachineModuleInfo.h"
18#include "llvm/CodeGen/MachineFunctionPass.h"
19#include "llvm/Target/TargetInstrInfo.h"
Evan Cheng86cbfea2007-05-18 00:20:58 +000020#include "llvm/Target/TargetLowering.h"
Evan Cheng4e654852007-05-16 02:00:57 +000021#include "llvm/Target/TargetMachine.h"
Evan Chengedf48962007-06-08 19:10:51 +000022#include "llvm/Support/CommandLine.h"
Evan Cheng4e654852007-05-16 02:00:57 +000023#include "llvm/Support/Debug.h"
Evan Cheng36489bb2007-05-18 19:26:33 +000024#include "llvm/ADT/DepthFirstIterator.h"
Evan Cheng4e654852007-05-16 02:00:57 +000025#include "llvm/ADT/Statistic.h"
26using namespace llvm;
27
Evan Chengedf48962007-06-08 19:10:51 +000028namespace {
29 // Hidden options for help debugging.
30 cl::opt<int> IfCvtFnStart("ifcvt-fn-start", cl::init(-1), cl::Hidden);
31 cl::opt<int> IfCvtFnStop("ifcvt-fn-stop", cl::init(-1), cl::Hidden);
32 cl::opt<int> IfCvtLimit("ifcvt-limit", cl::init(-1), cl::Hidden);
33 cl::opt<bool> DisableSimple("disable-ifcvt-simple",
34 cl::init(false), cl::Hidden);
Evan Cheng1c9f91d2007-06-09 01:03:43 +000035 cl::opt<bool> DisableSimpleF("disable-ifcvt-simple-false",
36 cl::init(false), cl::Hidden);
Evan Chengedf48962007-06-08 19:10:51 +000037 cl::opt<bool> DisableTriangle("disable-ifcvt-triangle",
38 cl::init(false), cl::Hidden);
Evan Chengcc8fb462007-06-12 23:54:05 +000039 cl::opt<bool> DisableTriangleR("disable-ifcvt-triangle-rev",
40 cl::init(false), cl::Hidden);
Evan Cheng1c9f91d2007-06-09 01:03:43 +000041 cl::opt<bool> DisableTriangleF("disable-ifcvt-triangle-false",
42 cl::init(false), cl::Hidden);
43 cl::opt<bool> DisableTriangleFR("disable-ifcvt-triangle-false-rev",
44 cl::init(false), cl::Hidden);
Evan Chengedf48962007-06-08 19:10:51 +000045 cl::opt<bool> DisableDiamond("disable-ifcvt-diamond",
46 cl::init(false), cl::Hidden);
47}
48
Evan Cheng1c9f91d2007-06-09 01:03:43 +000049STATISTIC(NumSimple, "Number of simple if-conversions performed");
50STATISTIC(NumSimpleFalse, "Number of simple (F) if-conversions performed");
51STATISTIC(NumTriangle, "Number of triangle if-conversions performed");
Evan Chengcc8fb462007-06-12 23:54:05 +000052STATISTIC(NumTriangleRev, "Number of triangle (R) if-conversions performed");
Evan Cheng1c9f91d2007-06-09 01:03:43 +000053STATISTIC(NumTriangleFalse,"Number of triangle (F) if-conversions performed");
54STATISTIC(NumTriangleFRev, "Number of triangle (F/R) if-conversions performed");
55STATISTIC(NumDiamonds, "Number of diamond if-conversions performed");
56STATISTIC(NumIfConvBBs, "Number of if-converted blocks");
Evan Cheng2c8c3a42007-06-15 07:36:12 +000057STATISTIC(NumDupBBs, "Number of duplicated blocks");
Evan Cheng4e654852007-05-16 02:00:57 +000058
59namespace {
60 class IfConverter : public MachineFunctionPass {
Evan Chenge882fca2007-06-16 09:34:52 +000061 enum IfcvtKind {
Evan Cheng4e654852007-05-16 02:00:57 +000062 ICNotClassfied, // BB data valid, but not classified.
Evan Chengb6665f62007-06-04 06:47:22 +000063 ICSimpleFalse, // Same as ICSimple, but on the false path.
Evan Chenge882fca2007-06-16 09:34:52 +000064 ICSimple, // BB is entry of an one split, no rejoin sub-CFG.
65 ICTriangleFRev, // Same as ICTriangleFalse, but false path rev condition.
Evan Chengcc8fb462007-06-12 23:54:05 +000066 ICTriangleRev, // Same as ICTriangle, but true path rev condition.
Evan Cheng1c9f91d2007-06-09 01:03:43 +000067 ICTriangleFalse, // Same as ICTriangle, but on the false path.
Evan Chenge882fca2007-06-16 09:34:52 +000068 ICTriangle, // BB is entry of a triangle sub-CFG.
Evan Cheng9618bca2007-06-11 22:26:22 +000069 ICDiamond // BB is entry of a diamond sub-CFG.
Evan Cheng4e654852007-05-16 02:00:57 +000070 };
71
72 /// BBInfo - One per MachineBasicBlock, this is used to cache the result
73 /// if-conversion feasibility analysis. This includes results from
74 /// TargetInstrInfo::AnalyzeBranch() (i.e. TBB, FBB, and Cond), and its
Evan Cheng86cbfea2007-05-18 00:20:58 +000075 /// classification, and common tail block of its successors (if it's a
Evan Chengcf6cc112007-05-18 18:14:37 +000076 /// diamond shape), its size, whether it's predicable, and whether any
77 /// instruction can clobber the 'would-be' predicate.
Evan Chenga13aa952007-05-23 07:23:16 +000078 ///
Evan Cheng9618bca2007-06-11 22:26:22 +000079 /// IsDone - True if BB is not to be considered for ifcvt.
80 /// IsBeingAnalyzed - True if BB is currently being analyzed.
81 /// IsAnalyzed - True if BB has been analyzed (info is still valid).
82 /// IsEnqueued - True if BB has been enqueued to be ifcvt'ed.
83 /// IsBrAnalyzable - True if AnalyzeBranch() returns false.
84 /// HasFallThrough - True if BB may fallthrough to the following BB.
85 /// IsUnpredicable - True if BB is known to be unpredicable.
86 /// ClobbersPredicate- True if BB would modify the predicate (e.g. has
Evan Cheng7a655472007-06-06 10:16:17 +000087 /// cmp, call, etc.)
Evan Cheng9618bca2007-06-11 22:26:22 +000088 /// NonPredSize - Number of non-predicated instructions.
Evan Chenga13aa952007-05-23 07:23:16 +000089 /// BB - Corresponding MachineBasicBlock.
90 /// TrueBB / FalseBB- See AnalyzeBranch().
91 /// BrCond - Conditions for end of block conditional branches.
92 /// Predicate - Predicate used in the BB.
Evan Cheng4e654852007-05-16 02:00:57 +000093 struct BBInfo {
Evan Cheng9618bca2007-06-11 22:26:22 +000094 bool IsDone : 1;
95 bool IsBeingAnalyzed : 1;
96 bool IsAnalyzed : 1;
97 bool IsEnqueued : 1;
98 bool IsBrAnalyzable : 1;
99 bool HasFallThrough : 1;
100 bool IsUnpredicable : 1;
Evan Chenga2acf842007-06-15 21:18:05 +0000101 bool CannotBeCopied : 1;
Evan Cheng9618bca2007-06-11 22:26:22 +0000102 bool ClobbersPred : 1;
Evan Chenga13aa952007-05-23 07:23:16 +0000103 unsigned NonPredSize;
Evan Cheng86cbfea2007-05-18 00:20:58 +0000104 MachineBasicBlock *BB;
105 MachineBasicBlock *TrueBB;
106 MachineBasicBlock *FalseBB;
Evan Chenga13aa952007-05-23 07:23:16 +0000107 std::vector<MachineOperand> BrCond;
108 std::vector<MachineOperand> Predicate;
Evan Chenge882fca2007-06-16 09:34:52 +0000109 BBInfo() : IsDone(false), IsBeingAnalyzed(false),
Evan Cheng9618bca2007-06-11 22:26:22 +0000110 IsAnalyzed(false), IsEnqueued(false), IsBrAnalyzable(false),
111 HasFallThrough(false), IsUnpredicable(false),
Evan Chenga2acf842007-06-15 21:18:05 +0000112 CannotBeCopied(false), ClobbersPred(false), NonPredSize(0),
Evan Chenge882fca2007-06-16 09:34:52 +0000113 BB(0), TrueBB(0), FalseBB(0) {}
114 };
115
116 /// IfcvtToken - Record information about pending if-conversions to attemp:
117 /// BBI - Corresponding BBInfo.
118 /// Kind - Type of block. See IfcvtKind.
119 /// NeedSubsumsion - True if the to be predicated BB has already been
120 /// predicated.
121 /// Duplicates - Number of instructions that would be duplicated due
122 /// to this if-conversion.
123 struct IfcvtToken {
124 BBInfo &BBI;
125 IfcvtKind Kind;
126 bool NeedSubsumsion;
127 unsigned Duplicates;
128 IfcvtToken(BBInfo &b, IfcvtKind k, bool s, unsigned d)
129 : BBI(b), Kind(k), NeedSubsumsion(s), Duplicates(d) {}
Evan Cheng4e654852007-05-16 02:00:57 +0000130 };
131
Evan Cheng82582102007-05-30 19:49:19 +0000132 /// Roots - Basic blocks that do not have successors. These are the starting
133 /// points of Graph traversal.
134 std::vector<MachineBasicBlock*> Roots;
135
Evan Cheng4e654852007-05-16 02:00:57 +0000136 /// BBAnalysis - Results of if-conversion feasibility analysis indexed by
137 /// basic block number.
138 std::vector<BBInfo> BBAnalysis;
139
Evan Cheng86cbfea2007-05-18 00:20:58 +0000140 const TargetLowering *TLI;
Evan Cheng4e654852007-05-16 02:00:57 +0000141 const TargetInstrInfo *TII;
142 bool MadeChange;
143 public:
144 static char ID;
145 IfConverter() : MachineFunctionPass((intptr_t)&ID) {}
146
147 virtual bool runOnMachineFunction(MachineFunction &MF);
148 virtual const char *getPassName() const { return "If converter"; }
149
150 private:
Evan Chengb6665f62007-06-04 06:47:22 +0000151 bool ReverseBranchCondition(BBInfo &BBI);
Evan Chenge882fca2007-06-16 09:34:52 +0000152 bool ValidSimple(BBInfo &TrueBBI, unsigned &Dups) const;
Evan Chengac5f1422007-06-08 09:36:04 +0000153 bool ValidTriangle(BBInfo &TrueBBI, BBInfo &FalseBBI,
Evan Chenge882fca2007-06-16 09:34:52 +0000154 bool FalseBranch, unsigned &Dups) const;
Evan Chengd4de6d92007-06-07 02:12:15 +0000155 bool ValidDiamond(BBInfo &TrueBBI, BBInfo &FalseBBI) const;
Evan Chengac5f1422007-06-08 09:36:04 +0000156 void ScanInstructions(BBInfo &BBI);
Evan Chenge882fca2007-06-16 09:34:52 +0000157 BBInfo &AnalyzeBlock(MachineBasicBlock *BB,
158 std::vector<IfcvtToken*> &Tokens);
Evan Chengac5f1422007-06-08 09:36:04 +0000159 bool FeasibilityAnalysis(BBInfo &BBI, std::vector<MachineOperand> &Cond,
160 bool isTriangle = false, bool RevBranch = false);
Evan Cheng82582102007-05-30 19:49:19 +0000161 bool AnalyzeBlocks(MachineFunction &MF,
Evan Chenge882fca2007-06-16 09:34:52 +0000162 std::vector<IfcvtToken*> &Tokens);
Evan Cheng82582102007-05-30 19:49:19 +0000163 void ReTryPreds(MachineBasicBlock *BB);
Evan Cheng7e75ba82007-06-08 22:01:07 +0000164 void RemoveExtraEdges(BBInfo &BBI);
Evan Chenge882fca2007-06-16 09:34:52 +0000165 bool IfConvertSimple(BBInfo &BBI, IfcvtKind Kind);
166 bool IfConvertTriangle(BBInfo &BBI, IfcvtKind Kind);
167 bool IfConvertDiamond(BBInfo &BBI, IfcvtKind Kind);
Evan Chenga13aa952007-05-23 07:23:16 +0000168 void PredicateBlock(BBInfo &BBI,
Evan Cheng4e654852007-05-16 02:00:57 +0000169 std::vector<MachineOperand> &Cond,
170 bool IgnoreTerm = false);
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000171 void CopyAndPredicateBlock(BBInfo &ToBBI, BBInfo &FromBBI,
172 std::vector<MachineOperand> &Cond,
173 bool IgnoreBr = false);
174 void MergeBlocks(BBInfo &ToBBI, BBInfo &FromBBI);
Evan Cheng5f702182007-06-01 00:12:12 +0000175
Evan Chengd4de6d92007-06-07 02:12:15 +0000176 // blockAlwaysFallThrough - Block ends without a terminator.
177 bool blockAlwaysFallThrough(BBInfo &BBI) const {
Evan Cheng9618bca2007-06-11 22:26:22 +0000178 return BBI.IsBrAnalyzable && BBI.TrueBB == NULL;
Evan Cheng7a655472007-06-06 10:16:17 +0000179 }
180
Evan Chenge882fca2007-06-16 09:34:52 +0000181 // IfcvtTokenCmp - Used to sort if-conversion candidates.
182 static bool IfcvtTokenCmp(IfcvtToken *C1, IfcvtToken *C2) {
183 // Favors subsumsion.
184 if (C1->NeedSubsumsion == false && C2->NeedSubsumsion == true)
185 return true;
186 else if (C1->NeedSubsumsion == C2->NeedSubsumsion) {
187 if (C1->Duplicates > C2->Duplicates)
188 return true;
189 else if (C1->Duplicates == C2->Duplicates) {
190 // Favors diamond over triangle, etc.
191 if ((unsigned)C1->Kind < (unsigned)C2->Kind)
192 return true;
193 else if (C1->Kind == C2->Kind)
194 return C1->BBI.BB->getNumber() < C2->BBI.BB->getNumber();
195 }
196 }
197 return false;
Evan Cheng5f702182007-06-01 00:12:12 +0000198 }
Evan Cheng4e654852007-05-16 02:00:57 +0000199 };
Evan Chenge882fca2007-06-16 09:34:52 +0000200
Evan Cheng4e654852007-05-16 02:00:57 +0000201 char IfConverter::ID = 0;
202}
203
204FunctionPass *llvm::createIfConverterPass() { return new IfConverter(); }
205
206bool IfConverter::runOnMachineFunction(MachineFunction &MF) {
Evan Cheng86cbfea2007-05-18 00:20:58 +0000207 TLI = MF.getTarget().getTargetLowering();
Evan Cheng4e654852007-05-16 02:00:57 +0000208 TII = MF.getTarget().getInstrInfo();
209 if (!TII) return false;
210
Evan Chengedf48962007-06-08 19:10:51 +0000211 static int FnNum = -1;
212 DOUT << "\nIfcvt: function (" << ++FnNum << ") \'"
213 << MF.getFunction()->getName() << "\'";
214
215 if (FnNum < IfCvtFnStart || (IfCvtFnStop != -1 && FnNum > IfCvtFnStop)) {
216 DOUT << " skipped\n";
217 return false;
218 }
219 DOUT << "\n";
Evan Cheng5f702182007-06-01 00:12:12 +0000220
Evan Cheng4e654852007-05-16 02:00:57 +0000221 MF.RenumberBlocks();
Evan Cheng5f702182007-06-01 00:12:12 +0000222 BBAnalysis.resize(MF.getNumBlockIDs());
Evan Cheng4e654852007-05-16 02:00:57 +0000223
Evan Cheng82582102007-05-30 19:49:19 +0000224 // Look for root nodes, i.e. blocks without successors.
225 for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I)
226 if (I->succ_size() == 0)
227 Roots.push_back(I);
228
Evan Chenge882fca2007-06-16 09:34:52 +0000229 std::vector<IfcvtToken*> Tokens;
Evan Cheng47d25022007-05-18 01:55:58 +0000230 MadeChange = false;
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000231 unsigned NumIfCvts = NumSimple + NumSimpleFalse + NumTriangle +
232 NumTriangleRev + NumTriangleFalse + NumTriangleFRev + NumDiamonds;
233 while (IfCvtLimit == -1 || (int)NumIfCvts < IfCvtLimit) {
Evan Chenga13aa952007-05-23 07:23:16 +0000234 // Do an intial analysis for each basic block and finding all the potential
235 // candidates to perform if-convesion.
Evan Chenge882fca2007-06-16 09:34:52 +0000236 bool Change = AnalyzeBlocks(MF, Tokens);
237 while (!Tokens.empty()) {
238 IfcvtToken *Token = Tokens.back();
239 Tokens.pop_back();
240 BBInfo &BBI = Token->BBI;
241 IfcvtKind Kind = Token->Kind;
Evan Chengb6665f62007-06-04 06:47:22 +0000242
Evan Cheng9618bca2007-06-11 22:26:22 +0000243 // If the block has been evicted out of the queue or it has already been
244 // marked dead (due to it being predicated), then skip it.
Evan Chenge882fca2007-06-16 09:34:52 +0000245 if (BBI.IsDone)
246 BBI.IsEnqueued = false;
247 if (!BBI.IsEnqueued)
Evan Cheng9618bca2007-06-11 22:26:22 +0000248 continue;
Evan Chenge882fca2007-06-16 09:34:52 +0000249
Evan Cheng86ff2962007-06-14 20:28:52 +0000250 BBI.IsEnqueued = false;
Evan Cheng9618bca2007-06-11 22:26:22 +0000251
Evan Chengb6665f62007-06-04 06:47:22 +0000252 bool RetVal = false;
Evan Chenge882fca2007-06-16 09:34:52 +0000253 switch (Kind) {
Evan Chenga13aa952007-05-23 07:23:16 +0000254 default: assert(false && "Unexpected!");
255 break;
Evan Chengb6665f62007-06-04 06:47:22 +0000256 case ICSimple:
Evan Chengcb78d672007-06-06 01:12:44 +0000257 case ICSimpleFalse: {
Evan Chenge882fca2007-06-16 09:34:52 +0000258 bool isFalse = Kind == ICSimpleFalse;
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000259 if ((isFalse && DisableSimpleF) || (!isFalse && DisableSimple)) break;
Evan Chenge882fca2007-06-16 09:34:52 +0000260 DOUT << "Ifcvt (Simple" << (Kind == ICSimpleFalse ? " false" :"")
Evan Cheng7a655472007-06-06 10:16:17 +0000261 << "): BB#" << BBI.BB->getNumber() << " ("
Evan Chenge882fca2007-06-16 09:34:52 +0000262 << ((Kind == ICSimpleFalse)
263 ? BBI.FalseBB->getNumber()
264 : BBI.TrueBB->getNumber()) << ") ";
265 RetVal = IfConvertSimple(BBI, Kind);
Evan Chengb6665f62007-06-04 06:47:22 +0000266 DOUT << (RetVal ? "succeeded!" : "failed!") << "\n";
267 if (RetVal)
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000268 if (isFalse) NumSimpleFalse++;
269 else NumSimple++;
Evan Chengb6665f62007-06-04 06:47:22 +0000270 break;
Evan Chengcb78d672007-06-06 01:12:44 +0000271 }
Evan Chenga13aa952007-05-23 07:23:16 +0000272 case ICTriangle:
Evan Chengcc8fb462007-06-12 23:54:05 +0000273 case ICTriangleRev:
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000274 case ICTriangleFalse:
Reid Spencera9bf49c2007-06-10 00:19:17 +0000275 case ICTriangleFRev: {
Evan Chenge882fca2007-06-16 09:34:52 +0000276 bool isFalse = Kind == ICTriangleFalse;
277 bool isRev = (Kind == ICTriangleRev || Kind == ICTriangleFRev);
Evan Chengcc8fb462007-06-12 23:54:05 +0000278 if (DisableTriangle && !isFalse && !isRev) break;
279 if (DisableTriangleR && !isFalse && isRev) break;
280 if (DisableTriangleF && isFalse && !isRev) break;
281 if (DisableTriangleFR && isFalse && isRev) break;
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000282 DOUT << "Ifcvt (Triangle";
283 if (isFalse)
284 DOUT << " false";
Evan Chengcc8fb462007-06-12 23:54:05 +0000285 if (isRev)
286 DOUT << " rev";
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000287 DOUT << "): BB#" << BBI.BB->getNumber() << " (T:"
Evan Chenge882fca2007-06-16 09:34:52 +0000288 << BBI.TrueBB->getNumber() << ",F:"
289 << BBI.FalseBB->getNumber() << ") ";
290 RetVal = IfConvertTriangle(BBI, Kind);
Evan Chengb6665f62007-06-04 06:47:22 +0000291 DOUT << (RetVal ? "succeeded!" : "failed!") << "\n";
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000292 if (RetVal) {
Evan Chengcc8fb462007-06-12 23:54:05 +0000293 if (isFalse) {
294 if (isRev) NumTriangleFRev++;
295 else NumTriangleFalse++;
296 } else {
297 if (isRev) NumTriangleRev++;
298 else NumTriangle++;
299 }
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000300 }
Evan Chenga13aa952007-05-23 07:23:16 +0000301 break;
Reid Spencera9bf49c2007-06-10 00:19:17 +0000302 }
Evan Chenge882fca2007-06-16 09:34:52 +0000303 case ICDiamond: {
Evan Chengedf48962007-06-08 19:10:51 +0000304 if (DisableDiamond) break;
Evan Cheng7a655472007-06-06 10:16:17 +0000305 DOUT << "Ifcvt (Diamond): BB#" << BBI.BB->getNumber() << " (T:"
Evan Chenge882fca2007-06-16 09:34:52 +0000306 << BBI.TrueBB->getNumber() << ",F:"
307 << BBI.FalseBB->getNumber() << ") ";
308 RetVal = IfConvertDiamond(BBI, Kind);
Evan Chengb6665f62007-06-04 06:47:22 +0000309 DOUT << (RetVal ? "succeeded!" : "failed!") << "\n";
310 if (RetVal) NumDiamonds++;
Evan Chenga13aa952007-05-23 07:23:16 +0000311 break;
312 }
Evan Chenge882fca2007-06-16 09:34:52 +0000313 }
314
Evan Chengb6665f62007-06-04 06:47:22 +0000315 Change |= RetVal;
Evan Chengedf48962007-06-08 19:10:51 +0000316
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000317 NumIfCvts = NumSimple + NumSimpleFalse + NumTriangle + NumTriangleRev +
318 NumTriangleFalse + NumTriangleFRev + NumDiamonds;
319 if (IfCvtLimit != -1 && (int)NumIfCvts >= IfCvtLimit)
Evan Chengedf48962007-06-08 19:10:51 +0000320 break;
Evan Cheng4e654852007-05-16 02:00:57 +0000321 }
Evan Chenga13aa952007-05-23 07:23:16 +0000322
Evan Chenga13aa952007-05-23 07:23:16 +0000323 if (!Change)
324 break;
Evan Chengb6665f62007-06-04 06:47:22 +0000325 MadeChange |= Change;
Evan Cheng4e654852007-05-16 02:00:57 +0000326 }
Evan Cheng47d25022007-05-18 01:55:58 +0000327
Evan Chenge882fca2007-06-16 09:34:52 +0000328 // Delete tokens in case of early exit.
329 while (!Tokens.empty()) {
330 IfcvtToken *Token = Tokens.back();
331 Tokens.pop_back();
332 delete Token;
333 }
334
335 Tokens.clear();
Evan Cheng82582102007-05-30 19:49:19 +0000336 Roots.clear();
Evan Cheng47d25022007-05-18 01:55:58 +0000337 BBAnalysis.clear();
338
Evan Cheng4e654852007-05-16 02:00:57 +0000339 return MadeChange;
340}
341
Evan Chenge882fca2007-06-16 09:34:52 +0000342/// findFalseBlock - BB has a fallthrough. Find its 'false' successor given
343/// its 'true' successor.
Evan Cheng4e654852007-05-16 02:00:57 +0000344static MachineBasicBlock *findFalseBlock(MachineBasicBlock *BB,
Evan Cheng86cbfea2007-05-18 00:20:58 +0000345 MachineBasicBlock *TrueBB) {
Evan Cheng4e654852007-05-16 02:00:57 +0000346 for (MachineBasicBlock::succ_iterator SI = BB->succ_begin(),
347 E = BB->succ_end(); SI != E; ++SI) {
348 MachineBasicBlock *SuccBB = *SI;
Evan Cheng86cbfea2007-05-18 00:20:58 +0000349 if (SuccBB != TrueBB)
Evan Cheng4e654852007-05-16 02:00:57 +0000350 return SuccBB;
351 }
352 return NULL;
353}
354
Evan Chenge882fca2007-06-16 09:34:52 +0000355/// ReverseBranchCondition - Reverse the condition of the end of the block
356/// branchs. Swap block's 'true' and 'false' successors.
Evan Chengb6665f62007-06-04 06:47:22 +0000357bool IfConverter::ReverseBranchCondition(BBInfo &BBI) {
358 if (!TII->ReverseBranchCondition(BBI.BrCond)) {
359 TII->RemoveBranch(*BBI.BB);
360 TII->InsertBranch(*BBI.BB, BBI.FalseBB, BBI.TrueBB, BBI.BrCond);
361 std::swap(BBI.TrueBB, BBI.FalseBB);
362 return true;
363 }
364 return false;
365}
366
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000367/// getNextBlock - Returns the next block in the function blocks ordering. If
368/// it is the end, returns NULL.
369static inline MachineBasicBlock *getNextBlock(MachineBasicBlock *BB) {
370 MachineFunction::iterator I = BB;
371 MachineFunction::iterator E = BB->getParent()->end();
372 if (++I == E)
373 return NULL;
374 return I;
375}
376
Evan Chengac5f1422007-06-08 09:36:04 +0000377/// ValidSimple - Returns true if the 'true' block (along with its
Evan Chenge882fca2007-06-16 09:34:52 +0000378/// predecessor) forms a valid simple shape for ifcvt. It also returns the
379/// number of instructions that the ifcvt would need to duplicate if performed
380/// in Dups.
381bool IfConverter::ValidSimple(BBInfo &TrueBBI, unsigned &Dups) const {
382 Dups = 0;
Evan Cheng9618bca2007-06-11 22:26:22 +0000383 if (TrueBBI.IsBeingAnalyzed)
384 return false;
385
Evan Chenga2acf842007-06-15 21:18:05 +0000386 if (TrueBBI.BB->pred_size() > 1) {
387 if (TrueBBI.CannotBeCopied ||
388 TrueBBI.NonPredSize > TLI->getIfCvtDupBlockSizeLimit())
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000389 return false;
Evan Chenge882fca2007-06-16 09:34:52 +0000390 Dups = TrueBBI.NonPredSize;
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000391 }
392
393 return !blockAlwaysFallThrough(TrueBBI) && TrueBBI.BrCond.size() == 0;
Evan Cheng7a655472007-06-06 10:16:17 +0000394}
395
Evan Chengac5f1422007-06-08 09:36:04 +0000396/// ValidTriangle - Returns true if the 'true' and 'false' blocks (along
Evan Chengd4de6d92007-06-07 02:12:15 +0000397/// with their common predecessor) forms a valid triangle shape for ifcvt.
Evan Chenge882fca2007-06-16 09:34:52 +0000398/// If 'FalseBranch' is true, it checks if 'true' block's false branch
399/// branches to the false branch rather than the other way around. It also
400/// returns the number of instructions that the ifcvt would need to duplicate
401/// if performed in 'Dups'.
Evan Chengac5f1422007-06-08 09:36:04 +0000402bool IfConverter::ValidTriangle(BBInfo &TrueBBI, BBInfo &FalseBBI,
Evan Chenge882fca2007-06-16 09:34:52 +0000403 bool FalseBranch, unsigned &Dups) const {
404 Dups = 0;
Evan Cheng9618bca2007-06-11 22:26:22 +0000405 if (TrueBBI.IsBeingAnalyzed)
406 return false;
407
Evan Chenga2acf842007-06-15 21:18:05 +0000408 if (TrueBBI.BB->pred_size() > 1) {
409 if (TrueBBI.CannotBeCopied)
410 return false;
411
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000412 unsigned Size = TrueBBI.NonPredSize;
Evan Chenge882fca2007-06-16 09:34:52 +0000413 if (TrueBBI.IsBrAnalyzable) {
414 if (TrueBBI.TrueBB && TrueBBI.BrCond.size() == 0)
415 // End with an unconditional branch. It will be removed.
416 --Size;
417 else {
418 MachineBasicBlock *FExit = FalseBranch
419 ? TrueBBI.TrueBB : TrueBBI.FalseBB;
420 if (FExit)
421 // Require a conditional branch
422 ++Size;
423 }
424 }
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000425 if (Size > TLI->getIfCvtDupBlockSizeLimit())
426 return false;
Evan Chenge882fca2007-06-16 09:34:52 +0000427 Dups = Size;
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000428 }
Evan Chengd4de6d92007-06-07 02:12:15 +0000429
Evan Chengac5f1422007-06-08 09:36:04 +0000430 MachineBasicBlock *TExit = FalseBranch ? TrueBBI.FalseBB : TrueBBI.TrueBB;
431 if (!TExit && blockAlwaysFallThrough(TrueBBI)) {
Evan Chengd4de6d92007-06-07 02:12:15 +0000432 MachineFunction::iterator I = TrueBBI.BB;
433 if (++I == TrueBBI.BB->getParent()->end())
434 return false;
Evan Chengac5f1422007-06-08 09:36:04 +0000435 TExit = I;
Evan Chengd4de6d92007-06-07 02:12:15 +0000436 }
Evan Chengac5f1422007-06-08 09:36:04 +0000437 return TExit && TExit == FalseBBI.BB;
Evan Chengd4de6d92007-06-07 02:12:15 +0000438}
439
Evan Chengac5f1422007-06-08 09:36:04 +0000440/// ValidDiamond - Returns true if the 'true' and 'false' blocks (along
Evan Chengd4de6d92007-06-07 02:12:15 +0000441/// with their common predecessor) forms a valid diamond shape for ifcvt.
442bool IfConverter::ValidDiamond(BBInfo &TrueBBI, BBInfo &FalseBBI) const {
Evan Cheng9618bca2007-06-11 22:26:22 +0000443 if (TrueBBI.IsBeingAnalyzed || FalseBBI.IsBeingAnalyzed)
444 return false;
445
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000446 MachineBasicBlock *TT = TrueBBI.TrueBB;
447 MachineBasicBlock *FT = FalseBBI.TrueBB;
448
449 if (!TT && blockAlwaysFallThrough(TrueBBI))
450 TT = getNextBlock(TrueBBI.BB);
451 if (!FT && blockAlwaysFallThrough(FalseBBI))
452 FT = getNextBlock(FalseBBI.BB);
453 if (TT != FT)
454 return false;
Evan Cheng9618bca2007-06-11 22:26:22 +0000455 if (TT == NULL && (TrueBBI.IsBrAnalyzable || FalseBBI.IsBrAnalyzable))
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000456 return false;
457 // FIXME: Allow false block to have an early exit?
458 return (TrueBBI.BB->pred_size() == 1 &&
Evan Chengd4de6d92007-06-07 02:12:15 +0000459 FalseBBI.BB->pred_size() == 1 &&
Evan Chengd4de6d92007-06-07 02:12:15 +0000460 !TrueBBI.FalseBB && !FalseBBI.FalseBB);
461}
462
Evan Cheng9618bca2007-06-11 22:26:22 +0000463/// ScanInstructions - Scan all the instructions in the block to determine if
464/// the block is predicable. In most cases, that means all the instructions
465/// in the block has M_PREDICABLE flag. Also checks if the block contains any
466/// instruction which can clobber a predicate (e.g. condition code register).
467/// If so, the block is not predicable unless it's the last instruction.
468void IfConverter::ScanInstructions(BBInfo &BBI) {
469 if (BBI.IsDone)
470 return;
471
472 // First analyze the end of BB branches.
Evan Chengb7c908b2007-06-14 21:26:08 +0000473 BBI.TrueBB = BBI.FalseBB = NULL;
Evan Cheng9618bca2007-06-11 22:26:22 +0000474 BBI.BrCond.clear();
475 BBI.IsBrAnalyzable =
476 !TII->AnalyzeBranch(*BBI.BB, BBI.TrueBB, BBI.FalseBB, BBI.BrCond);
477 BBI.HasFallThrough = BBI.IsBrAnalyzable && BBI.FalseBB == NULL;
478
479 if (BBI.BrCond.size()) {
480 // No false branch. This BB must end with a conditional branch and a
481 // fallthrough.
482 if (!BBI.FalseBB)
483 BBI.FalseBB = findFalseBlock(BBI.BB, BBI.TrueBB);
484 assert(BBI.FalseBB && "Expected to find the fallthrough block!");
485 }
486
487 // Then scan all the instructions.
488 BBI.NonPredSize = 0;
489 BBI.ClobbersPred = false;
490 bool SeenCondBr = false;
491 for (MachineBasicBlock::iterator I = BBI.BB->begin(), E = BBI.BB->end();
492 I != E; ++I) {
Evan Chenga2acf842007-06-15 21:18:05 +0000493 if (!BBI.CannotBeCopied && !TII->CanBeDuplicated(I))
494 BBI.CannotBeCopied = true;
495
Evan Cheng9618bca2007-06-11 22:26:22 +0000496 const TargetInstrDescriptor *TID = I->getInstrDescriptor();
497 bool isPredicated = TII->isPredicated(I);
498 bool isCondBr = BBI.IsBrAnalyzable &&
499 (TID->Flags & M_BRANCH_FLAG) != 0 && (TID->Flags & M_BARRIER_FLAG) == 0;
500
501 if (!isPredicated && !isCondBr)
502 BBI.NonPredSize++;
503
504 if (BBI.ClobbersPred && !isPredicated) {
505 // Predicate modification instruction should end the block (except for
506 // already predicated instructions and end of block branches).
507 if (isCondBr) {
508 SeenCondBr = true;
509
510 // Conditional branches is not predicable. But it may be eliminated.
511 continue;
512 }
513
514 // Predicate may have been modified, the subsequent (currently)
515 // unpredocated instructions cannot be correctly predicated.
516 BBI.IsUnpredicable = true;
517 return;
518 }
519
520 if (TID->Flags & M_CLOBBERS_PRED)
521 BBI.ClobbersPred = true;
522
Evan Chengbf9d02e2007-06-15 19:06:07 +0000523 if ((TID->Flags & M_PREDICABLE) == 0) {
Evan Cheng9618bca2007-06-11 22:26:22 +0000524 BBI.IsUnpredicable = true;
525 return;
526 }
527 }
528}
529
530/// FeasibilityAnalysis - Determine if the block is a suitable candidate to be
531/// predicated by the specified predicate.
532bool IfConverter::FeasibilityAnalysis(BBInfo &BBI,
533 std::vector<MachineOperand> &Pred,
534 bool isTriangle, bool RevBranch) {
Evan Chenge882fca2007-06-16 09:34:52 +0000535 // If the block is dead or unpredicable, then it cannot be predicated.
536 if (BBI.IsDone || BBI.IsUnpredicable)
Evan Cheng9618bca2007-06-11 22:26:22 +0000537 return false;
538
539 // Check predication threshold.
540 if (BBI.NonPredSize == 0 || BBI.NonPredSize > TLI->getIfCvtBlockSizeLimit())
541 return false;
542
543 // If it is already predicated, check if its predicate subsumes the new
544 // predicate.
545 if (BBI.Predicate.size() && !TII->SubsumesPredicate(BBI.Predicate, Pred))
546 return false;
547
548 if (BBI.BrCond.size()) {
549 if (!isTriangle)
550 return false;
551
552 // Test predicate subsumsion.
553 std::vector<MachineOperand> RevPred(Pred);
554 std::vector<MachineOperand> Cond(BBI.BrCond);
555 if (RevBranch) {
556 if (TII->ReverseBranchCondition(Cond))
557 return false;
558 }
559 if (TII->ReverseBranchCondition(RevPred) ||
560 !TII->SubsumesPredicate(Cond, RevPred))
561 return false;
562 }
563
564 return true;
565}
566
Evan Chengac5f1422007-06-08 09:36:04 +0000567/// AnalyzeBlock - Analyze the structure of the sub-CFG starting from
Evan Chengcf6cc112007-05-18 18:14:37 +0000568/// the specified block. Record its successors and whether it looks like an
569/// if-conversion candidate.
Evan Chenge882fca2007-06-16 09:34:52 +0000570IfConverter::BBInfo &IfConverter::AnalyzeBlock(MachineBasicBlock *BB,
571 std::vector<IfcvtToken*> &Tokens) {
Evan Cheng4e654852007-05-16 02:00:57 +0000572 BBInfo &BBI = BBAnalysis[BB->getNumber()];
573
Evan Cheng9618bca2007-06-11 22:26:22 +0000574 if (BBI.IsAnalyzed || BBI.IsBeingAnalyzed)
575 return BBI;
Evan Cheng86cbfea2007-05-18 00:20:58 +0000576
Evan Cheng9618bca2007-06-11 22:26:22 +0000577 BBI.BB = BB;
578 BBI.IsBeingAnalyzed = true;
Evan Cheng9618bca2007-06-11 22:26:22 +0000579
580 ScanInstructions(BBI);
581
Evan Chengd4de6d92007-06-07 02:12:15 +0000582 // Unanalyable or ends with fallthrough or unconditional branch.
Evan Cheng9618bca2007-06-11 22:26:22 +0000583 if (!BBI.IsBrAnalyzable || BBI.BrCond.size() == 0) {
584 BBI.IsBeingAnalyzed = false;
585 BBI.IsAnalyzed = true;
586 return BBI;
Evan Cheng7a655472007-06-06 10:16:17 +0000587 }
588
Evan Cheng9618bca2007-06-11 22:26:22 +0000589 // Do not ifcvt if either path is a back edge to the entry block.
590 if (BBI.TrueBB == BB || BBI.FalseBB == BB) {
591 BBI.IsBeingAnalyzed = false;
592 BBI.IsAnalyzed = true;
593 return BBI;
594 }
595
Evan Chenge882fca2007-06-16 09:34:52 +0000596 BBInfo &TrueBBI = AnalyzeBlock(BBI.TrueBB, Tokens);
597 BBInfo &FalseBBI = AnalyzeBlock(BBI.FalseBB, Tokens);
Evan Cheng9618bca2007-06-11 22:26:22 +0000598
599 if (TrueBBI.IsDone && FalseBBI.IsDone) {
600 BBI.IsBeingAnalyzed = false;
601 BBI.IsAnalyzed = true;
602 return BBI;
603 }
604
Evan Chengb6665f62007-06-04 06:47:22 +0000605 std::vector<MachineOperand> RevCond(BBI.BrCond);
606 bool CanRevCond = !TII->ReverseBranchCondition(RevCond);
Evan Chengac5f1422007-06-08 09:36:04 +0000607
Evan Chenge882fca2007-06-16 09:34:52 +0000608 unsigned Dups = 0;
609 bool TNeedSub = TrueBBI.Predicate.size() > 0;
610 bool FNeedSub = FalseBBI.Predicate.size() > 0;
611 bool Enqueued = false;
Evan Chengac5f1422007-06-08 09:36:04 +0000612 if (CanRevCond && ValidDiamond(TrueBBI, FalseBBI) &&
Evan Cheng9618bca2007-06-11 22:26:22 +0000613 !(TrueBBI.ClobbersPred && FalseBBI.ClobbersPred) &&
Evan Chengac5f1422007-06-08 09:36:04 +0000614 FeasibilityAnalysis(TrueBBI, BBI.BrCond) &&
615 FeasibilityAnalysis(FalseBBI, RevCond)) {
Evan Cheng4e654852007-05-16 02:00:57 +0000616 // Diamond:
617 // EBB
618 // / \_
619 // | |
620 // TBB FBB
621 // \ /
Evan Cheng86cbfea2007-05-18 00:20:58 +0000622 // TailBB
Evan Cheng993fc952007-06-05 23:46:14 +0000623 // Note TailBB can be empty.
Evan Chenge882fca2007-06-16 09:34:52 +0000624 Tokens.push_back(new IfcvtToken(BBI, ICDiamond, TNeedSub|FNeedSub, Dups));
625 Enqueued = true;
626 }
627
628 if (ValidTriangle(TrueBBI, FalseBBI, false, Dups) &&
629 FeasibilityAnalysis(TrueBBI, BBI.BrCond, true)) {
630 // Triangle:
631 // EBB
632 // | \_
633 // | |
634 // | TBB
635 // | /
636 // FBB
637 Tokens.push_back(new IfcvtToken(BBI, ICTriangle, TNeedSub, Dups));
638 Enqueued = true;
639 }
640
641 if (ValidTriangle(TrueBBI, FalseBBI, true, Dups) &&
642 FeasibilityAnalysis(TrueBBI, BBI.BrCond, true, true)) {
643 Tokens.push_back(new IfcvtToken(BBI, ICTriangleRev, TNeedSub, Dups));
644 Enqueued = true;
645 }
646
647 if (ValidSimple(TrueBBI, Dups) &&
648 FeasibilityAnalysis(TrueBBI, BBI.BrCond)) {
649 // Simple (split, no rejoin):
650 // EBB
651 // | \_
652 // | |
653 // | TBB---> exit
654 // |
655 // FBB
656 Tokens.push_back(new IfcvtToken(BBI, ICSimple, TNeedSub, Dups));
657 Enqueued = true;
658 }
659
660 if (CanRevCond) {
661 // Try the other path...
662 if (ValidTriangle(FalseBBI, TrueBBI, false, Dups) &&
663 FeasibilityAnalysis(FalseBBI, RevCond, true)) {
664 Tokens.push_back(new IfcvtToken(BBI, ICTriangleFalse, FNeedSub, Dups));
665 Enqueued = true;
666 }
667
668 if (ValidTriangle(FalseBBI, TrueBBI, true, Dups) &&
669 FeasibilityAnalysis(FalseBBI, RevCond, true, true)) {
670 Tokens.push_back(new IfcvtToken(BBI, ICTriangleFRev, FNeedSub, Dups));
671 Enqueued = true;
672 }
673
674 if (ValidSimple(FalseBBI, Dups) &&
675 FeasibilityAnalysis(FalseBBI, RevCond)) {
676 Tokens.push_back(new IfcvtToken(BBI, ICSimpleFalse, FNeedSub, Dups));
677 Enqueued = true;
Evan Chengb6665f62007-06-04 06:47:22 +0000678 }
Evan Cheng4e654852007-05-16 02:00:57 +0000679 }
Evan Cheng4e654852007-05-16 02:00:57 +0000680
Evan Chenge882fca2007-06-16 09:34:52 +0000681 BBI.IsEnqueued = Enqueued;
Evan Cheng9618bca2007-06-11 22:26:22 +0000682 BBI.IsBeingAnalyzed = false;
683 BBI.IsAnalyzed = true;
684 return BBI;
Evan Chengcf6cc112007-05-18 18:14:37 +0000685}
686
Evan Cheng82582102007-05-30 19:49:19 +0000687/// AnalyzeBlocks - Analyze all blocks and find entries for all if-conversion
688/// candidates. It returns true if any CFG restructuring is done to expose more
689/// if-conversion opportunities.
690bool IfConverter::AnalyzeBlocks(MachineFunction &MF,
Evan Chenge882fca2007-06-16 09:34:52 +0000691 std::vector<IfcvtToken*> &Tokens) {
Evan Cheng82582102007-05-30 19:49:19 +0000692 bool Change = false;
Evan Cheng36489bb2007-05-18 19:26:33 +0000693 std::set<MachineBasicBlock*> Visited;
Evan Cheng82582102007-05-30 19:49:19 +0000694 for (unsigned i = 0, e = Roots.size(); i != e; ++i) {
695 for (idf_ext_iterator<MachineBasicBlock*> I=idf_ext_begin(Roots[i],Visited),
696 E = idf_ext_end(Roots[i], Visited); I != E; ++I) {
697 MachineBasicBlock *BB = *I;
Evan Chenge882fca2007-06-16 09:34:52 +0000698 AnalyzeBlock(BB, Tokens);
Evan Chenga6b4f432007-05-21 22:22:58 +0000699 }
Evan Cheng4e654852007-05-16 02:00:57 +0000700 }
Evan Cheng82582102007-05-30 19:49:19 +0000701
Evan Cheng5f702182007-06-01 00:12:12 +0000702 // Sort to favor more complex ifcvt scheme.
Evan Chenge882fca2007-06-16 09:34:52 +0000703 std::stable_sort(Tokens.begin(), Tokens.end(), IfcvtTokenCmp);
Evan Cheng5f702182007-06-01 00:12:12 +0000704
Evan Cheng82582102007-05-30 19:49:19 +0000705 return Change;
Evan Cheng4e654852007-05-16 02:00:57 +0000706}
707
Evan Cheng7e75ba82007-06-08 22:01:07 +0000708/// canFallThroughTo - Returns true either if ToBB is the next block after BB or
709/// that all the intervening blocks are empty (given BB can fall through to its
710/// next block).
711static bool canFallThroughTo(MachineBasicBlock *BB, MachineBasicBlock *ToBB) {
Evan Chengb6665f62007-06-04 06:47:22 +0000712 MachineFunction::iterator I = BB;
Evan Chengc53ef582007-06-05 07:05:25 +0000713 MachineFunction::iterator TI = ToBB;
714 MachineFunction::iterator E = BB->getParent()->end();
715 while (++I != TI)
716 if (I == E || !I->empty())
Evan Chengb6665f62007-06-04 06:47:22 +0000717 return false;
718 return true;
Evan Chenga6b4f432007-05-21 22:22:58 +0000719}
720
Evan Cheng82582102007-05-30 19:49:19 +0000721/// ReTryPreds - Invalidate predecessor BB info so it would be re-analyzed
Evan Chenge37e2432007-06-14 23:13:19 +0000722/// to determine if it can be if-converted. If predecessor is already
723/// enqueud, dequeue it!
Evan Cheng82582102007-05-30 19:49:19 +0000724void IfConverter::ReTryPreds(MachineBasicBlock *BB) {
Evan Chenga13aa952007-05-23 07:23:16 +0000725 for (MachineBasicBlock::pred_iterator PI = BB->pred_begin(),
726 E = BB->pred_end(); PI != E; ++PI) {
727 BBInfo &PBBI = BBAnalysis[(*PI)->getNumber()];
Evan Chengbc198ee2007-06-14 23:34:09 +0000728 if (PBBI.IsDone || PBBI.BB == BB)
Evan Chenge37e2432007-06-14 23:13:19 +0000729 continue;
Evan Chengbc198ee2007-06-14 23:34:09 +0000730 PBBI.IsAnalyzed = false;
731 PBBI.IsEnqueued = false;
Evan Chenga13aa952007-05-23 07:23:16 +0000732 }
733}
734
Evan Chengc8ed9ba2007-05-29 22:31:16 +0000735/// InsertUncondBranch - Inserts an unconditional branch from BB to ToBB.
736///
737static void InsertUncondBranch(MachineBasicBlock *BB, MachineBasicBlock *ToBB,
738 const TargetInstrInfo *TII) {
739 std::vector<MachineOperand> NoCond;
740 TII->InsertBranch(*BB, ToBB, NULL, NoCond);
741}
742
Evan Cheng7e75ba82007-06-08 22:01:07 +0000743/// RemoveExtraEdges - Remove true / false edges if either / both are no longer
744/// successors.
745void IfConverter::RemoveExtraEdges(BBInfo &BBI) {
746 MachineBasicBlock *TBB = NULL, *FBB = NULL;
747 std::vector<MachineOperand> Cond;
748 bool isAnalyzable = !TII->AnalyzeBranch(*BBI.BB, TBB, FBB, Cond);
749 bool CanFallthrough = isAnalyzable && (TBB == NULL || FBB == NULL);
750 if (BBI.TrueBB && BBI.BB->isSuccessor(BBI.TrueBB))
751 if (!(BBI.TrueBB == TBB || BBI.TrueBB == FBB ||
752 (CanFallthrough && getNextBlock(BBI.BB) == BBI.TrueBB)))
753 BBI.BB->removeSuccessor(BBI.TrueBB);
754 if (BBI.FalseBB && BBI.BB->isSuccessor(BBI.FalseBB))
755 if (!(BBI.FalseBB == TBB || BBI.FalseBB == FBB ||
756 (CanFallthrough && getNextBlock(BBI.BB) == BBI.FalseBB)))
757 BBI.BB->removeSuccessor(BBI.FalseBB);
758}
759
Evan Chengb6665f62007-06-04 06:47:22 +0000760/// IfConvertSimple - If convert a simple (split, no rejoin) sub-CFG.
Evan Chenga6b4f432007-05-21 22:22:58 +0000761///
Evan Chenge882fca2007-06-16 09:34:52 +0000762bool IfConverter::IfConvertSimple(BBInfo &BBI, IfcvtKind Kind) {
Evan Chenga6b4f432007-05-21 22:22:58 +0000763 BBInfo &TrueBBI = BBAnalysis[BBI.TrueBB->getNumber()];
764 BBInfo &FalseBBI = BBAnalysis[BBI.FalseBB->getNumber()];
765 BBInfo *CvtBBI = &TrueBBI;
766 BBInfo *NextBBI = &FalseBBI;
Evan Chenga13aa952007-05-23 07:23:16 +0000767
Evan Chengb6665f62007-06-04 06:47:22 +0000768 std::vector<MachineOperand> Cond(BBI.BrCond);
Evan Chenge882fca2007-06-16 09:34:52 +0000769 if (Kind == ICSimpleFalse)
Evan Chengb5a06902007-06-01 20:29:21 +0000770 std::swap(CvtBBI, NextBBI);
Evan Chenga2acf842007-06-15 21:18:05 +0000771
772 if (CvtBBI->CannotBeCopied && CvtBBI->BB->pred_size() > 1) {
773 // Something has changed. It's no longer safe to predicate this block.
Evan Chenga2acf842007-06-15 21:18:05 +0000774 BBI.IsAnalyzed = false;
775 CvtBBI->IsAnalyzed = false;
776 return false;
Evan Chengb5a06902007-06-01 20:29:21 +0000777 }
Evan Chenga13aa952007-05-23 07:23:16 +0000778
Evan Chenge882fca2007-06-16 09:34:52 +0000779 if (Kind == ICSimpleFalse)
Evan Chenga2acf842007-06-15 21:18:05 +0000780 TII->ReverseBranchCondition(Cond);
781
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000782 if (CvtBBI->BB->pred_size() > 1) {
783 BBI.NonPredSize -= TII->RemoveBranch(*BBI.BB);
784 // Copy instructions in the true block, predicate them add them to
785 // the entry block.
786 CopyAndPredicateBlock(BBI, *CvtBBI, Cond);
787 } else {
788 PredicateBlock(*CvtBBI, Cond);
Evan Chenga6b4f432007-05-21 22:22:58 +0000789
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000790 // Merge converted block into entry block.
791 BBI.NonPredSize -= TII->RemoveBranch(*BBI.BB);
792 MergeBlocks(BBI, *CvtBBI);
793 }
Evan Chengd4de6d92007-06-07 02:12:15 +0000794
Evan Cheng3d6f60e2007-06-06 02:08:52 +0000795 bool IterIfcvt = true;
Evan Cheng7e75ba82007-06-08 22:01:07 +0000796 if (!canFallThroughTo(BBI.BB, NextBBI->BB)) {
Evan Chengc8ed9ba2007-05-29 22:31:16 +0000797 InsertUncondBranch(BBI.BB, NextBBI->BB, TII);
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000798 BBI.HasFallThrough = false;
Evan Chengac5f1422007-06-08 09:36:04 +0000799 // Now ifcvt'd block will look like this:
800 // BB:
801 // ...
802 // t, f = cmp
803 // if t op
804 // b BBf
805 //
806 // We cannot further ifcvt this block because the unconditional branch
807 // will have to be predicated on the new condition, that will not be
808 // available if cmp executes.
809 IterIfcvt = false;
Evan Cheng3d6f60e2007-06-06 02:08:52 +0000810 }
Evan Chenga6b4f432007-05-21 22:22:58 +0000811
Evan Cheng7e75ba82007-06-08 22:01:07 +0000812 RemoveExtraEdges(BBI);
813
Evan Chenga13aa952007-05-23 07:23:16 +0000814 // Update block info. BB can be iteratively if-converted.
Evan Cheng9618bca2007-06-11 22:26:22 +0000815 if (!IterIfcvt)
816 BBI.IsDone = true;
Evan Cheng82582102007-05-30 19:49:19 +0000817 ReTryPreds(BBI.BB);
Evan Cheng9618bca2007-06-11 22:26:22 +0000818 CvtBBI->IsDone = true;
Evan Chenga6b4f432007-05-21 22:22:58 +0000819
820 // FIXME: Must maintain LiveIns.
Evan Chenga6b4f432007-05-21 22:22:58 +0000821 return true;
822}
823
Evan Chengd6ddc302007-05-16 21:54:37 +0000824/// IfConvertTriangle - If convert a triangle sub-CFG.
825///
Evan Chenge882fca2007-06-16 09:34:52 +0000826bool IfConverter::IfConvertTriangle(BBInfo &BBI, IfcvtKind Kind) {
Evan Chenga13aa952007-05-23 07:23:16 +0000827 BBInfo &TrueBBI = BBAnalysis[BBI.TrueBB->getNumber()];
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000828 BBInfo &FalseBBI = BBAnalysis[BBI.FalseBB->getNumber()];
829 BBInfo *CvtBBI = &TrueBBI;
830 BBInfo *NextBBI = &FalseBBI;
831
832 std::vector<MachineOperand> Cond(BBI.BrCond);
Evan Chenge882fca2007-06-16 09:34:52 +0000833 if (Kind == ICTriangleFalse || Kind == ICTriangleFRev)
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000834 std::swap(CvtBBI, NextBBI);
Evan Chenga2acf842007-06-15 21:18:05 +0000835
836 if (CvtBBI->CannotBeCopied && CvtBBI->BB->pred_size() > 1) {
837 // Something has changed. It's no longer safe to predicate this block.
Evan Chenga2acf842007-06-15 21:18:05 +0000838 BBI.IsAnalyzed = false;
839 CvtBBI->IsAnalyzed = false;
840 return false;
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000841 }
Evan Chenga2acf842007-06-15 21:18:05 +0000842
Evan Chenge882fca2007-06-16 09:34:52 +0000843 if (Kind == ICTriangleFalse || Kind == ICTriangleFRev)
Evan Chenga2acf842007-06-15 21:18:05 +0000844 TII->ReverseBranchCondition(Cond);
845
Evan Chenge882fca2007-06-16 09:34:52 +0000846 if (Kind == ICTriangleRev || Kind == ICTriangleFRev) {
Evan Chengcc8fb462007-06-12 23:54:05 +0000847 ReverseBranchCondition(*CvtBBI);
848 // BB has been changed, modify its predecessors (except for this
849 // one) so they don't get ifcvt'ed based on bad intel.
850 for (MachineBasicBlock::pred_iterator PI = CvtBBI->BB->pred_begin(),
851 E = CvtBBI->BB->pred_end(); PI != E; ++PI) {
852 MachineBasicBlock *PBB = *PI;
853 if (PBB == BBI.BB)
854 continue;
855 BBInfo &PBBI = BBAnalysis[PBB->getNumber()];
Evan Chengbc198ee2007-06-14 23:34:09 +0000856 if (PBBI.IsEnqueued) {
Evan Chengbc198ee2007-06-14 23:34:09 +0000857 PBBI.IsAnalyzed = false;
Evan Chengcc8fb462007-06-12 23:54:05 +0000858 PBBI.IsEnqueued = false;
Evan Chengbc198ee2007-06-14 23:34:09 +0000859 }
Evan Chengcc8fb462007-06-12 23:54:05 +0000860 }
861 }
Evan Cheng8c529382007-06-01 07:41:07 +0000862
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000863 bool HasEarlyExit = CvtBBI->FalseBB != NULL;
864 bool DupBB = CvtBBI->BB->pred_size() > 1;
865 if (DupBB) {
866 BBI.NonPredSize -= TII->RemoveBranch(*BBI.BB);
867 // Copy instructions in the true block, predicate them add them to
868 // the entry block.
869 CopyAndPredicateBlock(BBI, *CvtBBI, Cond, true);
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000870 } else {
871 // Predicate the 'true' block after removing its branch.
872 CvtBBI->NonPredSize -= TII->RemoveBranch(*CvtBBI->BB);
873 PredicateBlock(*CvtBBI, Cond);
874 }
Evan Chenga6b4f432007-05-21 22:22:58 +0000875
Evan Chengb6665f62007-06-04 06:47:22 +0000876 // If 'true' block has a 'false' successor, add an exit branch to it.
Evan Cheng8ed680c2007-06-05 01:31:40 +0000877 if (HasEarlyExit) {
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000878 std::vector<MachineOperand> RevCond(CvtBBI->BrCond);
Evan Cheng8c529382007-06-01 07:41:07 +0000879 if (TII->ReverseBranchCondition(RevCond))
880 assert(false && "Unable to reverse branch condition!");
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000881 if (DupBB) {
882 TII->InsertBranch(*BBI.BB, CvtBBI->FalseBB, NULL, RevCond);
883 BBI.BB->addSuccessor(CvtBBI->FalseBB);
884 } else {
885 TII->InsertBranch(*CvtBBI->BB, CvtBBI->FalseBB, NULL, RevCond);
886 }
Evan Cheng8c529382007-06-01 07:41:07 +0000887 }
888
Evan Cheng2c8c3a42007-06-15 07:36:12 +0000889 if (!DupBB) {
890 // Now merge the entry of the triangle with the true block.
891 BBI.NonPredSize -= TII->RemoveBranch(*BBI.BB);
892 MergeBlocks(BBI, *CvtBBI);
893 }
Evan Chengac5f1422007-06-08 09:36:04 +0000894
895 // Merge in the 'false' block if the 'false' block has no other
896 // predecessors. Otherwise, add a unconditional branch from to 'false'.
Evan Chengf5305f92007-06-05 00:07:37 +0000897 bool FalseBBDead = false;
Evan Cheng3d6f60e2007-06-06 02:08:52 +0000898 bool IterIfcvt = true;
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000899 bool isFallThrough = canFallThroughTo(BBI.BB, NextBBI->BB);
Evan Chengf4769612007-06-07 08:13:00 +0000900 if (!isFallThrough) {
901 // Only merge them if the true block does not fallthrough to the false
902 // block. By not merging them, we make it possible to iteratively
903 // ifcvt the blocks.
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000904 if (!HasEarlyExit && NextBBI->BB->pred_size() == 1) {
905 MergeBlocks(BBI, *NextBBI);
Evan Chengf4769612007-06-07 08:13:00 +0000906 FalseBBDead = true;
Evan Chengf4769612007-06-07 08:13:00 +0000907 } else {
Evan Cheng1c9f91d2007-06-09 01:03:43 +0000908 InsertUncondBranch(BBI.BB, NextBBI->BB, TII);
909 BBI.HasFallThrough = false;
Evan Chengf4769612007-06-07 08:13:00 +0000910 }
Evan Chengac5f1422007-06-08 09:36:04 +0000911 // Mixed predicated and unpredicated code. This cannot be iteratively
912 // predicated.
913 IterIfcvt = false;
Evan Cheng3d6f60e2007-06-06 02:08:52 +0000914 }
Evan Chenga6b4f432007-05-21 22:22:58 +0000915
Evan Cheng7e75ba82007-06-08 22:01:07 +0000916 RemoveExtraEdges(BBI);
Evan Chenga6b4f432007-05-21 22:22:58 +0000917
Evan Chenga13aa952007-05-23 07:23:16 +0000918 // Update block info. BB can be iteratively if-converted.
Evan Cheng9618bca2007-06-11 22:26:22 +0000919 if (!IterIfcvt)
920 BBI.IsDone = true;
Evan Cheng82582102007-05-30 19:49:19 +0000921 ReTryPreds(BBI.BB);
Evan Cheng9618bca2007-06-11 22:26:22 +0000922 CvtBBI->IsDone = true;
Evan Chengf5305f92007-06-05 00:07:37 +0000923 if (FalseBBDead)
Evan Cheng9618bca2007-06-11 22:26:22 +0000924 NextBBI->IsDone = true;
Evan Chenga6b4f432007-05-21 22:22:58 +0000925
926 // FIXME: Must maintain LiveIns.
Evan Chenga6b4f432007-05-21 22:22:58 +0000927 return true;
Evan Cheng4e654852007-05-16 02:00:57 +0000928}
929
Evan Chengd6ddc302007-05-16 21:54:37 +0000930/// IfConvertDiamond - If convert a diamond sub-CFG.
931///
Evan Chenge882fca2007-06-16 09:34:52 +0000932bool IfConverter::IfConvertDiamond(BBInfo &BBI, IfcvtKind Kind) {
Evan Chengb6665f62007-06-04 06:47:22 +0000933 BBInfo &TrueBBI = BBAnalysis[BBI.TrueBB->getNumber()];
Evan Chengcf6cc112007-05-18 18:14:37 +0000934 BBInfo &FalseBBI = BBAnalysis[BBI.FalseBB->getNumber()];
Evan Chenge882fca2007-06-16 09:34:52 +0000935 MachineBasicBlock *TailBB = TrueBBI.TrueBB;
Evan Chengcf6cc112007-05-18 18:14:37 +0000936
Evan Chenga6b4f432007-05-21 22:22:58 +0000937 SmallVector<MachineInstr*, 2> Dups;
Evan Chenge882fca2007-06-16 09:34:52 +0000938 if (!TailBB) {
Evan Chengb6665f62007-06-04 06:47:22 +0000939 // No common merge block. Check if the terminators (e.g. return) are
940 // the same or predicable.
941 MachineBasicBlock::iterator TT = BBI.TrueBB->getFirstTerminator();
942 MachineBasicBlock::iterator FT = BBI.FalseBB->getFirstTerminator();
943 while (TT != BBI.TrueBB->end() && FT != BBI.FalseBB->end()) {
944 if (TT->isIdenticalTo(FT))
945 Dups.push_back(TT); // Will erase these later.
Evan Chengbf9d02e2007-06-15 19:06:07 +0000946 else if ((TT->getInstrDescriptor()->Flags & M_PREDICABLE) == 0 ||
947 (FT->getInstrDescriptor()->Flags & M_PREDICABLE) == 0)
Evan Chengb6665f62007-06-04 06:47:22 +0000948 return false; // Can't if-convert. Abort!
949 ++TT;
950 ++FT;
Evan Chengcf6cc112007-05-18 18:14:37 +0000951 }
Evan Chengcf6cc112007-05-18 18:14:37 +0000952
Evan Chengb6665f62007-06-04 06:47:22 +0000953 // One of the two pathes have more terminators, make sure they are
954 // all predicable.
955 while (TT != BBI.TrueBB->end()) {
Evan Chengbf9d02e2007-06-15 19:06:07 +0000956 if ((TT->getInstrDescriptor()->Flags & M_PREDICABLE) == 0)
Evan Chengb6665f62007-06-04 06:47:22 +0000957 return false; // Can't if-convert. Abort!
Evan Chengb6665f62007-06-04 06:47:22 +0000958 ++TT;
959 }
960 while (FT != BBI.FalseBB->end()) {
Evan Chengbf9d02e2007-06-15 19:06:07 +0000961 if ((FT->getInstrDescriptor()->Flags & M_PREDICABLE) == 0)
Evan Chengb6665f62007-06-04 06:47:22 +0000962 return false; // Can't if-convert. Abort!
Evan Chengb6665f62007-06-04 06:47:22 +0000963 ++FT;
Evan Cheng4e654852007-05-16 02:00:57 +0000964 }
Evan Cheng4e654852007-05-16 02:00:57 +0000965 }
Evan Chenga6b4f432007-05-21 22:22:58 +0000966
Evan Chenga6b4f432007-05-21 22:22:58 +0000967 // Remove the duplicated instructions from the 'true' block.
968 for (unsigned i = 0, e = Dups.size(); i != e; ++i) {
969 Dups[i]->eraseFromParent();
Evan Chenga13aa952007-05-23 07:23:16 +0000970 --TrueBBI.NonPredSize;
Evan Chenga6b4f432007-05-21 22:22:58 +0000971 }
972
Evan Chenga6b4f432007-05-21 22:22:58 +0000973 // Merge the 'true' and 'false' blocks by copying the instructions
974 // from the 'false' block to the 'true' block. That is, unless the true
975 // block would clobber the predicate, in that case, do the opposite.
Evan Cheng993fc952007-06-05 23:46:14 +0000976 BBInfo *BBI1 = &TrueBBI;
977 BBInfo *BBI2 = &FalseBBI;
Evan Chengb6665f62007-06-04 06:47:22 +0000978 std::vector<MachineOperand> RevCond(BBI.BrCond);
979 TII->ReverseBranchCondition(RevCond);
Evan Cheng993fc952007-06-05 23:46:14 +0000980 std::vector<MachineOperand> *Cond1 = &BBI.BrCond;
981 std::vector<MachineOperand> *Cond2 = &RevCond;
Evan Cheng993fc952007-06-05 23:46:14 +0000982 // Check the 'true' and 'false' blocks if either isn't ended with a branch.
983 // Either the block fallthrough to another block or it ends with a
984 // return. If it's the former, add a branch to its successor.
Evan Chenge882fca2007-06-16 09:34:52 +0000985 bool NeedBr1 = !BBI1->TrueBB && BBI1->BB->succ_size() > 0;
986 bool NeedBr2 = !BBI2->TrueBB && BBI2->BB->succ_size() > 0;
Evan Chenge7052132007-06-06 00:57:55 +0000987
Evan Chenge882fca2007-06-16 09:34:52 +0000988 // Figure out the more profitable ordering.
989 bool DoSwap = false;
990 if (TrueBBI.ClobbersPred && !FalseBBI.ClobbersPred)
991 DoSwap = true;
992 else if (TrueBBI.ClobbersPred == FalseBBI.ClobbersPred) {
993 if (!NeedBr1 && NeedBr2)
994 DoSwap = true;
995 else if (NeedBr1 == NeedBr2) {
996 if (TrueBBI.NonPredSize > FalseBBI.NonPredSize)
997 DoSwap = true;
998 }
999 }
1000 if (DoSwap) {
Evan Chenge7052132007-06-06 00:57:55 +00001001 std::swap(BBI1, BBI2);
1002 std::swap(Cond1, Cond2);
1003 std::swap(NeedBr1, NeedBr2);
1004 }
Evan Cheng993fc952007-06-05 23:46:14 +00001005
1006 // Predicate the 'true' block after removing its branch.
1007 BBI1->NonPredSize -= TII->RemoveBranch(*BBI1->BB);
1008 PredicateBlock(*BBI1, *Cond1);
1009
1010 // Add an early exit branch if needed.
1011 if (NeedBr1)
1012 TII->InsertBranch(*BBI1->BB, *BBI1->BB->succ_begin(), NULL, *Cond1);
1013
1014 // Predicate the 'false' block.
1015 PredicateBlock(*BBI2, *Cond2, true);
1016
1017 // Add an unconditional branch from 'false' to to 'false' successor if it
1018 // will not be the fallthrough block.
Evan Chengd4de6d92007-06-07 02:12:15 +00001019 if (NeedBr2 && !NeedBr1) {
1020 // If BBI2 isn't going to be merged in, then the existing fallthrough
1021 // or branch is fine.
Evan Cheng7e75ba82007-06-08 22:01:07 +00001022 if (!canFallThroughTo(BBI.BB, *BBI2->BB->succ_begin())) {
Evan Chengd4de6d92007-06-07 02:12:15 +00001023 InsertUncondBranch(BBI2->BB, *BBI2->BB->succ_begin(), TII);
Evan Cheng1c9f91d2007-06-09 01:03:43 +00001024 BBI2->HasFallThrough = false;
Evan Chengd4de6d92007-06-07 02:12:15 +00001025 }
1026 }
Evan Cheng993fc952007-06-05 23:46:14 +00001027
1028 // Keep them as two separate blocks if there is an early exit.
1029 if (!NeedBr1)
1030 MergeBlocks(*BBI1, *BBI2);
Evan Cheng993fc952007-06-05 23:46:14 +00001031
Evan Chenga6b4f432007-05-21 22:22:58 +00001032 // Remove the conditional branch from entry to the blocks.
Evan Chenga13aa952007-05-23 07:23:16 +00001033 BBI.NonPredSize -= TII->RemoveBranch(*BBI.BB);
Evan Chenga6b4f432007-05-21 22:22:58 +00001034
Evan Cheng993fc952007-06-05 23:46:14 +00001035 // Merge the combined block into the entry of the diamond.
1036 MergeBlocks(BBI, *BBI1);
Evan Chenga6b4f432007-05-21 22:22:58 +00001037
Evan Chenge7052132007-06-06 00:57:55 +00001038 // 'True' and 'false' aren't combined, see if we need to add a unconditional
1039 // branch to the 'false' block.
Evan Cheng7e75ba82007-06-08 22:01:07 +00001040 if (NeedBr1 && !canFallThroughTo(BBI.BB, BBI2->BB)) {
Evan Chengd4de6d92007-06-07 02:12:15 +00001041 InsertUncondBranch(BBI.BB, BBI2->BB, TII);
Evan Cheng1c9f91d2007-06-09 01:03:43 +00001042 BBI1->HasFallThrough = false;
Evan Chengd4de6d92007-06-07 02:12:15 +00001043 }
Evan Chenge7052132007-06-06 00:57:55 +00001044
Evan Cheng58fbb9f2007-05-29 23:37:20 +00001045 // If the if-converted block fallthrough or unconditionally branch into the
1046 // tail block, and the tail block does not have other predecessors, then
Evan Chenga6b4f432007-05-21 22:22:58 +00001047 // fold the tail block in as well.
Evan Cheng993fc952007-06-05 23:46:14 +00001048 BBInfo *CvtBBI = NeedBr1 ? BBI2 : &BBI;
Evan Chenge882fca2007-06-16 09:34:52 +00001049 if (TailBB &&
1050 TailBB->pred_size() == 1 && CvtBBI->BB->succ_size() == 1) {
Evan Chenga13aa952007-05-23 07:23:16 +00001051 CvtBBI->NonPredSize -= TII->RemoveBranch(*CvtBBI->BB);
Evan Chenge882fca2007-06-16 09:34:52 +00001052 BBInfo TailBBI = BBAnalysis[TailBB->getNumber()];
Evan Chenga6b4f432007-05-21 22:22:58 +00001053 MergeBlocks(*CvtBBI, TailBBI);
Evan Cheng9618bca2007-06-11 22:26:22 +00001054 TailBBI.IsDone = true;
Evan Chenga6b4f432007-05-21 22:22:58 +00001055 }
1056
Evan Cheng7e75ba82007-06-08 22:01:07 +00001057 RemoveExtraEdges(BBI);
1058
Evan Cheng993fc952007-06-05 23:46:14 +00001059 // Update block info.
Evan Cheng9618bca2007-06-11 22:26:22 +00001060 BBI.IsDone = TrueBBI.IsDone = FalseBBI.IsDone = true;
Evan Chenga6b4f432007-05-21 22:22:58 +00001061
1062 // FIXME: Must maintain LiveIns.
Evan Chenga6b4f432007-05-21 22:22:58 +00001063 return true;
Evan Cheng4e654852007-05-16 02:00:57 +00001064}
1065
Evan Cheng4e654852007-05-16 02:00:57 +00001066/// PredicateBlock - Predicate every instruction in the block with the specified
1067/// condition. If IgnoreTerm is true, skip over all terminator instructions.
Evan Chenga13aa952007-05-23 07:23:16 +00001068void IfConverter::PredicateBlock(BBInfo &BBI,
Evan Cheng4e654852007-05-16 02:00:57 +00001069 std::vector<MachineOperand> &Cond,
1070 bool IgnoreTerm) {
Evan Chenga13aa952007-05-23 07:23:16 +00001071 for (MachineBasicBlock::iterator I = BBI.BB->begin(), E = BBI.BB->end();
Evan Cheng4e654852007-05-16 02:00:57 +00001072 I != E; ++I) {
Evan Chengb6665f62007-06-04 06:47:22 +00001073 if (IgnoreTerm && TII->isTerminatorInstr(I->getOpcode()))
Evan Cheng4e654852007-05-16 02:00:57 +00001074 continue;
Evan Chengb6665f62007-06-04 06:47:22 +00001075 if (TII->isPredicated(I))
Evan Chenga13aa952007-05-23 07:23:16 +00001076 continue;
Evan Chengb6665f62007-06-04 06:47:22 +00001077 if (!TII->PredicateInstruction(I, Cond)) {
Evan Chengfe57a7e2007-06-01 00:55:26 +00001078 cerr << "Unable to predicate " << *I << "!\n";
Evan Chengd6ddc302007-05-16 21:54:37 +00001079 abort();
1080 }
Evan Cheng4e654852007-05-16 02:00:57 +00001081 }
Evan Chenga13aa952007-05-23 07:23:16 +00001082
Evan Cheng2acdbcc2007-06-08 19:17:12 +00001083 std::copy(Cond.begin(), Cond.end(), std::back_inserter(BBI.Predicate));
1084
Evan Cheng2c8c3a42007-06-15 07:36:12 +00001085 BBI.IsAnalyzed = false;
1086 BBI.NonPredSize = 0;
1087
Evan Chengb6665f62007-06-04 06:47:22 +00001088 NumIfConvBBs++;
1089}
1090
Evan Cheng2c8c3a42007-06-15 07:36:12 +00001091/// CopyAndPredicateBlock - Copy and predicate instructions from source BB to
1092/// the destination block. Skip end of block branches if IgnoreBr is true.
1093void IfConverter::CopyAndPredicateBlock(BBInfo &ToBBI, BBInfo &FromBBI,
1094 std::vector<MachineOperand> &Cond,
1095 bool IgnoreBr) {
1096 for (MachineBasicBlock::iterator I = FromBBI.BB->begin(),
1097 E = FromBBI.BB->end(); I != E; ++I) {
1098 const TargetInstrDescriptor *TID = I->getInstrDescriptor();
1099 bool isPredicated = TII->isPredicated(I);
1100 // Do not copy the end of the block branches.
1101 if (IgnoreBr && !isPredicated && (TID->Flags & M_BRANCH_FLAG) != 0)
1102 break;
1103
1104 MachineInstr *MI = I->clone();
1105 ToBBI.BB->insert(ToBBI.BB->end(), MI);
1106 ToBBI.NonPredSize++;
1107
1108 if (!isPredicated)
1109 if (!TII->PredicateInstruction(MI, Cond)) {
1110 cerr << "Unable to predicate " << *MI << "!\n";
1111 abort();
1112 }
1113 }
1114
1115 std::copy(FromBBI.Predicate.begin(), FromBBI.Predicate.end(),
1116 std::back_inserter(ToBBI.Predicate));
1117 std::copy(Cond.begin(), Cond.end(), std::back_inserter(ToBBI.Predicate));
1118
1119 ToBBI.ClobbersPred |= FromBBI.ClobbersPred;
1120 ToBBI.IsAnalyzed = false;
1121
1122 NumDupBBs++;
1123}
1124
Evan Cheng86cbfea2007-05-18 00:20:58 +00001125/// MergeBlocks - Move all instructions from FromBB to the end of ToBB.
Evan Cheng4e654852007-05-16 02:00:57 +00001126///
Evan Cheng86cbfea2007-05-18 00:20:58 +00001127void IfConverter::MergeBlocks(BBInfo &ToBBI, BBInfo &FromBBI) {
1128 ToBBI.BB->splice(ToBBI.BB->end(),
1129 FromBBI.BB, FromBBI.BB->begin(), FromBBI.BB->end());
Evan Chenga13aa952007-05-23 07:23:16 +00001130
Evan Chengb6665f62007-06-04 06:47:22 +00001131 // Redirect all branches to FromBB to ToBB.
Evan Chenga1a9f402007-06-05 22:03:53 +00001132 std::vector<MachineBasicBlock *> Preds(FromBBI.BB->pred_begin(),
1133 FromBBI.BB->pred_end());
Evan Chengd4de6d92007-06-07 02:12:15 +00001134 for (unsigned i = 0, e = Preds.size(); i != e; ++i) {
1135 MachineBasicBlock *Pred = Preds[i];
1136 if (Pred == ToBBI.BB)
1137 continue;
1138 Pred->ReplaceUsesOfBlockWith(FromBBI.BB, ToBBI.BB);
1139 }
1140
1141 std::vector<MachineBasicBlock *> Succs(FromBBI.BB->succ_begin(),
1142 FromBBI.BB->succ_end());
Evan Cheng7e75ba82007-06-08 22:01:07 +00001143 MachineBasicBlock *NBB = getNextBlock(FromBBI.BB);
Evan Cheng1c9f91d2007-06-09 01:03:43 +00001144 MachineBasicBlock *FallThrough = FromBBI.HasFallThrough ? NBB : NULL;
Evan Chengb6665f62007-06-04 06:47:22 +00001145
Evan Chengd4de6d92007-06-07 02:12:15 +00001146 for (unsigned i = 0, e = Succs.size(); i != e; ++i) {
1147 MachineBasicBlock *Succ = Succs[i];
Evan Cheng7e75ba82007-06-08 22:01:07 +00001148 // Fallthrough edge can't be transferred.
Evan Chengd4de6d92007-06-07 02:12:15 +00001149 if (Succ == FallThrough)
1150 continue;
1151 FromBBI.BB->removeSuccessor(Succ);
1152 if (!ToBBI.BB->isSuccessor(Succ))
1153 ToBBI.BB->addSuccessor(Succ);
1154 }
1155
Evan Cheng7e75ba82007-06-08 22:01:07 +00001156 // Now FromBBI always fall through to the next block!
1157 if (NBB)
1158 FromBBI.BB->addSuccessor(NBB);
1159
Evan Cheng2c8c3a42007-06-15 07:36:12 +00001160 std::copy(FromBBI.Predicate.begin(), FromBBI.Predicate.end(),
1161 std::back_inserter(ToBBI.Predicate));
1162 FromBBI.Predicate.clear();
1163
Evan Chenga13aa952007-05-23 07:23:16 +00001164 ToBBI.NonPredSize += FromBBI.NonPredSize;
1165 FromBBI.NonPredSize = 0;
Evan Cheng7a655472007-06-06 10:16:17 +00001166
Evan Cheng9618bca2007-06-11 22:26:22 +00001167 ToBBI.ClobbersPred |= FromBBI.ClobbersPred;
Evan Cheng1c9f91d2007-06-09 01:03:43 +00001168 ToBBI.HasFallThrough = FromBBI.HasFallThrough;
Evan Cheng86ff2962007-06-14 20:28:52 +00001169 ToBBI.IsAnalyzed = false;
1170 FromBBI.IsAnalyzed = false;
Evan Cheng4e654852007-05-16 02:00:57 +00001171}