test/Transforms/SimplifyCFG/rangereduce.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -simplifycfg -switch-to-lookup -S | FileCheck %s
   3 ; RUN: opt < %s -passes='simplify-cfg<switch-to-lookup>' -S | FileCheck %s
   4
   5 target datalayout = "e-n32"
   6
   7 define i32 @test1(i32 %a) {
   8 ; CHECK-LABEL: @test1(
   9 ; CHECK-NEXT:    [[TMP1:%.*]] = sub i32 [[A:%.*]], 97
  10 ; CHECK-NEXT:    [[TMP2:%.*]] = lshr i32 [[TMP1]], 2
  11 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i32 [[TMP1]], 30
  12 ; CHECK-NEXT:    [[TMP4:%.*]] = or i32 [[TMP2]], [[TMP3]]
  13 ; CHECK-NEXT:    switch i32 [[TMP4]], label [[DEF:%.*]] [
  14 ; CHECK-NEXT:    i32 0, label [[ONE:%.*]]
  15 ; CHECK-NEXT:    i32 1, label [[TWO:%.*]]
  16 ; CHECK-NEXT:    i32 2, label [[THREE:%.*]]
  17 ; CHECK-NEXT:    i32 3, label [[THREE]]
  18 ; CHECK-NEXT:    ]
  19 ; CHECK:       def:
  20 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
  21 ; CHECK-NEXT:    ret i32 [[MERGE]]
  22 ; CHECK:       one:
  23 ; CHECK-NEXT:    br label [[DEF]]
  24 ; CHECK:       two:
  25 ; CHECK-NEXT:    br label [[DEF]]
  26 ; CHECK:       three:
  27 ; CHECK-NEXT:    br label [[DEF]]
  28 ;
  29   switch i32 %a, label %def [
  30   i32 97, label %one
  31   i32 101, label %two
  32   i32 105, label %three
  33   i32 109, label %three
  34   ]
  35
  36 def:
  37   ret i32 8867
  38
  39 one:
  40   ret i32 11984
  41 two:
  42   ret i32 1143
  43 three:
  44   ret i32 99783
  45 }
  46
  47 ; Optimization shouldn't trigger; bitwidth > 64
  48 define i128 @test2(i128 %a) {
  49 ; CHECK-LABEL: @test2(
  50 ; CHECK-NEXT:    switch i128 [[A:%.*]], label [[DEF:%.*]] [
  51 ; CHECK-NEXT:    i128 97, label [[ONE:%.*]]
  52 ; CHECK-NEXT:    i128 101, label [[TWO:%.*]]
  53 ; CHECK-NEXT:    i128 105, label [[THREE:%.*]]
  54 ; CHECK-NEXT:    i128 109, label [[THREE]]
  55 ; CHECK-NEXT:    ]
  56 ; CHECK:       def:
  57 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i128 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
  58 ; CHECK-NEXT:    ret i128 [[MERGE]]
  59 ; CHECK:       one:
  60 ; CHECK-NEXT:    br label [[DEF]]
  61 ; CHECK:       two:
  62 ; CHECK-NEXT:    br label [[DEF]]
  63 ; CHECK:       three:
  64 ; CHECK-NEXT:    br label [[DEF]]
  65 ;
  66   switch i128 %a, label %def [
  67   i128 97, label %one
  68   i128 101, label %two
  69   i128 105, label %three
  70   i128 109, label %three
  71   ]
  72
  73 def:
  74   ret i128 8867
  75
  76 one:
  77   ret i128 11984
  78 two:
  79   ret i128 1143
  80 three:
  81   ret i128 99783
  82 }
  83
  84 ; Optimization shouldn't trigger; no holes present
  85 define i32 @test3(i32 %a) {
  86 ; CHECK-LABEL: @test3(
  87 ; CHECK-NEXT:    switch i32 [[A:%.*]], label [[DEF:%.*]] [
  88 ; CHECK-NEXT:    i32 97, label [[ONE:%.*]]
  89 ; CHECK-NEXT:    i32 98, label [[TWO:%.*]]
  90 ; CHECK-NEXT:    i32 99, label [[THREE:%.*]]
  91 ; CHECK-NEXT:    ]
  92 ; CHECK:       def:
  93 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
  94 ; CHECK-NEXT:    ret i32 [[MERGE]]
  95 ; CHECK:       one:
  96 ; CHECK-NEXT:    br label [[DEF]]
  97 ; CHECK:       two:
  98 ; CHECK-NEXT:    br label [[DEF]]
  99 ; CHECK:       three:
 100 ; CHECK-NEXT:    br label [[DEF]]
 101 ;
 102   switch i32 %a, label %def [
 103   i32 97, label %one
 104   i32 98, label %two
 105   i32 99, label %three
 106   ]
 107
 108 def:
 109   ret i32 8867
 110
 111 one:
 112   ret i32 11984
 113 two:
 114   ret i32 1143
 115 three:
 116   ret i32 99783
 117 }
 118
 119 ; Optimization shouldn't trigger; not an arithmetic progression
 120 define i32 @test4(i32 %a) {
 121 ; CHECK-LABEL: @test4(
 122 ; CHECK-NEXT:    switch i32 [[A:%.*]], label [[DEF:%.*]] [
 123 ; CHECK-NEXT:    i32 97, label [[ONE:%.*]]
 124 ; CHECK-NEXT:    i32 102, label [[TWO:%.*]]
 125 ; CHECK-NEXT:    i32 105, label [[THREE:%.*]]
 126 ; CHECK-NEXT:    i32 109, label [[THREE]]
 127 ; CHECK-NEXT:    ]
 128 ; CHECK:       def:
 129 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
 130 ; CHECK-NEXT:    ret i32 [[MERGE]]
 131 ; CHECK:       one:
 132 ; CHECK-NEXT:    br label [[DEF]]
 133 ; CHECK:       two:
 134 ; CHECK-NEXT:    br label [[DEF]]
 135 ; CHECK:       three:
 136 ; CHECK-NEXT:    br label [[DEF]]
 137 ;
 138   switch i32 %a, label %def [
 139   i32 97, label %one
 140   i32 102, label %two
 141   i32 105, label %three
 142   i32 109, label %three
 143   ]
 144
 145 def:
 146   ret i32 8867
 147
 148 one:
 149   ret i32 11984
 150 two:
 151   ret i32 1143
 152 three:
 153   ret i32 99783
 154 }
 155
 156 ; Optimization shouldn't trigger; not a power of two
 157 define i32 @test5(i32 %a) {
 158 ; CHECK-LABEL: @test5(
 159 ; CHECK-NEXT:    switch i32 [[A:%.*]], label [[DEF:%.*]] [
 160 ; CHECK-NEXT:    i32 97, label [[ONE:%.*]]
 161 ; CHECK-NEXT:    i32 102, label [[TWO:%.*]]
 162 ; CHECK-NEXT:    i32 107, label [[THREE:%.*]]
 163 ; CHECK-NEXT:    i32 112, label [[THREE]]
 164 ; CHECK-NEXT:    ]
 165 ; CHECK:       def:
 166 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
 167 ; CHECK-NEXT:    ret i32 [[MERGE]]
 168 ; CHECK:       one:
 169 ; CHECK-NEXT:    br label [[DEF]]
 170 ; CHECK:       two:
 171 ; CHECK-NEXT:    br label [[DEF]]
 172 ; CHECK:       three:
 173 ; CHECK-NEXT:    br label [[DEF]]
 174 ;
 175   switch i32 %a, label %def [
 176   i32 97, label %one
 177   i32 102, label %two
 178   i32 107, label %three
 179   i32 112, label %three
 180   ]
 181
 182 def:
 183   ret i32 8867
 184
 185 one:
 186   ret i32 11984
 187 two:
 188   ret i32 1143
 189 three:
 190   ret i32 99783
 191 }
 192
 193 define i32 @test6(i32 %a) optsize {
 194 ; CHECK-LABEL: @test6(
 195 ; CHECK-NEXT:    [[TMP1:%.*]] = sub i32 [[A:%.*]], -109
 196 ; CHECK-NEXT:    [[TMP2:%.*]] = lshr i32 [[TMP1]], 2
 197 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i32 [[TMP1]], 30
 198 ; CHECK-NEXT:    [[TMP4:%.*]] = or i32 [[TMP2]], [[TMP3]]
 199 ; CHECK-NEXT:    switch i32 [[TMP4]], label [[DEF:%.*]] [
 200 ; CHECK-NEXT:    i32 3, label [[ONE:%.*]]
 201 ; CHECK-NEXT:    i32 2, label [[TWO:%.*]]
 202 ; CHECK-NEXT:    i32 1, label [[THREE:%.*]]
 203 ; CHECK-NEXT:    i32 0, label [[THREE]]
 204 ; CHECK-NEXT:    ]
 205 ; CHECK:       def:
 206 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
 207 ; CHECK-NEXT:    ret i32 [[MERGE]]
 208 ; CHECK:       one:
 209 ; CHECK-NEXT:    br label [[DEF]]
 210 ; CHECK:       two:
 211 ; CHECK-NEXT:    br label [[DEF]]
 212 ; CHECK:       three:
 213 ; CHECK-NEXT:    br label [[DEF]]
 214 ;
 215   switch i32 %a, label %def [
 216   i32 -97, label %one
 217   i32 -101, label %two
 218   i32 -105, label %three
 219   i32 -109, label %three
 220   ]
 221
 222 def:
 223   ret i32 8867
 224
 225 one:
 226   ret i32 11984
 227 two:
 228   ret i32 1143
 229 three:
 230   ret i32 99783
 231 }
 232
 233 define i8 @test7(i8 %a) optsize {
 234 ; CHECK-LABEL: @test7(
 235 ; CHECK-NEXT:    [[TMP1:%.*]] = sub i8 [[A:%.*]], -36
 236 ; CHECK-NEXT:    [[TMP2:%.*]] = lshr i8 [[TMP1]], 2
 237 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i8 [[TMP1]], 6
 238 ; CHECK-NEXT:    [[TMP4:%.*]] = or i8 [[TMP2]], [[TMP3]]
 239 ; CHECK-NEXT:    [[TMP5:%.*]] = icmp ult i8 [[TMP4]], 4
 240 ; CHECK-NEXT:    br i1 [[TMP5]], label [[SWITCH_LOOKUP:%.*]], label [[DEF:%.*]]
 241 ; CHECK:       switch.lookup:
 242 ; CHECK-NEXT:    [[SWITCH_CAST:%.*]] = zext i8 [[TMP4]] to i32
 243 ; CHECK-NEXT:    [[SWITCH_SHIFTAMT:%.*]] = mul i32 [[SWITCH_CAST]], 8
 244 ; CHECK-NEXT:    [[SWITCH_DOWNSHIFT:%.*]] = lshr i32 -943228976, [[SWITCH_SHIFTAMT]]
 245 ; CHECK-NEXT:    [[SWITCH_MASKED:%.*]] = trunc i32 [[SWITCH_DOWNSHIFT]] to i8
 246 ; CHECK-NEXT:    ret i8 [[SWITCH_MASKED]]
 247 ; CHECK:       def:
 248 ; CHECK-NEXT:    ret i8 -93
 249 ;
 250   switch i8 %a, label %def [
 251   i8 220, label %one
 252   i8 224, label %two
 253   i8 228, label %three
 254   i8 232, label %three
 255   ]
 256
 257 def:
 258   ret i8 8867
 259
 260 one:
 261   ret i8 11984
 262 two:
 263   ret i8 1143
 264 three:
 265   ret i8 99783
 266 }
 267
 268 define i32 @test8(i32 %a) optsize {
 269 ; CHECK-LABEL: @test8(
 270 ; CHECK-NEXT:    [[TMP1:%.*]] = sub i32 [[A:%.*]], 97
 271 ; CHECK-NEXT:    [[TMP2:%.*]] = lshr i32 [[TMP1]], 2
 272 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i32 [[TMP1]], 30
 273 ; CHECK-NEXT:    [[TMP4:%.*]] = or i32 [[TMP2]], [[TMP3]]
 274 ; CHECK-NEXT:    switch i32 [[TMP4]], label [[DEF:%.*]] [
 275 ; CHECK-NEXT:    i32 0, label [[ONE:%.*]]
 276 ; CHECK-NEXT:    i32 1, label [[TWO:%.*]]
 277 ; CHECK-NEXT:    i32 2, label [[THREE:%.*]]
 278 ; CHECK-NEXT:    i32 4, label [[THREE]]
 279 ; CHECK-NEXT:    ]
 280 ; CHECK:       def:
 281 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
 282 ; CHECK-NEXT:    ret i32 [[MERGE]]
 283 ; CHECK:       one:
 284 ; CHECK-NEXT:    br label [[DEF]]
 285 ; CHECK:       two:
 286 ; CHECK-NEXT:    br label [[DEF]]
 287 ; CHECK:       three:
 288 ; CHECK-NEXT:    br label [[DEF]]
 289 ;
 290   switch i32 %a, label %def [
 291   i32 97, label %one
 292   i32 101, label %two
 293   i32 105, label %three
 294   i32 113, label %three
 295   ]
 296
 297 def:
 298   ret i32 8867
 299
 300 one:
 301   ret i32 11984
 302 two:
 303   ret i32 1143
 304 three:
 305   ret i32 99783
 306 }
 307
 308 define i32 @test9(i32 %a) {
 309 ; CHECK-LABEL: @test9(
 310 ; CHECK-NEXT:    [[TMP1:%.*]] = sub i32 [[A:%.*]], 6
 311 ; CHECK-NEXT:    [[TMP2:%.*]] = lshr i32 [[TMP1]], 1
 312 ; CHECK-NEXT:    [[TMP3:%.*]] = shl i32 [[TMP1]], 31
 313 ; CHECK-NEXT:    [[TMP4:%.*]] = or i32 [[TMP2]], [[TMP3]]
 314 ; CHECK-NEXT:    switch i32 [[TMP4]], label [[DEF:%.*]] [
 315 ; CHECK-NEXT:    i32 6, label [[ONE:%.*]]
 316 ; CHECK-NEXT:    i32 7, label [[TWO:%.*]]
 317 ; CHECK-NEXT:    i32 0, label [[THREE:%.*]]
 318 ; CHECK-NEXT:    i32 2, label [[THREE]]
 319 ; CHECK-NEXT:    ]
 320 ; CHECK:       def:
 321 ; CHECK-NEXT:    [[MERGE:%.*]] = phi i32 [ 8867, [[TMP0:%.*]] ], [ 11984, [[ONE]] ], [ 1143, [[TWO]] ], [ 99783, [[THREE]] ]
 322 ; CHECK-NEXT:    ret i32 [[MERGE]]
 323 ; CHECK:       one:
 324 ; CHECK-NEXT:    br label [[DEF]]
 325 ; CHECK:       two:
 326 ; CHECK-NEXT:    br label [[DEF]]
 327 ; CHECK:       three:
 328 ; CHECK-NEXT:    br label [[DEF]]
 329 ;
 330   switch i32 %a, label %def [
 331   i32 18, label %one
 332   i32 20, label %two
 333   i32 6, label %three
 334   i32 10, label %three
 335   ]
 336
 337 def:
 338   ret i32 8867
 339
 340 one:
 341   ret i32 11984
 342 two:
 343   ret i32 1143
 344 three:
 345   ret i32 99783
 346 }
 347