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) DTCriticalEdgeSplittingOptions162 : DT(DT), PDT(PDT), LI(LI), MSSAU(MSSAU) {} 163 setMergeIdenticalEdgesCriticalEdgeSplittingOptions164 CriticalEdgeSplittingOptions &setMergeIdenticalEdges() { 165 MergeIdenticalEdges = true; 166 return *this; 167 } 168 setKeepOneInputPHIsCriticalEdgeSplittingOptions169 CriticalEdgeSplittingOptions &setKeepOneInputPHIs() { 170 KeepOneInputPHIs = true; 171 return *this; 172 } 173 setPreserveLCSSACriticalEdgeSplittingOptions174 CriticalEdgeSplittingOptions &setPreserveLCSSA() { 175 PreserveLCSSA = true; 176 return *this; 177 } 178 setIgnoreUnreachableDestsCriticalEdgeSplittingOptions179 CriticalEdgeSplittingOptions &setIgnoreUnreachableDests() { 180 IgnoreUnreachableDests = true; 181 return *this; 182 } 183 unsetPreserveLoopSimplifyCriticalEdgeSplittingOptions184 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 591