summaryrefslogtreecommitdiffstats
path: root/llvm/test/Transforms/CodeGenPrepare/AArch64/widen_switch.ll
blob: cae1cc54a9615c6641b86fd13420dceebaec9c0a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
;; AArch64 is arbitralily chosen as a 32/64-bit RISC representative to show the transform in all tests.

; RUN: opt < %s -codegenprepare -S -mtriple=aarch64-unknown-unknown | FileCheck %s --check-prefix=ARM64

; AArch64 widens to 32-bit.

define i32 @widen_switch_i16(i32 %a)  {
entry:
  %trunc = trunc i32 %a to i16
  switch i16 %trunc, label %sw.default [
    i16 1, label %sw.bb0
    i16 -1, label %sw.bb1
  ]

sw.bb0:
  br label %return

sw.bb1:
  br label %return

sw.default:
  br label %return

return:
  %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  ret i32 %retval

; ARM64-LABEL: @widen_switch_i16(
; ARM64:       %0 = zext i16 %trunc to i32
; ARM64-NEXT:  switch i32 %0, label %sw.default [
; ARM64-NEXT:    i32 1, label %sw.bb0
; ARM64-NEXT:    i32 65535, label %sw.bb1
}

; Widen to 32-bit from a smaller, non-native type.

define i32 @widen_switch_i17(i32 %a)  {
entry:
  %trunc = trunc i32 %a to i17
  switch i17 %trunc, label %sw.default [
    i17 10, label %sw.bb0
    i17 -1, label %sw.bb1
  ]

sw.bb0:
  br label %return

sw.bb1:
  br label %return

sw.default:
  br label %return

return:
  %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  ret i32 %retval

; ARM64-LABEL: @widen_switch_i17(
; ARM64:       %0 = zext i17 %trunc to i32
; ARM64-NEXT:  switch i32 %0, label %sw.default [
; ARM64-NEXT:    i32 10, label %sw.bb0
; ARM64-NEXT:    i32 131071, label %sw.bb1
}

; If the switch condition is a sign-extended function argument, then the
; condition and cases should be sign-extended rather than zero-extended
; because the sign-extension can be optimized away.

define i32 @widen_switch_i16_sext(i2 signext %a)  {
entry:
  switch i2 %a, label %sw.default [
    i2 1, label %sw.bb0
    i2 -1, label %sw.bb1
  ]

sw.bb0:
  br label %return

sw.bb1:
  br label %return

sw.default:
  br label %return

return:
  %retval = phi i32 [ -1, %sw.default ], [ 0, %sw.bb0 ], [ 1, %sw.bb1 ]
  ret i32 %retval

; ARM64-LABEL: @widen_switch_i16_sext(
; ARM64:       %0 = sext i2 %a to i32
; ARM64-NEXT:  switch i32 %0, label %sw.default [
; ARM64-NEXT:    i32 1, label %sw.bb0
; ARM64-NEXT:    i32 -1, label %sw.bb1
}

OpenPOWER on IntegriCloud