1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc -mtriple=arm-eabi -mattr=+neon %s -o - -lower-interleaved-accesses=false | FileCheck %s
3
4define <8 x i8> @test_vextd(<8 x i8>* %A, <8 x i8>* %B) nounwind {
5; CHECK-LABEL: test_vextd:
6; CHECK:       @ %bb.0:
7; CHECK-NEXT:    vldr d16, [r1]
8; CHECK-NEXT:    vldr d17, [r0]
9; CHECK-NEXT:    vext.8 d16, d17, d16, #3
10; CHECK-NEXT:    vmov r0, r1, d16
11; CHECK-NEXT:    mov pc, lr
12	%tmp1 = load <8 x i8>, <8 x i8>* %A
13	%tmp2 = load <8 x i8>, <8 x i8>* %B
14	%tmp3 = shufflevector <8 x i8> %tmp1, <8 x i8> %tmp2, <8 x i32> <i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10>
15	ret <8 x i8> %tmp3
16}
17
18define <8 x i8> @test_vextRd(<8 x i8>* %A, <8 x i8>* %B) nounwind {
19; CHECK-LABEL: test_vextRd:
20; CHECK:       @ %bb.0:
21; CHECK-NEXT:    vldr d16, [r0]
22; CHECK-NEXT:    vldr d17, [r1]
23; CHECK-NEXT:    vext.8 d16, d17, d16, #5
24; CHECK-NEXT:    vmov r0, r1, d16
25; CHECK-NEXT:    mov pc, lr
26	%tmp1 = load <8 x i8>, <8 x i8>* %A
27	%tmp2 = load <8 x i8>, <8 x i8>* %B
28	%tmp3 = shufflevector <8 x i8> %tmp1, <8 x i8> %tmp2, <8 x i32> <i32 13, i32 14, i32 15, i32 0, i32 1, i32 2, i32 3, i32 4>
29	ret <8 x i8> %tmp3
30}
31
32define <16 x i8> @test_vextq(<16 x i8>* %A, <16 x i8>* %B) nounwind {
33; CHECK-LABEL: test_vextq:
34; CHECK:       @ %bb.0:
35; CHECK-NEXT:    vld1.64 {d16, d17}, [r1]
36; CHECK-NEXT:    vld1.64 {d18, d19}, [r0]
37; CHECK-NEXT:    vext.8 q8, q9, q8, #3
38; CHECK-NEXT:    vmov r0, r1, d16
39; CHECK-NEXT:    vmov r2, r3, d17
40; CHECK-NEXT:    mov pc, lr
41	%tmp1 = load <16 x i8>, <16 x i8>* %A
42	%tmp2 = load <16 x i8>, <16 x i8>* %B
43	%tmp3 = shufflevector <16 x i8> %tmp1, <16 x i8> %tmp2, <16 x i32> <i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15, i32 16, i32 17, i32 18>
44	ret <16 x i8> %tmp3
45}
46
47define <16 x i8> @test_vextRq(<16 x i8>* %A, <16 x i8>* %B) nounwind {
48; CHECK-LABEL: test_vextRq:
49; CHECK:       @ %bb.0:
50; CHECK-NEXT:    vld1.64 {d16, d17}, [r0]
51; CHECK-NEXT:    vld1.64 {d18, d19}, [r1]
52; CHECK-NEXT:    vext.8 q8, q9, q8, #7
53; CHECK-NEXT:    vmov r0, r1, d16
54; CHECK-NEXT:    vmov r2, r3, d17
55; CHECK-NEXT:    mov pc, lr
56	%tmp1 = load <16 x i8>, <16 x i8>* %A
57	%tmp2 = load <16 x i8>, <16 x i8>* %B
58	%tmp3 = shufflevector <16 x i8> %tmp1, <16 x i8> %tmp2, <16 x i32> <i32 23, i32 24, i32 25, i32 26, i32 27, i32 28, i32 29, i32 30, i32 31, i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6>
59	ret <16 x i8> %tmp3
60}
61
62define <4 x i16> @test_vextd16(<4 x i16>* %A, <4 x i16>* %B) nounwind {
63; CHECK-LABEL: test_vextd16:
64; CHECK:       @ %bb.0:
65; CHECK-NEXT:    vldr d16, [r1]
66; CHECK-NEXT:    vldr d17, [r0]
67; CHECK-NEXT:    vext.16 d16, d17, d16, #3
68; CHECK-NEXT:    vmov r0, r1, d16
69; CHECK-NEXT:    mov pc, lr
70	%tmp1 = load <4 x i16>, <4 x i16>* %A
71	%tmp2 = load <4 x i16>, <4 x i16>* %B
72	%tmp3 = shufflevector <4 x i16> %tmp1, <4 x i16> %tmp2, <4 x i32> <i32 3, i32 4, i32 5, i32 6>
73	ret <4 x i16> %tmp3
74}
75
76define <4 x i32> @test_vextq32(<4 x i32>* %A, <4 x i32>* %B) nounwind {
77; CHECK-LABEL: test_vextq32:
78; CHECK:       @ %bb.0:
79; CHECK-NEXT:    vld1.64 {d16, d17}, [r1]
80; CHECK-NEXT:    vld1.64 {d18, d19}, [r0]
81; CHECK-NEXT:    vext.32 q8, q9, q8, #3
82; CHECK-NEXT:    vmov r0, r1, d16
83; CHECK-NEXT:    vmov r2, r3, d17
84; CHECK-NEXT:    mov pc, lr
85	%tmp1 = load <4 x i32>, <4 x i32>* %A
86	%tmp2 = load <4 x i32>, <4 x i32>* %B
87	%tmp3 = shufflevector <4 x i32> %tmp1, <4 x i32> %tmp2, <4 x i32> <i32 3, i32 4, i32 5, i32 6>
88	ret <4 x i32> %tmp3
89}
90
91; Undef shuffle indices should not prevent matching to VEXT:
92
93define <8 x i8> @test_vextd_undef(<8 x i8>* %A, <8 x i8>* %B) nounwind {
94; CHECK-LABEL: test_vextd_undef:
95; CHECK:       @ %bb.0:
96; CHECK-NEXT:    vldr d16, [r1]
97; CHECK-NEXT:    vldr d17, [r0]
98; CHECK-NEXT:    vext.8 d16, d17, d16, #3
99; CHECK-NEXT:    vmov r0, r1, d16
100; CHECK-NEXT:    mov pc, lr
101	%tmp1 = load <8 x i8>, <8 x i8>* %A
102	%tmp2 = load <8 x i8>, <8 x i8>* %B
103	%tmp3 = shufflevector <8 x i8> %tmp1, <8 x i8> %tmp2, <8 x i32> <i32 3, i32 undef, i32 undef, i32 6, i32 7, i32 8, i32 9, i32 10>
104	ret <8 x i8> %tmp3
105}
106
107define <16 x i8> @test_vextRq_undef(<16 x i8>* %A, <16 x i8>* %B) nounwind {
108; CHECK-LABEL: test_vextRq_undef:
109; CHECK:       @ %bb.0:
110; CHECK-NEXT:    vld1.64 {d16, d17}, [r0]
111; CHECK-NEXT:    vld1.64 {d18, d19}, [r1]
112; CHECK-NEXT:    vext.8 q8, q9, q8, #7
113; CHECK-NEXT:    vmov r0, r1, d16
114; CHECK-NEXT:    vmov r2, r3, d17
115; CHECK-NEXT:    mov pc, lr
116	%tmp1 = load <16 x i8>, <16 x i8>* %A
117	%tmp2 = load <16 x i8>, <16 x i8>* %B
118	%tmp3 = shufflevector <16 x i8> %tmp1, <16 x i8> %tmp2, <16 x i32> <i32 23, i32 24, i32 25, i32 26, i32 undef, i32 undef, i32 29, i32 30, i32 31, i32 0, i32 1, i32 2, i32 3, i32 4, i32 undef, i32 6>
119	ret <16 x i8> %tmp3
120}
121
122define <16 x i8> @test_vextq_undef_op2(<16 x i8> %a) nounwind {
123; CHECK-LABEL: test_vextq_undef_op2:
124; CHECK:       @ %bb.0: @ %entry
125; CHECK-NEXT:    vmov d17, r2, r3
126; CHECK-NEXT:    vmov d16, r0, r1
127; CHECK-NEXT:    vext.8 q8, q8, q8, #2
128; CHECK-NEXT:    vmov r0, r1, d16
129; CHECK-NEXT:    vmov r2, r3, d17
130; CHECK-NEXT:    mov pc, lr
131entry:
132  %tmp1 = shufflevector <16 x i8> %a, <16 x i8> undef, <16 x i32> <i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15, i32 0, i32 1>
133  ret <16 x i8> %tmp1
134}
135
136define <8 x i8> @test_vextd_undef_op2(<8 x i8> %a) nounwind {
137; CHECK-LABEL: test_vextd_undef_op2:
138; CHECK:       @ %bb.0: @ %entry
139; CHECK-NEXT:    vmov d16, r0, r1
140; CHECK-NEXT:    vext.8 d16, d16, d16, #2
141; CHECK-NEXT:    vmov r0, r1, d16
142; CHECK-NEXT:    mov pc, lr
143entry:
144  %tmp1 = shufflevector <8 x i8> %a, <8 x i8> undef, <8 x i32> <i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 0, i32 1>
145  ret <8 x i8> %tmp1
146}
147
148
149define <16 x i8> @test_vextq_undef_op2_undef(<16 x i8> %a) nounwind {
150; CHECK-LABEL: test_vextq_undef_op2_undef:
151; CHECK:       @ %bb.0: @ %entry
152; CHECK-NEXT:    vmov d17, r2, r3
153; CHECK-NEXT:    vmov d16, r0, r1
154; CHECK-NEXT:    vext.8 q8, q8, q8, #2
155; CHECK-NEXT:    vmov r0, r1, d16
156; CHECK-NEXT:    vmov r2, r3, d17
157; CHECK-NEXT:    mov pc, lr
158entry:
159  %tmp1 = shufflevector <16 x i8> %a, <16 x i8> undef, <16 x i32> <i32 2, i32 3, i32 4, i32 undef, i32 undef, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15, i32 0, i32 1>
160  ret <16 x i8> %tmp1
161}
162
163define <8 x i8> @test_vextd_undef_op2_undef(<8 x i8> %a) nounwind {
164; CHECK-LABEL: test_vextd_undef_op2_undef:
165; CHECK:       @ %bb.0: @ %entry
166; CHECK-NEXT:    vmov d16, r0, r1
167; CHECK-NEXT:    vext.8 d16, d16, d16, #2
168; CHECK-NEXT:    vmov r0, r1, d16
169; CHECK-NEXT:    mov pc, lr
170entry:
171  %tmp1 = shufflevector <8 x i8> %a, <8 x i8> undef, <8 x i32> <i32 2, i32 3, i32 undef, i32 undef, i32 6, i32 7, i32 undef, i32 1>
172  ret <8 x i8> %tmp1
173}
174
175; Tests for ReconstructShuffle function. Indices have to be carefully
176; chosen to reach lowering phase as a BUILD_VECTOR.
177
178; One vector needs vext, the other can be handled by extract_subvector
179; Also checks interleaving of sources is handled correctly.
180; Essence: a vext is used on %A and something saner than stack load/store for final result.
181define <4 x i16> @test_interleaved(<8 x i16>* %A, <8 x i16>* %B) nounwind {
182; CHECK-LABEL: test_interleaved:
183; CHECK:       @ %bb.0:
184; CHECK-NEXT:    vld1.64 {d16, d17}, [r0]
185; CHECK-NEXT:    vext.16 d16, d16, d17, #3
186; CHECK-NEXT:    vorr d18, d16, d16
187; CHECK-NEXT:    vldr d17, [r1]
188; CHECK-NEXT:    vuzp.16 d16, d18
189; CHECK-NEXT:    vzip.16 d16, d17
190; CHECK-NEXT:    vmov r0, r1, d16
191; CHECK-NEXT:    mov pc, lr
192        %tmp1 = load <8 x i16>, <8 x i16>* %A
193        %tmp2 = load <8 x i16>, <8 x i16>* %B
194        %tmp3 = shufflevector <8 x i16> %tmp1, <8 x i16> %tmp2, <4 x i32> <i32 3, i32 8, i32 5, i32 9>
195        ret <4 x i16> %tmp3
196}
197
198; An undef in the shuffle list should still be optimizable
199define <4 x i16> @test_undef(<8 x i16>* %A, <8 x i16>* %B) nounwind {
200; CHECK-LABEL: test_undef:
201; CHECK:       @ %bb.0:
202; CHECK-NEXT:    vldr d16, [r1]
203; CHECK-NEXT:    vldr d17, [r0, #8]
204; CHECK-NEXT:    vzip.16 d17, d16
205; CHECK-NEXT:    vmov r0, r1, d17
206; CHECK-NEXT:    mov pc, lr
207        %tmp1 = load <8 x i16>, <8 x i16>* %A
208        %tmp2 = load <8 x i16>, <8 x i16>* %B
209        %tmp3 = shufflevector <8 x i16> %tmp1, <8 x i16> %tmp2, <4 x i32> <i32 undef, i32 8, i32 5, i32 9>
210        ret <4 x i16> %tmp3
211}
212
213; We should ignore a build_vector with more than two sources.
214; Use illegal <32 x i16> type to produce such a shuffle after legalizing types.
215; Try to look for fallback to by-element inserts.
216define <4 x i16> @test_multisource(<32 x i16>* %B) nounwind {
217; CHECK-LABEL: test_multisource:
218; CHECK:       @ %bb.0:
219; CHECK-NEXT:    vldr d18, [r0, #32]
220; CHECK-NEXT:    mov r1, r0
221; CHECK-NEXT:    vorr d22, d18, d18
222; CHECK-NEXT:    vld1.16 {d16, d17}, [r1:128]!
223; CHECK-NEXT:    vldr d19, [r0, #48]
224; CHECK-NEXT:    vld1.64 {d20, d21}, [r1:128]
225; CHECK-NEXT:    vzip.16 d22, d19
226; CHECK-NEXT:    vtrn.16 q8, q10
227; CHECK-NEXT:    vext.16 d18, d18, d22, #2
228; CHECK-NEXT:    vext.16 d16, d18, d16, #2
229; CHECK-NEXT:    vext.16 d16, d16, d16, #2
230; CHECK-NEXT:    vmov r0, r1, d16
231; CHECK-NEXT:    mov pc, lr
232        %tmp1 = load <32 x i16>, <32 x i16>* %B
233        %tmp2 = shufflevector <32 x i16> %tmp1, <32 x i16> undef, <4 x i32> <i32 0, i32 8, i32 16, i32 24>
234        ret <4 x i16> %tmp2
235}
236
237; We don't handle shuffles using more than half of a 128-bit vector.
238; Again, test for fallback to by-element inserts.
239define <4 x i16> @test_largespan(<8 x i16>* %B) nounwind {
240; CHECK-LABEL: test_largespan:
241; CHECK:       @ %bb.0:
242; CHECK-NEXT:    vld1.64 {d16, d17}, [r0]
243; CHECK-NEXT:    vorr d18, d16, d16
244; CHECK-NEXT:    vuzp.16 d18, d17
245; CHECK-NEXT:    vmov r0, r1, d18
246; CHECK-NEXT:    mov pc, lr
247        %tmp1 = load <8 x i16>, <8 x i16>* %B
248        %tmp2 = shufflevector <8 x i16> %tmp1, <8 x i16> undef, <4 x i32> <i32 0, i32 2, i32 4, i32 6>
249        ret <4 x i16> %tmp2
250}
251
252; The actual shuffle code only handles some cases, make sure we check
253; this rather than blindly emitting a VECTOR_SHUFFLE (infinite
254; lowering loop can result otherwise).
255; (There are probably better ways to lower this shuffle, but it's not
256; really important.)
257define <8 x i16> @test_illegal(<8 x i16>* %A, <8 x i16>* %B) nounwind {
258; CHECK-LABEL: test_illegal:
259; CHECK:       @ %bb.0:
260; CHECK-NEXT:    vld1.64 {d16, d17}, [r0]
261; CHECK-NEXT:    vorr d22, d16, d16
262; CHECK-NEXT:    vmov.u16 r0, d16[0]
263; CHECK-NEXT:    vorr d23, d16, d16
264; CHECK-NEXT:    vmov.u16 r2, d17[3]
265; CHECK-NEXT:    vmov.u16 r3, d17[1]
266; CHECK-NEXT:    vld1.64 {d18, d19}, [r1]
267; CHECK-NEXT:    vmov.u16 r1, d19[1]
268; CHECK-NEXT:    vuzp.16 d22, d23
269; CHECK-NEXT:    vuzp.16 d22, d18
270; CHECK-NEXT:    vmov.16 d20[0], r0
271; CHECK-NEXT:    vmov.16 d20[1], r2
272; CHECK-NEXT:    vmov.16 d20[2], r3
273; CHECK-NEXT:    vmov.16 d20[3], r1
274; CHECK-NEXT:    vext.16 d21, d16, d18, #3
275; CHECK-NEXT:    vmov r0, r1, d20
276; CHECK-NEXT:    vmov r2, r3, d21
277; CHECK-NEXT:    mov pc, lr
278       %tmp1 = load <8 x i16>, <8 x i16>* %A
279       %tmp2 = load <8 x i16>, <8 x i16>* %B
280       %tmp3 = shufflevector <8 x i16> %tmp1, <8 x i16> %tmp2, <8 x i32> <i32 0, i32 7, i32 5, i32 13, i32 3, i32 2, i32 2, i32 9>
281       ret <8 x i16> %tmp3
282}
283
284; PR11129
285; Make sure this doesn't crash
286define arm_aapcscc void @test_elem_mismatch(<2 x i64>* nocapture %src, <4 x i16>* nocapture %dest) nounwind {
287; CHECK-LABEL: test_elem_mismatch:
288; CHECK:       @ %bb.0:
289; CHECK-NEXT:    vld1.64 {d16, d17}, [r0:128]
290; CHECK-NEXT:    vmov.32 r0, d16[0]
291; CHECK-NEXT:    vmov.32 r2, d17[0]
292; CHECK-NEXT:    vmov.16 d16[0], r0
293; CHECK-NEXT:    vmov.16 d16[1], r2
294; CHECK-NEXT:    vstr d16, [r1]
295; CHECK-NEXT:    mov pc, lr
296  %tmp0 = load <2 x i64>, <2 x i64>* %src, align 16
297  %tmp1 = bitcast <2 x i64> %tmp0 to <4 x i32>
298  %tmp2 = extractelement <4 x i32> %tmp1, i32 0
299  %tmp3 = extractelement <4 x i32> %tmp1, i32 2
300  %tmp4 = trunc i32 %tmp2 to i16
301  %tmp5 = trunc i32 %tmp3 to i16
302  %tmp6 = insertelement <4 x i16> undef, i16 %tmp4, i32 0
303  %tmp7 = insertelement <4 x i16> %tmp6, i16 %tmp5, i32 1
304  store <4 x i16> %tmp7, <4 x i16>* %dest, align 4
305  ret void
306}
307
308define <4 x i32> @test_reverse_and_extract(<2 x i32>* %A) {
309; CHECK-LABEL: test_reverse_and_extract:
310; CHECK:       @ %bb.0: @ %entry
311; CHECK-NEXT:    vldr d16, [r0]
312; CHECK-NEXT:    vrev64.32 q9, q8
313; CHECK-NEXT:    vext.32 q8, q8, q9, #2
314; CHECK-NEXT:    vmov r0, r1, d16
315; CHECK-NEXT:    vmov r2, r3, d17
316; CHECK-NEXT:    mov pc, lr
317entry:
318	%tmp1 = load <2 x i32>, <2 x i32>* %A
319  %0 = shufflevector <2 x i32> %tmp1, <2 x i32> undef, <4 x i32> <i32 undef, i32 undef, i32 1, i32 0>
320  ret <4 x i32> %0
321}
322
323define <4 x i32> @test_dup_and_extract(<2 x i32>* %A) {
324; CHECK-LABEL: test_dup_and_extract:
325; CHECK:       @ %bb.0: @ %entry
326; CHECK-NEXT:    vldr d16, [r0]
327; CHECK-NEXT:    vdup.32 q9, d16[0]
328; CHECK-NEXT:    vext.32 q8, q9, q8, #2
329; CHECK-NEXT:    vmov r0, r1, d16
330; CHECK-NEXT:    vmov r2, r3, d17
331; CHECK-NEXT:    mov pc, lr
332entry:
333	%tmp1 = load <2 x i32>, <2 x i32>* %A
334  %0 = shufflevector <2 x i32> %tmp1, <2 x i32> undef, <4 x i32> <i32 0, i32 0, i32 0, i32 1>
335  ret <4 x i32> %0
336}
337
338define <4 x i32> @test_zip_and_extract(<2 x i32>* %A) {
339; CHECK-LABEL: test_zip_and_extract:
340; CHECK:       @ %bb.0: @ %entry
341; CHECK-NEXT:    vldr d16, [r0]
342; CHECK-NEXT:    vorr q9, q8, q8
343; CHECK-NEXT:    vorr q10, q8, q8
344; CHECK-NEXT:    vzip.32 q9, q10
345; CHECK-NEXT:    vext.32 q8, q9, q8, #2
346; CHECK-NEXT:    vmov r0, r1, d16
347; CHECK-NEXT:    vmov r2, r3, d17
348; CHECK-NEXT:    mov pc, lr
349entry:
350	%tmp1 = load <2 x i32>, <2 x i32>* %A
351  %0 = shufflevector <2 x i32> %tmp1, <2 x i32> undef, <4 x i32> <i32 1, i32 1, i32 0, i32 1>
352  ret <4 x i32> %0
353}
354