xref: /aosp_15_r20/external/XNNPACK/src/qs8-gemm/gen/8x8c4-minmax-rndnu-neondot.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36) 
1 // Auto-generated file. Do not edit!
2 //   Template: src/qs8-gemm/c4-neondot.c.in
3 //   Generator: tools/xngen
4 //
5 // Copyright 2020 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 <arm_neon.h>
13 
14 #include <xnnpack/gemm.h>
15 #include <xnnpack/math.h>
16 
17 
xnn_qs8_gemm_minmax_rndnu_ukernel_8x8c4__neondot(size_t mr,size_t nc,size_t kc,const int8_t * restrict a,size_t a_stride,const void * restrict w,int8_t * restrict c,size_t cm_stride,size_t cn_stride,const union xnn_qs8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])18 void xnn_qs8_gemm_minmax_rndnu_ukernel_8x8c4__neondot(
19     size_t mr,
20     size_t nc,
21     size_t kc,
22     const int8_t* restrict a,
23     size_t a_stride,
24     const void* restrict w,
25     int8_t* restrict c,
26     size_t cm_stride,
27     size_t cn_stride,
28     const union xnn_qs8_conv_minmax_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
29 {
30   assert(mr != 0);
31   assert(mr <= 8);
32   assert(nc != 0);
33   assert(kc != 0);
34   assert(kc % sizeof(int8_t) == 0);
35   assert(a != NULL);
36   assert(w != NULL);
37   assert(c != NULL);
38 
39   kc = round_up_po2(kc, 4 * sizeof(int8_t));
40   const int8_t* a0 = a;
41   int8_t* c0 = c;
42   const int8_t* a1 = (const int8_t*) ((uintptr_t) a0 + a_stride);
43   int8_t* c1 = (int8_t*) ((uintptr_t) c0 + cm_stride);
44   if XNN_UNPREDICTABLE(mr < 2) {
45     a1 = a0;
46     c1 = c0;
47   }
48   const int8_t* a2 = (const int8_t*) ((uintptr_t) a1 + a_stride);
49   int8_t* c2 = (int8_t*) ((uintptr_t) c1 + cm_stride);
50   if XNN_UNPREDICTABLE(mr <= 2) {
51     a2 = a1;
52     c2 = c1;
53   }
54   const int8_t* a3 = (const int8_t*) ((uintptr_t) a2 + a_stride);
55   int8_t* c3 = (int8_t*) ((uintptr_t) c2 + cm_stride);
56   if XNN_UNPREDICTABLE(mr < 4) {
57     a3 = a2;
58     c3 = c2;
59   }
60   const int8_t* a4 = (const int8_t*) ((uintptr_t) a3 + a_stride);
61   int8_t* c4 = (int8_t*) ((uintptr_t) c3 + cm_stride);
62   if XNN_UNPREDICTABLE(mr <= 4) {
63     a4 = a3;
64     c4 = c3;
65   }
66   const int8_t* a5 = (const int8_t*) ((uintptr_t) a4 + a_stride);
67   int8_t* c5 = (int8_t*) ((uintptr_t) c4 + cm_stride);
68   if XNN_UNPREDICTABLE(mr < 6) {
69     a5 = a4;
70     c5 = c4;
71   }
72   const int8_t* a6 = (const int8_t*) ((uintptr_t) a5 + a_stride);
73   int8_t* c6 = (int8_t*) ((uintptr_t) c5 + cm_stride);
74   if XNN_UNPREDICTABLE(mr <= 6) {
75     a6 = a5;
76     c6 = c5;
77   }
78   const int8_t* a7 = (const int8_t*) ((uintptr_t) a6 + a_stride);
79   int8_t* c7 = (int8_t*) ((uintptr_t) c6 + cm_stride);
80   if XNN_UNPREDICTABLE(mr != 8) {
81     a7 = a6;
82     c7 = c6;
83   }
84 
85   // Loop over groups of 8 columns.
86   do {
87     // Initialize accumulators with bias. 8 bias values are loaded from the
88     // weight matrix, at the start of the group of 8 columns.
89     int32x4_t vacc0x0123 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
90     int32x4_t vacc0x4567 = vld1q_s32(w); w = (const void*) ((const int32_t*) w + 4);
91     int32x4_t vacc1x0123 = vacc0x0123;
92     int32x4_t vacc1x4567 = vacc0x4567;
93     int32x4_t vacc2x0123 = vacc0x0123;
94     int32x4_t vacc2x4567 = vacc0x4567;
95     int32x4_t vacc3x0123 = vacc0x0123;
96     int32x4_t vacc3x4567 = vacc0x4567;
97     int32x4_t vacc4x0123 = vacc0x0123;
98     int32x4_t vacc4x4567 = vacc0x4567;
99     int32x4_t vacc5x0123 = vacc0x0123;
100     int32x4_t vacc5x4567 = vacc0x4567;
101     int32x4_t vacc6x0123 = vacc0x0123;
102     int32x4_t vacc6x4567 = vacc0x4567;
103     int32x4_t vacc7x0123 = vacc0x0123;
104     int32x4_t vacc7x4567 = vacc0x4567;
105 
106     // Inner accumulation loop along the 8 columns.
107     size_t k = kc;
108     // 2x partial unrolled loop to load 8 bytes at a time.
109     while (k >= 8 * sizeof(int8_t)) {
110       // Load a 8x8 block of activations.
111       const int8x8_t va0x01234567 = vld1_s8(a0); a0 += 8;
112       const int8x8_t va1x01234567 = vld1_s8(a1); a1 += 8;
113       const int8x8_t va2x01234567 = vld1_s8(a2); a2 += 8;
114       const int8x8_t va3x01234567 = vld1_s8(a3); a3 += 8;
115       const int8x8_t va4x01234567 = vld1_s8(a4); a4 += 8;
116       const int8x8_t va5x01234567 = vld1_s8(a5); a5 += 8;
117       const int8x8_t va6x01234567 = vld1_s8(a6); a6 += 8;
118       const int8x8_t va7x01234567 = vld1_s8(a7); a7 += 8;
119 
120       // Load a 8x8 block of weights.
121       const int8x16_t vb0123x0123 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
122       const int8x16_t vb0123x4567 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
123       const int8x16_t vb4567x0123 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
124       const int8x16_t vb4567x4567 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
125 
126       // Multiply-accumulate: 8x8 * 8x8 --> 8x8.
127       vacc0x0123 = vdotq_lane_s32(vacc0x0123, vb0123x0123, va0x01234567, 0);
128       vacc0x4567 = vdotq_lane_s32(vacc0x4567, vb0123x4567, va0x01234567, 0);
129       vacc1x0123 = vdotq_lane_s32(vacc1x0123, vb0123x0123, va1x01234567, 0);
130       vacc1x4567 = vdotq_lane_s32(vacc1x4567, vb0123x4567, va1x01234567, 0);
131       vacc2x0123 = vdotq_lane_s32(vacc2x0123, vb0123x0123, va2x01234567, 0);
132       vacc2x4567 = vdotq_lane_s32(vacc2x4567, vb0123x4567, va2x01234567, 0);
133       vacc3x0123 = vdotq_lane_s32(vacc3x0123, vb0123x0123, va3x01234567, 0);
134       vacc3x4567 = vdotq_lane_s32(vacc3x4567, vb0123x4567, va3x01234567, 0);
135       vacc4x0123 = vdotq_lane_s32(vacc4x0123, vb0123x0123, va4x01234567, 0);
136       vacc4x4567 = vdotq_lane_s32(vacc4x4567, vb0123x4567, va4x01234567, 0);
137       vacc5x0123 = vdotq_lane_s32(vacc5x0123, vb0123x0123, va5x01234567, 0);
138       vacc5x4567 = vdotq_lane_s32(vacc5x4567, vb0123x4567, va5x01234567, 0);
139       vacc6x0123 = vdotq_lane_s32(vacc6x0123, vb0123x0123, va6x01234567, 0);
140       vacc6x4567 = vdotq_lane_s32(vacc6x4567, vb0123x4567, va6x01234567, 0);
141       vacc7x0123 = vdotq_lane_s32(vacc7x0123, vb0123x0123, va7x01234567, 0);
142       vacc7x4567 = vdotq_lane_s32(vacc7x4567, vb0123x4567, va7x01234567, 0);
143       vacc0x0123 = vdotq_lane_s32(vacc0x0123, vb4567x0123, va0x01234567, 1);
144       vacc0x4567 = vdotq_lane_s32(vacc0x4567, vb4567x4567, va0x01234567, 1);
145       vacc1x0123 = vdotq_lane_s32(vacc1x0123, vb4567x0123, va1x01234567, 1);
146       vacc1x4567 = vdotq_lane_s32(vacc1x4567, vb4567x4567, va1x01234567, 1);
147       vacc2x0123 = vdotq_lane_s32(vacc2x0123, vb4567x0123, va2x01234567, 1);
148       vacc2x4567 = vdotq_lane_s32(vacc2x4567, vb4567x4567, va2x01234567, 1);
149       vacc3x0123 = vdotq_lane_s32(vacc3x0123, vb4567x0123, va3x01234567, 1);
150       vacc3x4567 = vdotq_lane_s32(vacc3x4567, vb4567x4567, va3x01234567, 1);
151       vacc4x0123 = vdotq_lane_s32(vacc4x0123, vb4567x0123, va4x01234567, 1);
152       vacc4x4567 = vdotq_lane_s32(vacc4x4567, vb4567x4567, va4x01234567, 1);
153       vacc5x0123 = vdotq_lane_s32(vacc5x0123, vb4567x0123, va5x01234567, 1);
154       vacc5x4567 = vdotq_lane_s32(vacc5x4567, vb4567x4567, va5x01234567, 1);
155       vacc6x0123 = vdotq_lane_s32(vacc6x0123, vb4567x0123, va6x01234567, 1);
156       vacc6x4567 = vdotq_lane_s32(vacc6x4567, vb4567x4567, va6x01234567, 1);
157       vacc7x0123 = vdotq_lane_s32(vacc7x0123, vb4567x0123, va7x01234567, 1);
158       vacc7x4567 = vdotq_lane_s32(vacc7x4567, vb4567x4567, va7x01234567, 1);
159 
160       k -= 8 * sizeof(int8_t);
161     }
162     // Handle up to 4 final positions of `k`
163     if XNN_UNLIKELY(k != 0) {
164       // Load a 8x4 block of activations.
165       const int8x8_t va0x01234567 = vld1_s8(a0); a0 += 4;
166       const int8x8_t va1x01234567 = vld1_s8(a1); a1 += 4;
167       const int8x8_t va2x01234567 = vld1_s8(a2); a2 += 4;
168       const int8x8_t va3x01234567 = vld1_s8(a3); a3 += 4;
169       const int8x8_t va4x01234567 = vld1_s8(a4); a4 += 4;
170       const int8x8_t va5x01234567 = vld1_s8(a5); a5 += 4;
171       const int8x8_t va6x01234567 = vld1_s8(a6); a6 += 4;
172       const int8x8_t va7x01234567 = vld1_s8(a7); a7 += 4;
173 
174       // Load a 4x8 block of weights.
175       const int8x16_t vb0123x0123 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
176       const int8x16_t vb0123x4567 = vld1q_s8(w); w = (const void*) ((const int8_t*) w + 16);
177 
178       // Multiply-accumulate: 8x4 * 4x8 --> 8x8.
179       vacc0x0123 = vdotq_lane_s32(vacc0x0123, vb0123x0123, va0x01234567, 0);
180       vacc0x4567 = vdotq_lane_s32(vacc0x4567, vb0123x4567, va0x01234567, 0);
181       vacc1x0123 = vdotq_lane_s32(vacc1x0123, vb0123x0123, va1x01234567, 0);
182       vacc1x4567 = vdotq_lane_s32(vacc1x4567, vb0123x4567, va1x01234567, 0);
183       vacc2x0123 = vdotq_lane_s32(vacc2x0123, vb0123x0123, va2x01234567, 0);
184       vacc2x4567 = vdotq_lane_s32(vacc2x4567, vb0123x4567, va2x01234567, 0);
185       vacc3x0123 = vdotq_lane_s32(vacc3x0123, vb0123x0123, va3x01234567, 0);
186       vacc3x4567 = vdotq_lane_s32(vacc3x4567, vb0123x4567, va3x01234567, 0);
187       vacc4x0123 = vdotq_lane_s32(vacc4x0123, vb0123x0123, va4x01234567, 0);
188       vacc4x4567 = vdotq_lane_s32(vacc4x4567, vb0123x4567, va4x01234567, 0);
189       vacc5x0123 = vdotq_lane_s32(vacc5x0123, vb0123x0123, va5x01234567, 0);
190       vacc5x4567 = vdotq_lane_s32(vacc5x4567, vb0123x4567, va5x01234567, 0);
191       vacc6x0123 = vdotq_lane_s32(vacc6x0123, vb0123x0123, va6x01234567, 0);
192       vacc6x4567 = vdotq_lane_s32(vacc6x4567, vb0123x4567, va6x01234567, 0);
193       vacc7x0123 = vdotq_lane_s32(vacc7x0123, vb0123x0123, va7x01234567, 0);
194       vacc7x4567 = vdotq_lane_s32(vacc7x4567, vb0123x4567, va7x01234567, 0);
195     }
196 
197     const int32x4_t vright_pre_shift = vld1q_dup_s32(&params->rndnu_neon.right_pre_shift);
198     const int32x4_t vmultiplier = vld1q_dup_s32(&params->rndnu_neon.multiplier);
199     const int32x4_t vright_post_shift = vld1q_dup_s32(&params->rndnu_neon.right_post_shift);
200 
201     vacc0x0123 = vqshlq_s32(vacc0x0123, vright_pre_shift);
202     vacc0x4567 = vqshlq_s32(vacc0x4567, vright_pre_shift);
203     vacc1x0123 = vqshlq_s32(vacc1x0123, vright_pre_shift);
204     vacc1x4567 = vqshlq_s32(vacc1x4567, vright_pre_shift);
205     vacc2x0123 = vqshlq_s32(vacc2x0123, vright_pre_shift);
206     vacc2x4567 = vqshlq_s32(vacc2x4567, vright_pre_shift);
207     vacc3x0123 = vqshlq_s32(vacc3x0123, vright_pre_shift);
208     vacc3x4567 = vqshlq_s32(vacc3x4567, vright_pre_shift);
209     vacc4x0123 = vqshlq_s32(vacc4x0123, vright_pre_shift);
210     vacc4x4567 = vqshlq_s32(vacc4x4567, vright_pre_shift);
211     vacc5x0123 = vqshlq_s32(vacc5x0123, vright_pre_shift);
212     vacc5x4567 = vqshlq_s32(vacc5x4567, vright_pre_shift);
213     vacc6x0123 = vqshlq_s32(vacc6x0123, vright_pre_shift);
214     vacc6x4567 = vqshlq_s32(vacc6x4567, vright_pre_shift);
215     vacc7x0123 = vqshlq_s32(vacc7x0123, vright_pre_shift);
216     vacc7x4567 = vqshlq_s32(vacc7x4567, vright_pre_shift);
217 
218     vacc0x0123 = vqdmulhq_s32(vacc0x0123, vmultiplier);
219     vacc0x4567 = vqdmulhq_s32(vacc0x4567, vmultiplier);
220     vacc1x0123 = vqdmulhq_s32(vacc1x0123, vmultiplier);
221     vacc1x4567 = vqdmulhq_s32(vacc1x4567, vmultiplier);
222     vacc2x0123 = vqdmulhq_s32(vacc2x0123, vmultiplier);
223     vacc2x4567 = vqdmulhq_s32(vacc2x4567, vmultiplier);
224     vacc3x0123 = vqdmulhq_s32(vacc3x0123, vmultiplier);
225     vacc3x4567 = vqdmulhq_s32(vacc3x4567, vmultiplier);
226     vacc4x0123 = vqdmulhq_s32(vacc4x0123, vmultiplier);
227     vacc4x4567 = vqdmulhq_s32(vacc4x4567, vmultiplier);
228     vacc5x0123 = vqdmulhq_s32(vacc5x0123, vmultiplier);
229     vacc5x4567 = vqdmulhq_s32(vacc5x4567, vmultiplier);
230     vacc6x0123 = vqdmulhq_s32(vacc6x0123, vmultiplier);
231     vacc6x4567 = vqdmulhq_s32(vacc6x4567, vmultiplier);
232     vacc7x0123 = vqdmulhq_s32(vacc7x0123, vmultiplier);
233     vacc7x4567 = vqdmulhq_s32(vacc7x4567, vmultiplier);
234 
235     vacc0x0123 = vrshlq_s32(vacc0x0123, vright_post_shift);
236     vacc0x4567 = vrshlq_s32(vacc0x4567, vright_post_shift);
237     vacc1x0123 = vrshlq_s32(vacc1x0123, vright_post_shift);
238     vacc1x4567 = vrshlq_s32(vacc1x4567, vright_post_shift);
239     vacc2x0123 = vrshlq_s32(vacc2x0123, vright_post_shift);
240     vacc2x4567 = vrshlq_s32(vacc2x4567, vright_post_shift);
241     vacc3x0123 = vrshlq_s32(vacc3x0123, vright_post_shift);
242     vacc3x4567 = vrshlq_s32(vacc3x4567, vright_post_shift);
243     vacc4x0123 = vrshlq_s32(vacc4x0123, vright_post_shift);
244     vacc4x4567 = vrshlq_s32(vacc4x4567, vright_post_shift);
245     vacc5x0123 = vrshlq_s32(vacc5x0123, vright_post_shift);
246     vacc5x4567 = vrshlq_s32(vacc5x4567, vright_post_shift);
247     vacc6x0123 = vrshlq_s32(vacc6x0123, vright_post_shift);
248     vacc6x4567 = vrshlq_s32(vacc6x4567, vright_post_shift);
249     vacc7x0123 = vrshlq_s32(vacc7x0123, vright_post_shift);
250     vacc7x4567 = vrshlq_s32(vacc7x4567, vright_post_shift);
251 
252     const int16x8_t voutput_zero_point = vld1q_dup_s16(&params->rndnu_neon.output_zero_point);
253 #if XNN_ARCH_ARM64
254     const int16x8_t vacc0x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc0x0123), vacc0x4567), voutput_zero_point);
255     const int16x8_t vacc1x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc1x0123), vacc1x4567), voutput_zero_point);
256     const int16x8_t vacc2x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc2x0123), vacc2x4567), voutput_zero_point);
257     const int16x8_t vacc3x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc3x0123), vacc3x4567), voutput_zero_point);
258     const int16x8_t vacc4x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc4x0123), vacc4x4567), voutput_zero_point);
259     const int16x8_t vacc5x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc5x0123), vacc5x4567), voutput_zero_point);
260     const int16x8_t vacc6x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc6x0123), vacc6x4567), voutput_zero_point);
261     const int16x8_t vacc7x01234567 = vqaddq_s16(vqmovn_high_s32(vqmovn_s32(vacc7x0123), vacc7x4567), voutput_zero_point);
262 
263     int8x16_t vout0x01234567_1x01234567 = vqmovn_high_s16(vqmovn_s16(vacc0x01234567), vacc1x01234567);
264     int8x16_t vout2x01234567_3x01234567 = vqmovn_high_s16(vqmovn_s16(vacc2x01234567), vacc3x01234567);
265     int8x16_t vout4x01234567_5x01234567 = vqmovn_high_s16(vqmovn_s16(vacc4x01234567), vacc5x01234567);
266     int8x16_t vout6x01234567_7x01234567 = vqmovn_high_s16(vqmovn_s16(vacc6x01234567), vacc7x01234567);
267 #else
268     const int16x8_t vacc0x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc0x0123), vqmovn_s32(vacc0x4567)), voutput_zero_point);
269     const int16x8_t vacc1x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc1x0123), vqmovn_s32(vacc1x4567)), voutput_zero_point);
270     const int16x8_t vacc2x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc2x0123), vqmovn_s32(vacc2x4567)), voutput_zero_point);
271     const int16x8_t vacc3x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc3x0123), vqmovn_s32(vacc3x4567)), voutput_zero_point);
272     const int16x8_t vacc4x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc4x0123), vqmovn_s32(vacc4x4567)), voutput_zero_point);
273     const int16x8_t vacc5x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc5x0123), vqmovn_s32(vacc5x4567)), voutput_zero_point);
274     const int16x8_t vacc6x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc6x0123), vqmovn_s32(vacc6x4567)), voutput_zero_point);
275     const int16x8_t vacc7x01234567 = vqaddq_s16(vcombine_s16(vqmovn_s32(vacc7x0123), vqmovn_s32(vacc7x4567)), voutput_zero_point);
276 
277     int8x16_t vout0x01234567_1x01234567 = vcombine_s8(vqmovn_s16(vacc0x01234567), vqmovn_s16(vacc1x01234567));
278     int8x16_t vout2x01234567_3x01234567 = vcombine_s8(vqmovn_s16(vacc2x01234567), vqmovn_s16(vacc3x01234567));
279     int8x16_t vout4x01234567_5x01234567 = vcombine_s8(vqmovn_s16(vacc4x01234567), vqmovn_s16(vacc5x01234567));
280     int8x16_t vout6x01234567_7x01234567 = vcombine_s8(vqmovn_s16(vacc6x01234567), vqmovn_s16(vacc7x01234567));
281 #endif
282     const int8x16_t voutput_min = vld1q_dup_s8(&params->rndnu_neon.output_min);
283     const int8x16_t voutput_max = vld1q_dup_s8(&params->rndnu_neon.output_max);
284 
285     vout0x01234567_1x01234567 = vmaxq_s8(vout0x01234567_1x01234567, voutput_min);
286     vout2x01234567_3x01234567 = vmaxq_s8(vout2x01234567_3x01234567, voutput_min);
287     vout4x01234567_5x01234567 = vmaxq_s8(vout4x01234567_5x01234567, voutput_min);
288     vout6x01234567_7x01234567 = vmaxq_s8(vout6x01234567_7x01234567, voutput_min);
289 
290     vout0x01234567_1x01234567 = vminq_s8(vout0x01234567_1x01234567, voutput_max);
291     vout2x01234567_3x01234567 = vminq_s8(vout2x01234567_3x01234567, voutput_max);
292     vout4x01234567_5x01234567 = vminq_s8(vout4x01234567_5x01234567, voutput_max);
293     vout6x01234567_7x01234567 = vminq_s8(vout6x01234567_7x01234567, voutput_max);
294 
295     if (nc >= 8) {
296       // Main case where there the 8 columns fit in the destination.
297       vst1_s8(c0 + 0, vget_low_s8(vout0x01234567_1x01234567));
298       vst1_s8(c1 + 0, vget_high_s8(vout0x01234567_1x01234567));
299       vst1_s8(c2 + 0, vget_low_s8(vout2x01234567_3x01234567));
300       vst1_s8(c3 + 0, vget_high_s8(vout2x01234567_3x01234567));
301       vst1_s8(c4 + 0, vget_low_s8(vout4x01234567_5x01234567));
302       vst1_s8(c5 + 0, vget_high_s8(vout4x01234567_5x01234567));
303       vst1_s8(c6 + 0, vget_low_s8(vout6x01234567_7x01234567));
304       vst1_s8(c7 + 0, vget_high_s8(vout6x01234567_7x01234567));
305 
306       // Advance to the next 8 columns.
307       c0 = (int8_t*) ((uintptr_t) c0 + cn_stride);
308       c1 = (int8_t*) ((uintptr_t) c1 + cn_stride);
309       c2 = (int8_t*) ((uintptr_t) c2 + cn_stride);
310       c3 = (int8_t*) ((uintptr_t) c3 + cn_stride);
311       c4 = (int8_t*) ((uintptr_t) c4 + cn_stride);
312       c5 = (int8_t*) ((uintptr_t) c5 + cn_stride);
313       c6 = (int8_t*) ((uintptr_t) c6 + cn_stride);
314       c7 = (int8_t*) ((uintptr_t) c7 + cn_stride);
315 
316       a0 = (const int8_t*) ((uintptr_t) a0 - kc);
317       a1 = (const int8_t*) ((uintptr_t) a1 - kc);
318       a2 = (const int8_t*) ((uintptr_t) a2 - kc);
319       a3 = (const int8_t*) ((uintptr_t) a3 - kc);
320       a4 = (const int8_t*) ((uintptr_t) a4 - kc);
321       a5 = (const int8_t*) ((uintptr_t) a5 - kc);
322       a6 = (const int8_t*) ((uintptr_t) a6 - kc);
323       a7 = (const int8_t*) ((uintptr_t) a7 - kc);
324 
325       nc -= 8;
326     } else {
327       // Final case where not all of the 8 columns fit in the destination.
328       if (nc & 4) {
329         vst1q_lane_u32((void*) c0, vreinterpretq_u32_s8(vout0x01234567_1x01234567), 0); c0 += 4;
330         vst1q_lane_u32((void*) c1, vreinterpretq_u32_s8(vout0x01234567_1x01234567), 2); c1 += 4;
331         vst1q_lane_u32((void*) c2, vreinterpretq_u32_s8(vout2x01234567_3x01234567), 0); c2 += 4;
332         vst1q_lane_u32((void*) c3, vreinterpretq_u32_s8(vout2x01234567_3x01234567), 2); c3 += 4;
333         vst1q_lane_u32((void*) c4, vreinterpretq_u32_s8(vout4x01234567_5x01234567), 0); c4 += 4;
334         vst1q_lane_u32((void*) c5, vreinterpretq_u32_s8(vout4x01234567_5x01234567), 2); c5 += 4;
335         vst1q_lane_u32((void*) c6, vreinterpretq_u32_s8(vout6x01234567_7x01234567), 0); c6 += 4;
336         vst1q_lane_u32((void*) c7, vreinterpretq_u32_s8(vout6x01234567_7x01234567), 2); c7 += 4;
337         vout0x01234567_1x01234567 = vextq_s8(vout0x01234567_1x01234567, vout0x01234567_1x01234567, 4);
338         vout2x01234567_3x01234567 = vextq_s8(vout2x01234567_3x01234567, vout2x01234567_3x01234567, 4);
339         vout4x01234567_5x01234567 = vextq_s8(vout4x01234567_5x01234567, vout4x01234567_5x01234567, 4);
340         vout6x01234567_7x01234567 = vextq_s8(vout6x01234567_7x01234567, vout6x01234567_7x01234567, 4);
341       }
342       if (nc & 2) {
343         vst1q_lane_u16((void*) c0, vreinterpretq_u16_s8(vout0x01234567_1x01234567), 0); c0 += 2;
344         vst1q_lane_u16((void*) c1, vreinterpretq_u16_s8(vout0x01234567_1x01234567), 4); c1 += 2;
345         vst1q_lane_u16((void*) c2, vreinterpretq_u16_s8(vout2x01234567_3x01234567), 0); c2 += 2;
346         vst1q_lane_u16((void*) c3, vreinterpretq_u16_s8(vout2x01234567_3x01234567), 4); c3 += 2;
347         vst1q_lane_u16((void*) c4, vreinterpretq_u16_s8(vout4x01234567_5x01234567), 0); c4 += 2;
348         vst1q_lane_u16((void*) c5, vreinterpretq_u16_s8(vout4x01234567_5x01234567), 4); c5 += 2;
349         vst1q_lane_u16((void*) c6, vreinterpretq_u16_s8(vout6x01234567_7x01234567), 0); c6 += 2;
350         vst1q_lane_u16((void*) c7, vreinterpretq_u16_s8(vout6x01234567_7x01234567), 4); c7 += 2;
351         vout0x01234567_1x01234567 = vextq_s8(vout0x01234567_1x01234567, vout0x01234567_1x01234567, 2);
352         vout2x01234567_3x01234567 = vextq_s8(vout2x01234567_3x01234567, vout2x01234567_3x01234567, 2);
353         vout4x01234567_5x01234567 = vextq_s8(vout4x01234567_5x01234567, vout4x01234567_5x01234567, 2);
354         vout6x01234567_7x01234567 = vextq_s8(vout6x01234567_7x01234567, vout6x01234567_7x01234567, 2);
355       }
356       if (nc & 1) {
357         vst1q_lane_s8(c0, vout0x01234567_1x01234567, 0);
358         vst1q_lane_s8(c1, vout0x01234567_1x01234567, 8);
359         vst1q_lane_s8(c2, vout2x01234567_3x01234567, 0);
360         vst1q_lane_s8(c3, vout2x01234567_3x01234567, 8);
361         vst1q_lane_s8(c4, vout4x01234567_5x01234567, 0);
362         vst1q_lane_s8(c5, vout4x01234567_5x01234567, 8);
363         vst1q_lane_s8(c6, vout6x01234567_7x01234567, 0);
364         vst1q_lane_s8(c7, vout6x01234567_7x01234567, 8);
365       }
366 
367       nc = 0;
368     }
369   } while (nc != 0);
370 }
371