[InstCombine] Add tests for icmp of select fold (NFC)

For D150360.

diff --git a/llvm/test/Transforms/InstCombine/icmp-with-selects.ll b/llvm/test/Transforms/InstCombine/icmp-with-selects.ll
new file mode 100644 (file)
index 0000000..540ecce
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/icmp-with-selects.ll
@@ -0,0 +1,184 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
+; RUN: opt < %s -passes=instcombine -S | FileCheck %s
+
+declare void @use(i32,...)
+
+define i1 @both_sides_fold_slt(i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @both_sides_fold_slt
+; CHECK-SAME: (i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 1, i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 9, i32 [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 1, i32 %param
+  %cond6 = select i1 %cond, i32 9, i32 %param
+  %cmp = icmp slt i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @both_sides_fold_eq(i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @both_sides_fold_eq
+; CHECK-SAME: (i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 1, i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 9, i32 [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 1, i32 %param
+  %cond6 = select i1 %cond, i32 9, i32 %param
+  %cmp = icmp eq i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @one_side_fold_slt(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @one_side_fold_slt
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %val1, i32 %param
+  %cond6 = select i1 %cond, i32 %val2, i32 %param
+  %cmp = icmp slt i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @one_side_fold_sgt(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @one_side_fold_sgt
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[PARAM]], i32 [[VAL1]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[PARAM]], i32 [[VAL2]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp sgt i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %param, i32 %val1
+  %cond6 = select i1 %cond, i32 %param, i32 %val2
+  %cmp = icmp sgt i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @one_side_fold_eq(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @one_side_fold_eq
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %val1, i32 %param
+  %cond6 = select i1 %cond, i32 %val2, i32 %param
+  %cmp = icmp eq i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @no_side_fold_cond(i32 %val1, i32 %val2, i32 %param, i1 %cond1, i1 %cond2) {
+; CHECK-LABEL: define i1 @no_side_fold_cond
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND1:%.*]], i1 [[COND2:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND3:%.*]] = select i1 [[COND1]], i32 [[VAL1]], i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND2]], i32 [[VAL2]], i32 [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp sle i32 [[COND6]], [[COND3]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond3 = select i1 %cond1, i32 %val1, i32 %param
+  %cond6 = select i1 %cond2, i32 %val2, i32 %param
+  %cmp = icmp sle i32 %cond6, %cond3
+  ret i1 %cmp
+}
+
+define i1 @no_side_fold_op(i32 %val1, i32 %val2, i32 %val3, i1 %cond) {
+; CHECK-LABEL: define i1 @no_side_fold_op
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[VAL3:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[VAL2]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[VAL3]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp sge i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %val1, i32 %val2
+  %cond6 = select i1 %cond, i32 %val2, i32 %val3
+  %cmp = icmp sge i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @one_select_mult_use(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @one_select_mult_use
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[PARAM]]
+; CHECK-NEXT:    call void @use(i32 [[COND1]])
+; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %val1, i32 %param
+  %cond6 = select i1 %cond, i32 %val2, i32 %param
+  call void @use(i32 %cond1)
+  %cmp = icmp slt i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define i1 @both_select_mult_use(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
+; CHECK-LABEL: define i1 @both_select_mult_use
+; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[PARAM]]
+; CHECK-NEXT:    call void @use(i32 [[COND1]], i32 [[COND6]])
+; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, i32 %val1, i32 %param
+  %cond6 = select i1 %cond, i32 %val2, i32 %param
+  call void @use(i32 %cond1, i32 %cond6)
+  %cmp = icmp slt i32 %cond6, %cond1
+  ret i1 %cmp
+}
+
+define <4 x i1> @fold_vector_ops(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %param, i1 %cond) {
+; CHECK-LABEL: define <4 x i1> @fold_vector_ops
+; CHECK-SAME: (<4 x i32> [[VAL1:%.*]], <4 x i32> [[VAL2:%.*]], <4 x i32> [[PARAM:%.*]], i1 [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], <4 x i32> [[VAL1]], <4 x i32> [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], <4 x i32> [[VAL2]], <4 x i32> [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp eq <4 x i32> [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret <4 x i1> [[CMP]]
+;
+entry:
+  %cond1 = select i1 %cond, <4 x i32> %val1, <4 x i32> %param
+  %cond6 = select i1 %cond, <4 x i32> %val2, <4 x i32> %param
+  %cmp = icmp eq <4 x i32> %cond6, %cond1
+  ret <4 x i1> %cmp
+}
+
+define <8 x i1> @fold_vector_cond_ops(<8 x i32> %val1, <8 x i32> %val2, <8 x i32> %param, <8 x i1> %cond) {
+; CHECK-LABEL: define <8 x i1> @fold_vector_cond_ops
+; CHECK-SAME: (<8 x i32> [[VAL1:%.*]], <8 x i32> [[VAL2:%.*]], <8 x i32> [[PARAM:%.*]], <8 x i1> [[COND:%.*]]) {
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[COND1:%.*]] = select <8 x i1> [[COND]], <8 x i32> [[VAL1]], <8 x i32> [[PARAM]]
+; CHECK-NEXT:    [[COND6:%.*]] = select <8 x i1> [[COND]], <8 x i32> [[VAL2]], <8 x i32> [[PARAM]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp sgt <8 x i32> [[COND6]], [[COND1]]
+; CHECK-NEXT:    ret <8 x i1> [[CMP]]
+;
+entry:
+  %cond1 = select <8 x i1> %cond, <8 x i32> %val1, <8 x i32> %param
+  %cond6 = select <8 x i1> %cond, <8 x i32> %val2, <8 x i32> %param
+  %cmp = icmp sgt <8 x i32> %cond6, %cond1
+  ret <8 x i1> %cmp
+}