diff --git a/llvm/lib/Analysis/LazyValueInfo.cpp b/llvm/lib/Analysis/LazyValueInfo.cpp index 7ee35b212d33..ec2c798e6c6e 100644 --- a/llvm/lib/Analysis/LazyValueInfo.cpp +++ b/llvm/lib/Analysis/LazyValueInfo.cpp @@ -954,6 +954,7 @@ bool LazyValueInfoCache::getEdgeValue(Value *Val, BasicBlock *BBFrom, // Try to intersect ranges of the BB and the constraint on the edge. LVILatticeVal InBlock = getBlockValue(Val, BBFrom); + mergeAssumeBlockValueConstantRange(Val, InBlock, BBFrom->getTerminator()); mergeAssumeBlockValueConstantRange(Val, InBlock, CxtI); if (!InBlock.isConstantRange()) return true; @@ -971,6 +972,7 @@ bool LazyValueInfoCache::getEdgeValue(Value *Val, BasicBlock *BBFrom, // if we couldn't compute the value on the edge, use the value from the BB Result = getBlockValue(Val, BBFrom); + mergeAssumeBlockValueConstantRange(Val, Result, BBFrom->getTerminator()); mergeAssumeBlockValueConstantRange(Val, Result, CxtI); return true; } diff --git a/llvm/test/Transforms/JumpThreading/assume-edge-dom.ll b/llvm/test/Transforms/JumpThreading/assume-edge-dom.ll new file mode 100644 index 000000000000..f1d0f41e2505 --- /dev/null +++ b/llvm/test/Transforms/JumpThreading/assume-edge-dom.ll @@ -0,0 +1,39 @@ +; RUN: opt -S -jump-threading < %s | FileCheck %s + +declare i8* @escape() +declare void @llvm.assume(i1) + +define i1 @test1(i1 %cond) { +entry: + br i1 %cond, label %taken, label %not_taken + +; CHECK-LABEL: @test1 +; CHECK: br i1 %cond, label %no, label %yes +; CHECK: ret i1 true + +taken: + %res1 = call i8* @escape() + %a = icmp eq i8* %res1, null + tail call void @llvm.assume(i1 %a) + br label %done +not_taken: + %res2 = call i8* @escape() + %b = icmp ne i8* %res2, null + tail call void @llvm.assume(i1 %b) + br label %done + +; An assume that can be used to simplify this comparison dominates each +; predecessor branch (although no assume dominates the cmp itself). Make sure +; this still can be simplified. + +done: + %res = phi i8* [ %res1, %taken ], [ %res2, %not_taken ] + %cnd = icmp ne i8* %res, null + br i1 %cnd, label %yes, label %no + +yes: + ret i1 true +no: + ret i1 false +} +