24.04
x1.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2017-2020 Arm Limited.
3
*
4
* SPDX-License-Identifier: MIT
5
*
6
* Permission is hereby granted, free of charge, to any person obtaining a copy
7
* of this software and associated documentation files (the "Software"), to
8
* deal in the Software without restriction, including without limitation the
9
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10
* sell copies of the Software, and to permit persons to whom the Software is
11
* furnished to do so, subject to the following conditions:
12
*
13
* The above copyright notice and this permission notice shall be included in all
14
* copies or substantial portions of the Software.
15
*
16
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22
* SOFTWARE.
23
*/
24
#ifdef __aarch64__
25
26
#include <arm_neon.h>
27
28
#include "../../asmlib.hpp"
29
30
// Kernel implementation.
31
//
32
// Assume that "Apanel" points to a chunk of A blocks (each size 8xK) in read-order.
33
// Assume that "Bpanel" points to a chunk of B blocks (each size 12xK) in read-order.
34
// Assume that "Cpanel" points to a chunk of C output blocks (each size
35
// 12x8), the chunks being arranged in a row major fashion.
36
//
37
// Note that the intent of this is that either ablocks or bblocks will be 1
38
// - this construction allows the output loop to proceed in either order.
39
40
namespace
arm_gemm
{
41
42
void
a64_sgemm_asimd_8x12_x1(
const
float
*Apanel,
const
float
*Bpanel,
float
*Cpanel,
int
ablocks,
int
bblocks,
int
K
) {
43
const
float
*a_ptr = Apanel;
44
float
*c_ptr = Cpanel;
45
46
for
(
int
yb=0; yb<ablocks; yb++) {
47
const
float
*a_ptr0 = a_ptr;
48
const
float
*b_ptr = Bpanel;
49
50
for
(
int
xb=0; xb<bblocks; xb++) {
51
a_ptr = a_ptr0;
52
// Fix up for odd lengths - set a flag if K is odd, but make
53
// sure we round up the iteration count.
54
int
oddk = (
K
& 1);
55
int
k = ((
K
+1)/2) - 1;
56
57
register
float32x4_t a0
asm
(
"v0"
);
58
register
float32x4_t a1
asm
(
"v1"
);
59
register
float32x4_t b0
asm
(
"v2"
);
60
register
float32x4_t b1
asm
(
"v3"
);
61
register
float32x4_t b2
asm
(
"v4"
);
62
63
__asm __volatile (
64
// Initialize result registers, load initial operands, prime prefetches.
65
"movi v8.4s, #0x0\n"
66
"ldr %q[a0], [%[a_ptr]]\n"
67
"movi v9.4s, #0x0\n"
68
"ldr %q[b0], [%[b_ptr]]\n"
69
"movi v10.4s, #0x0\n"
70
"ldr %q[a1], [%[a_ptr], #16]\n"
71
"movi v11.4s, #0x0\n"
72
"ldr %q[b1], [%[b_ptr], #16]\n"
73
"movi v12.4s, #0x0\n"
74
ASM_PREFETCH
(
"[%[b_ptr], #64]"
)
75
"movi v13.4s, #0x0\n"
76
ASM_PREFETCH
(
"[%[a_ptr], #64]"
)
77
"movi v14.4s, #0x0\n"
78
ASM_PREFETCH
(
"[%[b_ptr], #128]"
)
79
"movi v15.4s, #0x0\n"
80
ASM_PREFETCH
(
"[%[a_ptr], #128]"
)
81
"movi v16.4s, #0x0\n"
82
ASM_PREFETCH
(
"[%[b_ptr], #192]"
)
83
"movi v17.4s, #0x0\n"
84
ASM_PREFETCH
(
"[%[b_ptr], #256]"
)
85
"movi v18.4s, #0x0\n"
86
ASM_PREFETCH
(
"[%[a_ptr], #192]"
)
87
"movi v19.4s, #0x0\n"
88
ASM_PREFETCH
(
"[%[b_ptr], #320]"
)
89
"movi v20.4s, #0x0\n"
90
ASM_PREFETCH
(
"[%[a_ptr], #256]"
)
91
"movi v21.4s, #0x0\n"
92
ASM_PREFETCH
(
"[%[b_ptr], #384]"
)
93
"movi v22.4s, #0x0\n"
94
"movi v23.4s, #0x0\n"
95
"movi v24.4s, #0x0\n"
96
"movi v25.4s, #0x0\n"
97
"movi v26.4s, #0x0\n"
98
"movi v27.4s, #0x0\n"
99
"movi v28.4s, #0x0\n"
100
"movi v29.4s, #0x0\n"
101
"movi v30.4s, #0x0\n"
102
"movi v31.4s, #0x0\n"
103
104
// Skip loop if we are doing zero iterations of it.
105
"cbz %w[k], 4f\n"
106
107
// Loop proper
108
"1:\n"
109
"fmla v8.4s , %[b0].4s, %[a0].s[0]\n"
110
"fmla v9.4s , %[b0].4s, %[a0].s[1]\n"
111
"ldr %q[b2], [%[b_ptr], #32]\n"
112
"fmla v10.4s, %[b0].4s, %[a0].s[2]\n"
113
"fmla v11.4s, %[b0].4s, %[a0].s[3]\n"
114
"fmla v12.4s, %[b0].4s, %[a1].s[0]\n"
115
"fmla v13.4s, %[b0].4s, %[a1].s[1]\n"
116
"fmla v14.4s, %[b0].4s, %[a1].s[2]\n"
117
"fmla v15.4s, %[b0].4s, %[a1].s[3]\n"
118
"ldr %q[b0], [%[b_ptr], #48]\n"
119
120
"fmla v16.4s, %[b1].4s, %[a0].s[0]\n"
121
"fmla v17.4s, %[b1].4s, %[a0].s[1]\n"
122
ASM_PREFETCH
(
"[%[a_ptr], #320]"
)
123
"fmla v18.4s, %[b1].4s, %[a0].s[2]\n"
124
"fmla v19.4s, %[b1].4s, %[a0].s[3]\n"
125
"fmla v20.4s, %[b1].4s, %[a1].s[0]\n"
126
"fmla v21.4s, %[b1].4s, %[a1].s[1]\n"
127
"fmla v22.4s, %[b1].4s, %[a1].s[2]\n"
128
"fmla v23.4s, %[b1].4s, %[a1].s[3]\n"
129
"ldr %q[b1], [%[b_ptr], #64]\n"
130
131
"fmla v24.4s, %[b2].4s, %[a0].s[0]\n"
132
"fmla v25.4s, %[b2].4s, %[a0].s[1]\n"
133
ASM_PREFETCH
(
"[%[b_ptr], #448]"
)
134
"fmla v26.4s, %[b2].4s, %[a0].s[2]\n"
135
"fmla v27.4s, %[b2].4s, %[a0].s[3]\n"
136
"ldr %q[a0], [%[a_ptr], #32]\n"
137
"fmla v28.4s, %[b2].4s, %[a1].s[0]\n"
138
"fmla v29.4s, %[b2].4s, %[a1].s[1]\n"
139
"fmla v30.4s, %[b2].4s, %[a1].s[2]\n"
140
"fmla v31.4s, %[b2].4s, %[a1].s[3]\n"
141
"ldr %q[a1], [%[a_ptr], #48]\n"
142
143
"fmla v8.4s , %[b0].4s, %[a0].s[0]\n"
144
"fmla v9.4s , %[b0].4s, %[a0].s[1]\n"
145
"ldr %q[b2], [%[b_ptr], #80]\n"
146
"fmla v10.4s, %[b0].4s, %[a0].s[2]\n"
147
"fmla v11.4s, %[b0].4s, %[a0].s[3]\n"
148
"fmla v12.4s, %[b0].4s, %[a1].s[0]\n"
149
"fmla v13.4s, %[b0].4s, %[a1].s[1]\n"
150
"fmla v14.4s, %[b0].4s, %[a1].s[2]\n"
151
"fmla v15.4s, %[b0].4s, %[a1].s[3]\n"
152
"ldr %q[b0], [%[b_ptr], #96]\n"
153
154
"fmla v16.4s, %[b1].4s, %[a0].s[0]\n"
155
"fmla v17.4s, %[b1].4s, %[a0].s[1]\n"
156
"fmla v18.4s, %[b1].4s, %[a0].s[2]\n"
157
"fmla v19.4s, %[b1].4s, %[a0].s[3]\n"
158
ASM_PREFETCH
(
"[%[b_ptr], #512]"
)
159
"fmla v20.4s, %[b1].4s, %[a1].s[0]\n"
160
"fmla v21.4s, %[b1].4s, %[a1].s[1]\n"
161
"subs %w[k], %w[k], #1\n"
162
"fmla v22.4s, %[b1].4s, %[a1].s[2]\n"
163
"fmla v23.4s, %[b1].4s, %[a1].s[3]\n"
164
"ldr %q[b1], [%[b_ptr], #112]\n"
165
166
"fmla v24.4s, %[b2].4s, %[a0].s[0]\n"
167
"fmla v25.4s, %[b2].4s, %[a0].s[1]\n"
168
"add %[a_ptr], %[a_ptr], #64\n"
169
"fmla v26.4s, %[b2].4s, %[a0].s[2]\n"
170
"fmla v27.4s, %[b2].4s, %[a0].s[3]\n"
171
"ldr %q[a0], [%[a_ptr]]\n"
172
"fmla v28.4s, %[b2].4s, %[a1].s[0]\n"
173
"fmla v29.4s, %[b2].4s, %[a1].s[1]\n"
174
"add %[b_ptr], %[b_ptr], #96\n"
175
"fmla v30.4s, %[b2].4s, %[a1].s[2]\n"
176
"fmla v31.4s, %[b2].4s, %[a1].s[3]\n"
177
"ldr %q[a1], [%[a_ptr], #16]\n"
178
"bne 1b\n"
179
180
// Target to use when K is 1 or 2 (i.e. zero iterations of main loop)
181
"4:\n"
182
183
// Branch to alternative tail for odd K
184
"cbnz %w[oddk], 2f\n"
185
186
// Detached final iteration (even K)
187
"fmla v8.4s , %[b0].4s, %[a0].s[0]\n"
188
"fmla v9.4s , %[b0].4s, %[a0].s[1]\n"
189
"ldr %q[b2], [%[b_ptr], #32]\n"
190
"fmla v10.4s, %[b0].4s, %[a0].s[2]\n"
191
"fmla v11.4s, %[b0].4s, %[a0].s[3]\n"
192
"fmla v12.4s, %[b0].4s, %[a1].s[0]\n"
193
"fmla v13.4s, %[b0].4s, %[a1].s[1]\n"
194
"fmla v14.4s, %[b0].4s, %[a1].s[2]\n"
195
"fmla v15.4s, %[b0].4s, %[a1].s[3]\n"
196
"ldr %q[b0], [%[b_ptr], #48]\n"
197
198
"fmla v16.4s, %[b1].4s, %[a0].s[0]\n"
199
"fmla v17.4s, %[b1].4s, %[a0].s[1]\n"
200
"fmla v18.4s, %[b1].4s, %[a0].s[2]\n"
201
"fmla v19.4s, %[b1].4s, %[a0].s[3]\n"
202
"fmla v20.4s, %[b1].4s, %[a1].s[0]\n"
203
"fmla v21.4s, %[b1].4s, %[a1].s[1]\n"
204
"fmla v22.4s, %[b1].4s, %[a1].s[2]\n"
205
"fmla v23.4s, %[b1].4s, %[a1].s[3]\n"
206
"ldr %q[b1], [%[b_ptr], #64]\n"
207
208
"fmla v24.4s, %[b2].4s, %[a0].s[0]\n"
209
"fmla v25.4s, %[b2].4s, %[a0].s[1]\n"
210
"add %[a_ptr], %[a_ptr], #64\n"
211
"fmla v26.4s, %[b2].4s, %[a0].s[2]\n"
212
"fmla v27.4s, %[b2].4s, %[a0].s[3]\n"
213
"ldr %q[a0], [%[a_ptr], #-32]\n"
214
"fmla v28.4s, %[b2].4s, %[a1].s[0]\n"
215
"fmla v29.4s, %[b2].4s, %[a1].s[1]\n"
216
"add %[b_ptr], %[b_ptr], #96\n"
217
"fmla v30.4s, %[b2].4s, %[a1].s[2]\n"
218
"fmla v31.4s, %[b2].4s, %[a1].s[3]\n"
219
"ldr %q[a1], [%[a_ptr], #-16]\n"
220
221
"fmla v8.4s , %[b0].4s, %[a0].s[0]\n"
222
"fmla v16.4s, %[b1].4s, %[a0].s[0]\n"
223
"ldr %q[b2], [%[b_ptr], #-16]\n"
224
"fmla v9.4s , %[b0].4s, %[a0].s[1]\n"
225
"str q8, [%[c_ptr], #0]\n"
226
"fmla v17.4s, %[b1].4s, %[a0].s[1]\n"
227
"str q16, [%[c_ptr], #16]\n"
228
"fmla v24.4s, %[b2].4s, %[a0].s[0]\n"
229
"str q24, [%[c_ptr], #32]\n"
230
231
"fmla v25.4s, %[b2].4s, %[a0].s[1]\n"
232
"str q9, [%[c_ptr], #48]\n"
233
"fmla v10.4s, %[b0].4s, %[a0].s[2]\n"
234
"str q17, [%[c_ptr], #64]\n"
235
"fmla v18.4s, %[b1].4s, %[a0].s[2]\n"
236
"str q25, [%[c_ptr], #80]\n"
237
"fmla v26.4s, %[b2].4s, %[a0].s[2]\n"
238
"str q10, [%[c_ptr], #96]\n"
239
240
"fmla v11.4s, %[b0].4s, %[a0].s[3]\n"
241
"str q18, [%[c_ptr], #112]\n"
242
"fmla v19.4s, %[b1].4s, %[a0].s[3]\n"
243
"str q26, [%[c_ptr], #128]\n"
244
"fmla v27.4s, %[b2].4s, %[a0].s[3]\n"
245
"str q11, [%[c_ptr], #144]\n"
246
247
"fmla v12.4s, %[b0].4s, %[a1].s[0]\n"
248
"str q19, [%[c_ptr], #160]\n"
249
"fmla v20.4s, %[b1].4s, %[a1].s[0]\n"
250
"str q27, [%[c_ptr], #176]\n"
251
"fmla v28.4s, %[b2].4s, %[a1].s[0]\n"
252
"str q12, [%[c_ptr], #192]\n"
253
254
"fmla v13.4s, %[b0].4s, %[a1].s[1]\n"
255
"str q20, [%[c_ptr], #208]\n"
256
"fmla v21.4s, %[b1].4s, %[a1].s[1]\n"
257
"str q28, [%[c_ptr], #224]\n"
258
"fmla v29.4s, %[b2].4s, %[a1].s[1]\n"
259
"str q13, [%[c_ptr], #240]\n"
260
261
"fmla v14.4s, %[b0].4s, %[a1].s[2]\n"
262
"str q21, [%[c_ptr], #256]\n"
263
"fmla v22.4s, %[b1].4s, %[a1].s[2]\n"
264
"str q29, [%[c_ptr], #272]\n"
265
"fmla v30.4s, %[b2].4s, %[a1].s[2]\n"
266
"str q14, [%[c_ptr], #288]\n"
267
268
"fmla v15.4s, %[b0].4s, %[a1].s[3]\n"
269
"str q22, [%[c_ptr], #304]\n"
270
"fmla v23.4s, %[b1].4s, %[a1].s[3]\n"
271
"str q30, [%[c_ptr], #320]\n"
272
"fmla v31.4s, %[b2].4s, %[a1].s[3]\n"
273
"str q15, [%[c_ptr], #336]\n"
274
275
"b 3f\n"
276
277
// Detached final iteration (odd K)
278
"2:\n"
279
"fmla v8.4s , %[b0].4s, %[a0].s[0]\n"
280
"ldr %q[b2], [%[b_ptr], #32]\n"
281
"fmla v16.4s, %[b1].4s, %[a0].s[0]\n"
282
"fmla v9.4s , %[b0].4s, %[a0].s[1]\n"
283
"str q8, [%[c_ptr], #0]\n"
284
"fmla v17.4s, %[b1].4s, %[a0].s[1]\n"
285
"str q16, [%[c_ptr], #16]\n"
286
"fmla v24.4s, %[b2].4s, %[a0].s[0]\n"
287
"add %[b_ptr], %[b_ptr], #48\n"
288
"add %[a_ptr], %[a_ptr], #32\n"
289
"str q24, [%[c_ptr], #32]\n"
290
"fmla v25.4s, %[b2].4s, %[a0].s[1]\n"
291
"str q9, [%[c_ptr], #48]\n"
292
293
"fmla v10.4s, %[b0].4s, %[a0].s[2]\n"
294
"str q17, [%[c_ptr], #64]\n"
295
"fmla v18.4s, %[b1].4s, %[a0].s[2]\n"
296
"str q25, [%[c_ptr], #80]\n"
297
"fmla v26.4s, %[b2].4s, %[a0].s[2]\n"
298
"str q10, [%[c_ptr], #96]\n"
299
300
"fmla v11.4s, %[b0].4s, %[a0].s[3]\n"
301
"str q18, [%[c_ptr], #112]\n"
302
"fmla v19.4s, %[b1].4s, %[a0].s[3]\n"
303
"str q26, [%[c_ptr], #128]\n"
304
"fmla v27.4s, %[b2].4s, %[a0].s[3]\n"
305
"str q11, [%[c_ptr], #144]\n"
306
307
"fmla v12.4s, %[b0].4s, %[a1].s[0]\n"
308
"str q19, [%[c_ptr], #160]\n"
309
"fmla v20.4s, %[b1].4s, %[a1].s[0]\n"
310
"str q27, [%[c_ptr], #176]\n"
311
"fmla v28.4s, %[b2].4s, %[a1].s[0]\n"
312
"str q12, [%[c_ptr], #192]\n"
313
314
"fmla v13.4s, %[b0].4s, %[a1].s[1]\n"
315
"str q20, [%[c_ptr], #208]\n"
316
"fmla v21.4s, %[b1].4s, %[a1].s[1]\n"
317
"str q28, [%[c_ptr], #224]\n"
318
"fmla v29.4s, %[b2].4s, %[a1].s[1]\n"
319
"str q13, [%[c_ptr], #240]\n"
320
321
"fmla v14.4s, %[b0].4s, %[a1].s[2]\n"
322
"str q21, [%[c_ptr], #256]\n"
323
"fmla v22.4s, %[b1].4s, %[a1].s[2]\n"
324
"str q29, [%[c_ptr], #272]\n"
325
"fmla v30.4s, %[b2].4s, %[a1].s[2]\n"
326
"str q14, [%[c_ptr], #288]\n"
327
328
"fmla v15.4s, %[b0].4s, %[a1].s[3]\n"
329
"str q22, [%[c_ptr], #304]\n"
330
"fmla v23.4s, %[b1].4s, %[a1].s[3]\n"
331
"str q30, [%[c_ptr], #320]\n"
332
"fmla v31.4s, %[b2].4s, %[a1].s[3]\n"
333
"str q15, [%[c_ptr], #336]\n"
334
335
// Common tail
336
"3:\n"
337
"str q23, [%[c_ptr], #352]\n"
338
"str q31, [%[c_ptr], #368]\n"
339
"add %[c_ptr], %[c_ptr], #384\n"
340
:
341
[a_ptr]
"+r"
(a_ptr), [b_ptr]
"+r"
(b_ptr), [c_ptr]
"+r"
(c_ptr),
342
[a0]
"+w"
(a0), [a1]
"+w"
(a1),
343
[b0]
"+w"
(b0), [b1]
"+w"
(b1), [b2]
"+w"
(b2), [k]
"+r"
(k)
344
: [oddk]
"r"
(oddk)
345
:
"x20"
,
"x21"
,
"v8"
,
"v9"
,
"v10"
,
"v11"
,
"v12"
,
"v13"
,
"v14"
,
"v15"
,
"v16"
,
"v17"
,
"v18"
,
346
"v19"
,
"v20"
,
"v21"
,
"v22"
,
"v23"
,
"v24"
,
"v25"
,
"v26"
,
"v27"
,
"v28"
,
"v29"
,
"v30"
,
"v31"
,
"cc"
347
);
348
}
349
}
350
}
351
352
}
// namespace arm_gemm
353
354
#endif
ASM_PREFETCH
#define ASM_PREFETCH(address)
Definition:
asmlib.hpp:45
arm_gemm
Definition:
barrier.hpp:30
K
unsigned int K
Definition:
CpuGemmAssemblyDispatch.cpp:106
src
core
NEON
kernels
arm_gemm
kernels
a64_sgemm_8x12
x1.cpp
Generated on Mon Apr 29 2024 10:53:55 for Compute Library by
1.8.17