test/Transforms/CodeGenPrepare/AArch64/widen_switch.ll

   1 ;; AArch64 is arbitralily chosen as a 32/64-bit RISC representative to show the transform in all tests.
   2
   3 ; RUN: opt < %s -codegenprepare -S -mtriple=aarch64-unknown-unknown | FileCheck %s --check-prefix=ARM64
   4
   5 ; AArch64 widens to 32-bit.
   6
   7 define i32 @widen_switch_i16(i32 %a)  {
   8 entry:
   9   %trunc = trunc i32 %a to i16
  10   switch i16 %trunc, label %sw.default [
  11     i16 1, label %sw.bb0
  12     i16 -1, label %sw.bb1
  13   ]
  14
  15 sw.bb0:
  16   br label %return
  17
  18 sw.bb1:
  19   br label %return
  20
  21 sw.default:
  22   br label %return
  23
  24 return:
  25   %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  26   ret i32 %retval
  27
  28 ; ARM64-LABEL: @widen_switch_i16(
  29 ; ARM64:       %0 = zext i16 %trunc to i32
  30 ; ARM64-NEXT:  switch i32 %0, label %sw.default [
  31 ; ARM64-NEXT:    i32 1, label %return
  32 ; ARM64-NEXT:    i32 65535, label %sw.bb1
  33 }
  34
  35 ; Widen to 32-bit from a smaller, non-native type.
  36
  37 define i32 @widen_switch_i17(i32 %a)  {
  38 entry:
  39   %trunc = trunc i32 %a to i17
  40   switch i17 %trunc, label %sw.default [
  41     i17 10, label %sw.bb0
  42     i17 -1, label %sw.bb1
  43   ]
  44
  45 sw.bb0:
  46   br label %return
  47
  48 sw.bb1:
  49   br label %return
  50
  51 sw.default:
  52   br label %return
  53
  54 return:
  55   %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  56   ret i32 %retval
  57
  58 ; ARM64-LABEL: @widen_switch_i17(
  59 ; ARM64:       %0 = zext i17 %trunc to i32
  60 ; ARM64-NEXT:  switch i32 %0, label %sw.default [
  61 ; ARM64-NEXT:    i32 10, label %return
  62 ; ARM64-NEXT:    i32 131071, label %sw.bb1
  63 }
  64
  65 ; If the switch condition is a sign-extended function argument, then the
  66 ; condition and cases should be sign-extended rather than zero-extended
  67 ; because the sign-extension can be optimized away.
  68
  69 define i32 @widen_switch_i16_sext(i2 signext %a)  {
  70 entry:
  71   switch i2 %a, label %sw.default [
  72     i2 1, label %sw.bb0
  73     i2 -1, label %sw.bb1
  74   ]
  75
  76 sw.bb0:
  77   br label %return
  78
  79 sw.bb1:
  80   br label %return
  81
  82 sw.default:
  83   br label %return
  84
  85 return:
  86   %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  87   ret i32 %retval
  88
  89 ; ARM64-LABEL: @widen_switch_i16_sext(
  90 ; ARM64:       %0 = sext i2 %a to i32
  91 ; ARM64-NEXT:  switch i32 %0, label %sw.default [
  92 ; ARM64-NEXT:    i32 1, label %return
  93 ; ARM64-NEXT:    i32 -1, label %sw.bb1
  94 }
  95