[llvm] r281270 - add more tests for PR30273

[Sanjay Patel via llvm-commits]

Author: spatel
Date: Mon Sep 12 17:28:29 2016
New Revision: 281270

URL: http://llvm.org/viewvc/llvm-project?rev=281270&view=rev
Log:
add more tests for PR30273

Modified:
    llvm/trunk/test/Transforms/InstCombine/zext-bool-add-sub.ll

Modified: llvm/trunk/test/Transforms/InstCombine/zext-bool-add-sub.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/InstCombine/zext-bool-add-sub.ll?rev=281270&r1=281269&r2=281270&view=diff
==============================================================================
--- llvm/trunk/test/Transforms/InstCombine/zext-bool-add-sub.ll (original)
+++ llvm/trunk/test/Transforms/InstCombine/zext-bool-add-sub.ll Mon Sep 12 17:28:29 2016
@@ -18,8 +18,8 @@ define i32 @a(i1 zeroext %x, i1 zeroext
   ret i32 %add
 }
 
-define i32 @PR30273(i1 %a, i1 %b) {
-; CHECK-LABEL: @PR30273(
+define i32 @PR30273_select(i1 %a, i1 %b) {
+; CHECK-LABEL: @PR30273_select(
 ; CHECK-NEXT:    [[ZEXT:%.*]] = zext i1 %a to i32
 ; CHECK-NEXT:    [[SEL1:%.*]] = select i1 %a, i32 2, i32 1
 ; CHECK-NEXT:    [[SEL2:%.*]] = select i1 %b, i32 [[SEL1]], i32 [[ZEXT]]
@@ -31,3 +31,33 @@ define i32 @PR30273(i1 %a, i1 %b) {
   ret i32 %sel2
 }
 
+define i32 @PR30273_zext_add(i1 %a, i1 %b) {
+; CHECK-LABEL: @PR30273_zext_add(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i1 %a to i32
+; CHECK-NEXT:    [[CONV3:%.*]] = zext i1 %b to i32
+; CHECK-NEXT:    [[ADD:%.*]] = add nuw nsw i32 [[CONV3]], [[CONV]]
+; CHECK-NEXT:    ret i32 [[ADD]]
+;
+  %conv = zext i1 %a to i32
+  %conv3 = zext i1 %b to i32
+  %add = add nuw nsw i32 %conv3, %conv
+  ret i32 %add
+}
+
+define i32 @PR30273_three_bools(i1 %x, i1 %y, i1 %z) {
+; CHECK-LABEL: @PR30273_three_bools(
+; CHECK-NEXT:    [[FROMBOOL:%.*]] = zext i1 %x to i32
+; CHECK-NEXT:    [[ADD1:%.*]] = select i1 %x, i32 2, i32 1
+; CHECK-NEXT:    [[SEL1:%.*]] = select i1 %y, i32 [[ADD1]], i32 [[FROMBOOL]]
+; CHECK-NEXT:    [[ADD2:%.*]] = zext i1 %z to i32
+; CHECK-NEXT:    [[SEL2:%.*]] = add nuw nsw i32 [[SEL1]], [[ADD2]]
+; CHECK-NEXT:    ret i32 [[SEL2]]
+;
+  %frombool = zext i1 %x to i32
+  %add1 = add nsw i32 %frombool, 1
+  %sel1 = select i1 %y, i32 %add1, i32 %frombool
+  %add2 = add nsw i32 %sel1, 1
+  %sel2 = select i1 %z, i32 %add2, i32 %sel1
+  ret i32 %sel2
+}
+