llvm/test/Transforms/InstCombine/icmp-with-selects.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3
   4 declare void @use(i32,...)
   5
   6 define i1 @both_sides_fold_slt(i32 %param, i1 %cond) {
   7 ; CHECK-LABEL: define i1 @both_sides_fold_slt
   8 ; CHECK-SAME: (i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
   9 ; CHECK-NEXT:  entry:
  10 ; CHECK-NEXT:    ret i1 false
  11 ;
  12 entry:
  13   %cond1 = select i1 %cond, i32 1, i32 %param
  14   %cond6 = select i1 %cond, i32 9, i32 %param
  15   %cmp = icmp slt i32 %cond6, %cond1
  16   ret i1 %cmp
  17 }
  18
  19 define i1 @both_sides_fold_eq(i32 %param, i1 %cond) {
  20 ; CHECK-LABEL: define i1 @both_sides_fold_eq
  21 ; CHECK-SAME: (i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
  22 ; CHECK-NEXT:  entry:
  23 ; CHECK-NEXT:    [[NOT_COND:%.*]] = xor i1 [[COND]], true
  24 ; CHECK-NEXT:    ret i1 [[NOT_COND]]
  25 ;
  26 entry:
  27   %cond1 = select i1 %cond, i32 1, i32 %param
  28   %cond6 = select i1 %cond, i32 9, i32 %param
  29   %cmp = icmp eq i32 %cond6, %cond1
  30   ret i1 %cmp
  31 }
  32
  33 define i1 @one_side_fold_slt(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
  34 ; CHECK-LABEL: define i1 @one_side_fold_slt
  35 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
  36 ; CHECK-NEXT:  entry:
  37 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp slt i32 [[VAL2]], [[VAL1]]
  38 ; CHECK-NEXT:    [[CMP:%.*]] = select i1 [[COND]], i1 [[TMP0]], i1 false
  39 ; CHECK-NEXT:    ret i1 [[CMP]]
  40 ;
  41 entry:
  42   %cond1 = select i1 %cond, i32 %val1, i32 %param
  43   %cond6 = select i1 %cond, i32 %val2, i32 %param
  44   %cmp = icmp slt i32 %cond6, %cond1
  45   ret i1 %cmp
  46 }
  47
  48 define i1 @one_side_fold_sgt(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
  49 ; CHECK-LABEL: define i1 @one_side_fold_sgt
  50 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
  51 ; CHECK-NEXT:  entry:
  52 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp sgt i32 [[VAL2]], [[VAL1]]
  53 ; CHECK-NEXT:    [[NOT_COND:%.*]] = xor i1 [[COND]], true
  54 ; CHECK-NEXT:    [[CMP:%.*]] = select i1 [[NOT_COND]], i1 [[TMP0]], i1 false
  55 ; CHECK-NEXT:    ret i1 [[CMP]]
  56 ;
  57 entry:
  58   %cond1 = select i1 %cond, i32 %param, i32 %val1
  59   %cond6 = select i1 %cond, i32 %param, i32 %val2
  60   %cmp = icmp sgt i32 %cond6, %cond1
  61   ret i1 %cmp
  62 }
  63
  64 define i1 @one_side_fold_eq(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
  65 ; CHECK-LABEL: define i1 @one_side_fold_eq
  66 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
  67 ; CHECK-NEXT:  entry:
  68 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp eq i32 [[VAL2]], [[VAL1]]
  69 ; CHECK-NEXT:    [[NOT_COND:%.*]] = xor i1 [[COND]], true
  70 ; CHECK-NEXT:    [[CMP:%.*]] = select i1 [[NOT_COND]], i1 true, i1 [[TMP0]]
  71 ; CHECK-NEXT:    ret i1 [[CMP]]
  72 ;
  73 entry:
  74   %cond1 = select i1 %cond, i32 %val1, i32 %param
  75   %cond6 = select i1 %cond, i32 %val2, i32 %param
  76   %cmp = icmp eq i32 %cond6, %cond1
  77   ret i1 %cmp
  78 }
  79
  80 define i1 @no_side_fold_cond(i32 %val1, i32 %val2, i32 %param, i1 %cond1, i1 %cond2) {
  81 ; CHECK-LABEL: define i1 @no_side_fold_cond
  82 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND1:%.*]], i1 [[COND2:%.*]]) {
  83 ; CHECK-NEXT:  entry:
  84 ; CHECK-NEXT:    [[COND3:%.*]] = select i1 [[COND1]], i32 [[VAL1]], i32 [[PARAM]]
  85 ; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND2]], i32 [[VAL2]], i32 [[PARAM]]
  86 ; CHECK-NEXT:    [[CMP:%.*]] = icmp sle i32 [[COND6]], [[COND3]]
  87 ; CHECK-NEXT:    ret i1 [[CMP]]
  88 ;
  89 entry:
  90   %cond3 = select i1 %cond1, i32 %val1, i32 %param
  91   %cond6 = select i1 %cond2, i32 %val2, i32 %param
  92   %cmp = icmp sle i32 %cond6, %cond3
  93   ret i1 %cmp
  94 }
  95
  96 define i1 @no_side_fold_op(i32 %val1, i32 %val2, i32 %val3, i1 %cond) {
  97 ; CHECK-LABEL: define i1 @no_side_fold_op
  98 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[VAL3:%.*]], i1 [[COND:%.*]]) {
  99 ; CHECK-NEXT:  entry:
 100 ; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[VAL2]]
 101 ; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[VAL3]]
 102 ; CHECK-NEXT:    [[CMP:%.*]] = icmp sge i32 [[COND6]], [[COND1]]
 103 ; CHECK-NEXT:    ret i1 [[CMP]]
 104 ;
 105 entry:
 106   %cond1 = select i1 %cond, i32 %val1, i32 %val2
 107   %cond6 = select i1 %cond, i32 %val2, i32 %val3
 108   %cmp = icmp sge i32 %cond6, %cond1
 109   ret i1 %cmp
 110 }
 111
 112 define i1 @one_select_mult_use(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
 113 ; CHECK-LABEL: define i1 @one_select_mult_use
 114 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
 115 ; CHECK-NEXT:  entry:
 116 ; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
 117 ; CHECK-NEXT:    call void @use(i32 [[COND1]])
 118 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp slt i32 [[VAL2]], [[VAL1]]
 119 ; CHECK-NEXT:    [[CMP:%.*]] = select i1 [[COND]], i1 [[TMP0]], i1 false
 120 ; CHECK-NEXT:    ret i1 [[CMP]]
 121 ;
 122 entry:
 123   %cond1 = select i1 %cond, i32 %val1, i32 %param
 124   %cond6 = select i1 %cond, i32 %val2, i32 %param
 125   call void @use(i32 %cond1)
 126   %cmp = icmp slt i32 %cond6, %cond1
 127   ret i1 %cmp
 128 }
 129
 130 define i1 @both_select_mult_use(i32 %val1, i32 %val2, i32 %param, i1 %cond) {
 131 ; CHECK-LABEL: define i1 @both_select_mult_use
 132 ; CHECK-SAME: (i32 [[VAL1:%.*]], i32 [[VAL2:%.*]], i32 [[PARAM:%.*]], i1 [[COND:%.*]]) {
 133 ; CHECK-NEXT:  entry:
 134 ; CHECK-NEXT:    [[COND1:%.*]] = select i1 [[COND]], i32 [[VAL1]], i32 [[PARAM]]
 135 ; CHECK-NEXT:    [[COND6:%.*]] = select i1 [[COND]], i32 [[VAL2]], i32 [[PARAM]]
 136 ; CHECK-NEXT:    call void @use(i32 [[COND1]], i32 [[COND6]])
 137 ; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[COND6]], [[COND1]]
 138 ; CHECK-NEXT:    ret i1 [[CMP]]
 139 ;
 140 entry:
 141   %cond1 = select i1 %cond, i32 %val1, i32 %param
 142   %cond6 = select i1 %cond, i32 %val2, i32 %param
 143   call void @use(i32 %cond1, i32 %cond6)
 144   %cmp = icmp slt i32 %cond6, %cond1
 145   ret i1 %cmp
 146 }
 147
 148 define <4 x i1> @fold_vector_ops(<4 x i32> %val1, <4 x i32> %val2, <4 x i32> %param, i1 %cond) {
 149 ; CHECK-LABEL: define <4 x i1> @fold_vector_ops
 150 ; CHECK-SAME: (<4 x i32> [[VAL1:%.*]], <4 x i32> [[VAL2:%.*]], <4 x i32> [[PARAM:%.*]], i1 [[COND:%.*]]) {
 151 ; CHECK-NEXT:  entry:
 152 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp eq <4 x i32> [[VAL2]], [[VAL1]]
 153 ; CHECK-NEXT:    [[CMP:%.*]] = select i1 [[COND]], <4 x i1> [[TMP0]], <4 x i1> splat (i1 true)
 154 ; CHECK-NEXT:    ret <4 x i1> [[CMP]]
 155 ;
 156 entry:
 157   %cond1 = select i1 %cond, <4 x i32> %val1, <4 x i32> %param
 158   %cond6 = select i1 %cond, <4 x i32> %val2, <4 x i32> %param
 159   %cmp = icmp eq <4 x i32> %cond6, %cond1
 160   ret <4 x i1> %cmp
 161 }
 162
 163 define <8 x i1> @fold_vector_cond_ops(<8 x i32> %val1, <8 x i32> %val2, <8 x i32> %param, <8 x i1> %cond) {
 164 ; CHECK-LABEL: define <8 x i1> @fold_vector_cond_ops
 165 ; CHECK-SAME: (<8 x i32> [[VAL1:%.*]], <8 x i32> [[VAL2:%.*]], <8 x i32> [[PARAM:%.*]], <8 x i1> [[COND:%.*]]) {
 166 ; CHECK-NEXT:  entry:
 167 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp sgt <8 x i32> [[VAL2]], [[VAL1]]
 168 ; CHECK-NEXT:    [[CMP:%.*]] = select <8 x i1> [[COND]], <8 x i1> [[TMP0]], <8 x i1> zeroinitializer
 169 ; CHECK-NEXT:    ret <8 x i1> [[CMP]]
 170 ;
 171 entry:
 172   %cond1 = select <8 x i1> %cond, <8 x i32> %val1, <8 x i32> %param
 173   %cond6 = select <8 x i1> %cond, <8 x i32> %val2, <8 x i32> %param
 174   %cmp = icmp sgt <8 x i32> %cond6, %cond1
 175   ret <8 x i1> %cmp
 176 }