28 #if defined(ARM_COMPUTE_ENABLE_SME2)
34 const __fp16 *
const *
const input_ptrs,
35 __fp16 *
const *
const outptrs,
37 unsigned int n_channels,
44 __fp16 *
const *outptrs;
46 const __fp16 min, max;
47 const __fp16 *inptrs[25];
50 const __fp16 *
const *
const input_ptrs,
51 __fp16 *
const *
const outptrs,
52 const void *
const params,
55 ) : outptrs(outptrs), params(params), min(min), max(max)
57 inptrs[0] = input_ptrs[12];
58 inptrs[1] = input_ptrs[0];
59 inptrs[2] = input_ptrs[1];
60 inptrs[3] = input_ptrs[3];
61 inptrs[4] = input_ptrs[4];
62 inptrs[5] = input_ptrs[5];
63 inptrs[6] = input_ptrs[6];
64 inptrs[7] = input_ptrs[2];
65 inptrs[8] = input_ptrs[8];
66 inptrs[9] = input_ptrs[9];
67 inptrs[10] = input_ptrs[7];
68 inptrs[11] = input_ptrs[15];
69 inptrs[12] = input_ptrs[10];
70 inptrs[13] = input_ptrs[16];
71 inptrs[14] = input_ptrs[11];
72 inptrs[15] = input_ptrs[18];
73 inptrs[16] = input_ptrs[13];
74 inptrs[17] = input_ptrs[19];
75 inptrs[18] = input_ptrs[20];
76 inptrs[19] = input_ptrs[14];
77 inptrs[20] = input_ptrs[21];
78 inptrs[21] = input_ptrs[17];
79 inptrs[22] = input_ptrs[23];
80 inptrs[23] = input_ptrs[22];
81 inptrs[24] = input_ptrs[24];
86 Args params_struct(input_ptrs, outptrs, params,
90 "ldr x20, [%x[params_struct], %[offsetof_args_outptrs]]\n"
91 ".inst 0xd503477f // SMSTART ZA\n"
92 "add x16, %x[params_struct], %[offsetof_Args_inptrs]\n"
94 "ldr x14, [%x[params_struct], %[offsetof_args_params]]\n"
96 ".inst 0x25207810 // ptrue pn8.b\n"
98 "whilelt p2.h, XZR, %x[n_channels]\n"
99 "ld1rh { z19.h }, p3/Z, [%x[params_struct], %[offsetof_args_min]]\n"
100 "ldp x12, x11, [x20, #0x0]\n"
101 "ldp x10, x9, [x20, #0x10]\n"
102 "cmp x13, %x[n_channels]\n"
103 "ld1rh { z18.h }, p3/Z, [%x[params_struct], %[offsetof_args_max]]\n"
104 "sub x28, XZR, x13\n"
105 "ld1h { z17.h }, p3/Z, [x14]\n"
106 "addvl x14, x14, #1\n"
107 "ldp x27, x26, [x16, #0x0]\n"
108 "ldp x25, x24, [x16, #0x10]\n"
109 ".inst 0xa040a1c0 // ld1h { z0.h-z3.h }, pn8.b/Z, [x14]\n"
110 "addvl x14, x14, #4\n"
111 "ldp x23, x22, [x16, #0x20]\n"
112 ".inst 0xa040a1c4 // ld1h { z4.h-z7.h }, pn8.b/Z, [x14]\n"
113 "addvl x14, x14, #4\n"
114 "ldp x21, x20, [x16, #0x30]\n"
115 "ld1h { z8.h }, p3/Z, [x14]\n"
116 "addvl x14, x14, #1\n"
117 "ld1h { z9.h }, p2/Z, [x27, x15, LSL #1]\n"
118 "ld1h { z10.h }, p2/Z, [x26, x15, LSL #1]\n"
119 "ld1h { z11.h }, p2/Z, [x25, x15, LSL #1]\n"
120 "ld1h { z12.h }, p2/Z, [x24, x15, LSL #1]\n"
121 "ld1h { z13.h }, p2/Z, [x23, x15, LSL #1]\n"
122 "ld1h { z14.h }, p2/Z, [x22, x15, LSL #1]\n"
123 "ld1h { z15.h }, p2/Z, [x21, x15, LSL #1]\n"
124 "ld1h { z16.h }, p2/Z, [x20, x15, LSL #1]\n"
127 "movprfx z28, z17\n fmla z28.h, p3/M, z8.h, z9.h\n"
128 "movprfx z29, z17\n fmla z29.h, p3/M, z6.h, z9.h\n"
129 "ldr x27, [x16, #0x40]\n"
130 "whilelt p1.h, x13, %x[n_channels]\n"
131 "ldr x26, [x16, #0x48]\n"
132 "movprfx z30, z17\n fmla z30.h, p3/M, z2.h, z9.h\n"
133 "movprfx z31, z17\n fmla z31.h, p3/M, z0.h, z9.h\n"
134 "ld1h { z17.h }, p3/Z, [x14]\n"
135 "ldr x25, [x16, #0x50]\n"
136 "addvl x14, x14, #1\n"
138 "ldr x24, [x16, #0x58]\n"
140 "fmla z28.h, p3/M, z0.h, z10.h\n"
141 "fmla z29.h, p3/M, z1.h, z12.h\n"
142 "ld1h { z12.h }, p2/Z, [x26, x15, LSL #1]\n"
143 "ldr x20, [x16, #0x78]\n"
144 "ldr x23, [x16, #0x60]\n"
145 "ldr x22, [x16, #0x68]\n"
146 "fmla z28.h, p3/M, z1.h, z11.h\n"
147 "ld1h { z11.h }, p2/Z, [x27, x15, LSL #1]\n"
148 "fmla z29.h, p3/M, z2.h, z13.h\n"
149 "ld1h { z13.h }, p2/Z, [x25, x15, LSL #1]\n"
150 "ldr x27, [x16, #0x80]\n"
151 "ldr x26, [x16, #0x88]\n"
152 "ldr x21, [x16, #0x70]\n"
153 "fmla z28.h, p3/M, z3.h, z14.h\n"
154 "ld1h { z14.h }, p2/Z, [x24, x15, LSL #1]\n"
155 "fmla z29.h, p3/M, z0.h, z16.h\n"
156 "ldr x24, [x16, #0x98]\n"
157 "ldr x25, [x16, #0x90]\n"
158 "fmla z30.h, p3/M, z3.h, z14.h\n"
159 "ld1h { z14.h }, p2/Z, [x26, x15, LSL #1]\n"
160 "fmla z28.h, p3/M, z4.h, z15.h\n"
161 "ld1h { z15.h }, p2/Z, [x23, x15, LSL #1]\n"
162 "ldr x23, [x16, #0xa0]\n"
163 "fmla z29.h, p3/M, z4.h, z11.h\n"
164 "ld1h { z11.h }, p2/Z, [x22, x15, LSL #1]\n"
165 "ldr x22, [x16, #0xa8]\n"
166 "fmla z28.h, p3/M, z2.h, z16.h\n"
167 "ld1h { z16.h }, p2/Z, [x21, x15, LSL #1]\n"
168 "ldr x21, [x16, #0xb0]\n"
169 "fmla z30.h, p3/M, z0.h, z15.h\n"
170 "fmla z29.h, p3/M, z5.h, z12.h\n"
171 "ld1h { z12.h }, p2/Z, [x27, x15, LSL #1]\n"
172 "ldr x27, [x16, #0xc0]\n"
173 "fmla z28.h, p3/M, z5.h, z13.h\n"
174 "fmla z29.h, p3/M, z3.h, z13.h\n"
175 "ld1h { z13.h }, p2/Z, [x20, x15, LSL #1]\n"
176 "ldr x20, [x16, #0xb8]\n"
177 "fmla z30.h, p3/M, z4.h, z11.h\n"
178 "ld1h { z11.h }, p2/Z, [x24, x15, LSL #1]\n"
179 "fmla z31.h, p3/M, z4.h, z13.h\n"
180 "ld1h { z13.h }, p2/Z, [x23, x15, LSL #1]\n"
181 "fmla z28.h, p3/M, z6.h, z15.h\n"
182 "ld1h { z15.h }, p2/Z, [x25, x15, LSL #1]\n"
183 "fmla z29.h, p3/M, z7.h, z12.h\n"
184 "fmla z30.h, p3/M, z1.h, z16.h\n"
185 "fmla z31.h, p3/M, z1.h, z12.h\n"
186 "fmla z28.h, p3/M, z7.h, z16.h\n"
187 "ld1h { z16.h }, p2/Z, [x22, x15, LSL #1]\n"
188 "fmla z30.h, p3/M, z6.h, z15.h\n"
189 "ld1h { z15.h }, p2/Z, [x20, x15, LSL #1]\n"
190 "fmla z29.h, p3/M, z8.h, z11.h\n"
191 "fmla z31.h, p3/M, z5.h, z14.h\n"
192 "ld1h { z14.h }, p2/Z, [x21, x15, LSL #1]\n"
193 "fmla z30.h, p3/M, z7.h, z13.h\n"
194 "fmla z31.h, p3/M, z2.h, z11.h\n"
195 "ld1h { z11.h }, p2/Z, [x27, x15, LSL #1]\n"
196 "ldp x27, x26, [x16, #0x0]\n"
198 "ldp x25, x24, [x16, #0x10]\n"
199 "whilelt p2.h, x15, %x[n_channels]\n"
200 "ldp x23, x22, [x16, #0x20]\n"
201 "fmla z30.h, p3/M, z5.h, z16.h\n"
202 "ldp x21, x20, [x16, #0x30]\n"
203 "ld1h { z9.h }, p1/Z, [x27, x13, LSL #1]\n"
204 "fmla z31.h, p3/M, z3.h, z16.h\n"
205 "ld1h { z10.h }, p1/Z, [x26, x13, LSL #1]\n"
206 "ld1h { z12.h }, p1/Z, [x24, x13, LSL #1]\n"
207 "fmla z30.h, p3/M, z8.h, z15.h\n"
208 "ld1h { z13.h }, p1/Z, [x23, x13, LSL #1]\n"
209 "fmla z31.h, p3/M, z7.h, z14.h\n"
210 "ld1h { z14.h }, p1/Z, [x22, x13, LSL #1]\n"
211 "ld1h { z16.h }, p1/Z, [x20, x13, LSL #1]\n"
212 ".inst 0xa040a1c0 // ld1h { z0.h-z3.h }, pn8.b/Z, [x14]\n"
213 "addvl x14, x14, #4\n"
214 "fmla z31.h, p3/M, z6.h, z15.h\n"
215 "ld1h { z15.h }, p1/Z, [x21, x13, LSL #1]\n"
216 ".inst 0xa040a1c4 // ld1h { z4.h-z7.h }, pn8.b/Z, [x14]\n"
217 "addvl x14, x14, #4\n"
218 "fmla z31.h, p3/M, z8.h, z11.h\n"
219 "ld1h { z11.h }, p1/Z, [x25, x13, LSL #1]\n"
221 "cmp x13, %x[n_channels]\n"
222 "ld1h { z8.h }, p3/Z, [x14]\n"
223 "addvl x14, x14, #1\n"
224 ".inst 0xc172ca7c // fclamp { z28.h-z31.h }, z19.h, z18.h\n"
225 "st1h { z28.h }, p0, [x12, x28, LSL #1]\n"
226 "st1h { z29.h }, p0, [x11, x28, LSL #1]\n"
227 "st1h { z30.h }, p0, [x10, x28, LSL #1]\n"
228 "st1h { z31.h }, p0, [x9, x28, LSL #1]\n"
231 "movprfx z28, z17\n fmla z28.h, p3/M, z8.h, z9.h\n"
232 "movprfx z29, z17\n fmla z29.h, p3/M, z6.h, z9.h\n"
233 "ldr x27, [x16, #0x40]\n"
235 "ldr x26, [x16, #0x48]\n"
236 "movprfx z30, z17\n fmla z30.h, p3/M, z2.h, z9.h\n"
237 "movprfx z31, z17\n fmla z31.h, p3/M, z0.h, z9.h\n"
239 "ldr x25, [x16, #0x50]\n"
240 "ldr x24, [x16, #0x58]\n"
241 "fmla z28.h, p3/M, z0.h, z10.h\n"
242 "fmla z29.h, p3/M, z1.h, z12.h\n"
243 "ld1h { z12.h }, p2/Z, [x26, x15, LSL #1]\n"
244 "ldr x20, [x16, #0x78]\n"
245 "ldr x23, [x16, #0x60]\n"
246 "ldr x22, [x16, #0x68]\n"
247 "fmla z28.h, p3/M, z1.h, z11.h\n"
248 "ld1h { z11.h }, p2/Z, [x27, x15, LSL #1]\n"
249 "fmla z29.h, p3/M, z2.h, z13.h\n"
250 "ld1h { z13.h }, p2/Z, [x25, x15, LSL #1]\n"
251 "ldr x27, [x16, #0x80]\n"
252 "ldr x26, [x16, #0x88]\n"
253 "ldr x21, [x16, #0x70]\n"
254 "fmla z28.h, p3/M, z3.h, z14.h\n"
255 "ld1h { z14.h }, p2/Z, [x24, x15, LSL #1]\n"
256 "fmla z29.h, p3/M, z0.h, z16.h\n"
257 "ldr x24, [x16, #0x98]\n"
258 "ldr x25, [x16, #0x90]\n"
259 "fmla z30.h, p3/M, z3.h, z14.h\n"
260 "ld1h { z14.h }, p2/Z, [x26, x15, LSL #1]\n"
261 "fmla z28.h, p3/M, z4.h, z15.h\n"
262 "ld1h { z15.h }, p2/Z, [x23, x15, LSL #1]\n"
263 "ldr x23, [x16, #0xa0]\n"
264 "fmla z29.h, p3/M, z4.h, z11.h\n"
265 "ld1h { z11.h }, p2/Z, [x22, x15, LSL #1]\n"
266 "ldr x22, [x16, #0xa8]\n"
267 "fmla z28.h, p3/M, z2.h, z16.h\n"
268 "ld1h { z16.h }, p2/Z, [x21, x15, LSL #1]\n"
269 "ldr x21, [x16, #0xb0]\n"
270 "fmla z30.h, p3/M, z0.h, z15.h\n"
271 "fmla z29.h, p3/M, z5.h, z12.h\n"
272 "ld1h { z12.h }, p2/Z, [x27, x15, LSL #1]\n"
273 "ldr x27, [x16, #0xc0]\n"
274 "fmla z28.h, p3/M, z5.h, z13.h\n"
275 "fmla z29.h, p3/M, z3.h, z13.h\n"
276 "ld1h { z13.h }, p2/Z, [x20, x15, LSL #1]\n"
277 "ldr x20, [x16, #0xb8]\n"
278 "fmla z30.h, p3/M, z4.h, z11.h\n"
279 "ld1h { z11.h }, p2/Z, [x24, x15, LSL #1]\n"
280 "fmla z31.h, p3/M, z4.h, z13.h\n"
281 "ld1h { z13.h }, p2/Z, [x23, x15, LSL #1]\n"
282 "fmla z28.h, p3/M, z6.h, z15.h\n"
283 "ld1h { z15.h }, p2/Z, [x25, x15, LSL #1]\n"
284 "fmla z29.h, p3/M, z7.h, z12.h\n"
285 "fmla z30.h, p3/M, z1.h, z16.h\n"
286 "fmla z31.h, p3/M, z1.h, z12.h\n"
287 "fmla z28.h, p3/M, z7.h, z16.h\n"
288 "ld1h { z16.h }, p2/Z, [x22, x15, LSL #1]\n"
289 "fmla z30.h, p3/M, z6.h, z15.h\n"
290 "ld1h { z15.h }, p2/Z, [x20, x15, LSL #1]\n"
291 "fmla z29.h, p3/M, z8.h, z11.h\n"
292 "fmla z31.h, p3/M, z5.h, z14.h\n"
293 "ld1h { z14.h }, p2/Z, [x21, x15, LSL #1]\n"
294 "fmla z30.h, p3/M, z7.h, z13.h\n"
295 "fmla z31.h, p3/M, z2.h, z11.h\n"
296 "ld1h { z11.h }, p2/Z, [x27, x15, LSL #1]\n"
297 "fmla z30.h, p3/M, z5.h, z16.h\n"
298 "fmla z31.h, p3/M, z3.h, z16.h\n"
299 "fmla z30.h, p3/M, z8.h, z15.h\n"
300 "fmla z31.h, p3/M, z7.h, z14.h\n"
301 "fmla z31.h, p3/M, z6.h, z15.h\n"
302 "fmla z31.h, p3/M, z8.h, z11.h\n"
303 ".inst 0xc172ca7c // fclamp { z28.h-z31.h }, z19.h, z18.h\n"
304 "st1h { z28.h }, p0, [x12, x28, LSL #1]\n"
305 "st1h { z29.h }, p0, [x11, x28, LSL #1]\n"
306 "st1h { z30.h }, p0, [x10, x28, LSL #1]\n"
307 "st1h { z31.h }, p0, [x9, x28, LSL #1]\n"
308 ".inst 0xd503467f // SMSTOP\n"
310 : [n_channels]
"r" ((
unsigned long) n_channels), [offsetof_Args_inptrs]
"I" (offsetof(Args, inptrs)), [offsetof_args_max]
"I" (offsetof(Args, max)), [offsetof_args_min]
"I" (offsetof(Args, min)), [offsetof_args_outptrs]
"I" (offsetof(Args, outptrs)), [offsetof_args_params]
"I" (offsetof(Args, params)), [params_struct]
"r" (¶ms_struct)
311 :
"cc",
"memory",
"p0",
"p1",
"p2",
"p3",
"p4",
"p5",
"p6",
"p7",
"p8",
"p9",
"p10",
"p11",
"p12",
"p13",
"p14",
"p15",
"x9",
"x10",
"x11",
"x12",
"x13",
"x14",
"x15",
"x16",
"x20",
"x21",
"x22",
"x23",
"x24",
"x25",
"x26",
"x27",
"x28",
"z0",
"z1",
"z2",
"z3",
"z4",
"z5",
"z6",
"z7",
"z8",
"z9",
"z10",
"z11",
"z12",
"z13",
"z14",
"z15",
"z16",
"z17",
"z18",
"z19",
"z20",
"z21",
"z22",
"z23",
"z24",
"z25",
"z26",
"z27",
"z28",
"z29",
"z30",
"z31"
318 #endif // defined(ARM_COMPUTE_ENABLE_SME2)