diff options
Diffstat (limited to 'llvm')
| -rw-r--r-- | llvm/lib/Transforms/Scalar/NewGVN.cpp | 8 | ||||
| -rw-r--r-- | llvm/test/Transforms/NewGVN/assumes.ll | 26 |
2 files changed, 30 insertions, 4 deletions
diff --git a/llvm/lib/Transforms/Scalar/NewGVN.cpp b/llvm/lib/Transforms/Scalar/NewGVN.cpp index 0cf9979b40a..94973c986e9 100644 --- a/llvm/lib/Transforms/Scalar/NewGVN.cpp +++ b/llvm/lib/Transforms/Scalar/NewGVN.cpp @@ -1585,11 +1585,11 @@ NewGVN::performSymbolicPredicateInfoEvaluation(Instruction *I) const { SwappedOps ? Cmp->getSwappedPredicate() : Cmp->getPredicate(); if (isa<PredicateAssume>(PI)) { - // If the comparison is true when the operands are equal, then we know the - // operands are equal, because assumes must always be true. - if (CmpInst::isTrueWhenEqual(Predicate)) { + // If we assume the operands are equal, then they are equal. + if (Predicate == CmpInst::ICMP_EQ) { addPredicateUsers(PI, I); - addAdditionalUsers(Cmp->getOperand(0), I); + addAdditionalUsers(SwappedOps ? Cmp->getOperand(1) : Cmp->getOperand(0), + I); return createVariableOrConstant(FirstOp); } } diff --git a/llvm/test/Transforms/NewGVN/assumes.ll b/llvm/test/Transforms/NewGVN/assumes.ll new file mode 100644 index 00000000000..065cc0fb62e --- /dev/null +++ b/llvm/test/Transforms/NewGVN/assumes.ll @@ -0,0 +1,26 @@ +; RUN: opt < %s -newgvn -S | FileCheck %s + +; CHECK-LABEL: @test1 +; CHECK: ret i32 %arg +define i32 @test1(i32 %arg) { + %cmp = icmp sge i32 %arg, 5 + call void @llvm.assume(i1 %cmp) + ret i32 %arg +} + +; CHECK-LABEL: @test2 +; CHECK: ret i32 %arg +define i32 @test2(i32 %arg, i1 %b) { + br label %bb + +bb: + %a = phi i32 [ 1, %0 ], [ 2, %bb ] + %cmp = icmp eq i32 %arg, %a + call void @llvm.assume(i1 %cmp) + br i1 %b, label %bb, label %end + +end: + ret i32 %arg +} + +declare void @llvm.assume(i1 %cond) |
