1 //===-- TailDuplication.cpp - Duplicate blocks into predecessors' tails ---===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This pass duplicates basic blocks ending in unconditional branches into 11 // the tails of their predecessors, using the TailDuplicator utility class. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "llvm/CodeGen/MachineFunctionPass.h" 16 #include "llvm/CodeGen/Passes.h" 17 #include "llvm/CodeGen/TailDuplicator.h" 18 #include "llvm/IR/Function.h" 19 #include "llvm/Support/Debug.h" 20 using namespace llvm; 21 22 #define DEBUG_TYPE "tailduplication" 23 24 namespace { 25 /// Perform tail duplication. Delegates to TailDuplicator 26 class TailDuplicatePass : public MachineFunctionPass { 27 TailDuplicator Duplicator; 28 29 public: 30 static char ID; 31 explicit TailDuplicatePass() : MachineFunctionPass(ID) {} 32 33 bool runOnMachineFunction(MachineFunction &MF) override; 34 35 void getAnalysisUsage(AnalysisUsage &AU) const override; 36 }; 37 38 char TailDuplicatePass::ID = 0; 39 } 40 41 char &llvm::TailDuplicateID = TailDuplicatePass::ID; 42 43 INITIALIZE_PASS(TailDuplicatePass, DEBUG_TYPE, "Tail Duplication", false, false) 44 45 bool TailDuplicatePass::runOnMachineFunction(MachineFunction &MF) { 46 if (skipFunction(*MF.getFunction())) 47 return false; 48 49 auto MBPI = &getAnalysis<MachineBranchProbabilityInfo>(); 50 51 Duplicator.initMF(MF, MBPI, /* LayoutMode */ false); 52 53 bool MadeChange = false; 54 while (Duplicator.tailDuplicateBlocks()) 55 MadeChange = true; 56 57 return MadeChange; 58 } 59 60 void TailDuplicatePass::getAnalysisUsage(AnalysisUsage &AU) const { 61 AU.addRequired<MachineBranchProbabilityInfo>(); 62 MachineFunctionPass::getAnalysisUsage(AU); 63 } 64