diff options
Diffstat (limited to 'llvm/lib/CodeGen/PreAllocSplitting.cpp')
| -rw-r--r-- | llvm/lib/CodeGen/PreAllocSplitting.cpp | 12 | 
1 files changed, 11 insertions, 1 deletions
| diff --git a/llvm/lib/CodeGen/PreAllocSplitting.cpp b/llvm/lib/CodeGen/PreAllocSplitting.cpp index 160144599c7..fe1efe5c957 100644 --- a/llvm/lib/CodeGen/PreAllocSplitting.cpp +++ b/llvm/lib/CodeGen/PreAllocSplitting.cpp @@ -442,6 +442,9 @@ PreAllocSplitting::UpdateRegisterInterval(VNInfo *ValNo, unsigned SpillIndex,           SE = MBB->succ_end(); SI != SE; ++SI)      WorkList.push_back(*SI); +  SmallPtrSet<MachineBasicBlock*, 4> ProcessedBlocks; +  ProcessedBlocks.insert(MBB); +    while (!WorkList.empty()) {      MBB = WorkList.back();      WorkList.pop_back(); @@ -459,6 +462,13 @@ PreAllocSplitting::UpdateRegisterInterval(VNInfo *ValNo, unsigned SpillIndex,        }        Processed.insert(LR);      } +     +    ProcessedBlocks.insert(MBB); +    if (LR) +      for (MachineBasicBlock::succ_iterator SI = MBB->succ_begin(), +            SE = MBB->succ_end(); SI != SE; ++SI) +        if (!ProcessedBlocks.count(*SI)) +          WorkList.push_back(*SI);    }    for (LiveInterval::iterator I = CurrLI->begin(), E = CurrLI->end(); @@ -1039,7 +1049,7 @@ bool PreAllocSplitting::runOnMachineFunction(MachineFunction &MF) {    // Make sure blocks are numbered in order.    MF.RenumberBlocks(); -#if 0 +#if 1    // FIXME: Go top down.    MachineBasicBlock *Entry = MF.begin();    SmallPtrSet<MachineBasicBlock*,16> Visited; | 

