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[16];
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[5];
58 inptrs[1] = input_ptrs[0];
59 inptrs[2] = input_ptrs[3];
60 inptrs[3] = input_ptrs[6];
61 inptrs[4] = input_ptrs[9];
62 inptrs[5] = input_ptrs[12];
63 inptrs[6] = input_ptrs[15];
64 inptrs[7] = input_ptrs[1];
65 inptrs[8] = input_ptrs[2];
66 inptrs[9] = input_ptrs[10];
67 inptrs[10] = input_ptrs[4];
68 inptrs[11] = input_ptrs[7];
69 inptrs[12] = input_ptrs[8];
70 inptrs[13] = input_ptrs[11];
71 inptrs[14] = input_ptrs[13];
72 inptrs[15] = input_ptrs[14];
77 Args params_struct(input_ptrs, outptrs, params,
81 "ldr x20, [%x[params_struct], %[offsetof_args_outptrs]]\n"
82 ".inst 0xd503477f // SMSTART ZA\n"
83 "add x16, %x[params_struct], %[offsetof_Args_inptrs]\n"
85 "ldr x14, [%x[params_struct], %[offsetof_args_params]]\n"
87 ".inst 0x25207810 // ptrue pn8.b\n"
88 "ldr x13, [x16, #0x20]\n"
90 "whilelt p2.h, XZR, %x[n_channels]\n"
91 "ld1rh { z18.h }, p3/Z, [%x[params_struct], %[offsetof_args_min]]\n"
92 "ldp x11, x10, [x20, #0x0]\n"
93 "cmp x12, %x[n_channels]\n"
94 "ld1rh { z17.h }, p3/Z, [%x[params_struct], %[offsetof_args_max]]\n"
96 "ldp x28, x27, [x20, #0x10]\n"
97 "ld1h { z16.h }, p3/Z, [x14]\n"
98 "addvl x14, x14, #1\n"
99 "ldp x26, x25, [x16, #0x0]\n"
100 ".inst 0xa040a1c0 // ld1h { z0.h-z3.h }, pn8.b/Z, [x14]\n"
101 "addvl x14, x14, #4\n"
102 "ldp x24, x23, [x16, #0x10]\n"
103 ".inst 0xa040a1c4 // ld1h { z4.h-z7.h }, pn8.b/Z, [x14]\n"
104 "addvl x14, x14, #4\n"
105 "ld1h { z8.h }, p3/Z, [x14]\n"
106 "addvl x14, x14, #1\n"
107 "ld1h { z9.h }, p2/Z, [x26, x15, LSL #1]\n"
108 "ld1h { z10.h }, p2/Z, [x25, x15, LSL #1]\n"
109 "ld1h { z11.h }, p2/Z, [x24, x15, LSL #1]\n"
110 "ld1h { z12.h }, p2/Z, [x23, x15, LSL #1]\n"
111 "ld1h { z13.h }, p2/Z, [x13, x15, LSL #1]\n"
114 "movprfx z28, z16\n fmla z28.h, p3/M, z4.h, z9.h\n"
115 "movprfx z29, z16\n fmla z29.h, p3/M, z3.h, z9.h\n"
116 "ldr x22, [x16, #0x28]\n"
117 "whilelt p1.h, x12, %x[n_channels]\n"
118 "movprfx z30, z16\n fmla z30.h, p3/M, z1.h, z9.h\n"
119 "movprfx z31, z16\n fmla z31.h, p3/M, z0.h, z9.h\n"
120 "ldr x21, [x16, #0x30]\n"
121 "ld1h { z16.h }, p3/Z, [x14]\n"
122 "ldr x20, [x16, #0x38]\n"
123 "addvl x14, x14, #1\n"
125 "ld1h { z9.h }, p2/Z, [x22, x15, LSL #1]\n"
126 "ldr x25, [x16, #0x48]\n"
128 "fmla z28.h, p3/M, z0.h, z10.h\n"
129 "fmla z29.h, p3/M, z2.h, z11.h\n"
130 "ld1h { z11.h }, p2/Z, [x21, x15, LSL #1]\n"
131 "ldr x26, [x16, #0x40]\n"
132 "fmla z30.h, p3/M, z2.h, z12.h\n"
133 "fmla z31.h, p3/M, z1.h, z12.h\n"
134 "ldr x24, [x16, #0x50]\n"
135 "ld1h { z10.h }, p2/Z, [x25, x15, LSL #1]\n"
136 "ldr x23, [x16, #0x58]\n"
137 "fmla z28.h, p3/M, z5.h, z12.h\n"
138 "fmla z29.h, p3/M, z4.h, z12.h\n"
139 "ld1h { z12.h }, p2/Z, [x20, x15, LSL #1]\n"
140 "ldr x13, [x16, #0x60]\n"
141 "fmla z30.h, p3/M, z6.h, z9.h\n"
142 "ld1h { z9.h }, p2/Z, [x26, x15, LSL #1]\n"
143 "fmla z31.h, p3/M, z3.h, z13.h\n"
144 "ldr x22, [x16, #0x68]\n"
145 "ldr x21, [x16, #0x70]\n"
146 "fmla z28.h, p3/M, z7.h, z13.h\n"
147 "fmla z29.h, p3/M, z6.h, z13.h\n"
148 "ldr x20, [x16, #0x78]\n"
149 "fmla z30.h, p3/M, z4.h, z13.h\n"
150 "fmla z31.h, p3/M, z8.h, z11.h\n"
151 "ld1h { z11.h }, p2/Z, [x24, x15, LSL #1]\n"
152 "ldp x26, x25, [x16, #0x0]\n"
153 "fmla z28.h, p3/M, z1.h, z12.h\n"
154 "fmla z29.h, p3/M, z0.h, z12.h\n"
155 "ld1h { z12.h }, p2/Z, [x23, x15, LSL #1]\n"
156 "ldp x24, x23, [x16, #0x10]\n"
157 "fmla z30.h, p3/M, z5.h, z10.h\n"
158 "fmla z31.h, p3/M, z4.h, z10.h\n"
159 "fmla z28.h, p3/M, z2.h, z9.h\n"
160 "fmla z29.h, p3/M, z1.h, z9.h\n"
161 "ld1h { z9.h }, p2/Z, [x13, x15, LSL #1]\n"
162 "ldr x13, [x16, #0x20]\n"
163 "fmla z30.h, p3/M, z0.h, z11.h\n"
164 "fmla z31.h, p3/M, z2.h, z12.h\n"
165 "ld1h { z13.h }, p1/Z, [x13, x12, LSL #1]\n"
166 "fmla z28.h, p3/M, z8.h, z10.h\n"
167 "fmla z29.h, p3/M, z7.h, z10.h\n"
168 "ld1h { z10.h }, p2/Z, [x22, x15, LSL #1]\n"
169 "fmla z30.h, p3/M, z3.h, z9.h\n"
170 "fmla z31.h, p3/M, z5.h, z10.h\n"
171 "fmla z28.h, p3/M, z3.h, z11.h\n"
172 "ld1h { z11.h }, p2/Z, [x21, x15, LSL #1]\n"
173 ".inst 0xa040a1c0 // ld1h { z0.h-z3.h }, pn8.b/Z, [x14]\n"
174 "addvl x14, x14, #4\n"
175 "fmla z29.h, p3/M, z5.h, z12.h\n"
176 "ld1h { z12.h }, p2/Z, [x20, x15, LSL #1]\n"
178 "fmla z30.h, p3/M, z7.h, z11.h\n"
179 "fmla z31.h, p3/M, z6.h, z11.h\n"
180 "ld1h { z11.h }, p1/Z, [x24, x12, LSL #1]\n"
181 "whilelt p2.h, x15, %x[n_channels]\n"
182 "fmla z28.h, p3/M, z6.h, z9.h\n"
183 "ld1h { z9.h }, p1/Z, [x26, x12, LSL #1]\n"
184 "fmla z29.h, p3/M, z8.h, z10.h\n"
185 "ld1h { z10.h }, p1/Z, [x25, x12, LSL #1]\n"
186 "fmla z30.h, p3/M, z8.h, z12.h\n"
187 "fmla z31.h, p3/M, z7.h, z12.h\n"
188 "ld1h { z12.h }, p1/Z, [x23, x12, LSL #1]\n"
190 ".inst 0xa040a1c4 // ld1h { z4.h-z7.h }, pn8.b/Z, [x14]\n"
191 "addvl x14, x14, #4\n"
192 "cmp x12, %x[n_channels]\n"
193 "ld1h { z8.h }, p3/Z, [x14]\n"
194 "addvl x14, x14, #1\n"
195 ".inst 0xc171ca5c // fclamp { z28.h-z31.h }, z18.h, z17.h\n"
196 "st1h { z28.h }, p0, [x11, x9, LSL #1]\n"
197 "st1h { z29.h }, p0, [x10, x9, LSL #1]\n"
198 "st1h { z30.h }, p0, [x28, x9, LSL #1]\n"
199 "st1h { z31.h }, p0, [x27, x9, LSL #1]\n"
202 "movprfx z28, z16\n fmla z28.h, p3/M, z4.h, z9.h\n"
203 "movprfx z29, z16\n fmla z29.h, p3/M, z3.h, z9.h\n"
204 "ldr x22, [x16, #0x28]\n"
206 "movprfx z30, z16\n fmla z30.h, p3/M, z1.h, z9.h\n"
207 "movprfx z31, z16\n fmla z31.h, p3/M, z0.h, z9.h\n"
208 "ldr x21, [x16, #0x30]\n"
210 "ldr x20, [x16, #0x38]\n"
211 "ld1h { z9.h }, p2/Z, [x22, x15, LSL #1]\n"
212 "ldr x25, [x16, #0x48]\n"
213 "fmla z28.h, p3/M, z0.h, z10.h\n"
214 "fmla z29.h, p3/M, z2.h, z11.h\n"
215 "ld1h { z11.h }, p2/Z, [x21, x15, LSL #1]\n"
216 "ldr x26, [x16, #0x40]\n"
217 "fmla z30.h, p3/M, z2.h, z12.h\n"
218 "fmla z31.h, p3/M, z1.h, z12.h\n"
219 "ldr x24, [x16, #0x50]\n"
220 "ld1h { z10.h }, p2/Z, [x25, x15, LSL #1]\n"
221 "ldr x23, [x16, #0x58]\n"
222 "fmla z28.h, p3/M, z5.h, z12.h\n"
223 "fmla z29.h, p3/M, z4.h, z12.h\n"
224 "ld1h { z12.h }, p2/Z, [x20, x15, LSL #1]\n"
225 "ldr x13, [x16, #0x60]\n"
226 "fmla z30.h, p3/M, z6.h, z9.h\n"
227 "ld1h { z9.h }, p2/Z, [x26, x15, LSL #1]\n"
228 "fmla z31.h, p3/M, z3.h, z13.h\n"
229 "ldr x22, [x16, #0x68]\n"
230 "ldr x21, [x16, #0x70]\n"
231 "fmla z28.h, p3/M, z7.h, z13.h\n"
232 "fmla z29.h, p3/M, z6.h, z13.h\n"
233 "ldr x20, [x16, #0x78]\n"
234 "fmla z30.h, p3/M, z4.h, z13.h\n"
235 "fmla z31.h, p3/M, z8.h, z11.h\n"
236 "ld1h { z11.h }, p2/Z, [x24, x15, LSL #1]\n"
237 "fmla z28.h, p3/M, z1.h, z12.h\n"
238 "fmla z29.h, p3/M, z0.h, z12.h\n"
239 "ld1h { z12.h }, p2/Z, [x23, x15, LSL #1]\n"
240 "fmla z30.h, p3/M, z5.h, z10.h\n"
241 "fmla z31.h, p3/M, z4.h, z10.h\n"
242 "fmla z28.h, p3/M, z2.h, z9.h\n"
243 "fmla z29.h, p3/M, z1.h, z9.h\n"
244 "ld1h { z9.h }, p2/Z, [x13, x15, LSL #1]\n"
245 "fmla z30.h, p3/M, z0.h, z11.h\n"
246 "fmla z31.h, p3/M, z2.h, z12.h\n"
247 "fmla z28.h, p3/M, z8.h, z10.h\n"
248 "fmla z29.h, p3/M, z7.h, z10.h\n"
249 "ld1h { z10.h }, p2/Z, [x22, x15, LSL #1]\n"
250 "fmla z30.h, p3/M, z3.h, z9.h\n"
251 "fmla z31.h, p3/M, z5.h, z10.h\n"
252 "fmla z28.h, p3/M, z3.h, z11.h\n"
253 "ld1h { z11.h }, p2/Z, [x21, x15, LSL #1]\n"
254 "fmla z29.h, p3/M, z5.h, z12.h\n"
255 "ld1h { z12.h }, p2/Z, [x20, x15, LSL #1]\n"
256 "fmla z30.h, p3/M, z7.h, z11.h\n"
257 "fmla z31.h, p3/M, z6.h, z11.h\n"
258 "fmla z28.h, p3/M, z6.h, z9.h\n"
259 "fmla z29.h, p3/M, z8.h, z10.h\n"
260 "fmla z30.h, p3/M, z8.h, z12.h\n"
261 "fmla z31.h, p3/M, z7.h, z12.h\n"
262 ".inst 0xc171ca5c // fclamp { z28.h-z31.h }, z18.h, z17.h\n"
263 "st1h { z28.h }, p0, [x11, x9, LSL #1]\n"
264 "st1h { z29.h }, p0, [x10, x9, LSL #1]\n"
265 "st1h { z30.h }, p0, [x28, x9, LSL #1]\n"
266 "st1h { z31.h }, p0, [x27, x9, LSL #1]\n"
267 ".inst 0xd503467f // SMSTOP\n"
269 : [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)
270 :
"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"
277 #endif // defined(ARM_COMPUTE_ENABLE_SME2)