Subversion Repositories QNX 8.QNX8 LLVM/Clang compiler suite

Rev

Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
14 pmbaty 1
//===- Transform/Utils/BasicBlockUtils.h - BasicBlock Utils -----*- C++ -*-===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8
//
9
// This family of functions perform manipulations on basic blocks, and
10
// instructions contained within basic blocks.
11
//
12
//===----------------------------------------------------------------------===//
13
 
14
#ifndef LLVM_TRANSFORMS_UTILS_BASICBLOCKUTILS_H
15
#define LLVM_TRANSFORMS_UTILS_BASICBLOCKUTILS_H
16
 
17
// FIXME: Move to this file: BasicBlock::removePredecessor, BB::splitBasicBlock
18
 
19
#include "llvm/ADT/ArrayRef.h"
20
#include "llvm/ADT/SetVector.h"
21
#include "llvm/IR/BasicBlock.h"
22
#include "llvm/IR/Dominators.h"
23
#include <cassert>
24
 
25
namespace llvm {
26
class BranchInst;
27
class LandingPadInst;
28
class Loop;
29
class PHINode;
30
template <typename PtrType> class SmallPtrSetImpl;
31
class BlockFrequencyInfo;
32
class BranchProbabilityInfo;
33
class DomTreeUpdater;
34
class Function;
35
class LoopInfo;
36
class MDNode;
37
class MemoryDependenceResults;
38
class MemorySSAUpdater;
39
class PostDominatorTree;
40
class ReturnInst;
41
class TargetLibraryInfo;
42
class Value;
43
 
44
/// Replace contents of every block in \p BBs with single unreachable
45
/// instruction. If \p Updates is specified, collect all necessary DT updates
46
/// into this vector. If \p KeepOneInputPHIs is true, one-input Phis in
47
/// successors of blocks being deleted will be preserved.
48
void detachDeadBlocks(ArrayRef <BasicBlock *> BBs,
49
                      SmallVectorImpl<DominatorTree::UpdateType> *Updates,
50
                      bool KeepOneInputPHIs = false);
51
 
52
/// Delete the specified block, which must have no predecessors.
53
void DeleteDeadBlock(BasicBlock *BB, DomTreeUpdater *DTU = nullptr,
54
                     bool KeepOneInputPHIs = false);
55
 
56
/// Delete the specified blocks from \p BB. The set of deleted blocks must have
57
/// no predecessors that are not being deleted themselves. \p BBs must have no
58
/// duplicating blocks. If there are loops among this set of blocks, all
59
/// relevant loop info updates should be done before this function is called.
60
/// If \p KeepOneInputPHIs is true, one-input Phis in successors of blocks
61
/// being deleted will be preserved.
62
void DeleteDeadBlocks(ArrayRef <BasicBlock *> BBs,
63
                      DomTreeUpdater *DTU = nullptr,
64
                      bool KeepOneInputPHIs = false);
65
 
66
/// Delete all basic blocks from \p F that are not reachable from its entry
67
/// node. If \p KeepOneInputPHIs is true, one-input Phis in successors of
68
/// blocks being deleted will be preserved.
69
bool EliminateUnreachableBlocks(Function &F, DomTreeUpdater *DTU = nullptr,
70
                                bool KeepOneInputPHIs = false);
71
 
72
/// We know that BB has one predecessor. If there are any single-entry PHI nodes
73
/// in it, fold them away. This handles the case when all entries to the PHI
74
/// nodes in a block are guaranteed equal, such as when the block has exactly
75
/// one predecessor.
76
bool FoldSingleEntryPHINodes(BasicBlock *BB,
77
                             MemoryDependenceResults *MemDep = nullptr);
78
 
79
/// Examine each PHI in the given block and delete it if it is dead. Also
80
/// recursively delete any operands that become dead as a result. This includes
81
/// tracing the def-use list from the PHI to see if it is ultimately unused or
82
/// if it reaches an unused cycle. Return true if any PHIs were deleted.
83
bool DeleteDeadPHIs(BasicBlock *BB, const TargetLibraryInfo *TLI = nullptr,
84
                    MemorySSAUpdater *MSSAU = nullptr);
85
 
86
/// Attempts to merge a block into its predecessor, if possible. The return
87
/// value indicates success or failure.
88
/// By default do not merge blocks if BB's predecessor has multiple successors.
89
/// If PredecessorWithTwoSuccessors = true, the blocks can only be merged
90
/// if BB's Pred has a branch to BB and to AnotherBB, and BB has a single
91
/// successor Sing. In this case the branch will be updated with Sing instead of
92
/// BB, and BB will still be merged into its predecessor and removed.
93
/// If \p DT is not nullptr, update it directly; in that case, DTU must be
94
/// nullptr.
95
bool MergeBlockIntoPredecessor(BasicBlock *BB, DomTreeUpdater *DTU = nullptr,
96
                               LoopInfo *LI = nullptr,
97
                               MemorySSAUpdater *MSSAU = nullptr,
98
                               MemoryDependenceResults *MemDep = nullptr,
99
                               bool PredecessorWithTwoSuccessors = false,
100
                               DominatorTree *DT = nullptr);
101
 
102
/// Merge block(s) sucessors, if possible. Return true if at least two
103
/// of the blocks were merged together.
104
/// In order to merge, each block must be terminated by an unconditional
105
/// branch. If L is provided, then the blocks merged into their predecessors
106
/// must be in L. In addition, This utility calls on another utility:
107
/// MergeBlockIntoPredecessor. Blocks are successfully merged when the call to
108
/// MergeBlockIntoPredecessor returns true.
109
bool MergeBlockSuccessorsIntoGivenBlocks(
110
    SmallPtrSetImpl<BasicBlock *> &MergeBlocks, Loop *L = nullptr,
111
    DomTreeUpdater *DTU = nullptr, LoopInfo *LI = nullptr);
112
 
113
/// Try to remove redundant dbg.value instructions from given basic block.
114
/// Returns true if at least one instruction was removed. Remove redundant
115
/// pseudo ops when RemovePseudoOp is true.
116
bool RemoveRedundantDbgInstrs(BasicBlock *BB);
117
 
118
/// Replace all uses of an instruction (specified by BI) with a value, then
119
/// remove and delete the original instruction.
120
void ReplaceInstWithValue(BasicBlock::iterator &BI, Value *V);
121
 
122
/// Replace the instruction specified by BI with the instruction specified by I.
123
/// Copies DebugLoc from BI to I, if I doesn't already have a DebugLoc. The
124
/// original instruction is deleted and BI is updated to point to the new
125
/// instruction.
126
void ReplaceInstWithInst(BasicBlock *BB, BasicBlock::iterator &BI,
127
                         Instruction *I);
128
 
129
/// Replace the instruction specified by From with the instruction specified by
130
/// To. Copies DebugLoc from BI to I, if I doesn't already have a DebugLoc.
131
void ReplaceInstWithInst(Instruction *From, Instruction *To);
132
 
133
/// Check if we can prove that all paths starting from this block converge
134
/// to a block that either has a @llvm.experimental.deoptimize call
135
/// prior to its terminating return instruction or is terminated by unreachable.
136
/// All blocks in the traversed sequence must have an unique successor, maybe
137
/// except for the last one.
138
bool IsBlockFollowedByDeoptOrUnreachable(const BasicBlock *BB);
139
 
140
/// Option class for critical edge splitting.
141
///
142
/// This provides a builder interface for overriding the default options used
143
/// during critical edge splitting.
144
struct CriticalEdgeSplittingOptions {
145
  DominatorTree *DT;
146
  PostDominatorTree *PDT;
147
  LoopInfo *LI;
148
  MemorySSAUpdater *MSSAU;
149
  bool MergeIdenticalEdges = false;
150
  bool KeepOneInputPHIs = false;
151
  bool PreserveLCSSA = false;
152
  bool IgnoreUnreachableDests = false;
153
  /// SplitCriticalEdge is guaranteed to preserve loop-simplify form if LI is
154
  /// provided. If it cannot be preserved, no splitting will take place. If it
155
  /// is not set, preserve loop-simplify form if possible.
156
  bool PreserveLoopSimplify = true;
157
 
158
  CriticalEdgeSplittingOptions(DominatorTree *DT = nullptr,
159
                               LoopInfo *LI = nullptr,
160
                               MemorySSAUpdater *MSSAU = nullptr,
161
                               PostDominatorTree *PDT = nullptr)
162
      : DT(DT), PDT(PDT), LI(LI), MSSAU(MSSAU) {}
163
 
164
  CriticalEdgeSplittingOptions &setMergeIdenticalEdges() {
165
    MergeIdenticalEdges = true;
166
    return *this;
167
  }
168
 
169
  CriticalEdgeSplittingOptions &setKeepOneInputPHIs() {
170
    KeepOneInputPHIs = true;
171
    return *this;
172
  }
173
 
174
  CriticalEdgeSplittingOptions &setPreserveLCSSA() {
175
    PreserveLCSSA = true;
176
    return *this;
177
  }
178
 
179
  CriticalEdgeSplittingOptions &setIgnoreUnreachableDests() {
180
    IgnoreUnreachableDests = true;
181
    return *this;
182
  }
183
 
184
  CriticalEdgeSplittingOptions &unsetPreserveLoopSimplify() {
185
    PreserveLoopSimplify = false;
186
    return *this;
187
  }
188
};
189
 
190
/// When a loop exit edge is split, LCSSA form may require new PHIs in the new
191
/// exit block. This function inserts the new PHIs, as needed. Preds is a list
192
/// of preds inside the loop, SplitBB is the new loop exit block, and DestBB is
193
/// the old loop exit, now the successor of SplitBB.
194
void createPHIsForSplitLoopExit(ArrayRef<BasicBlock *> Preds,
195
                                BasicBlock *SplitBB, BasicBlock *DestBB);
196
 
197
/// If this edge is a critical edge, insert a new node to split the critical
198
/// edge. This will update the analyses passed in through the option struct.
199
/// This returns the new block if the edge was split, null otherwise.
200
///
201
/// If MergeIdenticalEdges in the options struct is true (not the default),
202
/// *all* edges from TI to the specified successor will be merged into the same
203
/// critical edge block. This is most commonly interesting with switch
204
/// instructions, which may have many edges to any one destination.  This
205
/// ensures that all edges to that dest go to one block instead of each going
206
/// to a different block, but isn't the standard definition of a "critical
207
/// edge".
208
///
209
/// It is invalid to call this function on a critical edge that starts at an
210
/// IndirectBrInst.  Splitting these edges will almost always create an invalid
211
/// program because the address of the new block won't be the one that is jumped
212
/// to.
213
BasicBlock *SplitCriticalEdge(Instruction *TI, unsigned SuccNum,
214
                              const CriticalEdgeSplittingOptions &Options =
215
                                  CriticalEdgeSplittingOptions(),
216
                              const Twine &BBName = "");
217
 
218
/// If it is known that an edge is critical, SplitKnownCriticalEdge can be
219
/// called directly, rather than calling SplitCriticalEdge first.
220
BasicBlock *SplitKnownCriticalEdge(Instruction *TI, unsigned SuccNum,
221
                                   const CriticalEdgeSplittingOptions &Options =
222
                                       CriticalEdgeSplittingOptions(),
223
                                   const Twine &BBName = "");
224
 
225
/// If an edge from Src to Dst is critical, split the edge and return true,
226
/// otherwise return false. This method requires that there be an edge between
227
/// the two blocks. It updates the analyses passed in the options struct
228
inline BasicBlock *
229
SplitCriticalEdge(BasicBlock *Src, BasicBlock *Dst,
230
                  const CriticalEdgeSplittingOptions &Options =
231
                      CriticalEdgeSplittingOptions()) {
232
  Instruction *TI = Src->getTerminator();
233
  unsigned i = 0;
234
  while (true) {
235
    assert(i != TI->getNumSuccessors() && "Edge doesn't exist!");
236
    if (TI->getSuccessor(i) == Dst)
237
      return SplitCriticalEdge(TI, i, Options);
238
    ++i;
239
  }
240
}
241
 
242
/// Loop over all of the edges in the CFG, breaking critical edges as they are
243
/// found. Returns the number of broken edges.
244
unsigned SplitAllCriticalEdges(Function &F,
245
                               const CriticalEdgeSplittingOptions &Options =
246
                                   CriticalEdgeSplittingOptions());
247
 
248
/// Split the edge connecting the specified blocks, and return the newly created
249
/// basic block between \p From and \p To.
250
BasicBlock *SplitEdge(BasicBlock *From, BasicBlock *To,
251
                      DominatorTree *DT = nullptr, LoopInfo *LI = nullptr,
252
                      MemorySSAUpdater *MSSAU = nullptr,
253
                      const Twine &BBName = "");
254
 
255
/// Sets the unwind edge of an instruction to a particular successor.
256
void setUnwindEdgeTo(Instruction *TI, BasicBlock *Succ);
257
 
258
/// Replaces all uses of OldPred with the NewPred block in all PHINodes in a
259
/// block.
260
void updatePhiNodes(BasicBlock *DestBB, BasicBlock *OldPred,
261
                    BasicBlock *NewPred, PHINode *Until = nullptr);
262
 
263
/// Split the edge connect the specficed blocks in the case that \p Succ is an
264
/// Exception Handling Block
265
BasicBlock *ehAwareSplitEdge(BasicBlock *BB, BasicBlock *Succ,
266
                             LandingPadInst *OriginalPad = nullptr,
267
                             PHINode *LandingPadReplacement = nullptr,
268
                             const CriticalEdgeSplittingOptions &Options =
269
                                 CriticalEdgeSplittingOptions(),
270
                             const Twine &BBName = "");
271
 
272
/// Split the specified block at the specified instruction.
273
///
274
/// If \p Before is true, splitBlockBefore handles the block
275
/// splitting. Otherwise, execution proceeds as described below.
276
///
277
/// Everything before \p SplitPt stays in \p Old and everything starting with \p
278
/// SplitPt moves to a new block. The two blocks are joined by an unconditional
279
/// branch. The new block with name \p BBName is returned.
280
///
281
/// FIXME: deprecated, switch to the DomTreeUpdater-based one.
282
BasicBlock *SplitBlock(BasicBlock *Old, Instruction *SplitPt, DominatorTree *DT,
283
                       LoopInfo *LI = nullptr,
284
                       MemorySSAUpdater *MSSAU = nullptr,
285
                       const Twine &BBName = "", bool Before = false);
286
 
287
/// Split the specified block at the specified instruction.
288
///
289
/// If \p Before is true, splitBlockBefore handles the block
290
/// splitting. Otherwise, execution proceeds as described below.
291
///
292
/// Everything before \p SplitPt stays in \p Old and everything starting with \p
293
/// SplitPt moves to a new block. The two blocks are joined by an unconditional
294
/// branch. The new block with name \p BBName is returned.
295
BasicBlock *SplitBlock(BasicBlock *Old, Instruction *SplitPt,
296
                       DomTreeUpdater *DTU = nullptr, LoopInfo *LI = nullptr,
297
                       MemorySSAUpdater *MSSAU = nullptr,
298
                       const Twine &BBName = "", bool Before = false);
299
 
300
/// Split the specified block at the specified instruction \p SplitPt.
301
/// All instructions before \p SplitPt are moved to a new block and all
302
/// instructions after \p SplitPt stay in the old block. The new block and the
303
/// old block are joined by inserting an unconditional branch to the end of the
304
/// new block. The new block with name \p BBName is returned.
305
BasicBlock *splitBlockBefore(BasicBlock *Old, Instruction *SplitPt,
306
                             DomTreeUpdater *DTU, LoopInfo *LI,
307
                             MemorySSAUpdater *MSSAU, const Twine &BBName = "");
308
 
309
/// This method introduces at least one new basic block into the function and
310
/// moves some of the predecessors of BB to be predecessors of the new block.
311
/// The new predecessors are indicated by the Preds array. The new block is
312
/// given a suffix of 'Suffix'. Returns new basic block to which predecessors
313
/// from Preds are now pointing.
314
///
315
/// If BB is a landingpad block then additional basicblock might be introduced.
316
/// It will have Suffix+".split_lp". See SplitLandingPadPredecessors for more
317
/// details on this case.
318
///
319
/// This currently updates the LLVM IR, DominatorTree, LoopInfo, and LCCSA but
320
/// no other analyses. In particular, it does not preserve LoopSimplify
321
/// (because it's complicated to handle the case where one of the edges being
322
/// split is an exit of a loop with other exits).
323
///
324
/// FIXME: deprecated, switch to the DomTreeUpdater-based one.
325
BasicBlock *SplitBlockPredecessors(BasicBlock *BB, ArrayRef<BasicBlock *> Preds,
326
                                   const char *Suffix, DominatorTree *DT,
327
                                   LoopInfo *LI = nullptr,
328
                                   MemorySSAUpdater *MSSAU = nullptr,
329
                                   bool PreserveLCSSA = false);
330
 
331
/// This method introduces at least one new basic block into the function and
332
/// moves some of the predecessors of BB to be predecessors of the new block.
333
/// The new predecessors are indicated by the Preds array. The new block is
334
/// given a suffix of 'Suffix'. Returns new basic block to which predecessors
335
/// from Preds are now pointing.
336
///
337
/// If BB is a landingpad block then additional basicblock might be introduced.
338
/// It will have Suffix+".split_lp". See SplitLandingPadPredecessors for more
339
/// details on this case.
340
///
341
/// This currently updates the LLVM IR, DominatorTree, LoopInfo, and LCCSA but
342
/// no other analyses. In particular, it does not preserve LoopSimplify
343
/// (because it's complicated to handle the case where one of the edges being
344
/// split is an exit of a loop with other exits).
345
BasicBlock *SplitBlockPredecessors(BasicBlock *BB, ArrayRef<BasicBlock *> Preds,
346
                                   const char *Suffix,
347
                                   DomTreeUpdater *DTU = nullptr,
348
                                   LoopInfo *LI = nullptr,
349
                                   MemorySSAUpdater *MSSAU = nullptr,
350
                                   bool PreserveLCSSA = false);
351
 
352
/// This method transforms the landing pad, OrigBB, by introducing two new basic
353
/// blocks into the function. One of those new basic blocks gets the
354
/// predecessors listed in Preds. The other basic block gets the remaining
355
/// predecessors of OrigBB. The landingpad instruction OrigBB is clone into both
356
/// of the new basic blocks. The new blocks are given the suffixes 'Suffix1' and
357
/// 'Suffix2', and are returned in the NewBBs vector.
358
///
359
/// This currently updates the LLVM IR, DominatorTree, LoopInfo, and LCCSA but
360
/// no other analyses. In particular, it does not preserve LoopSimplify
361
/// (because it's complicated to handle the case where one of the edges being
362
/// split is an exit of a loop with other exits).
363
///
364
/// FIXME: deprecated, switch to the DomTreeUpdater-based one.
365
void SplitLandingPadPredecessors(BasicBlock *OrigBB,
366
                                 ArrayRef<BasicBlock *> Preds,
367
                                 const char *Suffix, const char *Suffix2,
368
                                 SmallVectorImpl<BasicBlock *> &NewBBs,
369
                                 DominatorTree *DT, LoopInfo *LI = nullptr,
370
                                 MemorySSAUpdater *MSSAU = nullptr,
371
                                 bool PreserveLCSSA = false);
372
 
373
/// This method transforms the landing pad, OrigBB, by introducing two new basic
374
/// blocks into the function. One of those new basic blocks gets the
375
/// predecessors listed in Preds. The other basic block gets the remaining
376
/// predecessors of OrigBB. The landingpad instruction OrigBB is clone into both
377
/// of the new basic blocks. The new blocks are given the suffixes 'Suffix1' and
378
/// 'Suffix2', and are returned in the NewBBs vector.
379
///
380
/// This currently updates the LLVM IR, DominatorTree, LoopInfo, and LCCSA but
381
/// no other analyses. In particular, it does not preserve LoopSimplify
382
/// (because it's complicated to handle the case where one of the edges being
383
/// split is an exit of a loop with other exits).
384
void SplitLandingPadPredecessors(
385
    BasicBlock *OrigBB, ArrayRef<BasicBlock *> Preds, const char *Suffix,
386
    const char *Suffix2, SmallVectorImpl<BasicBlock *> &NewBBs,
387
    DomTreeUpdater *DTU = nullptr, LoopInfo *LI = nullptr,
388
    MemorySSAUpdater *MSSAU = nullptr, bool PreserveLCSSA = false);
389
 
390
/// This method duplicates the specified return instruction into a predecessor
391
/// which ends in an unconditional branch. If the return instruction returns a
392
/// value defined by a PHI, propagate the right value into the return. It
393
/// returns the new return instruction in the predecessor.
394
ReturnInst *FoldReturnIntoUncondBranch(ReturnInst *RI, BasicBlock *BB,
395
                                       BasicBlock *Pred,
396
                                       DomTreeUpdater *DTU = nullptr);
397
 
398
/// Split the containing block at the specified instruction - everything before
399
/// SplitBefore stays in the old basic block, and the rest of the instructions
400
/// in the BB are moved to a new block. The two blocks are connected by a
401
/// conditional branch (with value of Cmp being the condition).
402
/// Before:
403
///   Head
404
///   SplitBefore
405
///   Tail
406
/// After:
407
///   Head
408
///   if (Cond)
409
///     ThenBlock
410
///   SplitBefore
411
///   Tail
412
///
413
/// If \p ThenBlock is not specified, a new block will be created for it.
414
/// If \p Unreachable is true, the newly created block will end with
415
/// UnreachableInst, otherwise it branches to Tail.
416
/// Returns the NewBasicBlock's terminator.
417
///
418
/// Updates DT and LI if given.
419
///
420
/// FIXME: deprecated, switch to the DomTreeUpdater-based one.
421
Instruction *SplitBlockAndInsertIfThen(Value *Cond, Instruction *SplitBefore,
422
                                       bool Unreachable, MDNode *BranchWeights,
423
                                       DominatorTree *DT,
424
                                       LoopInfo *LI = nullptr,
425
                                       BasicBlock *ThenBlock = nullptr);
426
 
427
/// Split the containing block at the specified instruction - everything before
428
/// SplitBefore stays in the old basic block, and the rest of the instructions
429
/// in the BB are moved to a new block. The two blocks are connected by a
430
/// conditional branch (with value of Cmp being the condition).
431
/// Before:
432
///   Head
433
///   SplitBefore
434
///   Tail
435
/// After:
436
///   Head
437
///   if (Cond)
438
///     ThenBlock
439
///   SplitBefore
440
///   Tail
441
///
442
/// If \p ThenBlock is not specified, a new block will be created for it.
443
/// If \p Unreachable is true, the newly created block will end with
444
/// UnreachableInst, otherwise it branches to Tail.
445
/// Returns the NewBasicBlock's terminator.
446
///
447
/// Updates DT and LI if given.
448
Instruction *SplitBlockAndInsertIfThen(Value *Cond, Instruction *SplitBefore,
449
                                       bool Unreachable,
450
                                       MDNode *BranchWeights = nullptr,
451
                                       DomTreeUpdater *DTU = nullptr,
452
                                       LoopInfo *LI = nullptr,
453
                                       BasicBlock *ThenBlock = nullptr);
454
 
455
/// SplitBlockAndInsertIfThenElse is similar to SplitBlockAndInsertIfThen,
456
/// but also creates the ElseBlock.
457
/// Before:
458
///   Head
459
///   SplitBefore
460
///   Tail
461
/// After:
462
///   Head
463
///   if (Cond)
464
///     ThenBlock
465
///   else
466
///     ElseBlock
467
///   SplitBefore
468
///   Tail
469
///
470
/// Updates DT if given.
471
void SplitBlockAndInsertIfThenElse(Value *Cond, Instruction *SplitBefore,
472
                                   Instruction **ThenTerm,
473
                                   Instruction **ElseTerm,
474
                                   MDNode *BranchWeights = nullptr,
475
                                   DomTreeUpdater *DTU = nullptr);
476
 
477
/// Check whether BB is the merge point of a if-region.
478
/// If so, return the branch instruction that determines which entry into
479
/// BB will be taken.  Also, return by references the block that will be
480
/// entered from if the condition is true, and the block that will be
481
/// entered if the condition is false.
482
///
483
/// This does no checking to see if the true/false blocks have large or unsavory
484
/// instructions in them.
485
BranchInst *GetIfCondition(BasicBlock *BB, BasicBlock *&IfTrue,
486
                           BasicBlock *&IfFalse);
487
 
488
// Split critical edges where the source of the edge is an indirectbr
489
// instruction. This isn't always possible, but we can handle some easy cases.
490
// This is useful because MI is unable to split such critical edges,
491
// which means it will not be able to sink instructions along those edges.
492
// This is especially painful for indirect branches with many successors, where
493
// we end up having to prepare all outgoing values in the origin block.
494
//
495
// Our normal algorithm for splitting critical edges requires us to update
496
// the outgoing edges of the edge origin block, but for an indirectbr this
497
// is hard, since it would require finding and updating the block addresses
498
// the indirect branch uses. But if a block only has a single indirectbr
499
// predecessor, with the others being regular branches, we can do it in a
500
// different way.
501
// Say we have A -> D, B -> D, I -> D where only I -> D is an indirectbr.
502
// We can split D into D0 and D1, where D0 contains only the PHIs from D,
503
// and D1 is the D block body. We can then duplicate D0 as D0A and D0B, and
504
// create the following structure:
505
// A -> D0A, B -> D0A, I -> D0B, D0A -> D1, D0B -> D1
506
// If BPI and BFI aren't non-null, BPI/BFI will be updated accordingly.
507
// When `IgnoreBlocksWithoutPHI` is set to `true` critical edges leading to a
508
// block without phi-instructions will not be split.
509
bool SplitIndirectBrCriticalEdges(Function &F, bool IgnoreBlocksWithoutPHI,
510
                                  BranchProbabilityInfo *BPI = nullptr,
511
                                  BlockFrequencyInfo *BFI = nullptr);
512
 
513
/// Given a set of incoming and outgoing blocks, create a "hub" such that every
514
/// edge from an incoming block InBB to an outgoing block OutBB is now split
515
/// into two edges, one from InBB to the hub and another from the hub to
516
/// OutBB. The hub consists of a series of guard blocks, one for each outgoing
517
/// block. Each guard block conditionally branches to the corresponding outgoing
518
/// block, or the next guard block in the chain. These guard blocks are returned
519
/// in the argument vector.
520
///
521
/// Since the control flow edges from InBB to OutBB have now been replaced, the
522
/// function also updates any PHINodes in OutBB. For each such PHINode, the
523
/// operands corresponding to incoming blocks are moved to a new PHINode in the
524
/// hub, and the hub is made an operand of the original PHINode.
525
///
526
/// Input CFG:
527
/// ----------
528
///
529
///                    Def
530
///                     |
531
///                     v
532
///           In1      In2
533
///            |        |
534
///            |        |
535
///            v        v
536
///  Foo ---> Out1     Out2
537
///                     |
538
///                     v
539
///                    Use
540
///
541
///
542
/// Create hub: Incoming = {In1, In2}, Outgoing = {Out1, Out2}
543
/// ----------------------------------------------------------
544
///
545
///             Def
546
///              |
547
///              v
548
///  In1        In2          Foo
549
///   |    Hub   |            |
550
///   |    + - - | - - +      |
551
///   |    '     v     '      V
552
///   +------> Guard1 -----> Out1
553
///        '     |     '
554
///        '     v     '
555
///        '   Guard2 -----> Out2
556
///        '           '      |
557
///        + - - - - - +      |
558
///                           v
559
///                          Use
560
///
561
/// Limitations:
562
/// -----------
563
/// 1. This assumes that all terminators in the CFG are direct branches (the
564
///    "br" instruction). The presence of any other control flow such as
565
///    indirectbr, switch or callbr will cause an assert.
566
///
567
/// 2. The updates to the PHINodes are not sufficient to restore SSA
568
///    form. Consider a definition Def, its use Use, incoming block In2 and
569
///    outgoing block Out2, such that:
570
///    a. In2 is reachable from D or contains D.
571
///    b. U is reachable from Out2 or is contained in Out2.
572
///    c. U is not a PHINode if U is contained in Out2.
573
///
574
///    Clearly, Def dominates Out2 since the program is valid SSA. But when the
575
///    hub is introduced, there is a new path through the hub along which Use is
576
///    reachable from entry without passing through Def, and SSA is no longer
577
///    valid. To fix this, we need to look at all the blocks post-dominated by
578
///    the hub on the one hand, and dominated by Out2 on the other. This is left
579
///    for the caller to accomplish, since each specific use of this function
580
///    may have additional information which simplifies this fixup. For example,
581
///    see restoreSSA() in the UnifyLoopExits pass.
582
BasicBlock *CreateControlFlowHub(
583
    DomTreeUpdater *DTU, SmallVectorImpl<BasicBlock *> &GuardBlocks,
584
    const SetVector<BasicBlock *> &Predecessors,
585
    const SetVector<BasicBlock *> &Successors, const StringRef Prefix,
586
    std::optional<unsigned> MaxControlFlowBooleans = std::nullopt);
587
 
588
} // end namespace llvm
589
 
590
#endif // LLVM_TRANSFORMS_UTILS_BASICBLOCKUTILS_H