[NFC][RemoveDIs] Prefer iterators over inst-pointers in InstCombine
[llvm-project.git] / llvm / test / CodeGen / AArch64 / arm64-shifted-sext.ll
blobda6499b7daa82e55a8cf38a10a5bfc18976163ff
1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2 ; RUN: llc < %s -mtriple=arm64-apple-ios | FileCheck %s
4 ; <rdar://problem/13820218>
6 define signext i16 @extendedLeftShiftcharToshortBy4(i8 signext %a) nounwind readnone ssp {
7 ; CHECK-LABEL: extendedLeftShiftcharToshortBy4:
8 ; CHECK:       ; %bb.0: ; %entry
9 ; CHECK-NEXT:    add w8, w0, #1
10 ; CHECK-NEXT:    sbfiz w0, w8, #4, #8
11 ; CHECK-NEXT:    ret
12 entry:
13   %inc = add i8 %a, 1
14   %conv1 = sext i8 %inc to i32
15   %shl = shl nsw i32 %conv1, 4
16   %conv2 = trunc i32 %shl to i16
17   ret i16 %conv2
20 define signext i16 @extendedRightShiftcharToshortBy4(i8 signext %a) nounwind readnone ssp {
21 ; CHECK-LABEL: extendedRightShiftcharToshortBy4:
22 ; CHECK:       ; %bb.0: ; %entry
23 ; CHECK-NEXT:    add w8, w0, #1
24 ; CHECK-NEXT:    sbfx w0, w8, #4, #4
25 ; CHECK-NEXT:    ret
26 entry:
27   %inc = add i8 %a, 1
28   %conv1 = sext i8 %inc to i32
29   %shr4 = lshr i32 %conv1, 4
30   %conv2 = trunc i32 %shr4 to i16
31   ret i16 %conv2
34 define signext i16 @extendedLeftShiftcharToshortBy8(i8 signext %a) nounwind readnone ssp {
35 ; CHECK-LABEL: extendedLeftShiftcharToshortBy8:
36 ; CHECK:       ; %bb.0: ; %entry
37 ; CHECK-NEXT:    add w8, w0, #1
38 ; CHECK-NEXT:    sbfiz w0, w8, #8, #8
39 ; CHECK-NEXT:    ret
40 entry:
41   %inc = add i8 %a, 1
42   %conv1 = sext i8 %inc to i32
43   %shl = shl nsw i32 %conv1, 8
44   %conv2 = trunc i32 %shl to i16
45   ret i16 %conv2
48 define signext i16 @extendedRightShiftcharToshortBy8(i8 signext %a) nounwind readnone ssp {
49 ; CHECK-LABEL: extendedRightShiftcharToshortBy8:
50 ; CHECK:       ; %bb.0: ; %entry
51 ; CHECK-NEXT:    add w8, w0, #1
52 ; CHECK-NEXT:    sxtb w8, w8
53 ; CHECK-NEXT:    asr w0, w8, #8
54 ; CHECK-NEXT:    ret
55 entry:
56   %inc = add i8 %a, 1
57   %conv1 = sext i8 %inc to i32
58   %shr4 = lshr i32 %conv1, 8
59   %conv2 = trunc i32 %shr4 to i16
60   ret i16 %conv2
63 define i32 @extendedLeftShiftcharTointBy4(i8 signext %a) nounwind readnone ssp {
64 ; CHECK-LABEL: extendedLeftShiftcharTointBy4:
65 ; CHECK:       ; %bb.0: ; %entry
66 ; CHECK-NEXT:    add w8, w0, #1
67 ; CHECK-NEXT:    sbfiz w0, w8, #4, #8
68 ; CHECK-NEXT:    ret
69 entry:
70   %inc = add i8 %a, 1
71   %conv = sext i8 %inc to i32
72   %shl = shl nsw i32 %conv, 4
73   ret i32 %shl
76 define i32 @extendedRightShiftcharTointBy4(i8 signext %a) nounwind readnone ssp {
77 ; CHECK-LABEL: extendedRightShiftcharTointBy4:
78 ; CHECK:       ; %bb.0: ; %entry
79 ; CHECK-NEXT:    add w8, w0, #1
80 ; CHECK-NEXT:    sbfx w0, w8, #4, #4
81 ; CHECK-NEXT:    ret
82 entry:
83   %inc = add i8 %a, 1
84   %conv = sext i8 %inc to i32
85   %shr = ashr i32 %conv, 4
86   ret i32 %shr
89 define i32 @extendedLeftShiftcharTointBy8(i8 signext %a) nounwind readnone ssp {
90 ; CHECK-LABEL: extendedLeftShiftcharTointBy8:
91 ; CHECK:       ; %bb.0: ; %entry
92 ; CHECK-NEXT:    add w8, w0, #1
93 ; CHECK-NEXT:    sbfiz w0, w8, #8, #8
94 ; CHECK-NEXT:    ret
95 entry:
96   %inc = add i8 %a, 1
97   %conv = sext i8 %inc to i32
98   %shl = shl nsw i32 %conv, 8
99   ret i32 %shl
102 define i32 @extendedRightShiftcharTointBy8(i8 signext %a) nounwind readnone ssp {
103 ; CHECK-LABEL: extendedRightShiftcharTointBy8:
104 ; CHECK:       ; %bb.0: ; %entry
105 ; CHECK-NEXT:    add w8, w0, #1
106 ; CHECK-NEXT:    sxtb w8, w8
107 ; CHECK-NEXT:    asr w0, w8, #8
108 ; CHECK-NEXT:    ret
109 entry:
110   %inc = add i8 %a, 1
111   %conv = sext i8 %inc to i32
112   %shr = ashr i32 %conv, 8
113   ret i32 %shr
116 define i64 @extendedLeftShiftcharToint64By4(i8 signext %a) nounwind readnone ssp {
117 ; CHECK-LABEL: extendedLeftShiftcharToint64By4:
118 ; CHECK:       ; %bb.0: ; %entry
119 ; CHECK-NEXT:    add w8, w0, #1
120 ; CHECK-NEXT:    sbfiz x0, x8, #4, #8
121 ; CHECK-NEXT:    ret
122 entry:
123   %inc = add i8 %a, 1
124   %conv = sext i8 %inc to i64
125   %shl = shl nsw i64 %conv, 4
126   ret i64 %shl
129 define i64 @extendedRightShiftcharToint64By4(i8 signext %a) nounwind readnone ssp {
130 ; CHECK-LABEL: extendedRightShiftcharToint64By4:
131 ; CHECK:       ; %bb.0: ; %entry
132 ; CHECK-NEXT:    add w8, w0, #1
133 ; CHECK-NEXT:    sbfx x0, x8, #4, #4
134 ; CHECK-NEXT:    ret
135 entry:
136   %inc = add i8 %a, 1
137   %conv = sext i8 %inc to i64
138   %shr = ashr i64 %conv, 4
139   ret i64 %shr
142 define i64 @extendedLeftShiftcharToint64By8(i8 signext %a) nounwind readnone ssp {
143 ; CHECK-LABEL: extendedLeftShiftcharToint64By8:
144 ; CHECK:       ; %bb.0: ; %entry
145 ; CHECK-NEXT:    add w8, w0, #1
146 ; CHECK-NEXT:    sbfiz x0, x8, #8, #8
147 ; CHECK-NEXT:    ret
148 entry:
149   %inc = add i8 %a, 1
150   %conv = sext i8 %inc to i64
151   %shl = shl nsw i64 %conv, 8
152   ret i64 %shl
155 define i64 @extendedRightShiftcharToint64By8(i8 signext %a) nounwind readnone ssp {
156 ; CHECK-LABEL: extendedRightShiftcharToint64By8:
157 ; CHECK:       ; %bb.0: ; %entry
158 ; CHECK-NEXT:    add w8, w0, #1
159 ; CHECK-NEXT:    sxtb x8, w8
160 ; CHECK-NEXT:    asr x0, x8, #8
161 ; CHECK-NEXT:    ret
162 entry:
163   %inc = add i8 %a, 1
164   %conv = sext i8 %inc to i64
165   %shr = ashr i64 %conv, 8
166   ret i64 %shr
169 define i32 @extendedLeftShiftshortTointBy4(i16 signext %a) nounwind readnone ssp {
170 ; CHECK-LABEL: extendedLeftShiftshortTointBy4:
171 ; CHECK:       ; %bb.0: ; %entry
172 ; CHECK-NEXT:    add w8, w0, #1
173 ; CHECK-NEXT:    sbfiz w0, w8, #4, #16
174 ; CHECK-NEXT:    ret
175 entry:
176   %inc = add i16 %a, 1
177   %conv = sext i16 %inc to i32
178   %shl = shl nsw i32 %conv, 4
179   ret i32 %shl
182 define i32 @extendedRightShiftshortTointBy4(i16 signext %a) nounwind readnone ssp {
183 ; CHECK-LABEL: extendedRightShiftshortTointBy4:
184 ; CHECK:       ; %bb.0: ; %entry
185 ; CHECK-NEXT:    add w8, w0, #1
186 ; CHECK-NEXT:    sbfx w0, w8, #4, #12
187 ; CHECK-NEXT:    ret
188 entry:
189   %inc = add i16 %a, 1
190   %conv = sext i16 %inc to i32
191   %shr = ashr i32 %conv, 4
192   ret i32 %shr
195 define i32 @extendedLeftShiftshortTointBy16(i16 signext %a) nounwind readnone ssp {
196 ; CHECK-LABEL: extendedLeftShiftshortTointBy16:
197 ; CHECK:       ; %bb.0: ; %entry
198 ; CHECK-NEXT:    lsl w8, w0, #16
199 ; CHECK-NEXT:    add w0, w8, #16, lsl #12 ; =65536
200 ; CHECK-NEXT:    ret
201 entry:
202   %inc = add i16 %a, 1
203   %conv2 = zext i16 %inc to i32
204   %shl = shl nuw i32 %conv2, 16
205   ret i32 %shl
208 define i32 @extendedRightShiftshortTointBy16(i16 signext %a) nounwind readnone ssp {
209 ; CHECK-LABEL: extendedRightShiftshortTointBy16:
210 ; CHECK:       ; %bb.0: ; %entry
211 ; CHECK-NEXT:    add w8, w0, #1
212 ; CHECK-NEXT:    sxth w8, w8
213 ; CHECK-NEXT:    asr w0, w8, #16
214 ; CHECK-NEXT:    ret
215 entry:
216   %inc = add i16 %a, 1
217   %conv = sext i16 %inc to i32
218   %shr = ashr i32 %conv, 16
219   ret i32 %shr
222 define i64 @extendedLeftShiftshortToint64By4(i16 signext %a) nounwind readnone ssp {
223 ; CHECK-LABEL: extendedLeftShiftshortToint64By4:
224 ; CHECK:       ; %bb.0: ; %entry
225 ; CHECK-NEXT:    add w8, w0, #1
226 ; CHECK-NEXT:    sbfiz x0, x8, #4, #16
227 ; CHECK-NEXT:    ret
228 entry:
229   %inc = add i16 %a, 1
230   %conv = sext i16 %inc to i64
231   %shl = shl nsw i64 %conv, 4
232   ret i64 %shl
235 define i64 @extendedRightShiftshortToint64By4(i16 signext %a) nounwind readnone ssp {
236 ; CHECK-LABEL: extendedRightShiftshortToint64By4:
237 ; CHECK:       ; %bb.0: ; %entry
238 ; CHECK-NEXT:    add w8, w0, #1
239 ; CHECK-NEXT:    sbfx x0, x8, #4, #12
240 ; CHECK-NEXT:    ret
241 entry:
242   %inc = add i16 %a, 1
243   %conv = sext i16 %inc to i64
244   %shr = ashr i64 %conv, 4
245   ret i64 %shr
248 define i64 @extendedLeftShiftshortToint64By16(i16 signext %a) nounwind readnone ssp {
249 ; CHECK-LABEL: extendedLeftShiftshortToint64By16:
250 ; CHECK:       ; %bb.0: ; %entry
251 ; CHECK-NEXT:    add w8, w0, #1
252 ; CHECK-NEXT:    sbfiz x0, x8, #16, #16
253 ; CHECK-NEXT:    ret
254 entry:
255   %inc = add i16 %a, 1
256   %conv = sext i16 %inc to i64
257   %shl = shl nsw i64 %conv, 16
258   ret i64 %shl
261 define i64 @extendedRightShiftshortToint64By16(i16 signext %a) nounwind readnone ssp {
262 ; CHECK-LABEL: extendedRightShiftshortToint64By16:
263 ; CHECK:       ; %bb.0: ; %entry
264 ; CHECK-NEXT:    add w8, w0, #1
265 ; CHECK-NEXT:    sxth x8, w8
266 ; CHECK-NEXT:    asr x0, x8, #16
267 ; CHECK-NEXT:    ret
268 entry:
269   %inc = add i16 %a, 1
270   %conv = sext i16 %inc to i64
271   %shr = ashr i64 %conv, 16
272   ret i64 %shr
275 define i64 @extendedLeftShiftintToint64By4(i32 %a) nounwind readnone ssp {
276 ; CHECK-LABEL: extendedLeftShiftintToint64By4:
277 ; CHECK:       ; %bb.0: ; %entry
278 ; CHECK-NEXT:    ; kill: def $w0 killed $w0 def $x0
279 ; CHECK-NEXT:    sbfiz x8, x0, #4, #32
280 ; CHECK-NEXT:    add x0, x8, #16
281 ; CHECK-NEXT:    ret
282 entry:
283   %inc = add nsw i32 %a, 1
284   %conv = sext i32 %inc to i64
285   %shl = shl nsw i64 %conv, 4
286   ret i64 %shl
289 define i64 @extendedRightShiftintToint64By4(i32 %a) nounwind readnone ssp {
290 ; CHECK-LABEL: extendedRightShiftintToint64By4:
291 ; CHECK:       ; %bb.0: ; %entry
292 ; CHECK-NEXT:    add w8, w0, #1
293 ; CHECK-NEXT:    sbfx x0, x8, #4, #28
294 ; CHECK-NEXT:    ret
295 entry:
296   %inc = add nsw i32 %a, 1
297   %conv = sext i32 %inc to i64
298   %shr = ashr i64 %conv, 4
299   ret i64 %shr
302 define i64 @extendedLeftShiftintToint64By32(i32 %a) nounwind readnone ssp {
303 ; CHECK-LABEL: extendedLeftShiftintToint64By32:
304 ; CHECK:       ; %bb.0: ; %entry
305 ; CHECK-NEXT:    add w8, w0, #1
306 ; CHECK-NEXT:    lsl x0, x8, #32
307 ; CHECK-NEXT:    ret
308 entry:
309   %inc = add nsw i32 %a, 1
310   %conv2 = zext i32 %inc to i64
311   %shl = shl nuw i64 %conv2, 32
312   ret i64 %shl
315 define i64 @extendedRightShiftintToint64By32(i32 %a) nounwind readnone ssp {
316 ; CHECK-LABEL: extendedRightShiftintToint64By32:
317 ; CHECK:       ; %bb.0: ; %entry
318 ; CHECK-NEXT:    add w8, w0, #1
319 ; CHECK-NEXT:    sxtw x8, w8
320 ; CHECK-NEXT:    asr x0, x8, #32
321 ; CHECK-NEXT:    ret
322 entry:
323   %inc = add nsw i32 %a, 1
324   %conv = sext i32 %inc to i64
325   %shr = ashr i64 %conv, 32
326   ret i64 %shr
329 define i64 @sign_extend_inreg_isdef32(i64) {
330 ; CHECK-LABEL: sign_extend_inreg_isdef32:
331 ; CHECK:       ; %bb.0:
332 ; CHECK-NEXT:    sbfx x8, x0, #32, #16
333 ; CHECK-NEXT:    mov w0, w8
334 ; CHECK-NEXT:    ret
335   %2 = lshr i64 %0, 32
336   %3 = shl i64 %2, 16
337   %4 = trunc i64 %3 to i32
338   %5 = ashr exact i32 %4, 16
339   %6 = zext i32 %5 to i64
340   ret i64 %6