diff options
-rw-r--r-- | llvm/include/llvm/CodeGen/TailDuplicator.h | 2 | ||||
-rw-r--r-- | llvm/lib/CodeGen/TailDuplication.cpp | 2 | ||||
-rw-r--r-- | llvm/lib/CodeGen/TailDuplicator.cpp | 3 |
3 files changed, 3 insertions, 4 deletions
diff --git a/llvm/include/llvm/CodeGen/TailDuplicator.h b/llvm/include/llvm/CodeGen/TailDuplicator.h index e70029e6de2..03fd4997794 100644 --- a/llvm/include/llvm/CodeGen/TailDuplicator.h +++ b/llvm/include/llvm/CodeGen/TailDuplicator.h @@ -49,7 +49,7 @@ class TailDuplicator { public: /// Prepare to run on a specific machine function. /// @param TailDupSize - Maxmimum size of blocks to tail-duplicate. - void initMF(MachineFunction &MF, const MachineModuleInfo *MMI, + void initMF(MachineFunction &MF, const MachineBranchProbabilityInfo *MBPI, unsigned TailDupSize = 0); bool tailDuplicateBlocks(); diff --git a/llvm/lib/CodeGen/TailDuplication.cpp b/llvm/lib/CodeGen/TailDuplication.cpp index d603400597f..6e8ee9e2634 100644 --- a/llvm/lib/CodeGen/TailDuplication.cpp +++ b/llvm/lib/CodeGen/TailDuplication.cpp @@ -49,7 +49,7 @@ bool TailDuplicatePass::runOnMachineFunction(MachineFunction &MF) { auto MBPI = &getAnalysis<MachineBranchProbabilityInfo>(); - Duplicator.initMF(MF, MMI, MBPI); + Duplicator.initMF(MF, MBPI); bool MadeChange = false; while (Duplicator.tailDuplicateBlocks()) diff --git a/llvm/lib/CodeGen/TailDuplicator.cpp b/llvm/lib/CodeGen/TailDuplicator.cpp index 2a25b3d1dbb..bf3379eb576 100644 --- a/llvm/lib/CodeGen/TailDuplicator.cpp +++ b/llvm/lib/CodeGen/TailDuplicator.cpp @@ -57,14 +57,13 @@ static cl::opt<unsigned> TailDupLimit("tail-dup-limit", cl::init(~0U), namespace llvm { void TailDuplicator::initMF(MachineFunction &MFin, - const MachineModuleInfo *MMIin, const MachineBranchProbabilityInfo *MBPIin, unsigned TailDupSizeIn) { MF = &MFin; TII = MF->getSubtarget().getInstrInfo(); TRI = MF->getSubtarget().getRegisterInfo(); MRI = &MF->getRegInfo(); - MMI = MMIin; + MMI = &MF->getMMI(); MBPI = MBPIin; TailDupSize = TailDupSizeIn; |