blob: 939ba13cfce390f1e9b2086a93a4faef6c858fa6 [file] [log] [blame]
Marat Dukhan479f87e2019-11-27 15:17:06 -08001// Auto-generated file. Do not edit!
2// Template: src/f32-dwconv/up-avx512.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>
Marat Dukhancfb31342019-12-05 10:42:57 -080015#include <xnnpack/intrinsics-polyfill.h>
Marat Dukhan479f87e2019-11-27 15:17:06 -080016
17
Marat Dukhande06f492020-04-09 00:19:31 -070018void xnn_f32_dwconv_minmax_ukernel_up16x25__avx512f_acc2(
Marat Dukhan479f87e2019-11-27 15:17:06 -080019 size_t channels,
20 size_t output_width,
21 const float** input,
22 const float* weights,
23 float* output,
24 size_t input_stride,
25 size_t output_increment,
Frank Barchardd5360722020-05-17 16:10:36 -070026 size_t input_offset,
27 const float* zero,
Marat Dukhanf196d012020-04-15 11:50:03 -070028 const union xnn_f32_minmax_params params[restrict XNN_MIN_ELEMENTS(1)])
Marat Dukhan479f87e2019-11-27 15:17:06 -080029{
30 assert(channels != 0);
31 assert(output_width != 0);
32
33 const __m512 vmax = _mm512_broadcast_f32x4(_mm_load_ps(params->sse.max));
34 const __m512 vmin = _mm512_broadcast_f32x4(_mm_load_ps(params->sse.min));
35 do {
36 const float* i0 = input[0];
Marat Dukhan68660992020-02-03 13:31:12 -080037 assert(i0 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070038 if XNN_UNPREDICTABLE(i0 != zero) {
39 i0 = (const float*) ((uintptr_t) i0 + input_offset);
40 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080041 const float* i1 = input[1];
Marat Dukhan68660992020-02-03 13:31:12 -080042 assert(i1 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070043 if XNN_UNPREDICTABLE(i1 != zero) {
44 i1 = (const float*) ((uintptr_t) i1 + input_offset);
45 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080046 const float* i2 = input[2];
Marat Dukhan68660992020-02-03 13:31:12 -080047 assert(i2 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070048 if XNN_UNPREDICTABLE(i2 != zero) {
49 i2 = (const float*) ((uintptr_t) i2 + input_offset);
50 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080051 const float* i3 = input[3];
Marat Dukhan68660992020-02-03 13:31:12 -080052 assert(i3 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070053 if XNN_UNPREDICTABLE(i3 != zero) {
54 i3 = (const float*) ((uintptr_t) i3 + input_offset);
55 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080056 const float* i4 = input[4];
Marat Dukhan68660992020-02-03 13:31:12 -080057 assert(i4 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070058 if XNN_UNPREDICTABLE(i4 != zero) {
59 i4 = (const float*) ((uintptr_t) i4 + input_offset);
60 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080061 const float* i5 = input[5];
Marat Dukhan68660992020-02-03 13:31:12 -080062 assert(i5 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070063 if XNN_UNPREDICTABLE(i5 != zero) {
64 i5 = (const float*) ((uintptr_t) i5 + input_offset);
65 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080066 const float* i6 = input[6];
Marat Dukhan68660992020-02-03 13:31:12 -080067 assert(i6 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070068 if XNN_UNPREDICTABLE(i6 != zero) {
69 i6 = (const float*) ((uintptr_t) i6 + input_offset);
70 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080071 const float* i7 = input[7];
Marat Dukhan68660992020-02-03 13:31:12 -080072 assert(i7 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070073 if XNN_UNPREDICTABLE(i7 != zero) {
74 i7 = (const float*) ((uintptr_t) i7 + input_offset);
75 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080076 const float* i8 = input[8];
Marat Dukhan68660992020-02-03 13:31:12 -080077 assert(i8 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070078 if XNN_UNPREDICTABLE(i8 != zero) {
79 i8 = (const float*) ((uintptr_t) i8 + input_offset);
80 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080081 const float* i9 = input[9];
Marat Dukhan68660992020-02-03 13:31:12 -080082 assert(i9 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070083 if XNN_UNPREDICTABLE(i9 != zero) {
84 i9 = (const float*) ((uintptr_t) i9 + input_offset);
85 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080086 const float* i10 = input[10];
Marat Dukhan68660992020-02-03 13:31:12 -080087 assert(i10 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070088 if XNN_UNPREDICTABLE(i10 != zero) {
89 i10 = (const float*) ((uintptr_t) i10 + input_offset);
90 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080091 const float* i11 = input[11];
Marat Dukhan68660992020-02-03 13:31:12 -080092 assert(i11 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070093 if XNN_UNPREDICTABLE(i11 != zero) {
94 i11 = (const float*) ((uintptr_t) i11 + input_offset);
95 }
Marat Dukhan479f87e2019-11-27 15:17:06 -080096 const float* i12 = input[12];
Marat Dukhan68660992020-02-03 13:31:12 -080097 assert(i12 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -070098 if XNN_UNPREDICTABLE(i12 != zero) {
99 i12 = (const float*) ((uintptr_t) i12 + input_offset);
100 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800101 const float* i13 = input[13];
Marat Dukhan68660992020-02-03 13:31:12 -0800102 assert(i13 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700103 if XNN_UNPREDICTABLE(i13 != zero) {
104 i13 = (const float*) ((uintptr_t) i13 + input_offset);
105 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800106 const float* i14 = input[14];
Marat Dukhan68660992020-02-03 13:31:12 -0800107 assert(i14 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700108 if XNN_UNPREDICTABLE(i14 != zero) {
109 i14 = (const float*) ((uintptr_t) i14 + input_offset);
110 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800111 const float* i15 = input[15];
Marat Dukhan68660992020-02-03 13:31:12 -0800112 assert(i15 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700113 if XNN_UNPREDICTABLE(i15 != zero) {
114 i15 = (const float*) ((uintptr_t) i15 + input_offset);
115 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800116 const float* i16 = input[16];
Marat Dukhan68660992020-02-03 13:31:12 -0800117 assert(i16 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700118 if XNN_UNPREDICTABLE(i16 != zero) {
119 i16 = (const float*) ((uintptr_t) i16 + input_offset);
120 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800121 const float* i17 = input[17];
Marat Dukhan68660992020-02-03 13:31:12 -0800122 assert(i17 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700123 if XNN_UNPREDICTABLE(i17 != zero) {
124 i17 = (const float*) ((uintptr_t) i17 + input_offset);
125 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800126 const float* i18 = input[18];
Marat Dukhan68660992020-02-03 13:31:12 -0800127 assert(i18 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700128 if XNN_UNPREDICTABLE(i18 != zero) {
129 i18 = (const float*) ((uintptr_t) i18 + input_offset);
130 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800131 const float* i19 = input[19];
Marat Dukhan68660992020-02-03 13:31:12 -0800132 assert(i19 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700133 if XNN_UNPREDICTABLE(i19 != zero) {
134 i19 = (const float*) ((uintptr_t) i19 + input_offset);
135 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800136 const float* i20 = input[20];
Marat Dukhan68660992020-02-03 13:31:12 -0800137 assert(i20 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700138 if XNN_UNPREDICTABLE(i20 != zero) {
139 i20 = (const float*) ((uintptr_t) i20 + input_offset);
140 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800141 const float* i21 = input[21];
Marat Dukhan68660992020-02-03 13:31:12 -0800142 assert(i21 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700143 if XNN_UNPREDICTABLE(i21 != zero) {
144 i21 = (const float*) ((uintptr_t) i21 + input_offset);
145 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800146 const float* i22 = input[22];
Marat Dukhan68660992020-02-03 13:31:12 -0800147 assert(i22 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700148 if XNN_UNPREDICTABLE(i22 != zero) {
149 i22 = (const float*) ((uintptr_t) i22 + input_offset);
150 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800151 const float* i23 = input[23];
Marat Dukhan68660992020-02-03 13:31:12 -0800152 assert(i23 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700153 if XNN_UNPREDICTABLE(i23 != zero) {
154 i23 = (const float*) ((uintptr_t) i23 + input_offset);
155 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800156 const float* i24 = input[24];
Marat Dukhan68660992020-02-03 13:31:12 -0800157 assert(i24 != NULL);
Frank Barchardd5360722020-05-17 16:10:36 -0700158 if XNN_UNPREDICTABLE(i24 != zero) {
159 i24 = (const float*) ((uintptr_t) i24 + input_offset);
160 }
Marat Dukhan479f87e2019-11-27 15:17:06 -0800161 input = (const float**) ((uintptr_t) input + input_stride);
162
163 size_t c = channels;
164 const float* w = weights;
165 for (; c >= 16; c -= 16) {
166 __m512 vacc0123456789ABCDEFp0 = _mm512_load_ps(w);
167
168
169 const __m512 vi0x0123456789ABCDEF = _mm512_loadu_ps(i0);
170 i0 += 16;
171
172 const __m512 vk0x0123456789ABCDEF = _mm512_load_ps(w + 16);
173 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi0x0123456789ABCDEF, vk0x0123456789ABCDEF, vacc0123456789ABCDEFp0);
174
175 const __m512 vi1x0123456789ABCDEF = _mm512_loadu_ps(i1);
176 i1 += 16;
177
178 const __m512 vk1x0123456789ABCDEF = _mm512_load_ps(w + 32);
179 __m512 vacc0123456789ABCDEFp1 = _mm512_mul_ps(vi1x0123456789ABCDEF, vk1x0123456789ABCDEF);
180
181 const __m512 vi2x0123456789ABCDEF = _mm512_loadu_ps(i2);
182 i2 += 16;
183
184 const __m512 vk2x0123456789ABCDEF = _mm512_load_ps(w + 48);
185 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi2x0123456789ABCDEF, vk2x0123456789ABCDEF, vacc0123456789ABCDEFp0);
186
187 const __m512 vi3x0123456789ABCDEF = _mm512_loadu_ps(i3);
188 i3 += 16;
189
190 const __m512 vk3x0123456789ABCDEF = _mm512_load_ps(w + 64);
191 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi3x0123456789ABCDEF, vk3x0123456789ABCDEF, vacc0123456789ABCDEFp1);
192
193 const __m512 vi4x0123456789ABCDEF = _mm512_loadu_ps(i4);
194 i4 += 16;
195
196 const __m512 vk4x0123456789ABCDEF = _mm512_load_ps(w + 80);
197 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi4x0123456789ABCDEF, vk4x0123456789ABCDEF, vacc0123456789ABCDEFp0);
198
199 const __m512 vi5x0123456789ABCDEF = _mm512_loadu_ps(i5);
200 i5 += 16;
201
202 const __m512 vk5x0123456789ABCDEF = _mm512_load_ps(w + 96);
203 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi5x0123456789ABCDEF, vk5x0123456789ABCDEF, vacc0123456789ABCDEFp1);
204
205 const __m512 vi6x0123456789ABCDEF = _mm512_loadu_ps(i6);
206 i6 += 16;
207
208 const __m512 vk6x0123456789ABCDEF = _mm512_load_ps(w + 112);
209 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi6x0123456789ABCDEF, vk6x0123456789ABCDEF, vacc0123456789ABCDEFp0);
210
211 const __m512 vi7x0123456789ABCDEF = _mm512_loadu_ps(i7);
212 i7 += 16;
213
214 const __m512 vk7x0123456789ABCDEF = _mm512_load_ps(w + 128);
215 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi7x0123456789ABCDEF, vk7x0123456789ABCDEF, vacc0123456789ABCDEFp1);
216
217 const __m512 vi8x0123456789ABCDEF = _mm512_loadu_ps(i8);
218 i8 += 16;
219
220 const __m512 vk8x0123456789ABCDEF = _mm512_load_ps(w + 144);
221 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi8x0123456789ABCDEF, vk8x0123456789ABCDEF, vacc0123456789ABCDEFp0);
222
223 const __m512 vi9x0123456789ABCDEF = _mm512_loadu_ps(i9);
224 i9 += 16;
225
226 const __m512 vk9x0123456789ABCDEF = _mm512_load_ps(w + 160);
227 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi9x0123456789ABCDEF, vk9x0123456789ABCDEF, vacc0123456789ABCDEFp1);
228
229 const __m512 vi10x0123456789ABCDEF = _mm512_loadu_ps(i10);
230 i10 += 16;
231
232 const __m512 vk10x0123456789ABCDEF = _mm512_load_ps(w + 176);
233 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi10x0123456789ABCDEF, vk10x0123456789ABCDEF, vacc0123456789ABCDEFp0);
234
235 const __m512 vi11x0123456789ABCDEF = _mm512_loadu_ps(i11);
236 i11 += 16;
237
238 const __m512 vk11x0123456789ABCDEF = _mm512_load_ps(w + 192);
239 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi11x0123456789ABCDEF, vk11x0123456789ABCDEF, vacc0123456789ABCDEFp1);
240
241 const __m512 vi12x0123456789ABCDEF = _mm512_loadu_ps(i12);
242 i12 += 16;
243
244 const __m512 vk12x0123456789ABCDEF = _mm512_load_ps(w + 208);
245 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi12x0123456789ABCDEF, vk12x0123456789ABCDEF, vacc0123456789ABCDEFp0);
246
247 const __m512 vi13x0123456789ABCDEF = _mm512_loadu_ps(i13);
248 i13 += 16;
249
250 const __m512 vk13x0123456789ABCDEF = _mm512_load_ps(w + 224);
251 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi13x0123456789ABCDEF, vk13x0123456789ABCDEF, vacc0123456789ABCDEFp1);
252
253 const __m512 vi14x0123456789ABCDEF = _mm512_loadu_ps(i14);
254 i14 += 16;
255
256 const __m512 vk14x0123456789ABCDEF = _mm512_load_ps(w + 240);
257 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi14x0123456789ABCDEF, vk14x0123456789ABCDEF, vacc0123456789ABCDEFp0);
258
259 const __m512 vi15x0123456789ABCDEF = _mm512_loadu_ps(i15);
260 i15 += 16;
261
262 const __m512 vk15x0123456789ABCDEF = _mm512_load_ps(w + 256);
263 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi15x0123456789ABCDEF, vk15x0123456789ABCDEF, vacc0123456789ABCDEFp1);
264
265 const __m512 vi16x0123456789ABCDEF = _mm512_loadu_ps(i16);
266 i16 += 16;
267
268 const __m512 vk16x0123456789ABCDEF = _mm512_load_ps(w + 272);
269 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi16x0123456789ABCDEF, vk16x0123456789ABCDEF, vacc0123456789ABCDEFp0);
270
271 const __m512 vi17x0123456789ABCDEF = _mm512_loadu_ps(i17);
272 i17 += 16;
273
274 const __m512 vk17x0123456789ABCDEF = _mm512_load_ps(w + 288);
275 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi17x0123456789ABCDEF, vk17x0123456789ABCDEF, vacc0123456789ABCDEFp1);
276
277 const __m512 vi18x0123456789ABCDEF = _mm512_loadu_ps(i18);
278 i18 += 16;
279
280 const __m512 vk18x0123456789ABCDEF = _mm512_load_ps(w + 304);
281 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi18x0123456789ABCDEF, vk18x0123456789ABCDEF, vacc0123456789ABCDEFp0);
282
283 const __m512 vi19x0123456789ABCDEF = _mm512_loadu_ps(i19);
284 i19 += 16;
285
286 const __m512 vk19x0123456789ABCDEF = _mm512_load_ps(w + 320);
287 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi19x0123456789ABCDEF, vk19x0123456789ABCDEF, vacc0123456789ABCDEFp1);
288
289 const __m512 vi20x0123456789ABCDEF = _mm512_loadu_ps(i20);
290 i20 += 16;
291
292 const __m512 vk20x0123456789ABCDEF = _mm512_load_ps(w + 336);
293 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi20x0123456789ABCDEF, vk20x0123456789ABCDEF, vacc0123456789ABCDEFp0);
294
295 const __m512 vi21x0123456789ABCDEF = _mm512_loadu_ps(i21);
296 i21 += 16;
297
298 const __m512 vk21x0123456789ABCDEF = _mm512_load_ps(w + 352);
299 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi21x0123456789ABCDEF, vk21x0123456789ABCDEF, vacc0123456789ABCDEFp1);
300
301 const __m512 vi22x0123456789ABCDEF = _mm512_loadu_ps(i22);
302 i22 += 16;
303
304 const __m512 vk22x0123456789ABCDEF = _mm512_load_ps(w + 368);
305 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi22x0123456789ABCDEF, vk22x0123456789ABCDEF, vacc0123456789ABCDEFp0);
306
307 const __m512 vi23x0123456789ABCDEF = _mm512_loadu_ps(i23);
308 i23 += 16;
309
310 const __m512 vk23x0123456789ABCDEF = _mm512_load_ps(w + 384);
311 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi23x0123456789ABCDEF, vk23x0123456789ABCDEF, vacc0123456789ABCDEFp1);
312
313 const __m512 vi24x0123456789ABCDEF = _mm512_loadu_ps(i24);
314 i24 += 16;
315
316 const __m512 vk24x0123456789ABCDEF = _mm512_load_ps(w + 400);
317 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi24x0123456789ABCDEF, vk24x0123456789ABCDEF, vacc0123456789ABCDEFp0);
318
319 w += 416;
320
321 // Add up all accumulators to vacc0123456789ABCDEFp0
322 vacc0123456789ABCDEFp0 = _mm512_add_ps(vacc0123456789ABCDEFp0, vacc0123456789ABCDEFp1);
323
324 __m512 vacc0123456789ABCDEF = _mm512_max_ps(vacc0123456789ABCDEFp0, vmin);
325 vacc0123456789ABCDEF = _mm512_min_ps(vacc0123456789ABCDEF, vmax);
326
327 _mm512_storeu_ps(output, vacc0123456789ABCDEF);
328 output += 16;
329 }
330 if XNN_UNLIKELY(c != 0) {
331 assert(c >= 1);
332 assert(c <= 16);
333 // Prepare mask for valid 32-bit elements (depends on nc).
334 const __mmask16 vmask = _cvtu32_mask16((uint16_t) ((uint32_t) (UINT32_C(1) << c) - UINT32_C(1)));
335
Marat Dukhan90dff802020-02-10 17:23:53 -0800336 __m512 vacc0123456789ABCDEFp0 = _mm512_maskz_loadu_ps(vmask, w);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800337
338 const __m512 vi0x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i0);
Marat Dukhan90dff802020-02-10 17:23:53 -0800339 const __m512 vk0x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 16);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800340 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi0x0123456789ABCDEF, vk0x0123456789ABCDEF, vacc0123456789ABCDEFp0);
341
342 const __m512 vi1x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i1);
Marat Dukhan90dff802020-02-10 17:23:53 -0800343 const __m512 vk1x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 32);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800344 __m512 vacc0123456789ABCDEFp1 = _mm512_mul_ps(vi1x0123456789ABCDEF, vk1x0123456789ABCDEF);
345
346 const __m512 vi2x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i2);
Marat Dukhan90dff802020-02-10 17:23:53 -0800347 const __m512 vk2x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 48);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800348 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi2x0123456789ABCDEF, vk2x0123456789ABCDEF, vacc0123456789ABCDEFp0);
349
350 const __m512 vi3x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i3);
Marat Dukhan90dff802020-02-10 17:23:53 -0800351 const __m512 vk3x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 64);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800352 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi3x0123456789ABCDEF, vk3x0123456789ABCDEF, vacc0123456789ABCDEFp1);
353
354 const __m512 vi4x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i4);
Marat Dukhan90dff802020-02-10 17:23:53 -0800355 const __m512 vk4x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 80);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800356 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi4x0123456789ABCDEF, vk4x0123456789ABCDEF, vacc0123456789ABCDEFp0);
357
358 const __m512 vi5x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i5);
Marat Dukhan90dff802020-02-10 17:23:53 -0800359 const __m512 vk5x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 96);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800360 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi5x0123456789ABCDEF, vk5x0123456789ABCDEF, vacc0123456789ABCDEFp1);
361
362 const __m512 vi6x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i6);
Marat Dukhan90dff802020-02-10 17:23:53 -0800363 const __m512 vk6x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 112);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800364 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi6x0123456789ABCDEF, vk6x0123456789ABCDEF, vacc0123456789ABCDEFp0);
365
366 const __m512 vi7x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i7);
Marat Dukhan90dff802020-02-10 17:23:53 -0800367 const __m512 vk7x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 128);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800368 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi7x0123456789ABCDEF, vk7x0123456789ABCDEF, vacc0123456789ABCDEFp1);
369
370 const __m512 vi8x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i8);
Marat Dukhan90dff802020-02-10 17:23:53 -0800371 const __m512 vk8x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 144);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800372 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi8x0123456789ABCDEF, vk8x0123456789ABCDEF, vacc0123456789ABCDEFp0);
373
374 const __m512 vi9x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i9);
Marat Dukhan90dff802020-02-10 17:23:53 -0800375 const __m512 vk9x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 160);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800376 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi9x0123456789ABCDEF, vk9x0123456789ABCDEF, vacc0123456789ABCDEFp1);
377
378 const __m512 vi10x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i10);
Marat Dukhan90dff802020-02-10 17:23:53 -0800379 const __m512 vk10x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 176);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800380 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi10x0123456789ABCDEF, vk10x0123456789ABCDEF, vacc0123456789ABCDEFp0);
381
382 const __m512 vi11x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i11);
Marat Dukhan90dff802020-02-10 17:23:53 -0800383 const __m512 vk11x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 192);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800384 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi11x0123456789ABCDEF, vk11x0123456789ABCDEF, vacc0123456789ABCDEFp1);
385
386 const __m512 vi12x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i12);
Marat Dukhan90dff802020-02-10 17:23:53 -0800387 const __m512 vk12x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 208);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800388 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi12x0123456789ABCDEF, vk12x0123456789ABCDEF, vacc0123456789ABCDEFp0);
389
390 const __m512 vi13x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i13);
Marat Dukhan90dff802020-02-10 17:23:53 -0800391 const __m512 vk13x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 224);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800392 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi13x0123456789ABCDEF, vk13x0123456789ABCDEF, vacc0123456789ABCDEFp1);
393
394 const __m512 vi14x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i14);
Marat Dukhan90dff802020-02-10 17:23:53 -0800395 const __m512 vk14x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 240);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800396 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi14x0123456789ABCDEF, vk14x0123456789ABCDEF, vacc0123456789ABCDEFp0);
397
398 const __m512 vi15x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i15);
Marat Dukhan90dff802020-02-10 17:23:53 -0800399 const __m512 vk15x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 256);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800400 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi15x0123456789ABCDEF, vk15x0123456789ABCDEF, vacc0123456789ABCDEFp1);
401
402 const __m512 vi16x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i16);
Marat Dukhan90dff802020-02-10 17:23:53 -0800403 const __m512 vk16x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 272);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800404 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi16x0123456789ABCDEF, vk16x0123456789ABCDEF, vacc0123456789ABCDEFp0);
405
406 const __m512 vi17x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i17);
Marat Dukhan90dff802020-02-10 17:23:53 -0800407 const __m512 vk17x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 288);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800408 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi17x0123456789ABCDEF, vk17x0123456789ABCDEF, vacc0123456789ABCDEFp1);
409
410 const __m512 vi18x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i18);
Marat Dukhan90dff802020-02-10 17:23:53 -0800411 const __m512 vk18x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 304);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800412 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi18x0123456789ABCDEF, vk18x0123456789ABCDEF, vacc0123456789ABCDEFp0);
413
414 const __m512 vi19x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i19);
Marat Dukhan90dff802020-02-10 17:23:53 -0800415 const __m512 vk19x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 320);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800416 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi19x0123456789ABCDEF, vk19x0123456789ABCDEF, vacc0123456789ABCDEFp1);
417
418 const __m512 vi20x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i20);
Marat Dukhan90dff802020-02-10 17:23:53 -0800419 const __m512 vk20x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 336);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800420 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi20x0123456789ABCDEF, vk20x0123456789ABCDEF, vacc0123456789ABCDEFp0);
421
422 const __m512 vi21x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i21);
Marat Dukhan90dff802020-02-10 17:23:53 -0800423 const __m512 vk21x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 352);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800424 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi21x0123456789ABCDEF, vk21x0123456789ABCDEF, vacc0123456789ABCDEFp1);
425
426 const __m512 vi22x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i22);
Marat Dukhan90dff802020-02-10 17:23:53 -0800427 const __m512 vk22x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 368);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800428 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi22x0123456789ABCDEF, vk22x0123456789ABCDEF, vacc0123456789ABCDEFp0);
429
430 const __m512 vi23x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i23);
Marat Dukhan90dff802020-02-10 17:23:53 -0800431 const __m512 vk23x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 384);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800432 vacc0123456789ABCDEFp1 = _mm512_fmadd_ps(vi23x0123456789ABCDEF, vk23x0123456789ABCDEF, vacc0123456789ABCDEFp1);
433
434 const __m512 vi24x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, i24);
Marat Dukhan90dff802020-02-10 17:23:53 -0800435 const __m512 vk24x0123456789ABCDEF = _mm512_maskz_loadu_ps(vmask, w + 400);
Marat Dukhan479f87e2019-11-27 15:17:06 -0800436 vacc0123456789ABCDEFp0 = _mm512_fmadd_ps(vi24x0123456789ABCDEF, vk24x0123456789ABCDEF, vacc0123456789ABCDEFp0);
437
438 // Add up all accumulators to vacc0123456789ABCDEFp0
439 vacc0123456789ABCDEFp0 = _mm512_add_ps(vacc0123456789ABCDEFp0, vacc0123456789ABCDEFp1);
440
441 __m512 vacc0123456789ABCDEF = _mm512_max_ps(vacc0123456789ABCDEFp0, vmin);
442 vacc0123456789ABCDEF = _mm512_min_ps(vacc0123456789ABCDEF, vmax);
443
444 _mm512_mask_storeu_ps(output, vmask, vacc0123456789ABCDEF);
445 output += c;
446 }
447
448 output = (float*) ((uintptr_t) output + output_increment);
449 } while (--output_width != 0);
450}