• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Auto-generated file. Do not edit!
2 //   Template: src/f32-dwconv/up-avx.c.in
3 //   Generator: tools/xngen
4 //
5 // Copyright 2019 Google LLC
6 //
7 // This source code is licensed under the BSD-style license found in the
8 // LICENSE file in the root directory of this source tree.
9 
10 #include <assert.h>
11 
12 #include <immintrin.h>
13 
14 #include <xnnpack/dwconv.h>
15 
16 
17 static const int32_t mask_table[14] = {-1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0};
18 
xnn_f32_dwconv_ukernel_up16x25__avx_acc2(size_t channels,size_t output_width,const float ** input,const float * weights,float * output,size_t input_stride,size_t output_increment,const union xnn_f32_output_params params[restrict static1])19 void xnn_f32_dwconv_ukernel_up16x25__avx_acc2(
20     size_t channels,
21     size_t output_width,
22     const float** input,
23     const float* weights,
24     float* output,
25     size_t input_stride,
26     size_t output_increment,
27     const union xnn_f32_output_params params[restrict static 1])
28 {
29   assert(channels != 0);
30   assert(output_width != 0);
31 
32   const __m256 vmax = _mm256_broadcast_ps((const __m128*) params->sse.max);
33   const __m256 vmin = _mm256_broadcast_ps((const __m128*) params->sse.min);
34   do {
35     const float* i0 = input[0];
36     assert(i0 != NULL);
37     const float* i1 = input[1];
38     assert(i1 != NULL);
39     const float* i2 = input[2];
40     assert(i2 != NULL);
41     const float* i3 = input[3];
42     assert(i3 != NULL);
43     const float* i4 = input[4];
44     assert(i4 != NULL);
45     const float* i5 = input[5];
46     assert(i5 != NULL);
47     const float* i6 = input[6];
48     assert(i6 != NULL);
49     const float* i7 = input[7];
50     assert(i7 != NULL);
51     const float* i8 = input[8];
52     assert(i8 != NULL);
53     const float* i9 = input[9];
54     assert(i9 != NULL);
55     const float* i10 = input[10];
56     assert(i10 != NULL);
57     const float* i11 = input[11];
58     assert(i11 != NULL);
59     const float* i12 = input[12];
60     assert(i12 != NULL);
61     const float* i13 = input[13];
62     assert(i13 != NULL);
63     const float* i14 = input[14];
64     assert(i14 != NULL);
65     const float* i15 = input[15];
66     assert(i15 != NULL);
67     const float* i16 = input[16];
68     assert(i16 != NULL);
69     const float* i17 = input[17];
70     assert(i17 != NULL);
71     const float* i18 = input[18];
72     assert(i18 != NULL);
73     const float* i19 = input[19];
74     assert(i19 != NULL);
75     const float* i20 = input[20];
76     assert(i20 != NULL);
77     const float* i21 = input[21];
78     assert(i21 != NULL);
79     const float* i22 = input[22];
80     assert(i22 != NULL);
81     const float* i23 = input[23];
82     assert(i23 != NULL);
83     const float* i24 = input[24];
84     assert(i24 != NULL);
85     input = (const float**) ((uintptr_t) input + input_stride);
86 
87     size_t c = channels;
88     const float* w = weights;
89     for (; c >= 16; c -= 16) {
90       __m256 vacc01234567p0 = _mm256_load_ps(w);
91       __m256 vacc89ABCDEFp0 = _mm256_load_ps(w + 8);
92 
93 
94       const __m256 vi0x01234567 = _mm256_loadu_ps(i0);
95       const __m256 vi0x89ABCDEF = _mm256_loadu_ps(i0 + 8);
96       i0 += 16;
97 
98       const __m256 vk0x01234567 = _mm256_load_ps(w + 16);
99       const __m256 vk0x89ABCDEF = _mm256_load_ps(w + 24);
100       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi0x01234567, vk0x01234567));
101       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi0x89ABCDEF, vk0x89ABCDEF));
102 
103       const __m256 vi1x01234567 = _mm256_loadu_ps(i1);
104       const __m256 vi1x89ABCDEF = _mm256_loadu_ps(i1 + 8);
105       i1 += 16;
106 
107       const __m256 vk1x01234567 = _mm256_load_ps(w + 32);
108       const __m256 vk1x89ABCDEF = _mm256_load_ps(w + 40);
109       __m256 vacc01234567p1 = _mm256_mul_ps(vi1x01234567, vk1x01234567);
110       __m256 vacc89ABCDEFp1 = _mm256_mul_ps(vi1x89ABCDEF, vk1x89ABCDEF);
111 
112       const __m256 vi2x01234567 = _mm256_loadu_ps(i2);
113       const __m256 vi2x89ABCDEF = _mm256_loadu_ps(i2 + 8);
114       i2 += 16;
115 
116       const __m256 vk2x01234567 = _mm256_load_ps(w + 48);
117       const __m256 vk2x89ABCDEF = _mm256_load_ps(w + 56);
118       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi2x01234567, vk2x01234567));
119       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi2x89ABCDEF, vk2x89ABCDEF));
120 
121       const __m256 vi3x01234567 = _mm256_loadu_ps(i3);
122       const __m256 vi3x89ABCDEF = _mm256_loadu_ps(i3 + 8);
123       i3 += 16;
124 
125       const __m256 vk3x01234567 = _mm256_load_ps(w + 64);
126       const __m256 vk3x89ABCDEF = _mm256_load_ps(w + 72);
127       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi3x01234567, vk3x01234567));
128       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi3x89ABCDEF, vk3x89ABCDEF));
129 
130       const __m256 vi4x01234567 = _mm256_loadu_ps(i4);
131       const __m256 vi4x89ABCDEF = _mm256_loadu_ps(i4 + 8);
132       i4 += 16;
133 
134       const __m256 vk4x01234567 = _mm256_load_ps(w + 80);
135       const __m256 vk4x89ABCDEF = _mm256_load_ps(w + 88);
136       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi4x01234567, vk4x01234567));
137       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi4x89ABCDEF, vk4x89ABCDEF));
138 
139       const __m256 vi5x01234567 = _mm256_loadu_ps(i5);
140       const __m256 vi5x89ABCDEF = _mm256_loadu_ps(i5 + 8);
141       i5 += 16;
142 
143       const __m256 vk5x01234567 = _mm256_load_ps(w + 96);
144       const __m256 vk5x89ABCDEF = _mm256_load_ps(w + 104);
145       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi5x01234567, vk5x01234567));
146       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi5x89ABCDEF, vk5x89ABCDEF));
147 
148       const __m256 vi6x01234567 = _mm256_loadu_ps(i6);
149       const __m256 vi6x89ABCDEF = _mm256_loadu_ps(i6 + 8);
150       i6 += 16;
151 
152       const __m256 vk6x01234567 = _mm256_load_ps(w + 112);
153       const __m256 vk6x89ABCDEF = _mm256_load_ps(w + 120);
154       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi6x01234567, vk6x01234567));
155       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi6x89ABCDEF, vk6x89ABCDEF));
156 
157       const __m256 vi7x01234567 = _mm256_loadu_ps(i7);
158       const __m256 vi7x89ABCDEF = _mm256_loadu_ps(i7 + 8);
159       i7 += 16;
160 
161       const __m256 vk7x01234567 = _mm256_load_ps(w + 128);
162       const __m256 vk7x89ABCDEF = _mm256_load_ps(w + 136);
163       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi7x01234567, vk7x01234567));
164       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi7x89ABCDEF, vk7x89ABCDEF));
165 
166       const __m256 vi8x01234567 = _mm256_loadu_ps(i8);
167       const __m256 vi8x89ABCDEF = _mm256_loadu_ps(i8 + 8);
168       i8 += 16;
169 
170       const __m256 vk8x01234567 = _mm256_load_ps(w + 144);
171       const __m256 vk8x89ABCDEF = _mm256_load_ps(w + 152);
172       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi8x01234567, vk8x01234567));
173       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi8x89ABCDEF, vk8x89ABCDEF));
174 
175       const __m256 vi9x01234567 = _mm256_loadu_ps(i9);
176       const __m256 vi9x89ABCDEF = _mm256_loadu_ps(i9 + 8);
177       i9 += 16;
178 
179       const __m256 vk9x01234567 = _mm256_load_ps(w + 160);
180       const __m256 vk9x89ABCDEF = _mm256_load_ps(w + 168);
181       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi9x01234567, vk9x01234567));
182       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi9x89ABCDEF, vk9x89ABCDEF));
183 
184       const __m256 vi10x01234567 = _mm256_loadu_ps(i10);
185       const __m256 vi10x89ABCDEF = _mm256_loadu_ps(i10 + 8);
186       i10 += 16;
187 
188       const __m256 vk10x01234567 = _mm256_load_ps(w + 176);
189       const __m256 vk10x89ABCDEF = _mm256_load_ps(w + 184);
190       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi10x01234567, vk10x01234567));
191       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi10x89ABCDEF, vk10x89ABCDEF));
192 
193       const __m256 vi11x01234567 = _mm256_loadu_ps(i11);
194       const __m256 vi11x89ABCDEF = _mm256_loadu_ps(i11 + 8);
195       i11 += 16;
196 
197       const __m256 vk11x01234567 = _mm256_load_ps(w + 192);
198       const __m256 vk11x89ABCDEF = _mm256_load_ps(w + 200);
199       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi11x01234567, vk11x01234567));
200       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi11x89ABCDEF, vk11x89ABCDEF));
201 
202       const __m256 vi12x01234567 = _mm256_loadu_ps(i12);
203       const __m256 vi12x89ABCDEF = _mm256_loadu_ps(i12 + 8);
204       i12 += 16;
205 
206       const __m256 vk12x01234567 = _mm256_load_ps(w + 208);
207       const __m256 vk12x89ABCDEF = _mm256_load_ps(w + 216);
208       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi12x01234567, vk12x01234567));
209       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi12x89ABCDEF, vk12x89ABCDEF));
210 
211       const __m256 vi13x01234567 = _mm256_loadu_ps(i13);
212       const __m256 vi13x89ABCDEF = _mm256_loadu_ps(i13 + 8);
213       i13 += 16;
214 
215       const __m256 vk13x01234567 = _mm256_load_ps(w + 224);
216       const __m256 vk13x89ABCDEF = _mm256_load_ps(w + 232);
217       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi13x01234567, vk13x01234567));
218       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi13x89ABCDEF, vk13x89ABCDEF));
219 
220       const __m256 vi14x01234567 = _mm256_loadu_ps(i14);
221       const __m256 vi14x89ABCDEF = _mm256_loadu_ps(i14 + 8);
222       i14 += 16;
223 
224       const __m256 vk14x01234567 = _mm256_load_ps(w + 240);
225       const __m256 vk14x89ABCDEF = _mm256_load_ps(w + 248);
226       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi14x01234567, vk14x01234567));
227       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi14x89ABCDEF, vk14x89ABCDEF));
228 
229       const __m256 vi15x01234567 = _mm256_loadu_ps(i15);
230       const __m256 vi15x89ABCDEF = _mm256_loadu_ps(i15 + 8);
231       i15 += 16;
232 
233       const __m256 vk15x01234567 = _mm256_load_ps(w + 256);
234       const __m256 vk15x89ABCDEF = _mm256_load_ps(w + 264);
235       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi15x01234567, vk15x01234567));
236       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi15x89ABCDEF, vk15x89ABCDEF));
237 
238       const __m256 vi16x01234567 = _mm256_loadu_ps(i16);
239       const __m256 vi16x89ABCDEF = _mm256_loadu_ps(i16 + 8);
240       i16 += 16;
241 
242       const __m256 vk16x01234567 = _mm256_load_ps(w + 272);
243       const __m256 vk16x89ABCDEF = _mm256_load_ps(w + 280);
244       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi16x01234567, vk16x01234567));
245       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi16x89ABCDEF, vk16x89ABCDEF));
246 
247       const __m256 vi17x01234567 = _mm256_loadu_ps(i17);
248       const __m256 vi17x89ABCDEF = _mm256_loadu_ps(i17 + 8);
249       i17 += 16;
250 
251       const __m256 vk17x01234567 = _mm256_load_ps(w + 288);
252       const __m256 vk17x89ABCDEF = _mm256_load_ps(w + 296);
253       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi17x01234567, vk17x01234567));
254       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi17x89ABCDEF, vk17x89ABCDEF));
255 
256       const __m256 vi18x01234567 = _mm256_loadu_ps(i18);
257       const __m256 vi18x89ABCDEF = _mm256_loadu_ps(i18 + 8);
258       i18 += 16;
259 
260       const __m256 vk18x01234567 = _mm256_load_ps(w + 304);
261       const __m256 vk18x89ABCDEF = _mm256_load_ps(w + 312);
262       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi18x01234567, vk18x01234567));
263       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi18x89ABCDEF, vk18x89ABCDEF));
264 
265       const __m256 vi19x01234567 = _mm256_loadu_ps(i19);
266       const __m256 vi19x89ABCDEF = _mm256_loadu_ps(i19 + 8);
267       i19 += 16;
268 
269       const __m256 vk19x01234567 = _mm256_load_ps(w + 320);
270       const __m256 vk19x89ABCDEF = _mm256_load_ps(w + 328);
271       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi19x01234567, vk19x01234567));
272       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi19x89ABCDEF, vk19x89ABCDEF));
273 
274       const __m256 vi20x01234567 = _mm256_loadu_ps(i20);
275       const __m256 vi20x89ABCDEF = _mm256_loadu_ps(i20 + 8);
276       i20 += 16;
277 
278       const __m256 vk20x01234567 = _mm256_load_ps(w + 336);
279       const __m256 vk20x89ABCDEF = _mm256_load_ps(w + 344);
280       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi20x01234567, vk20x01234567));
281       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi20x89ABCDEF, vk20x89ABCDEF));
282 
283       const __m256 vi21x01234567 = _mm256_loadu_ps(i21);
284       const __m256 vi21x89ABCDEF = _mm256_loadu_ps(i21 + 8);
285       i21 += 16;
286 
287       const __m256 vk21x01234567 = _mm256_load_ps(w + 352);
288       const __m256 vk21x89ABCDEF = _mm256_load_ps(w + 360);
289       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi21x01234567, vk21x01234567));
290       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi21x89ABCDEF, vk21x89ABCDEF));
291 
292       const __m256 vi22x01234567 = _mm256_loadu_ps(i22);
293       const __m256 vi22x89ABCDEF = _mm256_loadu_ps(i22 + 8);
294       i22 += 16;
295 
296       const __m256 vk22x01234567 = _mm256_load_ps(w + 368);
297       const __m256 vk22x89ABCDEF = _mm256_load_ps(w + 376);
298       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi22x01234567, vk22x01234567));
299       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi22x89ABCDEF, vk22x89ABCDEF));
300 
301       const __m256 vi23x01234567 = _mm256_loadu_ps(i23);
302       const __m256 vi23x89ABCDEF = _mm256_loadu_ps(i23 + 8);
303       i23 += 16;
304 
305       const __m256 vk23x01234567 = _mm256_load_ps(w + 384);
306       const __m256 vk23x89ABCDEF = _mm256_load_ps(w + 392);
307       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi23x01234567, vk23x01234567));
308       vacc89ABCDEFp1 = _mm256_add_ps(vacc89ABCDEFp1, _mm256_mul_ps(vi23x89ABCDEF, vk23x89ABCDEF));
309 
310       const __m256 vi24x01234567 = _mm256_loadu_ps(i24);
311       const __m256 vi24x89ABCDEF = _mm256_loadu_ps(i24 + 8);
312       i24 += 16;
313 
314       const __m256 vk24x01234567 = _mm256_load_ps(w + 400);
315       const __m256 vk24x89ABCDEF = _mm256_load_ps(w + 408);
316       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi24x01234567, vk24x01234567));
317       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, _mm256_mul_ps(vi24x89ABCDEF, vk24x89ABCDEF));
318 
319       w += 416;
320 
321       // Add up all accumulators to vacc0123456789ABCDEFp0
322       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, vacc01234567p1);
323       vacc89ABCDEFp0 = _mm256_add_ps(vacc89ABCDEFp0, vacc89ABCDEFp1);
324 
325       __m256 vacc01234567 = _mm256_max_ps(vacc01234567p0, vmin);
326       __m256 vacc89ABCDEF = _mm256_max_ps(vacc89ABCDEFp0, vmin);
327       vacc01234567 = _mm256_min_ps(vacc01234567, vmax);
328       vacc89ABCDEF = _mm256_min_ps(vacc89ABCDEF, vmax);
329 
330       _mm256_storeu_ps(output, vacc01234567);
331       _mm256_storeu_ps(output + 8, vacc89ABCDEF);
332       output += 16;
333     }
334     for (; c >= 8; c -= 8) {
335       __m256 vacc01234567p0 = _mm256_load_ps(w);
336 
337       const __m256 vi0x01234567 = _mm256_loadu_ps(i0);
338       i0 += 8;
339 
340       const __m256 vk0x01234567 = _mm256_load_ps(w + 16);
341       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi0x01234567, vk0x01234567));
342 
343       const __m256 vi1x01234567 = _mm256_loadu_ps(i1);
344       i1 += 8;
345 
346       const __m256 vk1x01234567 = _mm256_load_ps(w + 32);
347       __m256 vacc01234567p1 = _mm256_mul_ps(vi1x01234567, vk1x01234567);
348 
349       const __m256 vi2x01234567 = _mm256_loadu_ps(i2);
350       i2 += 8;
351 
352       const __m256 vk2x01234567 = _mm256_load_ps(w + 48);
353       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi2x01234567, vk2x01234567));
354 
355       const __m256 vi3x01234567 = _mm256_loadu_ps(i3);
356       i3 += 8;
357 
358       const __m256 vk3x01234567 = _mm256_load_ps(w + 64);
359       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi3x01234567, vk3x01234567));
360 
361       const __m256 vi4x01234567 = _mm256_loadu_ps(i4);
362       i4 += 8;
363 
364       const __m256 vk4x01234567 = _mm256_load_ps(w + 80);
365       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi4x01234567, vk4x01234567));
366 
367       const __m256 vi5x01234567 = _mm256_loadu_ps(i5);
368       i5 += 8;
369 
370       const __m256 vk5x01234567 = _mm256_load_ps(w + 96);
371       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi5x01234567, vk5x01234567));
372 
373       const __m256 vi6x01234567 = _mm256_loadu_ps(i6);
374       i6 += 8;
375 
376       const __m256 vk6x01234567 = _mm256_load_ps(w + 112);
377       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi6x01234567, vk6x01234567));
378 
379       const __m256 vi7x01234567 = _mm256_loadu_ps(i7);
380       i7 += 8;
381 
382       const __m256 vk7x01234567 = _mm256_load_ps(w + 128);
383       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi7x01234567, vk7x01234567));
384 
385       const __m256 vi8x01234567 = _mm256_loadu_ps(i8);
386       i8 += 8;
387 
388       const __m256 vk8x01234567 = _mm256_load_ps(w + 144);
389       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi8x01234567, vk8x01234567));
390 
391       const __m256 vi9x01234567 = _mm256_loadu_ps(i9);
392       i9 += 8;
393 
394       const __m256 vk9x01234567 = _mm256_load_ps(w + 160);
395       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi9x01234567, vk9x01234567));
396 
397       const __m256 vi10x01234567 = _mm256_loadu_ps(i10);
398       i10 += 8;
399 
400       const __m256 vk10x01234567 = _mm256_load_ps(w + 176);
401       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi10x01234567, vk10x01234567));
402 
403       const __m256 vi11x01234567 = _mm256_loadu_ps(i11);
404       i11 += 8;
405 
406       const __m256 vk11x01234567 = _mm256_load_ps(w + 192);
407       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi11x01234567, vk11x01234567));
408 
409       const __m256 vi12x01234567 = _mm256_loadu_ps(i12);
410       i12 += 8;
411 
412       const __m256 vk12x01234567 = _mm256_load_ps(w + 208);
413       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi12x01234567, vk12x01234567));
414 
415       const __m256 vi13x01234567 = _mm256_loadu_ps(i13);
416       i13 += 8;
417 
418       const __m256 vk13x01234567 = _mm256_load_ps(w + 224);
419       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi13x01234567, vk13x01234567));
420 
421       const __m256 vi14x01234567 = _mm256_loadu_ps(i14);
422       i14 += 8;
423 
424       const __m256 vk14x01234567 = _mm256_load_ps(w + 240);
425       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi14x01234567, vk14x01234567));
426 
427       const __m256 vi15x01234567 = _mm256_loadu_ps(i15);
428       i15 += 8;
429 
430       const __m256 vk15x01234567 = _mm256_load_ps(w + 256);
431       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi15x01234567, vk15x01234567));
432 
433       const __m256 vi16x01234567 = _mm256_loadu_ps(i16);
434       i16 += 8;
435 
436       const __m256 vk16x01234567 = _mm256_load_ps(w + 272);
437       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi16x01234567, vk16x01234567));
438 
439       const __m256 vi17x01234567 = _mm256_loadu_ps(i17);
440       i17 += 8;
441 
442       const __m256 vk17x01234567 = _mm256_load_ps(w + 288);
443       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi17x01234567, vk17x01234567));
444 
445       const __m256 vi18x01234567 = _mm256_loadu_ps(i18);
446       i18 += 8;
447 
448       const __m256 vk18x01234567 = _mm256_load_ps(w + 304);
449       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi18x01234567, vk18x01234567));
450 
451       const __m256 vi19x01234567 = _mm256_loadu_ps(i19);
452       i19 += 8;
453 
454       const __m256 vk19x01234567 = _mm256_load_ps(w + 320);
455       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi19x01234567, vk19x01234567));
456 
457       const __m256 vi20x01234567 = _mm256_loadu_ps(i20);
458       i20 += 8;
459 
460       const __m256 vk20x01234567 = _mm256_load_ps(w + 336);
461       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi20x01234567, vk20x01234567));
462 
463       const __m256 vi21x01234567 = _mm256_loadu_ps(i21);
464       i21 += 8;
465 
466       const __m256 vk21x01234567 = _mm256_load_ps(w + 352);
467       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi21x01234567, vk21x01234567));
468 
469       const __m256 vi22x01234567 = _mm256_loadu_ps(i22);
470       i22 += 8;
471 
472       const __m256 vk22x01234567 = _mm256_load_ps(w + 368);
473       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi22x01234567, vk22x01234567));
474 
475       const __m256 vi23x01234567 = _mm256_loadu_ps(i23);
476       i23 += 8;
477 
478       const __m256 vk23x01234567 = _mm256_load_ps(w + 384);
479       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi23x01234567, vk23x01234567));
480 
481       const __m256 vi24x01234567 = _mm256_loadu_ps(i24);
482       i24 += 8;
483 
484       const __m256 vk24x01234567 = _mm256_load_ps(w + 400);
485       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi24x01234567, vk24x01234567));
486 
487       w += 8;
488 
489       // Add up all accumulators to vacc01234567p0
490       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, vacc01234567p1);
491 
492       __m256 vacc01234567 = _mm256_max_ps(vacc01234567p0, vmin);
493       vacc01234567 = _mm256_min_ps(vacc01234567, vmax);
494 
495       _mm256_storeu_ps(output, vacc01234567);
496       output += 8;
497     }
498     if XNN_UNLIKELY(c != 0) {
499       assert(c >= 1);
500       assert(c <= 7);
501       __m256i vmask = _mm256_loadu_si256((const __m256i*) &mask_table[7 - c]);
502 
503       __m256 vacc01234567p0 = _mm256_load_ps(w);
504 
505       const __m256 vi0x01234567 = _mm256_maskload_ps(i0, vmask);
506       const __m256 vk0x01234567 = _mm256_load_ps(w + 16);
507       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi0x01234567, vk0x01234567));
508 
509       const __m256 vi1x01234567 = _mm256_maskload_ps(i1, vmask);
510       const __m256 vk1x01234567 = _mm256_load_ps(w + 32);
511       __m256 vacc01234567p1 = _mm256_mul_ps(vi1x01234567, vk1x01234567);
512 
513       const __m256 vi2x01234567 = _mm256_maskload_ps(i2, vmask);
514       const __m256 vk2x01234567 = _mm256_load_ps(w + 48);
515       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi2x01234567, vk2x01234567));
516 
517       const __m256 vi3x01234567 = _mm256_maskload_ps(i3, vmask);
518       const __m256 vk3x01234567 = _mm256_load_ps(w + 64);
519       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi3x01234567, vk3x01234567));
520 
521       const __m256 vi4x01234567 = _mm256_maskload_ps(i4, vmask);
522       const __m256 vk4x01234567 = _mm256_load_ps(w + 80);
523       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi4x01234567, vk4x01234567));
524 
525       const __m256 vi5x01234567 = _mm256_maskload_ps(i5, vmask);
526       const __m256 vk5x01234567 = _mm256_load_ps(w + 96);
527       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi5x01234567, vk5x01234567));
528 
529       const __m256 vi6x01234567 = _mm256_maskload_ps(i6, vmask);
530       const __m256 vk6x01234567 = _mm256_load_ps(w + 112);
531       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi6x01234567, vk6x01234567));
532 
533       const __m256 vi7x01234567 = _mm256_maskload_ps(i7, vmask);
534       const __m256 vk7x01234567 = _mm256_load_ps(w + 128);
535       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi7x01234567, vk7x01234567));
536 
537       const __m256 vi8x01234567 = _mm256_maskload_ps(i8, vmask);
538       const __m256 vk8x01234567 = _mm256_load_ps(w + 144);
539       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi8x01234567, vk8x01234567));
540 
541       const __m256 vi9x01234567 = _mm256_maskload_ps(i9, vmask);
542       const __m256 vk9x01234567 = _mm256_load_ps(w + 160);
543       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi9x01234567, vk9x01234567));
544 
545       const __m256 vi10x01234567 = _mm256_maskload_ps(i10, vmask);
546       const __m256 vk10x01234567 = _mm256_load_ps(w + 176);
547       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi10x01234567, vk10x01234567));
548 
549       const __m256 vi11x01234567 = _mm256_maskload_ps(i11, vmask);
550       const __m256 vk11x01234567 = _mm256_load_ps(w + 192);
551       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi11x01234567, vk11x01234567));
552 
553       const __m256 vi12x01234567 = _mm256_maskload_ps(i12, vmask);
554       const __m256 vk12x01234567 = _mm256_load_ps(w + 208);
555       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi12x01234567, vk12x01234567));
556 
557       const __m256 vi13x01234567 = _mm256_maskload_ps(i13, vmask);
558       const __m256 vk13x01234567 = _mm256_load_ps(w + 224);
559       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi13x01234567, vk13x01234567));
560 
561       const __m256 vi14x01234567 = _mm256_maskload_ps(i14, vmask);
562       const __m256 vk14x01234567 = _mm256_load_ps(w + 240);
563       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi14x01234567, vk14x01234567));
564 
565       const __m256 vi15x01234567 = _mm256_maskload_ps(i15, vmask);
566       const __m256 vk15x01234567 = _mm256_load_ps(w + 256);
567       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi15x01234567, vk15x01234567));
568 
569       const __m256 vi16x01234567 = _mm256_maskload_ps(i16, vmask);
570       const __m256 vk16x01234567 = _mm256_load_ps(w + 272);
571       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi16x01234567, vk16x01234567));
572 
573       const __m256 vi17x01234567 = _mm256_maskload_ps(i17, vmask);
574       const __m256 vk17x01234567 = _mm256_load_ps(w + 288);
575       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi17x01234567, vk17x01234567));
576 
577       const __m256 vi18x01234567 = _mm256_maskload_ps(i18, vmask);
578       const __m256 vk18x01234567 = _mm256_load_ps(w + 304);
579       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi18x01234567, vk18x01234567));
580 
581       const __m256 vi19x01234567 = _mm256_maskload_ps(i19, vmask);
582       const __m256 vk19x01234567 = _mm256_load_ps(w + 320);
583       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi19x01234567, vk19x01234567));
584 
585       const __m256 vi20x01234567 = _mm256_maskload_ps(i20, vmask);
586       const __m256 vk20x01234567 = _mm256_load_ps(w + 336);
587       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi20x01234567, vk20x01234567));
588 
589       const __m256 vi21x01234567 = _mm256_maskload_ps(i21, vmask);
590       const __m256 vk21x01234567 = _mm256_load_ps(w + 352);
591       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi21x01234567, vk21x01234567));
592 
593       const __m256 vi22x01234567 = _mm256_maskload_ps(i22, vmask);
594       const __m256 vk22x01234567 = _mm256_load_ps(w + 368);
595       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi22x01234567, vk22x01234567));
596 
597       const __m256 vi23x01234567 = _mm256_maskload_ps(i23, vmask);
598       const __m256 vk23x01234567 = _mm256_load_ps(w + 384);
599       vacc01234567p1 = _mm256_add_ps(vacc01234567p1, _mm256_mul_ps(vi23x01234567, vk23x01234567));
600 
601       const __m256 vi24x01234567 = _mm256_maskload_ps(i24, vmask);
602       const __m256 vk24x01234567 = _mm256_load_ps(w + 400);
603       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, _mm256_mul_ps(vi24x01234567, vk24x01234567));
604 
605       // Add up all accumulators to vacc01234567p0
606       vacc01234567p0 = _mm256_add_ps(vacc01234567p0, vacc01234567p1);
607 
608       __m256 vacc01234567 = _mm256_max_ps(vacc01234567p0, vmin);
609       vacc01234567 = _mm256_min_ps(vacc01234567, vmax);
610 
611       // _mm256_maskstore_ps(output, vmask, vacc01234567); output += c; could be used here, but triggers msan failures (probably an msan bug).
612       __m128 vacc0123 = _mm256_castps256_ps128(vacc01234567);
613       if (c & 4) {
614         _mm_storeu_ps(output, vacc0123);
615         vacc0123 = _mm256_extractf128_ps(vacc01234567, 1);
616         output += 4;
617       }
618       if (c & 2) {
619         _mm_storel_pi((__m64*) output, vacc0123);
620         vacc0123 = _mm_movehl_ps(vacc0123, vacc0123);
621         output += 2;
622       }
623       if (c & 1) {
624         _mm_store_ss(output, vacc0123);
625         output += 1;
626       }
627     }
628 
629     output = (float*) ((uintptr_t) output + output_increment);
630   } while (--output_width != 0);
631 }
632