[llvm] 73bebc9 - [InstSimplify] add tests for folding bool select to logic; NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2020-04-10T09:08:00-04:00
New Revision: 73bebc944543aa93c0e4b559712318f8b00c1c20

URL: https://github.com/llvm/llvm-project/commit/73bebc944543aa93c0e4b559712318f8b00c1c20
DIFF: https://github.com/llvm/llvm-project/commit/73bebc944543aa93c0e4b559712318f8b00c1c20.diff

LOG: [InstSimplify] add tests for folding bool select to logic; NFC

Added: 
    

Modified: 
    llvm/test/Transforms/InstSimplify/select.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstSimplify/select.ll b/llvm/test/Transforms/InstSimplify/select.ll
index b3f215a44754..bafcb96461ff 100644
--- a/llvm/test/Transforms/InstSimplify/select.ll
+++ b/llvm/test/Transforms/InstSimplify/select.ll
@@ -646,3 +646,94 @@ define i32* @select_icmp_ne_0_gep_operand(i32* %base, i64 %n) {
   %r = select i1 %cond, i32* %gep, i32* %base
   ret i32* %r
 }
+
+define i1 @and_cmps(i32 %x) {
+; CHECK-LABEL: @and_cmps(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[X:%.*]], 92
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp slt i32 [[X]], 11
+; CHECK-NEXT:    [[R:%.*]] = select i1 [[CMP1]], i1 [[CMP2]], i1 false
+; CHECK-NEXT:    ret i1 [[R]]
+;
+  %cmp1 = icmp slt i32 %x, 92
+  %cmp2 = icmp slt i32 %x, 11
+  %r = select i1 %cmp1, i1 %cmp2, i1 false
+  ret i1 %r
+}
+
+define <2 x i1> @and_cmps_vector(<2 x i32> %x) {
+; CHECK-LABEL: @and_cmps_vector(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt <2 x i32> [[X:%.*]], <i32 92, i32 92>
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp slt <2 x i32> [[X]], <i32 11, i32 11>
+; CHECK-NEXT:    [[R:%.*]] = select <2 x i1> [[CMP1]], <2 x i1> [[CMP2]], <2 x i1> zeroinitializer
+; CHECK-NEXT:    ret <2 x i1> [[R]]
+;
+  %cmp1 = icmp slt <2 x i32> %x, <i32 92, i32 92>
+  %cmp2 = icmp slt <2 x i32> %x, <i32 11, i32 11>
+  %r = select <2 x i1> %cmp1, <2 x i1> %cmp2, <2 x i1> <i1 false, i1 false>
+  ret <2 x i1> %r
+}
+
+define i1 @or_cmps(float %x) {
+; CHECK-LABEL: @or_cmps(
+; CHECK-NEXT:    [[CMP1:%.*]] = fcmp uno float [[X:%.*]], 4.200000e+01
+; CHECK-NEXT:    [[CMP2:%.*]] = fcmp uno float [[X]], 5.200000e+01
+; CHECK-NEXT:    [[R:%.*]] = select i1 [[CMP1]], i1 true, i1 [[CMP2]]
+; CHECK-NEXT:    ret i1 [[R]]
+;
+  %cmp1 = fcmp uno float %x, 42.0
+  %cmp2 = fcmp uno float %x, 52.0
+  %r = select i1 %cmp1, i1 true, i1 %cmp2
+  ret i1 %r
+}
+
+define <2 x i1> @or_logic_vector(<2 x i1> %x, <2 x i1> %y) {
+; CHECK-LABEL: @or_logic_vector(
+; CHECK-NEXT:    [[A:%.*]] = and <2 x i1> [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT:    [[R:%.*]] = select <2 x i1> [[X]], <2 x i1> <i1 true, i1 true>, <2 x i1> [[A]]
+; CHECK-NEXT:    ret <2 x i1> [[R]]
+;
+  %a = and <2 x i1> %x, %y
+  %r = select <2 x i1> %x, <2 x i1> <i1 true, i1 true>, <2 x i1> %a
+  ret <2 x i1> %r
+}
+
+define i1 @and_not_cmps(i32 %x) {
+; CHECK-LABEL: @and_not_cmps(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[X:%.*]], 92
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp slt i32 [[X]], 11
+; CHECK-NEXT:    [[R:%.*]] = select i1 [[CMP1]], i1 false, i1 [[CMP2]]
+; CHECK-NEXT:    ret i1 [[R]]
+;
+  %cmp1 = icmp slt i32 %x, 92
+  %cmp2 = icmp slt i32 %x, 11
+  %r = select i1 %cmp1, i1 false, i1 %cmp2
+  ret i1 %r
+}
+
+define i1 @or_not_cmps(i32 %x) {
+; CHECK-LABEL: @or_not_cmps(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[X:%.*]], 92
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp slt i32 [[X]], 11
+; CHECK-NEXT:    [[R:%.*]] = select i1 [[CMP1]], i1 [[CMP2]], i1 true
+; CHECK-NEXT:    ret i1 [[R]]
+;
+  %cmp1 = icmp slt i32 %x, 92
+  %cmp2 = icmp slt i32 %x, 11
+  %r = select i1 %cmp1, i1 %cmp2, i1 true
+  ret i1 %r
+}
+
+define i8 @and_cmps_wrong_type(i32 %x) {
+; CHECK-LABEL: @and_cmps_wrong_type(
+; CHECK-NEXT:    [[CMP1:%.*]] = icmp slt i32 [[X:%.*]], 92
+; CHECK-NEXT:    [[CMP2:%.*]] = icmp slt i32 [[X]], 11
+; CHECK-NEXT:    [[S:%.*]] = sext i1 [[CMP2]] to i8
+; CHECK-NEXT:    [[R:%.*]] = select i1 [[CMP1]], i8 [[S]], i8 0
+; CHECK-NEXT:    ret i8 [[R]]
+;
+  %cmp1 = icmp slt i32 %x, 92
+  %cmp2 = icmp slt i32 %x, 11
+  %s = sext i1 %cmp2 to i8
+  %r = select i1 %cmp1, i8 %s, i8 0
+  ret i8 %r
+}