diff --git a/llvm/lib/CodeGen/PreAllocSplitting.cpp b/llvm/lib/CodeGen/PreAllocSplitting.cpp index 456be09ecd39cb82e65b4c32dfbd14f7da64b6d8..547c528ca2397463c417b49845bc90d8333bdeaf 100644 --- a/llvm/lib/CodeGen/PreAllocSplitting.cpp +++ b/llvm/lib/CodeGen/PreAllocSplitting.cpp @@ -122,7 +122,7 @@ namespace { SmallVector&, SmallPtrSet&); - void ShrinkWrapLiveInterval(VNInfo*, MachineBasicBlock*, + void ShrinkWrapLiveInterval(VNInfo*, MachineBasicBlock*, MachineBasicBlock*, MachineBasicBlock*, SmallPtrSet&, DenseMap >&, DenseMap >&, @@ -426,15 +426,29 @@ PreAllocSplitting::ShrinkWrapToLastUse(MachineBasicBlock *MBB, /// chain to find the new 'kills' and shrink wrap the live interval to the /// new kill indices. void -PreAllocSplitting::ShrinkWrapLiveInterval(VNInfo *ValNo, - MachineBasicBlock *MBB, MachineBasicBlock *DefMBB, +PreAllocSplitting::ShrinkWrapLiveInterval(VNInfo *ValNo, MachineBasicBlock *MBB, + MachineBasicBlock *SuccMBB, MachineBasicBlock *DefMBB, SmallPtrSet &Visited, DenseMap > &Uses, DenseMap > &UseMIs, SmallVector &UseMBBs) { - if (!Visited.insert(MBB)) + if (Visited.count(MBB)) return; + // If live interval is live in another successor path, then we can't process + // this block. But we may able to do so after all the successors have been + // processed. + for (MachineBasicBlock::succ_iterator SI = MBB->succ_begin(), + SE = MBB->succ_end(); SI != SE; ++SI) { + MachineBasicBlock *SMBB = *SI; + if (SMBB == SuccMBB) + continue; + if (CurrLI->liveAt(LIs->getMBBStartIdx(SMBB))) + return; + } + + Visited.insert(MBB); + DenseMap >::iterator UMII = Uses.find(MBB); if (UMII != Uses.end()) { @@ -480,7 +494,8 @@ PreAllocSplitting::ShrinkWrapLiveInterval(VNInfo *ValNo, // Pred is the def bb and the def reaches other val#s, we must // allow the value to be live out of the bb. continue; - ShrinkWrapLiveInterval(ValNo, Pred, DefMBB, Visited, Uses, UseMIs, UseMBBs); + ShrinkWrapLiveInterval(ValNo, Pred, MBB, DefMBB, Visited, + Uses, UseMIs, UseMBBs); } return; @@ -622,7 +637,7 @@ bool PreAllocSplitting::SplitRegLiveInterval(LiveInterval *LI) { // Walk up the predecessor chains. SmallPtrSet Visited; - ShrinkWrapLiveInterval(ValNo, BarrierMBB, DefMBB, Visited, + ShrinkWrapLiveInterval(ValNo, BarrierMBB, NULL, DefMBB, Visited, Uses, UseMIs, UseMBBs); // Remove live range from barrier to the restore. FIXME: Find a better diff --git a/llvm/test/CodeGen/X86/pre-split6.ll b/llvm/test/CodeGen/X86/pre-split6.ll new file mode 100644 index 0000000000000000000000000000000000000000..c075b2a39e7ef0f4056af2d554ba0e0de6a8fa34 --- /dev/null +++ b/llvm/test/CodeGen/X86/pre-split6.ll @@ -0,0 +1,36 @@ +; RUN: llvm-as < %s | llc -march=x86 -mattr=+sse2 -pre-alloc-split + +@current_surfaces.b = external global i1 ; [#uses=1] + +declare double @sin(double) nounwind readonly + +declare double @asin(double) nounwind readonly + +define fastcc void @trace_line(i32 %line) nounwind { +entry: + %.b3 = load i1* @current_surfaces.b ; [#uses=1] + br i1 %.b3, label %bb.nph, label %return + +bb.nph: ; preds = %entry + %0 = load double* null, align 8 ; [#uses=1] + %1 = load double* null, align 8 ; [#uses=2] + %2 = fcmp une double %0, 0.000000e+00 ; [#uses=1] + br i1 %2, label %bb9.i, label %bb13.i + +bb9.i: ; preds = %bb.nph + %3 = tail call double @asin(double 0.000000e+00) nounwind readonly ; [#uses=0] + %4 = fdiv double 1.000000e+00, %1 ; [#uses=1] + %5 = mul double %4, 0.000000e+00 ; [#uses=1] + %6 = tail call double @asin(double %5) nounwind readonly ; [#uses=0] + unreachable + +bb13.i: ; preds = %bb.nph + %7 = fdiv double 1.000000e+00, %1 ; [#uses=1] + %8 = tail call double @sin(double 0.000000e+00) nounwind readonly ; [#uses=1] + %9 = mul double %7, %8 ; [#uses=1] + %10 = tail call double @asin(double %9) nounwind readonly ; [#uses=0] + unreachable + +return: ; preds = %entry + ret void +}