Skip to content

Commit de7222b

Browse files
committed
[AArch64][SME]: Add streaming-compatible testing files.
Testing files: - subvector.ll - permute-rev.ll - permute-zip-uzp-trn.ll - vector-shuffle.ll Reviewed By: david-arm, sdesmalen Differential Revision: https://reviews.llvm.org/D138683
1 parent a3b7458 commit de7222b

4 files changed

+2234
-0
lines changed
Lines changed: 240 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,240 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2+
; RUN: llc -force-streaming-compatible-sve < %s | FileCheck %s
3+
4+
target triple = "aarch64-unknown-linux-gnu"
5+
6+
; REVB pattern for shuffle v32i8 -> v16i16
7+
define void @test_revbv16i16(ptr %a) #0 {
8+
; CHECK-LABEL: test_revbv16i16:
9+
; CHECK: // %bb.0:
10+
; CHECK-NEXT: ldp q0, q1, [x0]
11+
; CHECK-NEXT: ptrue p0.h
12+
; CHECK-NEXT: revb z0.h, p0/m, z0.h
13+
; CHECK-NEXT: revb z1.h, p0/m, z1.h
14+
; CHECK-NEXT: stp q0, q1, [x0]
15+
; CHECK-NEXT: ret
16+
%tmp1 = load <32 x i8>, ptr %a
17+
%tmp2 = shufflevector <32 x i8> %tmp1, <32 x i8> undef, <32 x i32> <i32 1, i32 0, i32 3, i32 2, i32 5, i32 4, i32 7, i32 6, i32 9, i32 8, i32 11, i32 10, i32 13, i32 12, i32 15, i32 14, i32 17, i32 16, i32 19, i32 18, i32 21, i32 20, i32 23, i32 22, i32 undef, i32 24, i32 27, i32 undef, i32 29, i32 28, i32 undef, i32 undef>
18+
store <32 x i8> %tmp2, ptr %a
19+
ret void
20+
}
21+
22+
; REVB pattern for shuffle v32i8 -> v8i32
23+
define void @test_revbv8i32(ptr %a) #0 {
24+
; CHECK-LABEL: test_revbv8i32:
25+
; CHECK: // %bb.0:
26+
; CHECK-NEXT: ldp q0, q1, [x0]
27+
; CHECK-NEXT: ptrue p0.s
28+
; CHECK-NEXT: revb z0.s, p0/m, z0.s
29+
; CHECK-NEXT: revb z1.s, p0/m, z1.s
30+
; CHECK-NEXT: stp q0, q1, [x0]
31+
; CHECK-NEXT: ret
32+
%tmp1 = load <32 x i8>, ptr %a
33+
%tmp2 = shufflevector <32 x i8> %tmp1, <32 x i8> undef, <32 x i32> <i32 3, i32 2, i32 1, i32 0, i32 7, i32 6, i32 5, i32 4, i32 11, i32 10, i32 9, i32 8, i32 15, i32 14, i32 13, i32 12, i32 19, i32 18, i32 17, i32 16, i32 23, i32 22, i32 21, i32 20, i32 27, i32 undef, i32 undef, i32 undef, i32 31, i32 30, i32 29, i32 undef>
34+
store <32 x i8> %tmp2, ptr %a
35+
ret void
36+
}
37+
38+
; REVB pattern for shuffle v32i8 -> v4i64
39+
define void @test_revbv4i64(ptr %a) #0 {
40+
; CHECK-LABEL: test_revbv4i64:
41+
; CHECK: // %bb.0:
42+
; CHECK-NEXT: ldp q0, q1, [x0]
43+
; CHECK-NEXT: ptrue p0.d
44+
; CHECK-NEXT: revb z0.d, p0/m, z0.d
45+
; CHECK-NEXT: revb z1.d, p0/m, z1.d
46+
; CHECK-NEXT: stp q0, q1, [x0]
47+
; CHECK-NEXT: ret
48+
%tmp1 = load <32 x i8>, ptr %a
49+
%tmp2 = shufflevector <32 x i8> %tmp1, <32 x i8> undef, <32 x i32> <i32 7, i32 6, i32 5, i32 4, i32 3, i32 2, i32 1, i32 0, i32 15, i32 14, i32 13, i32 12, i32 11, i32 10, i32 9, i32 8, i32 23, i32 22, i32 21, i32 20, i32 19, i32 18, i32 17, i32 16, i32 31, i32 30, i32 29, i32 undef, i32 27, i32 undef, i32 undef, i32 undef>
50+
store <32 x i8> %tmp2, ptr %a
51+
ret void
52+
}
53+
54+
; REVH pattern for shuffle v16i16 -> v8i32
55+
define void @test_revhv8i32(ptr %a) #0 {
56+
; CHECK-LABEL: test_revhv8i32:
57+
; CHECK: // %bb.0:
58+
; CHECK-NEXT: ldp q0, q1, [x0]
59+
; CHECK-NEXT: ptrue p0.s
60+
; CHECK-NEXT: revh z0.s, p0/m, z0.s
61+
; CHECK-NEXT: revh z1.s, p0/m, z1.s
62+
; CHECK-NEXT: stp q0, q1, [x0]
63+
; CHECK-NEXT: ret
64+
%tmp1 = load <16 x i16>, ptr %a
65+
%tmp2 = shufflevector <16 x i16> %tmp1, <16 x i16> undef, <16 x i32> <i32 1, i32 0, i32 3, i32 2, i32 5, i32 4, i32 7, i32 6, i32 9, i32 8, i32 11, i32 10, i32 13, i32 12, i32 15, i32 14>
66+
store <16 x i16> %tmp2, ptr %a
67+
ret void
68+
}
69+
70+
; REVH pattern for shuffle v16f16 -> v8f32
71+
define void @test_revhv8f32(ptr %a) #0 {
72+
; CHECK-LABEL: test_revhv8f32:
73+
; CHECK: // %bb.0:
74+
; CHECK-NEXT: ldp q0, q1, [x0]
75+
; CHECK-NEXT: ptrue p0.s
76+
; CHECK-NEXT: revh z0.s, p0/m, z0.s
77+
; CHECK-NEXT: revh z1.s, p0/m, z1.s
78+
; CHECK-NEXT: stp q0, q1, [x0]
79+
; CHECK-NEXT: ret
80+
%tmp1 = load <16 x half>, ptr %a
81+
%tmp2 = shufflevector <16 x half> %tmp1, <16 x half> undef, <16 x i32> <i32 1, i32 0, i32 3, i32 2, i32 5, i32 4, i32 7, i32 6, i32 9, i32 8, i32 11, i32 10, i32 13, i32 12, i32 15, i32 14>
82+
store <16 x half> %tmp2, ptr %a
83+
ret void
84+
}
85+
86+
; REVH pattern for shuffle v16i16 -> v4i64
87+
define void @test_revhv4i64(ptr %a) #0 {
88+
; CHECK-LABEL: test_revhv4i64:
89+
; CHECK: // %bb.0:
90+
; CHECK-NEXT: ldp q0, q1, [x0]
91+
; CHECK-NEXT: ptrue p0.d
92+
; CHECK-NEXT: revh z0.d, p0/m, z0.d
93+
; CHECK-NEXT: revh z1.d, p0/m, z1.d
94+
; CHECK-NEXT: stp q0, q1, [x0]
95+
; CHECK-NEXT: ret
96+
%tmp1 = load <16 x i16>, ptr %a
97+
%tmp2 = shufflevector <16 x i16> %tmp1, <16 x i16> undef, <16 x i32> <i32 3, i32 2, i32 1, i32 0, i32 7, i32 6, i32 5, i32 4, i32 11, i32 10, i32 9, i32 8, i32 15, i32 14, i32 13, i32 12>
98+
store <16 x i16> %tmp2, ptr %a
99+
ret void
100+
}
101+
102+
; REVW pattern for shuffle v8i32 -> v4i64
103+
define void @test_revwv4i64(ptr %a) #0 {
104+
; CHECK-LABEL: test_revwv4i64:
105+
; CHECK: // %bb.0:
106+
; CHECK-NEXT: ldp q0, q1, [x0]
107+
; CHECK-NEXT: ptrue p0.d
108+
; CHECK-NEXT: revw z0.d, p0/m, z0.d
109+
; CHECK-NEXT: revw z1.d, p0/m, z1.d
110+
; CHECK-NEXT: stp q0, q1, [x0]
111+
; CHECK-NEXT: ret
112+
%tmp1 = load <8 x i32>, ptr %a
113+
%tmp2 = shufflevector <8 x i32> %tmp1, <8 x i32> undef, <8 x i32> <i32 1, i32 0, i32 3, i32 2, i32 5, i32 4, i32 7, i32 6>
114+
store <8 x i32> %tmp2, ptr %a
115+
ret void
116+
}
117+
118+
; REVW pattern for shuffle v8f32 -> v4f64
119+
define void @test_revwv4f64(ptr %a) #0 {
120+
; CHECK-LABEL: test_revwv4f64:
121+
; CHECK: // %bb.0:
122+
; CHECK-NEXT: ldp q0, q1, [x0]
123+
; CHECK-NEXT: ptrue p0.d
124+
; CHECK-NEXT: revw z0.d, p0/m, z0.d
125+
; CHECK-NEXT: revw z1.d, p0/m, z1.d
126+
; CHECK-NEXT: stp q0, q1, [x0]
127+
; CHECK-NEXT: ret
128+
%tmp1 = load <8 x float>, ptr %a
129+
%tmp2 = shufflevector <8 x float> %tmp1, <8 x float> undef, <8 x i32> <i32 1, i32 0, i32 3, i32 2, i32 5, i32 4, i32 7, i32 6>
130+
store <8 x float> %tmp2, ptr %a
131+
ret void
132+
}
133+
134+
define <16 x i8> @test_revv16i8(ptr %a) #0 {
135+
; CHECK-LABEL: test_revv16i8:
136+
; CHECK: // %bb.0:
137+
; CHECK-NEXT: ldr q0, [x0]
138+
; CHECK-NEXT: ptrue p0.d
139+
; CHECK-NEXT: revb z0.d, p0/m, z0.d
140+
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
141+
; CHECK-NEXT: ret
142+
%tmp1 = load <16 x i8>, ptr %a
143+
%tmp2 = shufflevector <16 x i8> %tmp1, <16 x i8> undef, <16 x i32> <i32 7, i32 6, i32 5, i32 4, i32 3, i32 2, i32 1, i32 0, i32 15, i32 14, i32 13, i32 12, i32 11, i32 10, i32 9, i32 8>
144+
ret <16 x i8> %tmp2
145+
}
146+
147+
; REVW pattern for shuffle two v8i32 inputs with the second input available.
148+
define void @test_revwv8i32v8i32(ptr %a, ptr %b) #0 {
149+
; CHECK-LABEL: test_revwv8i32v8i32:
150+
; CHECK: // %bb.0:
151+
; CHECK-NEXT: ldp q0, q1, [x1]
152+
; CHECK-NEXT: ptrue p0.d
153+
; CHECK-NEXT: revw z0.d, p0/m, z0.d
154+
; CHECK-NEXT: revw z1.d, p0/m, z1.d
155+
; CHECK-NEXT: stp q0, q1, [x0]
156+
; CHECK-NEXT: ret
157+
%tmp1 = load <8 x i32>, ptr %a
158+
%tmp2 = load <8 x i32>, ptr %b
159+
%tmp3 = shufflevector <8 x i32> %tmp1, <8 x i32> %tmp2, <8 x i32> <i32 9, i32 8, i32 11, i32 10, i32 13, i32 12, i32 15, i32 14>
160+
store <8 x i32> %tmp3, ptr %a
161+
ret void
162+
}
163+
164+
define void @test_revhv32i16(ptr %a) #0 {
165+
; CHECK-LABEL: test_revhv32i16:
166+
; CHECK: // %bb.0:
167+
; CHECK-NEXT: ldp q0, q1, [x0, #32]
168+
; CHECK-NEXT: ptrue p0.d
169+
; CHECK-NEXT: revh z0.d, p0/m, z0.d
170+
; CHECK-NEXT: ldp q2, q3, [x0]
171+
; CHECK-NEXT: revh z1.d, p0/m, z1.d
172+
; CHECK-NEXT: stp q0, q1, [x0, #32]
173+
; CHECK-NEXT: revh z0.d, p0/m, z2.d
174+
; CHECK-NEXT: revh z1.d, p0/m, z3.d
175+
; CHECK-NEXT: stp q0, q1, [x0]
176+
; CHECK-NEXT: ret
177+
%tmp1 = load <32 x i16>, ptr %a
178+
%tmp2 = shufflevector <32 x i16> %tmp1, <32 x i16> undef, <32 x i32> <i32 3, i32 2, i32 1, i32 0, i32 7, i32 6, i32 5, i32 4, i32 11, i32 10, i32 9, i32 8, i32 15, i32 14, i32 13, i32 12, i32 19, i32 18, i32 17, i32 16, i32 23, i32 22, i32 21, i32 20, i32 27, i32 undef, i32 undef, i32 undef, i32 31, i32 30, i32 29, i32 undef>
179+
store <32 x i16> %tmp2, ptr %a
180+
ret void
181+
}
182+
183+
define void @test_rev_elts_fail(ptr %a) #0 {
184+
; CHECK-LABEL: test_rev_elts_fail:
185+
; CHECK: // %bb.0:
186+
; CHECK-NEXT: ldp q1, q0, [x0]
187+
; CHECK-NEXT: fmov x10, d1
188+
; CHECK-NEXT: mov z2.d, z0.d[1]
189+
; CHECK-NEXT: fmov x8, d0
190+
; CHECK-NEXT: fmov x9, d2
191+
; CHECK-NEXT: mov z0.d, z1.d[1]
192+
; CHECK-NEXT: fmov x11, d0
193+
; CHECK-NEXT: stp x9, x8, [sp, #-32]!
194+
; CHECK-NEXT: .cfi_def_cfa_offset 32
195+
; CHECK-NEXT: stp x11, x10, [sp, #16]
196+
; CHECK-NEXT: ldp q1, q0, [sp]
197+
; CHECK-NEXT: stp q0, q1, [x0]
198+
; CHECK-NEXT: add sp, sp, #32
199+
; CHECK-NEXT: ret
200+
%tmp1 = load <4 x i64>, ptr %a
201+
%tmp2 = shufflevector <4 x i64> %tmp1, <4 x i64> undef, <4 x i32> <i32 1, i32 0, i32 3, i32 2>
202+
store <4 x i64> %tmp2, ptr %a
203+
ret void
204+
}
205+
206+
define void @test_revv8i32(ptr %a) #0 {
207+
; CHECK-LABEL: test_revv8i32:
208+
; CHECK: // %bb.0:
209+
; CHECK-NEXT: sub sp, sp, #32
210+
; CHECK-NEXT: .cfi_def_cfa_offset 32
211+
; CHECK-NEXT: ldp q0, q1, [x0]
212+
; CHECK-NEXT: mov z2.s, z0.s[1]
213+
; CHECK-NEXT: mov z3.s, z0.s[2]
214+
; CHECK-NEXT: mov z4.s, z0.s[3]
215+
; CHECK-NEXT: fmov w8, s0
216+
; CHECK-NEXT: fmov w9, s2
217+
; CHECK-NEXT: fmov w10, s3
218+
; CHECK-NEXT: fmov w11, s4
219+
; CHECK-NEXT: mov z0.s, z1.s[1]
220+
; CHECK-NEXT: mov z2.s, z1.s[2]
221+
; CHECK-NEXT: mov z3.s, z1.s[3]
222+
; CHECK-NEXT: stp w9, w8, [sp, #24]
223+
; CHECK-NEXT: fmov w8, s1
224+
; CHECK-NEXT: fmov w9, s0
225+
; CHECK-NEXT: stp w11, w10, [sp, #16]
226+
; CHECK-NEXT: fmov w10, s2
227+
; CHECK-NEXT: fmov w11, s3
228+
; CHECK-NEXT: stp w9, w8, [sp, #8]
229+
; CHECK-NEXT: stp w11, w10, [sp]
230+
; CHECK-NEXT: ldp q0, q1, [sp]
231+
; CHECK-NEXT: stp q0, q1, [x0]
232+
; CHECK-NEXT: add sp, sp, #32
233+
; CHECK-NEXT: ret
234+
%tmp1 = load <8 x i32>, ptr %a
235+
%tmp2 = shufflevector <8 x i32> %tmp1, <8 x i32> undef, <8 x i32> <i32 7, i32 6, i32 5, i32 4, i32 3, i32 2, i32 1, i32 0>
236+
store <8 x i32> %tmp2, ptr %a
237+
ret void
238+
}
239+
240+
attributes #0 = { "target-features"="+sve" }

0 commit comments

Comments
 (0)