xref: /aosp_15_r20/external/XNNPACK/src/f32-spmm/gen/12x4-minmax-neonfma.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker //   Template: src/f32-spmm/neon-blocked.c.in
3*4bdc9457SAndroid Build Coastguard Worker //   Generator: tools/xngen
4*4bdc9457SAndroid Build Coastguard Worker //
5*4bdc9457SAndroid Build Coastguard Worker // Copyright 2019 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 <arm_neon.h>
13*4bdc9457SAndroid Build Coastguard Worker 
14*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/spmm.h>
15*4bdc9457SAndroid Build Coastguard Worker 
16*4bdc9457SAndroid Build Coastguard Worker 
xnn_f32_spmm_minmax_ukernel_12x4__neonfma(size_t mc,size_t nc,const float * restrict input,const float * restrict weights,const int32_t * restrict widx_dmap,const uint32_t * restrict nidx_nnzmap,float * restrict output,size_t output_stride,const union xnn_f32_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])17*4bdc9457SAndroid Build Coastguard Worker void xnn_f32_spmm_minmax_ukernel_12x4__neonfma(
18*4bdc9457SAndroid Build Coastguard Worker     size_t mc,
19*4bdc9457SAndroid Build Coastguard Worker     size_t nc,
20*4bdc9457SAndroid Build Coastguard Worker     const float*restrict input,
21*4bdc9457SAndroid Build Coastguard Worker     const float*restrict weights,
22*4bdc9457SAndroid Build Coastguard Worker     const int32_t*restrict widx_dmap,
23*4bdc9457SAndroid Build Coastguard Worker     const uint32_t*restrict nidx_nnzmap,
24*4bdc9457SAndroid Build Coastguard Worker     float*restrict output,
25*4bdc9457SAndroid Build Coastguard Worker     size_t output_stride,
26*4bdc9457SAndroid Build Coastguard Worker     const union xnn_f32_minmax_params params[restrict XNN_MIN_ELEMENTS(1)])
27*4bdc9457SAndroid Build Coastguard Worker {
28*4bdc9457SAndroid Build Coastguard Worker   assert(mc != 0);
29*4bdc9457SAndroid Build Coastguard Worker   assert(mc % sizeof(float) == 0);
30*4bdc9457SAndroid Build Coastguard Worker   assert(nc != 0);
31*4bdc9457SAndroid Build Coastguard Worker 
32*4bdc9457SAndroid Build Coastguard Worker   const float32x4_t vmin = vld1q_dup_f32(&params->scalar.min);
33*4bdc9457SAndroid Build Coastguard Worker   const float32x4_t vmax = vld1q_dup_f32(&params->scalar.max);
34*4bdc9457SAndroid Build Coastguard Worker   size_t output_decrement = output_stride * nc - 12 * sizeof(float);
35*4bdc9457SAndroid Build Coastguard Worker   while XNN_LIKELY(mc >= 12 * sizeof(float)) {
36*4bdc9457SAndroid Build Coastguard Worker     const float*restrict w = weights;
37*4bdc9457SAndroid Build Coastguard Worker     const int32_t* dmap = widx_dmap;
38*4bdc9457SAndroid Build Coastguard Worker     const uint32_t* nnzmap = nidx_nnzmap;
39*4bdc9457SAndroid Build Coastguard Worker     size_t n = nc;
40*4bdc9457SAndroid Build Coastguard Worker     while (n >= 4) {
41*4bdc9457SAndroid Build Coastguard Worker       uint32_t nnz = *nnzmap++;
42*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc0123n0 = vld1q_dup_f32(w); w += 1;
43*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc4567n0 = vacc0123n0;
44*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc89ABn0 = vacc0123n0;
45*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc0123n1 = vld1q_dup_f32(w); w += 1;
46*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc4567n1 = vacc0123n1;
47*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc89ABn1 = vacc0123n1;
48*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc0123n2 = vld1q_dup_f32(w); w += 1;
49*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc4567n2 = vacc0123n2;
50*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc89ABn2 = vacc0123n2;
51*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc0123n3 = vld1q_dup_f32(w); w += 1;
52*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc4567n3 = vacc0123n3;
53*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vacc89ABn3 = vacc0123n3;
54*4bdc9457SAndroid Build Coastguard Worker       if XNN_LIKELY(nnz != 0) {
55*4bdc9457SAndroid Build Coastguard Worker         do {
56*4bdc9457SAndroid Build Coastguard Worker           const intptr_t diff = *dmap++;
57*4bdc9457SAndroid Build Coastguard Worker           const float32x4_t vi0123 = vld1q_f32(input);
58*4bdc9457SAndroid Build Coastguard Worker           const float32x4_t vi4567 = vld1q_f32(input + 4);
59*4bdc9457SAndroid Build Coastguard Worker           const float32x4_t vi89AB = vld1q_f32(input + 8);
60*4bdc9457SAndroid Build Coastguard Worker           input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
61*4bdc9457SAndroid Build Coastguard Worker           __builtin_prefetch(input + 16);
62*4bdc9457SAndroid Build Coastguard Worker           const float32x4_t vw = vld1q_f32(w); w += 4;
63*4bdc9457SAndroid Build Coastguard Worker           __builtin_prefetch(w + 32);
64*4bdc9457SAndroid Build Coastguard Worker           vacc0123n0 = vfmaq_laneq_f32(vacc0123n0, vi0123, vw, 0);
65*4bdc9457SAndroid Build Coastguard Worker           vacc4567n0 = vfmaq_laneq_f32(vacc4567n0, vi4567, vw, 0);
66*4bdc9457SAndroid Build Coastguard Worker           vacc89ABn0 = vfmaq_laneq_f32(vacc89ABn0, vi89AB, vw, 0);
67*4bdc9457SAndroid Build Coastguard Worker           vacc0123n1 = vfmaq_laneq_f32(vacc0123n1, vi0123, vw, 1);
68*4bdc9457SAndroid Build Coastguard Worker           vacc4567n1 = vfmaq_laneq_f32(vacc4567n1, vi4567, vw, 1);
69*4bdc9457SAndroid Build Coastguard Worker           vacc89ABn1 = vfmaq_laneq_f32(vacc89ABn1, vi89AB, vw, 1);
70*4bdc9457SAndroid Build Coastguard Worker           vacc0123n2 = vfmaq_laneq_f32(vacc0123n2, vi0123, vw, 2);
71*4bdc9457SAndroid Build Coastguard Worker           vacc4567n2 = vfmaq_laneq_f32(vacc4567n2, vi4567, vw, 2);
72*4bdc9457SAndroid Build Coastguard Worker           vacc89ABn2 = vfmaq_laneq_f32(vacc89ABn2, vi89AB, vw, 2);
73*4bdc9457SAndroid Build Coastguard Worker           vacc0123n3 = vfmaq_laneq_f32(vacc0123n3, vi0123, vw, 3);
74*4bdc9457SAndroid Build Coastguard Worker           vacc4567n3 = vfmaq_laneq_f32(vacc4567n3, vi4567, vw, 3);
75*4bdc9457SAndroid Build Coastguard Worker           vacc89ABn3 = vfmaq_laneq_f32(vacc89ABn3, vi89AB, vw, 3);
76*4bdc9457SAndroid Build Coastguard Worker         } while (--nnz != 0);
77*4bdc9457SAndroid Build Coastguard Worker       }
78*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout0123n0 = vminq_f32(vacc0123n0, vmax);
79*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout4567n0 = vminq_f32(vacc4567n0, vmax);
80*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout89ABn0 = vminq_f32(vacc89ABn0, vmax);
81*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout0123n1 = vminq_f32(vacc0123n1, vmax);
82*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout4567n1 = vminq_f32(vacc4567n1, vmax);
83*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout89ABn1 = vminq_f32(vacc89ABn1, vmax);
84*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout0123n2 = vminq_f32(vacc0123n2, vmax);
85*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout4567n2 = vminq_f32(vacc4567n2, vmax);
86*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout89ABn2 = vminq_f32(vacc89ABn2, vmax);
87*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout0123n3 = vminq_f32(vacc0123n3, vmax);
88*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout4567n3 = vminq_f32(vacc4567n3, vmax);
89*4bdc9457SAndroid Build Coastguard Worker       float32x4_t vout89ABn3 = vminq_f32(vacc89ABn3, vmax);
90*4bdc9457SAndroid Build Coastguard Worker 
91*4bdc9457SAndroid Build Coastguard Worker       vout0123n0 = vmaxq_f32(vout0123n0, vmin);
92*4bdc9457SAndroid Build Coastguard Worker       vout4567n0 = vmaxq_f32(vout4567n0, vmin);
93*4bdc9457SAndroid Build Coastguard Worker       vout89ABn0 = vmaxq_f32(vout89ABn0, vmin);
94*4bdc9457SAndroid Build Coastguard Worker       vout0123n1 = vmaxq_f32(vout0123n1, vmin);
95*4bdc9457SAndroid Build Coastguard Worker       vout4567n1 = vmaxq_f32(vout4567n1, vmin);
96*4bdc9457SAndroid Build Coastguard Worker       vout89ABn1 = vmaxq_f32(vout89ABn1, vmin);
97*4bdc9457SAndroid Build Coastguard Worker       vout0123n2 = vmaxq_f32(vout0123n2, vmin);
98*4bdc9457SAndroid Build Coastguard Worker       vout4567n2 = vmaxq_f32(vout4567n2, vmin);
99*4bdc9457SAndroid Build Coastguard Worker       vout89ABn2 = vmaxq_f32(vout89ABn2, vmin);
100*4bdc9457SAndroid Build Coastguard Worker       vout0123n3 = vmaxq_f32(vout0123n3, vmin);
101*4bdc9457SAndroid Build Coastguard Worker       vout4567n3 = vmaxq_f32(vout4567n3, vmin);
102*4bdc9457SAndroid Build Coastguard Worker       vout89ABn3 = vmaxq_f32(vout89ABn3, vmin);
103*4bdc9457SAndroid Build Coastguard Worker 
104*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 0, vout0123n0);
105*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 4, vout4567n0);
106*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 8, vout89ABn0);
107*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output + output_stride);
108*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 0, vout0123n1);
109*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 4, vout4567n1);
110*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 8, vout89ABn1);
111*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output + output_stride);
112*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 0, vout0123n2);
113*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 4, vout4567n2);
114*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 8, vout89ABn2);
115*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output + output_stride);
116*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 0, vout0123n3);
117*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 4, vout4567n3);
118*4bdc9457SAndroid Build Coastguard Worker       vst1q_f32(output + 8, vout89ABn3);
119*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output + output_stride);
120*4bdc9457SAndroid Build Coastguard Worker       n -= 4;
121*4bdc9457SAndroid Build Coastguard Worker     }
122*4bdc9457SAndroid Build Coastguard Worker 
123*4bdc9457SAndroid Build Coastguard Worker     // clean up loop, fall back to nr=1
124*4bdc9457SAndroid Build Coastguard Worker     if XNN_UNLIKELY(n != 0) {
125*4bdc9457SAndroid Build Coastguard Worker       do {
126*4bdc9457SAndroid Build Coastguard Worker         uint32_t nnz = *nnzmap++;
127*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123 = vld1q_dup_f32(w); w += 1;
128*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc4567 = vacc0123;
129*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc89AB = vacc0123;
130*4bdc9457SAndroid Build Coastguard Worker         if XNN_LIKELY(nnz != 0) {
131*4bdc9457SAndroid Build Coastguard Worker           do {
132*4bdc9457SAndroid Build Coastguard Worker             const intptr_t diff = *dmap++;
133*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi0123 = vld1q_f32(input);
134*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi4567 = vld1q_f32(input + 4);
135*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi89AB = vld1q_f32(input + 8);
136*4bdc9457SAndroid Build Coastguard Worker             input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
137*4bdc9457SAndroid Build Coastguard Worker             __builtin_prefetch(input + 16);
138*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vw = vld1q_dup_f32(w); w += 1;
139*4bdc9457SAndroid Build Coastguard Worker             __builtin_prefetch(w + 32);
140*4bdc9457SAndroid Build Coastguard Worker             vacc0123 = vfmaq_f32(vacc0123, vi0123, vw);
141*4bdc9457SAndroid Build Coastguard Worker             vacc4567 = vfmaq_f32(vacc4567, vi4567, vw);
142*4bdc9457SAndroid Build Coastguard Worker             vacc89AB = vfmaq_f32(vacc89AB, vi89AB, vw);
143*4bdc9457SAndroid Build Coastguard Worker           } while (--nnz != 0);
144*4bdc9457SAndroid Build Coastguard Worker         }
145*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123 = vminq_f32(vacc0123, vmax);
146*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout4567 = vminq_f32(vacc4567, vmax);
147*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout89AB = vminq_f32(vacc89AB, vmax);
148*4bdc9457SAndroid Build Coastguard Worker 
149*4bdc9457SAndroid Build Coastguard Worker         vout0123 = vmaxq_f32(vout0123, vmin);
150*4bdc9457SAndroid Build Coastguard Worker         vout4567 = vmaxq_f32(vout4567, vmin);
151*4bdc9457SAndroid Build Coastguard Worker         vout89AB = vmaxq_f32(vout89AB, vmin);
152*4bdc9457SAndroid Build Coastguard Worker 
153*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123);
154*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 4, vout4567);
155*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 8, vout89AB);
156*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
157*4bdc9457SAndroid Build Coastguard Worker         n -= 1;
158*4bdc9457SAndroid Build Coastguard Worker       } while (n != 0);
159*4bdc9457SAndroid Build Coastguard Worker     }
160*4bdc9457SAndroid Build Coastguard Worker     output = (float*restrict) ((uintptr_t) output - output_decrement);
161*4bdc9457SAndroid Build Coastguard Worker     input += 12;
162*4bdc9457SAndroid Build Coastguard Worker     mc -= 12 * sizeof(float);
163*4bdc9457SAndroid Build Coastguard Worker   }
164*4bdc9457SAndroid Build Coastguard Worker   if XNN_UNLIKELY(mc != 0) {
165*4bdc9457SAndroid Build Coastguard Worker     output_decrement += 4 * sizeof(float);
166*4bdc9457SAndroid Build Coastguard Worker     if (mc & (8 * sizeof(float))) {
167*4bdc9457SAndroid Build Coastguard Worker       const float*restrict w = weights;
168*4bdc9457SAndroid Build Coastguard Worker       const int32_t* dmap = widx_dmap;
169*4bdc9457SAndroid Build Coastguard Worker       const uint32_t* nnzmap = nidx_nnzmap;
170*4bdc9457SAndroid Build Coastguard Worker       size_t n = nc;
171*4bdc9457SAndroid Build Coastguard Worker       while (n >= 4) {
172*4bdc9457SAndroid Build Coastguard Worker         uint32_t nnz = *nnzmap++;
173*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n0 = vld1q_dup_f32(w); w += 1;
174*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc4567n0 = vacc0123n0;
175*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n1 = vld1q_dup_f32(w); w += 1;
176*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc4567n1 = vacc0123n1;
177*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n2 = vld1q_dup_f32(w); w += 1;
178*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc4567n2 = vacc0123n2;
179*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n3 = vld1q_dup_f32(w); w += 1;
180*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc4567n3 = vacc0123n3;
181*4bdc9457SAndroid Build Coastguard Worker         if XNN_LIKELY(nnz != 0) {
182*4bdc9457SAndroid Build Coastguard Worker           do {
183*4bdc9457SAndroid Build Coastguard Worker             const intptr_t diff = *dmap++;
184*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi0123 = vld1q_f32(input);
185*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi4567 = vld1q_f32(input + 4);
186*4bdc9457SAndroid Build Coastguard Worker             input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
187*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vw = vld1q_f32(w); w += 4;
188*4bdc9457SAndroid Build Coastguard Worker 
189*4bdc9457SAndroid Build Coastguard Worker             vacc0123n0 = vfmaq_laneq_f32(vacc0123n0, vi0123, vw, 0);
190*4bdc9457SAndroid Build Coastguard Worker             vacc4567n0 = vfmaq_laneq_f32(vacc4567n0, vi4567, vw, 0);
191*4bdc9457SAndroid Build Coastguard Worker             vacc0123n1 = vfmaq_laneq_f32(vacc0123n1, vi0123, vw, 1);
192*4bdc9457SAndroid Build Coastguard Worker             vacc4567n1 = vfmaq_laneq_f32(vacc4567n1, vi4567, vw, 1);
193*4bdc9457SAndroid Build Coastguard Worker             vacc0123n2 = vfmaq_laneq_f32(vacc0123n2, vi0123, vw, 2);
194*4bdc9457SAndroid Build Coastguard Worker             vacc4567n2 = vfmaq_laneq_f32(vacc4567n2, vi4567, vw, 2);
195*4bdc9457SAndroid Build Coastguard Worker             vacc0123n3 = vfmaq_laneq_f32(vacc0123n3, vi0123, vw, 3);
196*4bdc9457SAndroid Build Coastguard Worker             vacc4567n3 = vfmaq_laneq_f32(vacc4567n3, vi4567, vw, 3);
197*4bdc9457SAndroid Build Coastguard Worker           } while (--nnz != 0);
198*4bdc9457SAndroid Build Coastguard Worker         }
199*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n0 = vminq_f32(vacc0123n0, vmax);
200*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout4567n0 = vminq_f32(vacc4567n0, vmax);
201*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n1 = vminq_f32(vacc0123n1, vmax);
202*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout4567n1 = vminq_f32(vacc4567n1, vmax);
203*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n2 = vminq_f32(vacc0123n2, vmax);
204*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout4567n2 = vminq_f32(vacc4567n2, vmax);
205*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n3 = vminq_f32(vacc0123n3, vmax);
206*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout4567n3 = vminq_f32(vacc4567n3, vmax);
207*4bdc9457SAndroid Build Coastguard Worker 
208*4bdc9457SAndroid Build Coastguard Worker         vout0123n0 = vmaxq_f32(vout0123n0, vmin);
209*4bdc9457SAndroid Build Coastguard Worker         vout4567n0 = vmaxq_f32(vout4567n0, vmin);
210*4bdc9457SAndroid Build Coastguard Worker         vout0123n1 = vmaxq_f32(vout0123n1, vmin);
211*4bdc9457SAndroid Build Coastguard Worker         vout4567n1 = vmaxq_f32(vout4567n1, vmin);
212*4bdc9457SAndroid Build Coastguard Worker         vout0123n2 = vmaxq_f32(vout0123n2, vmin);
213*4bdc9457SAndroid Build Coastguard Worker         vout4567n2 = vmaxq_f32(vout4567n2, vmin);
214*4bdc9457SAndroid Build Coastguard Worker         vout0123n3 = vmaxq_f32(vout0123n3, vmin);
215*4bdc9457SAndroid Build Coastguard Worker         vout4567n3 = vmaxq_f32(vout4567n3, vmin);
216*4bdc9457SAndroid Build Coastguard Worker 
217*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n0);
218*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 4, vout4567n0);
219*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
220*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n1);
221*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 4, vout4567n1);
222*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
223*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n2);
224*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 4, vout4567n2);
225*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
226*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n3);
227*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 4, vout4567n3);
228*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
229*4bdc9457SAndroid Build Coastguard Worker         n -= 4;
230*4bdc9457SAndroid Build Coastguard Worker       }
231*4bdc9457SAndroid Build Coastguard Worker 
232*4bdc9457SAndroid Build Coastguard Worker       // clean up loop, fall back to nr=1
233*4bdc9457SAndroid Build Coastguard Worker       if XNN_UNLIKELY(n != 0) {
234*4bdc9457SAndroid Build Coastguard Worker         do {
235*4bdc9457SAndroid Build Coastguard Worker           uint32_t nnz = *nnzmap++;
236*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vacc0123 = vld1q_dup_f32(w); w += 1;
237*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vacc4567 = vacc0123;
238*4bdc9457SAndroid Build Coastguard Worker           if XNN_LIKELY(nnz != 0) {
239*4bdc9457SAndroid Build Coastguard Worker             do {
240*4bdc9457SAndroid Build Coastguard Worker               const intptr_t diff = *dmap++;
241*4bdc9457SAndroid Build Coastguard Worker               const float32x4_t vi0123 = vld1q_f32(input);
242*4bdc9457SAndroid Build Coastguard Worker               const float32x4_t vi4567 = vld1q_f32(input + 4);
243*4bdc9457SAndroid Build Coastguard Worker               input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
244*4bdc9457SAndroid Build Coastguard Worker               const float32x4_t vw = vld1q_dup_f32(w); w += 1;
245*4bdc9457SAndroid Build Coastguard Worker               vacc0123 = vfmaq_f32(vacc0123, vi0123, vw);
246*4bdc9457SAndroid Build Coastguard Worker               vacc4567 = vfmaq_f32(vacc4567, vi4567, vw);
247*4bdc9457SAndroid Build Coastguard Worker             } while (--nnz != 0);
248*4bdc9457SAndroid Build Coastguard Worker           }
249*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vout0123 = vminq_f32(vacc0123, vmax);
250*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vout4567 = vminq_f32(vacc4567, vmax);
251*4bdc9457SAndroid Build Coastguard Worker 
252*4bdc9457SAndroid Build Coastguard Worker           vout0123 = vmaxq_f32(vout0123, vmin);
253*4bdc9457SAndroid Build Coastguard Worker           vout4567 = vmaxq_f32(vout4567, vmin);
254*4bdc9457SAndroid Build Coastguard Worker 
255*4bdc9457SAndroid Build Coastguard Worker           vst1q_f32(output + 0, vout0123);
256*4bdc9457SAndroid Build Coastguard Worker           vst1q_f32(output + 4, vout4567);
257*4bdc9457SAndroid Build Coastguard Worker           output = (float*restrict) ((uintptr_t) output + output_stride);
258*4bdc9457SAndroid Build Coastguard Worker           n -= 1;
259*4bdc9457SAndroid Build Coastguard Worker         } while (n != 0);
260*4bdc9457SAndroid Build Coastguard Worker       }
261*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output - output_decrement);
262*4bdc9457SAndroid Build Coastguard Worker       input += 8;
263*4bdc9457SAndroid Build Coastguard Worker     }
264*4bdc9457SAndroid Build Coastguard Worker     output_decrement += 4 * sizeof(float);
265*4bdc9457SAndroid Build Coastguard Worker     if (mc & (4 * sizeof(float))) {
266*4bdc9457SAndroid Build Coastguard Worker       const float*restrict w = weights;
267*4bdc9457SAndroid Build Coastguard Worker       const int32_t* dmap = widx_dmap;
268*4bdc9457SAndroid Build Coastguard Worker       const uint32_t* nnzmap = nidx_nnzmap;
269*4bdc9457SAndroid Build Coastguard Worker       size_t n = nc;
270*4bdc9457SAndroid Build Coastguard Worker       while (n >= 4) {
271*4bdc9457SAndroid Build Coastguard Worker         uint32_t nnz = *nnzmap++;
272*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n0 = vld1q_dup_f32(w); w += 1;
273*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n1 = vld1q_dup_f32(w); w += 1;
274*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n2 = vld1q_dup_f32(w); w += 1;
275*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vacc0123n3 = vld1q_dup_f32(w); w += 1;
276*4bdc9457SAndroid Build Coastguard Worker         if XNN_LIKELY(nnz != 0) {
277*4bdc9457SAndroid Build Coastguard Worker           do {
278*4bdc9457SAndroid Build Coastguard Worker             const intptr_t diff = *dmap++;
279*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vi0123 = vld1q_f32(input);
280*4bdc9457SAndroid Build Coastguard Worker             input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
281*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vw = vld1q_f32(w); w += 4;
282*4bdc9457SAndroid Build Coastguard Worker 
283*4bdc9457SAndroid Build Coastguard Worker             vacc0123n0 = vfmaq_laneq_f32(vacc0123n0, vi0123, vw, 0);
284*4bdc9457SAndroid Build Coastguard Worker             vacc0123n1 = vfmaq_laneq_f32(vacc0123n1, vi0123, vw, 1);
285*4bdc9457SAndroid Build Coastguard Worker             vacc0123n2 = vfmaq_laneq_f32(vacc0123n2, vi0123, vw, 2);
286*4bdc9457SAndroid Build Coastguard Worker             vacc0123n3 = vfmaq_laneq_f32(vacc0123n3, vi0123, vw, 3);
287*4bdc9457SAndroid Build Coastguard Worker           } while (--nnz != 0);
288*4bdc9457SAndroid Build Coastguard Worker         }
289*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n0 = vminq_f32(vacc0123n0, vmax);
290*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n1 = vminq_f32(vacc0123n1, vmax);
291*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n2 = vminq_f32(vacc0123n2, vmax);
292*4bdc9457SAndroid Build Coastguard Worker         float32x4_t vout0123n3 = vminq_f32(vacc0123n3, vmax);
293*4bdc9457SAndroid Build Coastguard Worker 
294*4bdc9457SAndroid Build Coastguard Worker         vout0123n0 = vmaxq_f32(vout0123n0, vmin);
295*4bdc9457SAndroid Build Coastguard Worker         vout0123n1 = vmaxq_f32(vout0123n1, vmin);
296*4bdc9457SAndroid Build Coastguard Worker         vout0123n2 = vmaxq_f32(vout0123n2, vmin);
297*4bdc9457SAndroid Build Coastguard Worker         vout0123n3 = vmaxq_f32(vout0123n3, vmin);
298*4bdc9457SAndroid Build Coastguard Worker 
299*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n0);
300*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
301*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n1);
302*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
303*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n2);
304*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
305*4bdc9457SAndroid Build Coastguard Worker         vst1q_f32(output + 0, vout0123n3);
306*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
307*4bdc9457SAndroid Build Coastguard Worker         n -= 4;
308*4bdc9457SAndroid Build Coastguard Worker       }
309*4bdc9457SAndroid Build Coastguard Worker 
310*4bdc9457SAndroid Build Coastguard Worker       // clean up loop, fall back to nr=1
311*4bdc9457SAndroid Build Coastguard Worker       if XNN_UNLIKELY(n != 0) {
312*4bdc9457SAndroid Build Coastguard Worker         do {
313*4bdc9457SAndroid Build Coastguard Worker           uint32_t nnz = *nnzmap++;
314*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vacc0123 = vld1q_dup_f32(w); w += 1;
315*4bdc9457SAndroid Build Coastguard Worker           if XNN_LIKELY(nnz != 0) {
316*4bdc9457SAndroid Build Coastguard Worker             do {
317*4bdc9457SAndroid Build Coastguard Worker               const intptr_t diff = *dmap++;
318*4bdc9457SAndroid Build Coastguard Worker               const float32x4_t vi0123 = vld1q_f32(input);
319*4bdc9457SAndroid Build Coastguard Worker               input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
320*4bdc9457SAndroid Build Coastguard Worker               const float32x4_t vw = vld1q_dup_f32(w); w += 1;
321*4bdc9457SAndroid Build Coastguard Worker               vacc0123 = vfmaq_f32(vacc0123, vi0123, vw);
322*4bdc9457SAndroid Build Coastguard Worker             } while (--nnz != 0);
323*4bdc9457SAndroid Build Coastguard Worker           }
324*4bdc9457SAndroid Build Coastguard Worker           float32x4_t vout0123 = vminq_f32(vacc0123, vmax);
325*4bdc9457SAndroid Build Coastguard Worker 
326*4bdc9457SAndroid Build Coastguard Worker           vout0123 = vmaxq_f32(vout0123, vmin);
327*4bdc9457SAndroid Build Coastguard Worker 
328*4bdc9457SAndroid Build Coastguard Worker           vst1q_f32(output + 0, vout0123);
329*4bdc9457SAndroid Build Coastguard Worker           output = (float*restrict) ((uintptr_t) output + output_stride);
330*4bdc9457SAndroid Build Coastguard Worker           n -= 1;
331*4bdc9457SAndroid Build Coastguard Worker         } while (n != 0);
332*4bdc9457SAndroid Build Coastguard Worker       }
333*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output - output_decrement);
334*4bdc9457SAndroid Build Coastguard Worker       input += 4;
335*4bdc9457SAndroid Build Coastguard Worker     }
336*4bdc9457SAndroid Build Coastguard Worker     output_decrement += 2 * sizeof(float);
337*4bdc9457SAndroid Build Coastguard Worker     if (mc & (2 * sizeof(float))) {
338*4bdc9457SAndroid Build Coastguard Worker       const float*restrict w = weights;
339*4bdc9457SAndroid Build Coastguard Worker       const int32_t* dmap = widx_dmap;
340*4bdc9457SAndroid Build Coastguard Worker       const uint32_t* nnzmap = nidx_nnzmap;
341*4bdc9457SAndroid Build Coastguard Worker       size_t n = nc;
342*4bdc9457SAndroid Build Coastguard Worker       while (n >= 4) {
343*4bdc9457SAndroid Build Coastguard Worker         uint32_t nnz = *nnzmap++;
344*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc01n0 = vld1_dup_f32(w); w += 1;
345*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc01n1 = vld1_dup_f32(w); w += 1;
346*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc01n2 = vld1_dup_f32(w); w += 1;
347*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc01n3 = vld1_dup_f32(w); w += 1;
348*4bdc9457SAndroid Build Coastguard Worker         if XNN_LIKELY(nnz != 0) {
349*4bdc9457SAndroid Build Coastguard Worker           do {
350*4bdc9457SAndroid Build Coastguard Worker             const intptr_t diff = *dmap++;
351*4bdc9457SAndroid Build Coastguard Worker             const float32x2_t vi01 = vld1_f32(input);
352*4bdc9457SAndroid Build Coastguard Worker             input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
353*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vw = vld1q_f32(w); w += 4;
354*4bdc9457SAndroid Build Coastguard Worker 
355*4bdc9457SAndroid Build Coastguard Worker             vacc01n0 = vfma_laneq_f32(vacc01n0, vi01, vw, 0);
356*4bdc9457SAndroid Build Coastguard Worker             vacc01n1 = vfma_laneq_f32(vacc01n1, vi01, vw, 1);
357*4bdc9457SAndroid Build Coastguard Worker             vacc01n2 = vfma_laneq_f32(vacc01n2, vi01, vw, 2);
358*4bdc9457SAndroid Build Coastguard Worker             vacc01n3 = vfma_laneq_f32(vacc01n3, vi01, vw, 3);
359*4bdc9457SAndroid Build Coastguard Worker           } while (--nnz != 0);
360*4bdc9457SAndroid Build Coastguard Worker         }
361*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout01n0 = vmin_f32(vacc01n0, vget_low_f32(vmax));
362*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout01n1 = vmin_f32(vacc01n1, vget_low_f32(vmax));
363*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout01n2 = vmin_f32(vacc01n2, vget_low_f32(vmax));
364*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout01n3 = vmin_f32(vacc01n3, vget_low_f32(vmax));
365*4bdc9457SAndroid Build Coastguard Worker 
366*4bdc9457SAndroid Build Coastguard Worker         vout01n0 = vmax_f32(vout01n0, vget_low_f32(vmin));
367*4bdc9457SAndroid Build Coastguard Worker         vout01n1 = vmax_f32(vout01n1, vget_low_f32(vmin));
368*4bdc9457SAndroid Build Coastguard Worker         vout01n2 = vmax_f32(vout01n2, vget_low_f32(vmin));
369*4bdc9457SAndroid Build Coastguard Worker         vout01n3 = vmax_f32(vout01n3, vget_low_f32(vmin));
370*4bdc9457SAndroid Build Coastguard Worker 
371*4bdc9457SAndroid Build Coastguard Worker         vst1_f32(output + 0, vout01n0);
372*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
373*4bdc9457SAndroid Build Coastguard Worker         vst1_f32(output + 0, vout01n1);
374*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
375*4bdc9457SAndroid Build Coastguard Worker         vst1_f32(output + 0, vout01n2);
376*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
377*4bdc9457SAndroid Build Coastguard Worker         vst1_f32(output + 0, vout01n3);
378*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
379*4bdc9457SAndroid Build Coastguard Worker         n -= 4;
380*4bdc9457SAndroid Build Coastguard Worker       }
381*4bdc9457SAndroid Build Coastguard Worker 
382*4bdc9457SAndroid Build Coastguard Worker       // clean up loop, fall back to nr=1
383*4bdc9457SAndroid Build Coastguard Worker       if XNN_UNLIKELY(n != 0) {
384*4bdc9457SAndroid Build Coastguard Worker         do {
385*4bdc9457SAndroid Build Coastguard Worker           uint32_t nnz = *nnzmap++;
386*4bdc9457SAndroid Build Coastguard Worker           float32x2_t vacc01 = vld1_dup_f32(w); w += 1;
387*4bdc9457SAndroid Build Coastguard Worker           if XNN_LIKELY(nnz != 0) {
388*4bdc9457SAndroid Build Coastguard Worker             do {
389*4bdc9457SAndroid Build Coastguard Worker               const intptr_t diff = *dmap++;
390*4bdc9457SAndroid Build Coastguard Worker               const float32x2_t vi01 = vld1_f32(input);
391*4bdc9457SAndroid Build Coastguard Worker               input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
392*4bdc9457SAndroid Build Coastguard Worker               const float32x2_t vw = vld1_dup_f32(w); w += 1;
393*4bdc9457SAndroid Build Coastguard Worker               vacc01 = vfma_f32(vacc01, vi01, vw);
394*4bdc9457SAndroid Build Coastguard Worker             } while (--nnz != 0);
395*4bdc9457SAndroid Build Coastguard Worker           }
396*4bdc9457SAndroid Build Coastguard Worker           float32x2_t vout01 = vmin_f32(vacc01, vget_low_f32(vmax));
397*4bdc9457SAndroid Build Coastguard Worker           vout01 = vmax_f32(vout01, vget_low_f32(vmin));
398*4bdc9457SAndroid Build Coastguard Worker 
399*4bdc9457SAndroid Build Coastguard Worker           vst1_f32(output, vout01);
400*4bdc9457SAndroid Build Coastguard Worker           output = (float*restrict) ((uintptr_t) output + output_stride);
401*4bdc9457SAndroid Build Coastguard Worker           n -= 1;
402*4bdc9457SAndroid Build Coastguard Worker         } while (n != 0);
403*4bdc9457SAndroid Build Coastguard Worker       }
404*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output - output_decrement);
405*4bdc9457SAndroid Build Coastguard Worker       input += 2;
406*4bdc9457SAndroid Build Coastguard Worker     }
407*4bdc9457SAndroid Build Coastguard Worker     output_decrement += 1 * sizeof(float);
408*4bdc9457SAndroid Build Coastguard Worker     if (mc & (1 * sizeof(float))) {
409*4bdc9457SAndroid Build Coastguard Worker       const float*restrict w = weights;
410*4bdc9457SAndroid Build Coastguard Worker       const int32_t* dmap = widx_dmap;
411*4bdc9457SAndroid Build Coastguard Worker       const uint32_t* nnzmap = nidx_nnzmap;
412*4bdc9457SAndroid Build Coastguard Worker       size_t n = nc;
413*4bdc9457SAndroid Build Coastguard Worker       while (n >= 4) {
414*4bdc9457SAndroid Build Coastguard Worker         uint32_t nnz = *nnzmap++;
415*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc0n0 = vld1_dup_f32(w); w += 1;
416*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc0n1 = vld1_dup_f32(w); w += 1;
417*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc0n2 = vld1_dup_f32(w); w += 1;
418*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vacc0n3 = vld1_dup_f32(w); w += 1;
419*4bdc9457SAndroid Build Coastguard Worker         if XNN_LIKELY(nnz != 0) {
420*4bdc9457SAndroid Build Coastguard Worker           do {
421*4bdc9457SAndroid Build Coastguard Worker             const intptr_t diff = *dmap++;
422*4bdc9457SAndroid Build Coastguard Worker             const float32x2_t vi0 = vld1_dup_f32(input);
423*4bdc9457SAndroid Build Coastguard Worker             input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
424*4bdc9457SAndroid Build Coastguard Worker             const float32x4_t vw = vld1q_f32(w); w += 4;
425*4bdc9457SAndroid Build Coastguard Worker 
426*4bdc9457SAndroid Build Coastguard Worker             vacc0n0 = vfma_laneq_f32(vacc0n0, vi0, vw, 0);
427*4bdc9457SAndroid Build Coastguard Worker             vacc0n1 = vfma_laneq_f32(vacc0n1, vi0, vw, 1);
428*4bdc9457SAndroid Build Coastguard Worker             vacc0n2 = vfma_laneq_f32(vacc0n2, vi0, vw, 2);
429*4bdc9457SAndroid Build Coastguard Worker             vacc0n3 = vfma_laneq_f32(vacc0n3, vi0, vw, 3);
430*4bdc9457SAndroid Build Coastguard Worker           } while (--nnz != 0);
431*4bdc9457SAndroid Build Coastguard Worker         }
432*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout0n0 = vmin_f32(vacc0n0, vget_low_f32(vmax));
433*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout0n1 = vmin_f32(vacc0n1, vget_low_f32(vmax));
434*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout0n2 = vmin_f32(vacc0n2, vget_low_f32(vmax));
435*4bdc9457SAndroid Build Coastguard Worker         float32x2_t vout0n3 = vmin_f32(vacc0n3, vget_low_f32(vmax));
436*4bdc9457SAndroid Build Coastguard Worker 
437*4bdc9457SAndroid Build Coastguard Worker         vout0n0 = vmax_f32(vout0n0, vget_low_f32(vmin));
438*4bdc9457SAndroid Build Coastguard Worker         vout0n1 = vmax_f32(vout0n1, vget_low_f32(vmin));
439*4bdc9457SAndroid Build Coastguard Worker         vout0n2 = vmax_f32(vout0n2, vget_low_f32(vmin));
440*4bdc9457SAndroid Build Coastguard Worker         vout0n3 = vmax_f32(vout0n3, vget_low_f32(vmin));
441*4bdc9457SAndroid Build Coastguard Worker 
442*4bdc9457SAndroid Build Coastguard Worker         vst1_lane_f32(output + 0, vout0n0, 0);
443*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
444*4bdc9457SAndroid Build Coastguard Worker         vst1_lane_f32(output + 0, vout0n1, 0);
445*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
446*4bdc9457SAndroid Build Coastguard Worker         vst1_lane_f32(output + 0, vout0n2, 0);
447*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
448*4bdc9457SAndroid Build Coastguard Worker         vst1_lane_f32(output + 0, vout0n3, 0);
449*4bdc9457SAndroid Build Coastguard Worker         output = (float*restrict) ((uintptr_t) output + output_stride);
450*4bdc9457SAndroid Build Coastguard Worker         n -= 4;
451*4bdc9457SAndroid Build Coastguard Worker       }
452*4bdc9457SAndroid Build Coastguard Worker 
453*4bdc9457SAndroid Build Coastguard Worker       // clean up loop, fall back to nr=1
454*4bdc9457SAndroid Build Coastguard Worker       if XNN_UNLIKELY(n != 0) {
455*4bdc9457SAndroid Build Coastguard Worker         do {
456*4bdc9457SAndroid Build Coastguard Worker           uint32_t nnz = *nnzmap++;
457*4bdc9457SAndroid Build Coastguard Worker           float32x2_t vacc0 = vld1_dup_f32(w); w += 1;
458*4bdc9457SAndroid Build Coastguard Worker           if XNN_LIKELY(nnz != 0) {
459*4bdc9457SAndroid Build Coastguard Worker             do {
460*4bdc9457SAndroid Build Coastguard Worker               const intptr_t diff = *dmap++;
461*4bdc9457SAndroid Build Coastguard Worker               const float32x2_t vi0 = vld1_dup_f32(input);
462*4bdc9457SAndroid Build Coastguard Worker               input = (const float*restrict) ((uintptr_t) input + (uintptr_t) diff);
463*4bdc9457SAndroid Build Coastguard Worker               const float32x2_t vw = vld1_dup_f32(w); w += 1;
464*4bdc9457SAndroid Build Coastguard Worker               vacc0 = vfma_f32(vacc0, vi0, vw);
465*4bdc9457SAndroid Build Coastguard Worker             } while (--nnz != 0);
466*4bdc9457SAndroid Build Coastguard Worker           }
467*4bdc9457SAndroid Build Coastguard Worker           float32x2_t vout0 = vmin_f32(vacc0, vget_low_f32(vmax));
468*4bdc9457SAndroid Build Coastguard Worker           vout0 = vmax_f32(vout0, vget_low_f32(vmin));
469*4bdc9457SAndroid Build Coastguard Worker 
470*4bdc9457SAndroid Build Coastguard Worker           vst1_lane_f32(output, vout0, 1);
471*4bdc9457SAndroid Build Coastguard Worker           output = (float*restrict) ((uintptr_t) output + output_stride);
472*4bdc9457SAndroid Build Coastguard Worker           n -= 1;
473*4bdc9457SAndroid Build Coastguard Worker         } while (n != 0);
474*4bdc9457SAndroid Build Coastguard Worker       }
475*4bdc9457SAndroid Build Coastguard Worker       output = (float*restrict) ((uintptr_t) output - output_decrement);
476*4bdc9457SAndroid Build Coastguard Worker       input += 1;
477*4bdc9457SAndroid Build Coastguard Worker     }
478*4bdc9457SAndroid Build Coastguard Worker     }
479*4bdc9457SAndroid Build Coastguard Worker }
480