1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker // Template: src/qs8-dwconv/unipass-avx2-mul32.c.in
3*4bdc9457SAndroid Build Coastguard Worker // Generator: tools/xngen
4*4bdc9457SAndroid Build Coastguard Worker //
5*4bdc9457SAndroid Build Coastguard Worker // Copyright 2020 Google LLC
6*4bdc9457SAndroid Build Coastguard Worker //
7*4bdc9457SAndroid Build Coastguard Worker // This source code is licensed under the BSD-style license found in the
8*4bdc9457SAndroid Build Coastguard Worker // LICENSE file in the root directory of this source tree.
9*4bdc9457SAndroid Build Coastguard Worker
10*4bdc9457SAndroid Build Coastguard Worker #include <assert.h>
11*4bdc9457SAndroid Build Coastguard Worker
12*4bdc9457SAndroid Build Coastguard Worker #include <immintrin.h>
13*4bdc9457SAndroid Build Coastguard Worker
14*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/dwconv.h>
15*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/unaligned.h>
16*4bdc9457SAndroid Build Coastguard Worker
17*4bdc9457SAndroid Build Coastguard Worker
xnn_qu8_dwconv_minmax_fp32_ukernel_up8x25__avx2_mul32(size_t channels,size_t output_width,const uint8_t ** input,const void * weights,uint8_t * output,size_t input_stride,size_t output_increment,size_t input_offset,const uint8_t * zero,const union xnn_qu8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])18*4bdc9457SAndroid Build Coastguard Worker void xnn_qu8_dwconv_minmax_fp32_ukernel_up8x25__avx2_mul32(
19*4bdc9457SAndroid Build Coastguard Worker size_t channels,
20*4bdc9457SAndroid Build Coastguard Worker size_t output_width,
21*4bdc9457SAndroid Build Coastguard Worker const uint8_t** input,
22*4bdc9457SAndroid Build Coastguard Worker const void* weights,
23*4bdc9457SAndroid Build Coastguard Worker uint8_t* output,
24*4bdc9457SAndroid Build Coastguard Worker size_t input_stride,
25*4bdc9457SAndroid Build Coastguard Worker size_t output_increment,
26*4bdc9457SAndroid Build Coastguard Worker size_t input_offset,
27*4bdc9457SAndroid Build Coastguard Worker const uint8_t* zero,
28*4bdc9457SAndroid Build Coastguard Worker const union xnn_qu8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
29*4bdc9457SAndroid Build Coastguard Worker {
30*4bdc9457SAndroid Build Coastguard Worker assert(channels != 0);
31*4bdc9457SAndroid Build Coastguard Worker assert(output_width != 0);
32*4bdc9457SAndroid Build Coastguard Worker
33*4bdc9457SAndroid Build Coastguard Worker const __m256i vk_zero_point = _mm256_cvtepu16_epi32(_mm_load_si128((const __m128i*) params->fp32_avx2.kernel_zero_point));
34*4bdc9457SAndroid Build Coastguard Worker do {
35*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i0 = input[0];
36*4bdc9457SAndroid Build Coastguard Worker assert(i0 != NULL);
37*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i0 != zero) {
38*4bdc9457SAndroid Build Coastguard Worker i0 = (const uint8_t*) ((uintptr_t) i0 + input_offset);
39*4bdc9457SAndroid Build Coastguard Worker }
40*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i1 = input[1];
41*4bdc9457SAndroid Build Coastguard Worker assert(i1 != NULL);
42*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i1 != zero) {
43*4bdc9457SAndroid Build Coastguard Worker i1 = (const uint8_t*) ((uintptr_t) i1 + input_offset);
44*4bdc9457SAndroid Build Coastguard Worker }
45*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i2 = input[2];
46*4bdc9457SAndroid Build Coastguard Worker assert(i2 != NULL);
47*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i2 != zero) {
48*4bdc9457SAndroid Build Coastguard Worker i2 = (const uint8_t*) ((uintptr_t) i2 + input_offset);
49*4bdc9457SAndroid Build Coastguard Worker }
50*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i3 = input[3];
51*4bdc9457SAndroid Build Coastguard Worker assert(i3 != NULL);
52*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i3 != zero) {
53*4bdc9457SAndroid Build Coastguard Worker i3 = (const uint8_t*) ((uintptr_t) i3 + input_offset);
54*4bdc9457SAndroid Build Coastguard Worker }
55*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i4 = input[4];
56*4bdc9457SAndroid Build Coastguard Worker assert(i4 != NULL);
57*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i4 != zero) {
58*4bdc9457SAndroid Build Coastguard Worker i4 = (const uint8_t*) ((uintptr_t) i4 + input_offset);
59*4bdc9457SAndroid Build Coastguard Worker }
60*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i5 = input[5];
61*4bdc9457SAndroid Build Coastguard Worker assert(i5 != NULL);
62*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i5 != zero) {
63*4bdc9457SAndroid Build Coastguard Worker i5 = (const uint8_t*) ((uintptr_t) i5 + input_offset);
64*4bdc9457SAndroid Build Coastguard Worker }
65*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i6 = input[6];
66*4bdc9457SAndroid Build Coastguard Worker assert(i6 != NULL);
67*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i6 != zero) {
68*4bdc9457SAndroid Build Coastguard Worker i6 = (const uint8_t*) ((uintptr_t) i6 + input_offset);
69*4bdc9457SAndroid Build Coastguard Worker }
70*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i7 = input[7];
71*4bdc9457SAndroid Build Coastguard Worker assert(i7 != NULL);
72*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i7 != zero) {
73*4bdc9457SAndroid Build Coastguard Worker i7 = (const uint8_t*) ((uintptr_t) i7 + input_offset);
74*4bdc9457SAndroid Build Coastguard Worker }
75*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i8 = input[8];
76*4bdc9457SAndroid Build Coastguard Worker assert(i8 != NULL);
77*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i8 != zero) {
78*4bdc9457SAndroid Build Coastguard Worker i8 = (const uint8_t*) ((uintptr_t) i8 + input_offset);
79*4bdc9457SAndroid Build Coastguard Worker }
80*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i9 = input[9];
81*4bdc9457SAndroid Build Coastguard Worker assert(i9 != NULL);
82*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i9 != zero) {
83*4bdc9457SAndroid Build Coastguard Worker i9 = (const uint8_t*) ((uintptr_t) i9 + input_offset);
84*4bdc9457SAndroid Build Coastguard Worker }
85*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i10 = input[10];
86*4bdc9457SAndroid Build Coastguard Worker assert(i10 != NULL);
87*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i10 != zero) {
88*4bdc9457SAndroid Build Coastguard Worker i10 = (const uint8_t*) ((uintptr_t) i10 + input_offset);
89*4bdc9457SAndroid Build Coastguard Worker }
90*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i11 = input[11];
91*4bdc9457SAndroid Build Coastguard Worker assert(i11 != NULL);
92*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i11 != zero) {
93*4bdc9457SAndroid Build Coastguard Worker i11 = (const uint8_t*) ((uintptr_t) i11 + input_offset);
94*4bdc9457SAndroid Build Coastguard Worker }
95*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i12 = input[12];
96*4bdc9457SAndroid Build Coastguard Worker assert(i12 != NULL);
97*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i12 != zero) {
98*4bdc9457SAndroid Build Coastguard Worker i12 = (const uint8_t*) ((uintptr_t) i12 + input_offset);
99*4bdc9457SAndroid Build Coastguard Worker }
100*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i13 = input[13];
101*4bdc9457SAndroid Build Coastguard Worker assert(i13 != NULL);
102*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i13 != zero) {
103*4bdc9457SAndroid Build Coastguard Worker i13 = (const uint8_t*) ((uintptr_t) i13 + input_offset);
104*4bdc9457SAndroid Build Coastguard Worker }
105*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i14 = input[14];
106*4bdc9457SAndroid Build Coastguard Worker assert(i14 != NULL);
107*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i14 != zero) {
108*4bdc9457SAndroid Build Coastguard Worker i14 = (const uint8_t*) ((uintptr_t) i14 + input_offset);
109*4bdc9457SAndroid Build Coastguard Worker }
110*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i15 = input[15];
111*4bdc9457SAndroid Build Coastguard Worker assert(i15 != NULL);
112*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i15 != zero) {
113*4bdc9457SAndroid Build Coastguard Worker i15 = (const uint8_t*) ((uintptr_t) i15 + input_offset);
114*4bdc9457SAndroid Build Coastguard Worker }
115*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i16 = input[16];
116*4bdc9457SAndroid Build Coastguard Worker assert(i16 != NULL);
117*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i16 != zero) {
118*4bdc9457SAndroid Build Coastguard Worker i16 = (const uint8_t*) ((uintptr_t) i16 + input_offset);
119*4bdc9457SAndroid Build Coastguard Worker }
120*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i17 = input[17];
121*4bdc9457SAndroid Build Coastguard Worker assert(i17 != NULL);
122*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i17 != zero) {
123*4bdc9457SAndroid Build Coastguard Worker i17 = (const uint8_t*) ((uintptr_t) i17 + input_offset);
124*4bdc9457SAndroid Build Coastguard Worker }
125*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i18 = input[18];
126*4bdc9457SAndroid Build Coastguard Worker assert(i18 != NULL);
127*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i18 != zero) {
128*4bdc9457SAndroid Build Coastguard Worker i18 = (const uint8_t*) ((uintptr_t) i18 + input_offset);
129*4bdc9457SAndroid Build Coastguard Worker }
130*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i19 = input[19];
131*4bdc9457SAndroid Build Coastguard Worker assert(i19 != NULL);
132*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i19 != zero) {
133*4bdc9457SAndroid Build Coastguard Worker i19 = (const uint8_t*) ((uintptr_t) i19 + input_offset);
134*4bdc9457SAndroid Build Coastguard Worker }
135*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i20 = input[20];
136*4bdc9457SAndroid Build Coastguard Worker assert(i20 != NULL);
137*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i20 != zero) {
138*4bdc9457SAndroid Build Coastguard Worker i20 = (const uint8_t*) ((uintptr_t) i20 + input_offset);
139*4bdc9457SAndroid Build Coastguard Worker }
140*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i21 = input[21];
141*4bdc9457SAndroid Build Coastguard Worker assert(i21 != NULL);
142*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i21 != zero) {
143*4bdc9457SAndroid Build Coastguard Worker i21 = (const uint8_t*) ((uintptr_t) i21 + input_offset);
144*4bdc9457SAndroid Build Coastguard Worker }
145*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i22 = input[22];
146*4bdc9457SAndroid Build Coastguard Worker assert(i22 != NULL);
147*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i22 != zero) {
148*4bdc9457SAndroid Build Coastguard Worker i22 = (const uint8_t*) ((uintptr_t) i22 + input_offset);
149*4bdc9457SAndroid Build Coastguard Worker }
150*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i23 = input[23];
151*4bdc9457SAndroid Build Coastguard Worker assert(i23 != NULL);
152*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i23 != zero) {
153*4bdc9457SAndroid Build Coastguard Worker i23 = (const uint8_t*) ((uintptr_t) i23 + input_offset);
154*4bdc9457SAndroid Build Coastguard Worker }
155*4bdc9457SAndroid Build Coastguard Worker const uint8_t* i24 = input[24];
156*4bdc9457SAndroid Build Coastguard Worker assert(i24 != NULL);
157*4bdc9457SAndroid Build Coastguard Worker if XNN_UNPREDICTABLE(i24 != zero) {
158*4bdc9457SAndroid Build Coastguard Worker i24 = (const uint8_t*) ((uintptr_t) i24 + input_offset);
159*4bdc9457SAndroid Build Coastguard Worker }
160*4bdc9457SAndroid Build Coastguard Worker input = (const uint8_t**) ((uintptr_t) input + input_stride);
161*4bdc9457SAndroid Build Coastguard Worker
162*4bdc9457SAndroid Build Coastguard Worker size_t c = channels;
163*4bdc9457SAndroid Build Coastguard Worker const void* w = weights;
164*4bdc9457SAndroid Build Coastguard Worker for (; c >= 8; c -= 8) {
165*4bdc9457SAndroid Build Coastguard Worker __m256i vacc01234567 = _mm256_loadu_si256((const __m256i*) w);
166*4bdc9457SAndroid Build Coastguard Worker
167*4bdc9457SAndroid Build Coastguard Worker
168*4bdc9457SAndroid Build Coastguard Worker const __m256i vi0x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i0));
169*4bdc9457SAndroid Build Coastguard Worker const __m256i vk0x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 0 * sizeof(uint8_t)))), vk_zero_point);
170*4bdc9457SAndroid Build Coastguard Worker i0 += 8;
171*4bdc9457SAndroid Build Coastguard Worker
172*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi0x01234567, vk0x01234567));
173*4bdc9457SAndroid Build Coastguard Worker
174*4bdc9457SAndroid Build Coastguard Worker const __m256i vi1x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i1));
175*4bdc9457SAndroid Build Coastguard Worker const __m256i vk1x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 8 * sizeof(uint8_t)))), vk_zero_point);
176*4bdc9457SAndroid Build Coastguard Worker i1 += 8;
177*4bdc9457SAndroid Build Coastguard Worker
178*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi1x01234567, vk1x01234567));
179*4bdc9457SAndroid Build Coastguard Worker
180*4bdc9457SAndroid Build Coastguard Worker const __m256i vi2x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i2));
181*4bdc9457SAndroid Build Coastguard Worker const __m256i vk2x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 16 * sizeof(uint8_t)))), vk_zero_point);
182*4bdc9457SAndroid Build Coastguard Worker i2 += 8;
183*4bdc9457SAndroid Build Coastguard Worker
184*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi2x01234567, vk2x01234567));
185*4bdc9457SAndroid Build Coastguard Worker
186*4bdc9457SAndroid Build Coastguard Worker const __m256i vi3x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i3));
187*4bdc9457SAndroid Build Coastguard Worker const __m256i vk3x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 24 * sizeof(uint8_t)))), vk_zero_point);
188*4bdc9457SAndroid Build Coastguard Worker i3 += 8;
189*4bdc9457SAndroid Build Coastguard Worker
190*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi3x01234567, vk3x01234567));
191*4bdc9457SAndroid Build Coastguard Worker
192*4bdc9457SAndroid Build Coastguard Worker const __m256i vi4x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i4));
193*4bdc9457SAndroid Build Coastguard Worker const __m256i vk4x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 32 * sizeof(uint8_t)))), vk_zero_point);
194*4bdc9457SAndroid Build Coastguard Worker i4 += 8;
195*4bdc9457SAndroid Build Coastguard Worker
196*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi4x01234567, vk4x01234567));
197*4bdc9457SAndroid Build Coastguard Worker
198*4bdc9457SAndroid Build Coastguard Worker const __m256i vi5x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i5));
199*4bdc9457SAndroid Build Coastguard Worker const __m256i vk5x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 40 * sizeof(uint8_t)))), vk_zero_point);
200*4bdc9457SAndroid Build Coastguard Worker i5 += 8;
201*4bdc9457SAndroid Build Coastguard Worker
202*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi5x01234567, vk5x01234567));
203*4bdc9457SAndroid Build Coastguard Worker
204*4bdc9457SAndroid Build Coastguard Worker const __m256i vi6x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i6));
205*4bdc9457SAndroid Build Coastguard Worker const __m256i vk6x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 48 * sizeof(uint8_t)))), vk_zero_point);
206*4bdc9457SAndroid Build Coastguard Worker i6 += 8;
207*4bdc9457SAndroid Build Coastguard Worker
208*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi6x01234567, vk6x01234567));
209*4bdc9457SAndroid Build Coastguard Worker
210*4bdc9457SAndroid Build Coastguard Worker const __m256i vi7x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i7));
211*4bdc9457SAndroid Build Coastguard Worker const __m256i vk7x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 56 * sizeof(uint8_t)))), vk_zero_point);
212*4bdc9457SAndroid Build Coastguard Worker i7 += 8;
213*4bdc9457SAndroid Build Coastguard Worker
214*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi7x01234567, vk7x01234567));
215*4bdc9457SAndroid Build Coastguard Worker
216*4bdc9457SAndroid Build Coastguard Worker const __m256i vi8x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i8));
217*4bdc9457SAndroid Build Coastguard Worker const __m256i vk8x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 64 * sizeof(uint8_t)))), vk_zero_point);
218*4bdc9457SAndroid Build Coastguard Worker i8 += 8;
219*4bdc9457SAndroid Build Coastguard Worker
220*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi8x01234567, vk8x01234567));
221*4bdc9457SAndroid Build Coastguard Worker
222*4bdc9457SAndroid Build Coastguard Worker const __m256i vi9x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i9));
223*4bdc9457SAndroid Build Coastguard Worker const __m256i vk9x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 72 * sizeof(uint8_t)))), vk_zero_point);
224*4bdc9457SAndroid Build Coastguard Worker i9 += 8;
225*4bdc9457SAndroid Build Coastguard Worker
226*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi9x01234567, vk9x01234567));
227*4bdc9457SAndroid Build Coastguard Worker
228*4bdc9457SAndroid Build Coastguard Worker const __m256i vi10x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i10));
229*4bdc9457SAndroid Build Coastguard Worker const __m256i vk10x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 80 * sizeof(uint8_t)))), vk_zero_point);
230*4bdc9457SAndroid Build Coastguard Worker i10 += 8;
231*4bdc9457SAndroid Build Coastguard Worker
232*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi10x01234567, vk10x01234567));
233*4bdc9457SAndroid Build Coastguard Worker
234*4bdc9457SAndroid Build Coastguard Worker const __m256i vi11x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i11));
235*4bdc9457SAndroid Build Coastguard Worker const __m256i vk11x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 88 * sizeof(uint8_t)))), vk_zero_point);
236*4bdc9457SAndroid Build Coastguard Worker i11 += 8;
237*4bdc9457SAndroid Build Coastguard Worker
238*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi11x01234567, vk11x01234567));
239*4bdc9457SAndroid Build Coastguard Worker
240*4bdc9457SAndroid Build Coastguard Worker const __m256i vi12x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i12));
241*4bdc9457SAndroid Build Coastguard Worker const __m256i vk12x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 96 * sizeof(uint8_t)))), vk_zero_point);
242*4bdc9457SAndroid Build Coastguard Worker i12 += 8;
243*4bdc9457SAndroid Build Coastguard Worker
244*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi12x01234567, vk12x01234567));
245*4bdc9457SAndroid Build Coastguard Worker
246*4bdc9457SAndroid Build Coastguard Worker const __m256i vi13x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i13));
247*4bdc9457SAndroid Build Coastguard Worker const __m256i vk13x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 104 * sizeof(uint8_t)))), vk_zero_point);
248*4bdc9457SAndroid Build Coastguard Worker i13 += 8;
249*4bdc9457SAndroid Build Coastguard Worker
250*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi13x01234567, vk13x01234567));
251*4bdc9457SAndroid Build Coastguard Worker
252*4bdc9457SAndroid Build Coastguard Worker const __m256i vi14x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i14));
253*4bdc9457SAndroid Build Coastguard Worker const __m256i vk14x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 112 * sizeof(uint8_t)))), vk_zero_point);
254*4bdc9457SAndroid Build Coastguard Worker i14 += 8;
255*4bdc9457SAndroid Build Coastguard Worker
256*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi14x01234567, vk14x01234567));
257*4bdc9457SAndroid Build Coastguard Worker
258*4bdc9457SAndroid Build Coastguard Worker const __m256i vi15x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i15));
259*4bdc9457SAndroid Build Coastguard Worker const __m256i vk15x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 120 * sizeof(uint8_t)))), vk_zero_point);
260*4bdc9457SAndroid Build Coastguard Worker i15 += 8;
261*4bdc9457SAndroid Build Coastguard Worker
262*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi15x01234567, vk15x01234567));
263*4bdc9457SAndroid Build Coastguard Worker
264*4bdc9457SAndroid Build Coastguard Worker const __m256i vi16x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i16));
265*4bdc9457SAndroid Build Coastguard Worker const __m256i vk16x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 128 * sizeof(uint8_t)))), vk_zero_point);
266*4bdc9457SAndroid Build Coastguard Worker i16 += 8;
267*4bdc9457SAndroid Build Coastguard Worker
268*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi16x01234567, vk16x01234567));
269*4bdc9457SAndroid Build Coastguard Worker
270*4bdc9457SAndroid Build Coastguard Worker const __m256i vi17x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i17));
271*4bdc9457SAndroid Build Coastguard Worker const __m256i vk17x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 136 * sizeof(uint8_t)))), vk_zero_point);
272*4bdc9457SAndroid Build Coastguard Worker i17 += 8;
273*4bdc9457SAndroid Build Coastguard Worker
274*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi17x01234567, vk17x01234567));
275*4bdc9457SAndroid Build Coastguard Worker
276*4bdc9457SAndroid Build Coastguard Worker const __m256i vi18x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i18));
277*4bdc9457SAndroid Build Coastguard Worker const __m256i vk18x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 144 * sizeof(uint8_t)))), vk_zero_point);
278*4bdc9457SAndroid Build Coastguard Worker i18 += 8;
279*4bdc9457SAndroid Build Coastguard Worker
280*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi18x01234567, vk18x01234567));
281*4bdc9457SAndroid Build Coastguard Worker
282*4bdc9457SAndroid Build Coastguard Worker const __m256i vi19x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i19));
283*4bdc9457SAndroid Build Coastguard Worker const __m256i vk19x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 152 * sizeof(uint8_t)))), vk_zero_point);
284*4bdc9457SAndroid Build Coastguard Worker i19 += 8;
285*4bdc9457SAndroid Build Coastguard Worker
286*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi19x01234567, vk19x01234567));
287*4bdc9457SAndroid Build Coastguard Worker
288*4bdc9457SAndroid Build Coastguard Worker const __m256i vi20x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i20));
289*4bdc9457SAndroid Build Coastguard Worker const __m256i vk20x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 160 * sizeof(uint8_t)))), vk_zero_point);
290*4bdc9457SAndroid Build Coastguard Worker i20 += 8;
291*4bdc9457SAndroid Build Coastguard Worker
292*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi20x01234567, vk20x01234567));
293*4bdc9457SAndroid Build Coastguard Worker
294*4bdc9457SAndroid Build Coastguard Worker const __m256i vi21x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i21));
295*4bdc9457SAndroid Build Coastguard Worker const __m256i vk21x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 168 * sizeof(uint8_t)))), vk_zero_point);
296*4bdc9457SAndroid Build Coastguard Worker i21 += 8;
297*4bdc9457SAndroid Build Coastguard Worker
298*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi21x01234567, vk21x01234567));
299*4bdc9457SAndroid Build Coastguard Worker
300*4bdc9457SAndroid Build Coastguard Worker const __m256i vi22x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i22));
301*4bdc9457SAndroid Build Coastguard Worker const __m256i vk22x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 176 * sizeof(uint8_t)))), vk_zero_point);
302*4bdc9457SAndroid Build Coastguard Worker i22 += 8;
303*4bdc9457SAndroid Build Coastguard Worker
304*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi22x01234567, vk22x01234567));
305*4bdc9457SAndroid Build Coastguard Worker
306*4bdc9457SAndroid Build Coastguard Worker const __m256i vi23x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i23));
307*4bdc9457SAndroid Build Coastguard Worker const __m256i vk23x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 184 * sizeof(uint8_t)))), vk_zero_point);
308*4bdc9457SAndroid Build Coastguard Worker i23 += 8;
309*4bdc9457SAndroid Build Coastguard Worker
310*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi23x01234567, vk23x01234567));
311*4bdc9457SAndroid Build Coastguard Worker
312*4bdc9457SAndroid Build Coastguard Worker const __m256i vi24x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i24));
313*4bdc9457SAndroid Build Coastguard Worker const __m256i vk24x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 192 * sizeof(uint8_t)))), vk_zero_point);
314*4bdc9457SAndroid Build Coastguard Worker i24 += 8;
315*4bdc9457SAndroid Build Coastguard Worker
316*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi24x01234567, vk24x01234567));
317*4bdc9457SAndroid Build Coastguard Worker
318*4bdc9457SAndroid Build Coastguard Worker w = (const void*) ((uintptr_t) w + 8 * sizeof(int32_t) + 200 * sizeof(uint8_t));
319*4bdc9457SAndroid Build Coastguard Worker
320*4bdc9457SAndroid Build Coastguard Worker __m256 vscaled01234567 = _mm256_cvtepi32_ps(vacc01234567);
321*4bdc9457SAndroid Build Coastguard Worker
322*4bdc9457SAndroid Build Coastguard Worker const __m256 vscale = _mm256_load_ps(params->fp32_avx2.scale);
323*4bdc9457SAndroid Build Coastguard Worker vscaled01234567 = _mm256_mul_ps(vscaled01234567, vscale);
324*4bdc9457SAndroid Build Coastguard Worker
325*4bdc9457SAndroid Build Coastguard Worker const __m256 voutput_max_less_zero_point = _mm256_load_ps(params->fp32_avx2.output_max_less_zero_point);
326*4bdc9457SAndroid Build Coastguard Worker vscaled01234567 = _mm256_min_ps(vscaled01234567, voutput_max_less_zero_point);
327*4bdc9457SAndroid Build Coastguard Worker
328*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_cvtps_epi32(vscaled01234567);
329*4bdc9457SAndroid Build Coastguard Worker
330*4bdc9457SAndroid Build Coastguard Worker const __m128i voutput_zero_point = _mm_load_si128((const __m128i*) params->fp32_avx2.output_zero_point);
331*4bdc9457SAndroid Build Coastguard Worker __m128i vout01234567 = _mm_adds_epi16(_mm_packs_epi32(_mm256_castsi256_si128(vacc01234567), _mm256_extracti128_si256(vacc01234567, 1)), voutput_zero_point);
332*4bdc9457SAndroid Build Coastguard Worker
333*4bdc9457SAndroid Build Coastguard Worker __m128i vout0123456701234567 = _mm_packus_epi16(vout01234567, vout01234567);
334*4bdc9457SAndroid Build Coastguard Worker
335*4bdc9457SAndroid Build Coastguard Worker const __m128i voutput_min = _mm_load_si128((const __m128i*) params->fp32_avx2.output_min);
336*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = _mm_max_epu8(vout0123456701234567, voutput_min);
337*4bdc9457SAndroid Build Coastguard Worker
338*4bdc9457SAndroid Build Coastguard Worker _mm_storel_epi64((__m128i*) output, vout0123456701234567);
339*4bdc9457SAndroid Build Coastguard Worker output += 8;
340*4bdc9457SAndroid Build Coastguard Worker }
341*4bdc9457SAndroid Build Coastguard Worker if XNN_UNLIKELY(c != 0) {
342*4bdc9457SAndroid Build Coastguard Worker {
343*4bdc9457SAndroid Build Coastguard Worker __m256i vacc01234567 = _mm256_loadu_si256((const __m256i*) w);
344*4bdc9457SAndroid Build Coastguard Worker
345*4bdc9457SAndroid Build Coastguard Worker
346*4bdc9457SAndroid Build Coastguard Worker const __m256i vi0x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i0));
347*4bdc9457SAndroid Build Coastguard Worker const __m256i vk0x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 0 * sizeof(uint8_t)))), vk_zero_point);
348*4bdc9457SAndroid Build Coastguard Worker
349*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi0x01234567, vk0x01234567));
350*4bdc9457SAndroid Build Coastguard Worker
351*4bdc9457SAndroid Build Coastguard Worker const __m256i vi1x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i1));
352*4bdc9457SAndroid Build Coastguard Worker const __m256i vk1x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 8 * sizeof(uint8_t)))), vk_zero_point);
353*4bdc9457SAndroid Build Coastguard Worker
354*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi1x01234567, vk1x01234567));
355*4bdc9457SAndroid Build Coastguard Worker
356*4bdc9457SAndroid Build Coastguard Worker const __m256i vi2x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i2));
357*4bdc9457SAndroid Build Coastguard Worker const __m256i vk2x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 16 * sizeof(uint8_t)))), vk_zero_point);
358*4bdc9457SAndroid Build Coastguard Worker
359*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi2x01234567, vk2x01234567));
360*4bdc9457SAndroid Build Coastguard Worker
361*4bdc9457SAndroid Build Coastguard Worker const __m256i vi3x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i3));
362*4bdc9457SAndroid Build Coastguard Worker const __m256i vk3x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 24 * sizeof(uint8_t)))), vk_zero_point);
363*4bdc9457SAndroid Build Coastguard Worker
364*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi3x01234567, vk3x01234567));
365*4bdc9457SAndroid Build Coastguard Worker
366*4bdc9457SAndroid Build Coastguard Worker const __m256i vi4x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i4));
367*4bdc9457SAndroid Build Coastguard Worker const __m256i vk4x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 32 * sizeof(uint8_t)))), vk_zero_point);
368*4bdc9457SAndroid Build Coastguard Worker
369*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi4x01234567, vk4x01234567));
370*4bdc9457SAndroid Build Coastguard Worker
371*4bdc9457SAndroid Build Coastguard Worker const __m256i vi5x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i5));
372*4bdc9457SAndroid Build Coastguard Worker const __m256i vk5x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 40 * sizeof(uint8_t)))), vk_zero_point);
373*4bdc9457SAndroid Build Coastguard Worker
374*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi5x01234567, vk5x01234567));
375*4bdc9457SAndroid Build Coastguard Worker
376*4bdc9457SAndroid Build Coastguard Worker const __m256i vi6x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i6));
377*4bdc9457SAndroid Build Coastguard Worker const __m256i vk6x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 48 * sizeof(uint8_t)))), vk_zero_point);
378*4bdc9457SAndroid Build Coastguard Worker
379*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi6x01234567, vk6x01234567));
380*4bdc9457SAndroid Build Coastguard Worker
381*4bdc9457SAndroid Build Coastguard Worker const __m256i vi7x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i7));
382*4bdc9457SAndroid Build Coastguard Worker const __m256i vk7x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 56 * sizeof(uint8_t)))), vk_zero_point);
383*4bdc9457SAndroid Build Coastguard Worker
384*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi7x01234567, vk7x01234567));
385*4bdc9457SAndroid Build Coastguard Worker
386*4bdc9457SAndroid Build Coastguard Worker const __m256i vi8x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i8));
387*4bdc9457SAndroid Build Coastguard Worker const __m256i vk8x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 64 * sizeof(uint8_t)))), vk_zero_point);
388*4bdc9457SAndroid Build Coastguard Worker
389*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi8x01234567, vk8x01234567));
390*4bdc9457SAndroid Build Coastguard Worker
391*4bdc9457SAndroid Build Coastguard Worker const __m256i vi9x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i9));
392*4bdc9457SAndroid Build Coastguard Worker const __m256i vk9x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 72 * sizeof(uint8_t)))), vk_zero_point);
393*4bdc9457SAndroid Build Coastguard Worker
394*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi9x01234567, vk9x01234567));
395*4bdc9457SAndroid Build Coastguard Worker
396*4bdc9457SAndroid Build Coastguard Worker const __m256i vi10x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i10));
397*4bdc9457SAndroid Build Coastguard Worker const __m256i vk10x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 80 * sizeof(uint8_t)))), vk_zero_point);
398*4bdc9457SAndroid Build Coastguard Worker
399*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi10x01234567, vk10x01234567));
400*4bdc9457SAndroid Build Coastguard Worker
401*4bdc9457SAndroid Build Coastguard Worker const __m256i vi11x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i11));
402*4bdc9457SAndroid Build Coastguard Worker const __m256i vk11x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 88 * sizeof(uint8_t)))), vk_zero_point);
403*4bdc9457SAndroid Build Coastguard Worker
404*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi11x01234567, vk11x01234567));
405*4bdc9457SAndroid Build Coastguard Worker
406*4bdc9457SAndroid Build Coastguard Worker const __m256i vi12x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i12));
407*4bdc9457SAndroid Build Coastguard Worker const __m256i vk12x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 96 * sizeof(uint8_t)))), vk_zero_point);
408*4bdc9457SAndroid Build Coastguard Worker
409*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi12x01234567, vk12x01234567));
410*4bdc9457SAndroid Build Coastguard Worker
411*4bdc9457SAndroid Build Coastguard Worker const __m256i vi13x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i13));
412*4bdc9457SAndroid Build Coastguard Worker const __m256i vk13x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 104 * sizeof(uint8_t)))), vk_zero_point);
413*4bdc9457SAndroid Build Coastguard Worker
414*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi13x01234567, vk13x01234567));
415*4bdc9457SAndroid Build Coastguard Worker
416*4bdc9457SAndroid Build Coastguard Worker const __m256i vi14x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i14));
417*4bdc9457SAndroid Build Coastguard Worker const __m256i vk14x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 112 * sizeof(uint8_t)))), vk_zero_point);
418*4bdc9457SAndroid Build Coastguard Worker
419*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi14x01234567, vk14x01234567));
420*4bdc9457SAndroid Build Coastguard Worker
421*4bdc9457SAndroid Build Coastguard Worker const __m256i vi15x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i15));
422*4bdc9457SAndroid Build Coastguard Worker const __m256i vk15x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 120 * sizeof(uint8_t)))), vk_zero_point);
423*4bdc9457SAndroid Build Coastguard Worker
424*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi15x01234567, vk15x01234567));
425*4bdc9457SAndroid Build Coastguard Worker
426*4bdc9457SAndroid Build Coastguard Worker const __m256i vi16x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i16));
427*4bdc9457SAndroid Build Coastguard Worker const __m256i vk16x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 128 * sizeof(uint8_t)))), vk_zero_point);
428*4bdc9457SAndroid Build Coastguard Worker
429*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi16x01234567, vk16x01234567));
430*4bdc9457SAndroid Build Coastguard Worker
431*4bdc9457SAndroid Build Coastguard Worker const __m256i vi17x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i17));
432*4bdc9457SAndroid Build Coastguard Worker const __m256i vk17x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 136 * sizeof(uint8_t)))), vk_zero_point);
433*4bdc9457SAndroid Build Coastguard Worker
434*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi17x01234567, vk17x01234567));
435*4bdc9457SAndroid Build Coastguard Worker
436*4bdc9457SAndroid Build Coastguard Worker const __m256i vi18x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i18));
437*4bdc9457SAndroid Build Coastguard Worker const __m256i vk18x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 144 * sizeof(uint8_t)))), vk_zero_point);
438*4bdc9457SAndroid Build Coastguard Worker
439*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi18x01234567, vk18x01234567));
440*4bdc9457SAndroid Build Coastguard Worker
441*4bdc9457SAndroid Build Coastguard Worker const __m256i vi19x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i19));
442*4bdc9457SAndroid Build Coastguard Worker const __m256i vk19x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 152 * sizeof(uint8_t)))), vk_zero_point);
443*4bdc9457SAndroid Build Coastguard Worker
444*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi19x01234567, vk19x01234567));
445*4bdc9457SAndroid Build Coastguard Worker
446*4bdc9457SAndroid Build Coastguard Worker const __m256i vi20x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i20));
447*4bdc9457SAndroid Build Coastguard Worker const __m256i vk20x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 160 * sizeof(uint8_t)))), vk_zero_point);
448*4bdc9457SAndroid Build Coastguard Worker
449*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi20x01234567, vk20x01234567));
450*4bdc9457SAndroid Build Coastguard Worker
451*4bdc9457SAndroid Build Coastguard Worker const __m256i vi21x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i21));
452*4bdc9457SAndroid Build Coastguard Worker const __m256i vk21x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 168 * sizeof(uint8_t)))), vk_zero_point);
453*4bdc9457SAndroid Build Coastguard Worker
454*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi21x01234567, vk21x01234567));
455*4bdc9457SAndroid Build Coastguard Worker
456*4bdc9457SAndroid Build Coastguard Worker const __m256i vi22x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i22));
457*4bdc9457SAndroid Build Coastguard Worker const __m256i vk22x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 176 * sizeof(uint8_t)))), vk_zero_point);
458*4bdc9457SAndroid Build Coastguard Worker
459*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi22x01234567, vk22x01234567));
460*4bdc9457SAndroid Build Coastguard Worker
461*4bdc9457SAndroid Build Coastguard Worker const __m256i vi23x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i23));
462*4bdc9457SAndroid Build Coastguard Worker const __m256i vk23x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 184 * sizeof(uint8_t)))), vk_zero_point);
463*4bdc9457SAndroid Build Coastguard Worker
464*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi23x01234567, vk23x01234567));
465*4bdc9457SAndroid Build Coastguard Worker
466*4bdc9457SAndroid Build Coastguard Worker const __m256i vi24x01234567 = _mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) i24));
467*4bdc9457SAndroid Build Coastguard Worker const __m256i vk24x01234567 = _mm256_sub_epi32(_mm256_cvtepu8_epi32(_mm_loadl_epi64((const __m128i*) ((uintptr_t) w + 8 * sizeof(int32_t) + 192 * sizeof(uint8_t)))), vk_zero_point);
468*4bdc9457SAndroid Build Coastguard Worker
469*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_add_epi32(vacc01234567, _mm256_mullo_epi32(vi24x01234567, vk24x01234567));
470*4bdc9457SAndroid Build Coastguard Worker
471*4bdc9457SAndroid Build Coastguard Worker
472*4bdc9457SAndroid Build Coastguard Worker __m256 vscaled01234567 = _mm256_cvtepi32_ps(vacc01234567);
473*4bdc9457SAndroid Build Coastguard Worker vscaled01234567 = _mm256_mul_ps(vscaled01234567, _mm256_load_ps(params->fp32_avx2.scale));
474*4bdc9457SAndroid Build Coastguard Worker vscaled01234567 = _mm256_min_ps(vscaled01234567, _mm256_load_ps(params->fp32_avx2.output_max_less_zero_point));
475*4bdc9457SAndroid Build Coastguard Worker vacc01234567 = _mm256_cvtps_epi32(vscaled01234567);
476*4bdc9457SAndroid Build Coastguard Worker
477*4bdc9457SAndroid Build Coastguard Worker
478*4bdc9457SAndroid Build Coastguard Worker const __m128i voutput_zero_point = _mm_load_si128((const __m128i*) params->fp32_avx2.output_zero_point);
479*4bdc9457SAndroid Build Coastguard Worker __m128i vout01234567 = _mm_adds_epi16(_mm_packs_epi32(_mm256_castsi256_si128(vacc01234567), _mm256_extracti128_si256(vacc01234567, 1)), voutput_zero_point);
480*4bdc9457SAndroid Build Coastguard Worker
481*4bdc9457SAndroid Build Coastguard Worker __m128i vout0123456701234567 = _mm_packus_epi16(vout01234567, vout01234567);
482*4bdc9457SAndroid Build Coastguard Worker
483*4bdc9457SAndroid Build Coastguard Worker const __m128i voutput_min = _mm_load_si128((const __m128i*) params->fp32_avx2.output_min);
484*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = _mm_max_epu8(vout0123456701234567, voutput_min);
485*4bdc9457SAndroid Build Coastguard Worker
486*4bdc9457SAndroid Build Coastguard Worker if (c & 4) {
487*4bdc9457SAndroid Build Coastguard Worker unaligned_store_u32(output, (uint32_t) _mm_cvtsi128_si32(vout0123456701234567));
488*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = _mm_srli_epi64(vout0123456701234567, 32);
489*4bdc9457SAndroid Build Coastguard Worker output += 4;
490*4bdc9457SAndroid Build Coastguard Worker }
491*4bdc9457SAndroid Build Coastguard Worker if (c & 2) {
492*4bdc9457SAndroid Build Coastguard Worker unaligned_store_u16(output, (uint16_t) _mm_extract_epi16(vout0123456701234567, 0));
493*4bdc9457SAndroid Build Coastguard Worker vout0123456701234567 = _mm_srli_epi32(vout0123456701234567, 16);
494*4bdc9457SAndroid Build Coastguard Worker output += 2;
495*4bdc9457SAndroid Build Coastguard Worker }
496*4bdc9457SAndroid Build Coastguard Worker if (c & 1) {
497*4bdc9457SAndroid Build Coastguard Worker *output = (uint8_t) _mm_extract_epi8(vout0123456701234567, 0);
498*4bdc9457SAndroid Build Coastguard Worker output += 1;
499*4bdc9457SAndroid Build Coastguard Worker }
500*4bdc9457SAndroid Build Coastguard Worker }
501*4bdc9457SAndroid Build Coastguard Worker }
502*4bdc9457SAndroid Build Coastguard Worker
503*4bdc9457SAndroid Build Coastguard Worker output = (uint8_t*) ((uintptr_t) output + output_increment);
504*4bdc9457SAndroid Build Coastguard Worker } while (--output_width != 0);
505*4bdc9457SAndroid Build Coastguard Worker }
506