Skip to content

Commit 2850ba8

Browse files
[AArch64][NFC] Add test for vector sdiv scalarization
1 parent 90606ae commit 2850ba8

File tree

1 file changed

+318
-0
lines changed

1 file changed

+318
-0
lines changed
Lines changed: 318 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,318 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
2+
; RUN: llc -mtriple=aarch64-none-linux-gnu < %s | FileCheck %s
3+
4+
define <8 x i16> @sdiv_v8i16_by_7(<8 x i16> %x) {
5+
; CHECK-LABEL: sdiv_v8i16_by_7:
6+
; CHECK: // %bb.0:
7+
; CHECK-NEXT: mov w8, #18725 // =0x4925
8+
; CHECK-NEXT: dup v1.8h, w8
9+
; CHECK-NEXT: smull2 v2.4s, v0.8h, v1.8h
10+
; CHECK-NEXT: smull v0.4s, v0.4h, v1.4h
11+
; CHECK-NEXT: uzp2 v0.8h, v0.8h, v2.8h
12+
; CHECK-NEXT: sshr v0.8h, v0.8h, #1
13+
; CHECK-NEXT: usra v0.8h, v0.8h, #15
14+
; CHECK-NEXT: ret
15+
%div = sdiv <8 x i16> %x, <i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7>
16+
ret <8 x i16> %div
17+
}
18+
19+
define <16 x i16> @sdiv_v16i16_by_7(<16 x i16> %x) {
20+
; CHECK-LABEL: sdiv_v16i16_by_7:
21+
; CHECK: // %bb.0:
22+
; CHECK-NEXT: smov x11, v0.h[1]
23+
; CHECK-NEXT: smov x10, v0.h[0]
24+
; CHECK-NEXT: mov x8, #-56173 // =0xffffffffffff2493
25+
; CHECK-NEXT: smov x13, v0.h[3]
26+
; CHECK-NEXT: smov x14, v1.h[1]
27+
; CHECK-NEXT: movk x8, #37449, lsl #16
28+
; CHECK-NEXT: smov x16, v1.h[0]
29+
; CHECK-NEXT: smov w12, v0.h[1]
30+
; CHECK-NEXT: smov w15, v0.h[0]
31+
; CHECK-NEXT: smov x18, v1.h[2]
32+
; CHECK-NEXT: smov w0, v0.h[3]
33+
; CHECK-NEXT: smov w1, v1.h[1]
34+
; CHECK-NEXT: smull x11, w11, w8
35+
; CHECK-NEXT: smov w2, v1.h[0]
36+
; CHECK-NEXT: smov x9, v0.h[2]
37+
; CHECK-NEXT: smull x10, w10, w8
38+
; CHECK-NEXT: smov w17, v0.h[2]
39+
; CHECK-NEXT: smov w3, v1.h[2]
40+
; CHECK-NEXT: smull x13, w13, w8
41+
; CHECK-NEXT: smull x14, w14, w8
42+
; CHECK-NEXT: add x12, x12, x11, lsr #32
43+
; CHECK-NEXT: smull x16, w16, w8
44+
; CHECK-NEXT: add x10, x15, x10, lsr #32
45+
; CHECK-NEXT: smull x15, w18, w8
46+
; CHECK-NEXT: add x11, x0, x13, lsr #32
47+
; CHECK-NEXT: smov x0, v0.h[4]
48+
; CHECK-NEXT: add x13, x1, x14, lsr #32
49+
; CHECK-NEXT: asr w18, w10, #2
50+
; CHECK-NEXT: smull x9, w9, w8
51+
; CHECK-NEXT: add x14, x2, x16, lsr #32
52+
; CHECK-NEXT: asr w16, w12, #2
53+
; CHECK-NEXT: smov x2, v1.h[3]
54+
; CHECK-NEXT: add w18, w18, w10, lsr #31
55+
; CHECK-NEXT: add x15, x3, x15, lsr #32
56+
; CHECK-NEXT: smov w10, v0.h[5]
57+
; CHECK-NEXT: add w12, w16, w12, lsr #31
58+
; CHECK-NEXT: asr w16, w14, #2
59+
; CHECK-NEXT: add x9, x17, x9, lsr #32
60+
; CHECK-NEXT: fmov s2, w18
61+
; CHECK-NEXT: smov w17, v0.h[4]
62+
; CHECK-NEXT: smull x0, w0, w8
63+
; CHECK-NEXT: add w14, w16, w14, lsr #31
64+
; CHECK-NEXT: asr w16, w13, #2
65+
; CHECK-NEXT: asr w1, w9, #2
66+
; CHECK-NEXT: smov x18, v0.h[5]
67+
; CHECK-NEXT: fmov s3, w14
68+
; CHECK-NEXT: mov v2.h[1], w12
69+
; CHECK-NEXT: add w12, w16, w13, lsr #31
70+
; CHECK-NEXT: smov w13, v1.h[3]
71+
; CHECK-NEXT: smov x14, v1.h[4]
72+
; CHECK-NEXT: smull x16, w2, w8
73+
; CHECK-NEXT: add w1, w1, w9, lsr #31
74+
; CHECK-NEXT: add x17, x17, x0, lsr #32
75+
; CHECK-NEXT: asr w0, w15, #2
76+
; CHECK-NEXT: mov v3.h[1], w12
77+
; CHECK-NEXT: smov w12, v1.h[4]
78+
; CHECK-NEXT: smull x18, w18, w8
79+
; CHECK-NEXT: mov v2.h[2], w1
80+
; CHECK-NEXT: asr w1, w11, #2
81+
; CHECK-NEXT: add w15, w0, w15, lsr #31
82+
; CHECK-NEXT: add x13, x13, x16, lsr #32
83+
; CHECK-NEXT: smov x16, v1.h[5]
84+
; CHECK-NEXT: smull x14, w14, w8
85+
; CHECK-NEXT: add w11, w1, w11, lsr #31
86+
; CHECK-NEXT: smov x0, v0.h[6]
87+
; CHECK-NEXT: add x10, x10, x18, lsr #32
88+
; CHECK-NEXT: asr w1, w13, #2
89+
; CHECK-NEXT: mov v3.h[2], w15
90+
; CHECK-NEXT: smov w15, v1.h[5]
91+
; CHECK-NEXT: add x12, x12, x14, lsr #32
92+
; CHECK-NEXT: mov v2.h[3], w11
93+
; CHECK-NEXT: asr w11, w17, #2
94+
; CHECK-NEXT: add w13, w1, w13, lsr #31
95+
; CHECK-NEXT: smull x16, w16, w8
96+
; CHECK-NEXT: smov x14, v1.h[6]
97+
; CHECK-NEXT: asr w18, w12, #2
98+
; CHECK-NEXT: add w11, w11, w17, lsr #31
99+
; CHECK-NEXT: smov w9, v0.h[6]
100+
; CHECK-NEXT: mov v3.h[3], w13
101+
; CHECK-NEXT: smull x17, w0, w8
102+
; CHECK-NEXT: smov x0, v1.h[7]
103+
; CHECK-NEXT: add x13, x15, x16, lsr #32
104+
; CHECK-NEXT: add w12, w18, w12, lsr #31
105+
; CHECK-NEXT: smov w16, v1.h[6]
106+
; CHECK-NEXT: mov v2.h[4], w11
107+
; CHECK-NEXT: smov x11, v0.h[7]
108+
; CHECK-NEXT: smull x14, w14, w8
109+
; CHECK-NEXT: asr w15, w10, #2
110+
; CHECK-NEXT: asr w18, w13, #2
111+
; CHECK-NEXT: smov w1, v0.h[7]
112+
; CHECK-NEXT: mov v3.h[4], w12
113+
; CHECK-NEXT: add x9, x9, x17, lsr #32
114+
; CHECK-NEXT: add w10, w15, w10, lsr #31
115+
; CHECK-NEXT: add w12, w18, w13, lsr #31
116+
; CHECK-NEXT: add x13, x16, x14, lsr #32
117+
; CHECK-NEXT: smov w14, v1.h[7]
118+
; CHECK-NEXT: smull x11, w11, w8
119+
; CHECK-NEXT: smull x8, w0, w8
120+
; CHECK-NEXT: mov v2.h[5], w10
121+
; CHECK-NEXT: asr w10, w9, #2
122+
; CHECK-NEXT: mov v3.h[5], w12
123+
; CHECK-NEXT: asr w12, w13, #2
124+
; CHECK-NEXT: add w9, w10, w9, lsr #31
125+
; CHECK-NEXT: add x10, x1, x11, lsr #32
126+
; CHECK-NEXT: add w11, w12, w13, lsr #31
127+
; CHECK-NEXT: add x8, x14, x8, lsr #32
128+
; CHECK-NEXT: mov v2.h[6], w9
129+
; CHECK-NEXT: asr w9, w10, #2
130+
; CHECK-NEXT: mov v3.h[6], w11
131+
; CHECK-NEXT: asr w11, w8, #2
132+
; CHECK-NEXT: add w9, w9, w10, lsr #31
133+
; CHECK-NEXT: add w8, w11, w8, lsr #31
134+
; CHECK-NEXT: mov v2.h[7], w9
135+
; CHECK-NEXT: mov v3.h[7], w8
136+
; CHECK-NEXT: mov v0.16b, v2.16b
137+
; CHECK-NEXT: mov v1.16b, v3.16b
138+
; CHECK-NEXT: ret
139+
%div = sdiv <16 x i16> %x, <i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7>
140+
ret <16 x i16> %div
141+
}
142+
143+
define <8 x i16> @srem_v8i16_by_7(<8 x i16> %x) {
144+
; CHECK-LABEL: srem_v8i16_by_7:
145+
; CHECK: // %bb.0:
146+
; CHECK-NEXT: mov w8, #18725 // =0x4925
147+
; CHECK-NEXT: dup v1.8h, w8
148+
; CHECK-NEXT: smull2 v2.4s, v0.8h, v1.8h
149+
; CHECK-NEXT: smull v1.4s, v0.4h, v1.4h
150+
; CHECK-NEXT: uzp2 v1.8h, v1.8h, v2.8h
151+
; CHECK-NEXT: movi v2.8h, #7
152+
; CHECK-NEXT: sshr v1.8h, v1.8h, #1
153+
; CHECK-NEXT: usra v1.8h, v1.8h, #15
154+
; CHECK-NEXT: mls v0.8h, v1.8h, v2.8h
155+
; CHECK-NEXT: ret
156+
%rem = srem <8 x i16> %x, <i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7>
157+
ret <8 x i16> %rem
158+
}
159+
160+
define <16 x i16> @srem_v16i16_by_7(<16 x i16> %x) {
161+
; CHECK-LABEL: srem_v16i16_by_7:
162+
; CHECK: // %bb.0:
163+
; CHECK-NEXT: str x19, [sp, #-16]! // 8-byte Folded Spill
164+
; CHECK-NEXT: .cfi_def_cfa_offset 16
165+
; CHECK-NEXT: .cfi_offset w19, -16
166+
; CHECK-NEXT: smov x10, v0.h[0]
167+
; CHECK-NEXT: smov x9, v0.h[1]
168+
; CHECK-NEXT: mov x8, #-56173 // =0xffffffffffff2493
169+
; CHECK-NEXT: smov x15, v0.h[2]
170+
; CHECK-NEXT: movk x8, #37449, lsl #16
171+
; CHECK-NEXT: smov x16, v0.h[3]
172+
; CHECK-NEXT: smov w13, v0.h[0]
173+
; CHECK-NEXT: smov w12, v0.h[1]
174+
; CHECK-NEXT: smov x18, v1.h[0]
175+
; CHECK-NEXT: smov w17, v0.h[2]
176+
; CHECK-NEXT: smov w14, v0.h[3]
177+
; CHECK-NEXT: smov w2, v1.h[0]
178+
; CHECK-NEXT: smull x10, w10, w8
179+
; CHECK-NEXT: smov x4, v1.h[1]
180+
; CHECK-NEXT: smov x0, v0.h[4]
181+
; CHECK-NEXT: smull x9, w9, w8
182+
; CHECK-NEXT: smov w11, v0.h[4]
183+
; CHECK-NEXT: smull x15, w15, w8
184+
; CHECK-NEXT: smull x1, w16, w8
185+
; CHECK-NEXT: add x10, x13, x10, lsr #32
186+
; CHECK-NEXT: smov x16, v0.h[5]
187+
; CHECK-NEXT: add x3, x12, x9, lsr #32
188+
; CHECK-NEXT: smull x18, w18, w8
189+
; CHECK-NEXT: smov w9, v0.h[5]
190+
; CHECK-NEXT: add x15, x17, x15, lsr #32
191+
; CHECK-NEXT: asr w6, w10, #2
192+
; CHECK-NEXT: smull x4, w4, w8
193+
; CHECK-NEXT: asr w5, w3, #2
194+
; CHECK-NEXT: add x1, x14, x1, lsr #32
195+
; CHECK-NEXT: smull x0, w0, w8
196+
; CHECK-NEXT: add x7, x2, x18, lsr #32
197+
; CHECK-NEXT: asr w19, w15, #2
198+
; CHECK-NEXT: smov w18, v1.h[1]
199+
; CHECK-NEXT: add w10, w6, w10, lsr #31
200+
; CHECK-NEXT: add w3, w5, w3, lsr #31
201+
; CHECK-NEXT: asr w5, w1, #2
202+
; CHECK-NEXT: add w6, w19, w15, lsr #31
203+
; CHECK-NEXT: asr w15, w7, #2
204+
; CHECK-NEXT: sub w19, w10, w10, lsl #3
205+
; CHECK-NEXT: add w1, w5, w1, lsr #31
206+
; CHECK-NEXT: smov x5, v1.h[2]
207+
; CHECK-NEXT: sub w3, w3, w3, lsl #3
208+
; CHECK-NEXT: add x4, x18, x4, lsr #32
209+
; CHECK-NEXT: add w7, w15, w7, lsr #31
210+
; CHECK-NEXT: add w13, w13, w19
211+
; CHECK-NEXT: sub w6, w6, w6, lsl #3
212+
; CHECK-NEXT: sub w1, w1, w1, lsl #3
213+
; CHECK-NEXT: fmov s2, w13
214+
; CHECK-NEXT: add w12, w12, w3
215+
; CHECK-NEXT: add x3, x11, x0, lsr #32
216+
; CHECK-NEXT: smov w0, v1.h[2]
217+
; CHECK-NEXT: asr w13, w4, #2
218+
; CHECK-NEXT: smull x5, w5, w8
219+
; CHECK-NEXT: sub w7, w7, w7, lsl #3
220+
; CHECK-NEXT: add w6, w17, w6
221+
; CHECK-NEXT: smov w17, v1.h[3]
222+
; CHECK-NEXT: mov v2.h[1], w12
223+
; CHECK-NEXT: add w13, w13, w4, lsr #31
224+
; CHECK-NEXT: smov x4, v1.h[3]
225+
; CHECK-NEXT: add w2, w2, w7
226+
; CHECK-NEXT: smov x7, v1.h[4]
227+
; CHECK-NEXT: add w14, w14, w1
228+
; CHECK-NEXT: add x5, x0, x5, lsr #32
229+
; CHECK-NEXT: sub w13, w13, w13, lsl #3
230+
; CHECK-NEXT: fmov s3, w2
231+
; CHECK-NEXT: smull x2, w16, w8
232+
; CHECK-NEXT: smov w16, v1.h[4]
233+
; CHECK-NEXT: asr w12, w3, #2
234+
; CHECK-NEXT: mov v2.h[2], w6
235+
; CHECK-NEXT: asr w6, w5, #2
236+
; CHECK-NEXT: smull x4, w4, w8
237+
; CHECK-NEXT: add w18, w18, w13
238+
; CHECK-NEXT: smov x15, v0.h[6]
239+
; CHECK-NEXT: add w3, w12, w3, lsr #31
240+
; CHECK-NEXT: add w5, w6, w5, lsr #31
241+
; CHECK-NEXT: smull x6, w7, w8
242+
; CHECK-NEXT: mov v3.h[1], w18
243+
; CHECK-NEXT: add x4, x17, x4, lsr #32
244+
; CHECK-NEXT: add x1, x9, x2, lsr #32
245+
; CHECK-NEXT: sub w3, w3, w3, lsl #3
246+
; CHECK-NEXT: sub w18, w5, w5, lsl #3
247+
; CHECK-NEXT: smov x5, v1.h[5]
248+
; CHECK-NEXT: mov v2.h[3], w14
249+
; CHECK-NEXT: asr w2, w4, #2
250+
; CHECK-NEXT: smov w10, v0.h[6]
251+
; CHECK-NEXT: smov x13, v0.h[7]
252+
; CHECK-NEXT: add w18, w0, w18
253+
; CHECK-NEXT: add x0, x16, x6, lsr #32
254+
; CHECK-NEXT: add w11, w11, w3
255+
; CHECK-NEXT: add w2, w2, w4, lsr #31
256+
; CHECK-NEXT: smov w4, v1.h[5]
257+
; CHECK-NEXT: mov v3.h[2], w18
258+
; CHECK-NEXT: asr w18, w0, #2
259+
; CHECK-NEXT: smull x5, w5, w8
260+
; CHECK-NEXT: smov x3, v1.h[7]
261+
; CHECK-NEXT: sub w14, w2, w2, lsl #3
262+
; CHECK-NEXT: smov x2, v1.h[6]
263+
; CHECK-NEXT: smull x15, w15, w8
264+
; CHECK-NEXT: add w18, w18, w0, lsr #31
265+
; CHECK-NEXT: asr w6, w1, #2
266+
; CHECK-NEXT: mov v2.h[4], w11
267+
; CHECK-NEXT: add w14, w17, w14
268+
; CHECK-NEXT: add x17, x4, x5, lsr #32
269+
; CHECK-NEXT: smov w12, v0.h[7]
270+
; CHECK-NEXT: mov v3.h[3], w14
271+
; CHECK-NEXT: sub w14, w18, w18, lsl #3
272+
; CHECK-NEXT: smov w18, v1.h[6]
273+
; CHECK-NEXT: smull x2, w2, w8
274+
; CHECK-NEXT: asr w0, w17, #2
275+
; CHECK-NEXT: add w1, w6, w1, lsr #31
276+
; CHECK-NEXT: add w11, w16, w14
277+
; CHECK-NEXT: add x15, x10, x15, lsr #32
278+
; CHECK-NEXT: smull x13, w13, w8
279+
; CHECK-NEXT: add w16, w0, w17, lsr #31
280+
; CHECK-NEXT: smov w17, v1.h[7]
281+
; CHECK-NEXT: smull x8, w3, w8
282+
; CHECK-NEXT: mov v3.h[4], w11
283+
; CHECK-NEXT: add x11, x18, x2, lsr #32
284+
; CHECK-NEXT: sub w14, w1, w1, lsl #3
285+
; CHECK-NEXT: asr w0, w15, #2
286+
; CHECK-NEXT: sub w16, w16, w16, lsl #3
287+
; CHECK-NEXT: add x13, x12, x13, lsr #32
288+
; CHECK-NEXT: asr w1, w11, #2
289+
; CHECK-NEXT: add w9, w9, w14
290+
; CHECK-NEXT: add w14, w0, w15, lsr #31
291+
; CHECK-NEXT: add w15, w4, w16
292+
; CHECK-NEXT: add x8, x17, x8, lsr #32
293+
; CHECK-NEXT: add w11, w1, w11, lsr #31
294+
; CHECK-NEXT: mov v2.h[5], w9
295+
; CHECK-NEXT: mov v3.h[5], w15
296+
; CHECK-NEXT: sub w9, w14, w14, lsl #3
297+
; CHECK-NEXT: asr w14, w13, #2
298+
; CHECK-NEXT: asr w15, w8, #2
299+
; CHECK-NEXT: sub w11, w11, w11, lsl #3
300+
; CHECK-NEXT: add w9, w10, w9
301+
; CHECK-NEXT: add w10, w14, w13, lsr #31
302+
; CHECK-NEXT: add w8, w15, w8, lsr #31
303+
; CHECK-NEXT: add w11, w18, w11
304+
; CHECK-NEXT: mov v2.h[6], w9
305+
; CHECK-NEXT: mov v3.h[6], w11
306+
; CHECK-NEXT: sub w9, w10, w10, lsl #3
307+
; CHECK-NEXT: sub w8, w8, w8, lsl #3
308+
; CHECK-NEXT: add w9, w12, w9
309+
; CHECK-NEXT: add w8, w17, w8
310+
; CHECK-NEXT: mov v2.h[7], w9
311+
; CHECK-NEXT: mov v3.h[7], w8
312+
; CHECK-NEXT: mov v0.16b, v2.16b
313+
; CHECK-NEXT: mov v1.16b, v3.16b
314+
; CHECK-NEXT: ldr x19, [sp], #16 // 8-byte Folded Reload
315+
; CHECK-NEXT: ret
316+
%rem = srem <16 x i16> %x, <i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7, i16 7>
317+
ret <16 x i16> %rem
318+
}

0 commit comments

Comments
 (0)