Simplify negated bit test

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@170020 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Transforms/InstCombine/InstCombineCompares.cpp b/lib/Transforms/InstCombine/InstCombineCompares.cpp
index 1b96c3cca4ee30142a8106e8f588de9676dea23a..b0fc82f6f66e57ada7e17861aab40a1640a83527 100644 (file)
--- a/lib/Transforms/InstCombine/InstCombineCompares.cpp
+++ b/lib/Transforms/InstCombine/InstCombineCompares.cpp
@@ -2034,6 +2034,15 @@ Instruction *InstCombiner::visitICmpInst(ICmpInst &I) {
                                                CI->countTrailingZeros()));
       }
 
+      // Turn x&~y == 0 into x&y != 0 if x is a power of 2.
+      Value *X = 0, *Y = 0;
+      if (match(Op0, m_And(m_Value(X), m_Not(m_Value(Y)))) &&
+          match(Op1, m_Zero()) && isPowerOfTwo(X, TD)) {
+        return new ICmpInst(ICmpInst::ICMP_NE,
+                            Builder->CreateAnd(X, Y),
+                            Op1);
+      }
+
       break;
     }
     case ICmpInst::ICMP_NE: {
@@ -2071,6 +2080,15 @@ Instruction *InstCombiner::visitICmpInst(ICmpInst &I) {
                                                CI->countTrailingZeros()));
       }
 
+      // Turn x&~y != 0 into x&y == 0 if x is a power of 2.
+      Value *X = 0, *Y = 0;
+      if (match(Op0, m_And(m_Value(X), m_Not(m_Value(Y)))) &&
+          match(Op1, m_Zero()) && isPowerOfTwo(X, TD)) {
+        return new ICmpInst(ICmpInst::ICMP_EQ,
+                            Builder->CreateAnd(X, Y),
+                            Op1);
+      }
+
       break;
     }
     case ICmpInst::ICMP_ULT:

diff --git a/test/Transforms/InstCombine/icmp.ll b/test/Transforms/InstCombine/icmp.ll
index 8e064a4f2fc940e2fb49c406a952b556fb2ec187..d11b6cc756f37869d64183f4a3c75d82358bd26a 100644 (file)
--- a/test/Transforms/InstCombine/icmp.ll
+++ b/test/Transforms/InstCombine/icmp.ll
@@ -677,3 +677,29 @@ define i1 @test66(i64 %A, i64 %B) {
 ; CHECK-NEXT: ret i1 true
   ret i1 %cmp
 }
+
+define i1 @test67(i32 %A, i32 %B) {
+  %neg = xor i32 %A, -1
+  %shl = shl i32 1, %B
+  %and = and i32 %shl, %neg
+  %cmp = icmp ne i32 %and, 0
+; CHECK: @test67
+; CHECK-NEXT: %shl = shl i32 1, %B
+; CHECK-NEXT: %1 = and i32 %shl, %A
+; CHECK-NEXT: %cmp = icmp eq i32 %1, 0
+; CHECK-NEXT: ret i1 %cmp
+  ret i1 %cmp
+}
+
+define i1 @test68(i32 %A, i32 %B) {
+  %neg = xor i32 %A, -1
+  %shl = shl i32 1, %B
+  %and = and i32 %shl, %neg
+  %cmp = icmp eq i32 %and, 0
+; CHECK: @test68
+; CHECK-NEXT: %shl = shl i32 1, %B
+; CHECK-NEXT: %1 = and i32 %shl, %A
+; CHECK-NEXT: %cmp = icmp ne i32 %1, 0
+; CHECK-NEXT: ret i1 %cmp
+  ret i1 %cmp
+}