xref: /aosp_15_r20/external/XNNPACK/src/qu8-vcvt/gen/vcvt-scalar-x4.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker //   Template: src/qs8-vcvt/scalar.c.in
3*4bdc9457SAndroid Build Coastguard Worker //   Generator: tools/xngen
4*4bdc9457SAndroid Build Coastguard Worker //
5*4bdc9457SAndroid Build Coastguard Worker // Copyright 2022 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 <xnnpack/math.h>
13*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/vcvt.h>
14*4bdc9457SAndroid Build Coastguard Worker 
15*4bdc9457SAndroid Build Coastguard Worker 
xnn_qu8_vcvt_ukernel__scalar_x4(size_t n,const uint8_t * x,uint8_t * y,const union xnn_qu8_cvt_params params[restrict XNN_MIN_ELEMENTS (1)])16*4bdc9457SAndroid Build Coastguard Worker void xnn_qu8_vcvt_ukernel__scalar_x4(
17*4bdc9457SAndroid Build Coastguard Worker     size_t n,
18*4bdc9457SAndroid Build Coastguard Worker     const uint8_t* x,
19*4bdc9457SAndroid Build Coastguard Worker     uint8_t* y,
20*4bdc9457SAndroid Build Coastguard Worker     const union xnn_qu8_cvt_params params[restrict XNN_MIN_ELEMENTS(1)])
21*4bdc9457SAndroid Build Coastguard Worker {
22*4bdc9457SAndroid Build Coastguard Worker   const int32_t vbias = params->scalar.bias;
23*4bdc9457SAndroid Build Coastguard Worker   const int32_t vmultiplier = params->scalar.multiplier;
24*4bdc9457SAndroid Build Coastguard Worker   for (; n >= 4 * sizeof(uint8_t); n -= 4 * sizeof(uint8_t)) {
25*4bdc9457SAndroid Build Coastguard Worker     int32_t vacc0 = x[0];
26*4bdc9457SAndroid Build Coastguard Worker     int32_t vacc1 = x[1];
27*4bdc9457SAndroid Build Coastguard Worker     int32_t vacc2 = x[2];
28*4bdc9457SAndroid Build Coastguard Worker     int32_t vacc3 = x[3];
29*4bdc9457SAndroid Build Coastguard Worker     x += 4;
30*4bdc9457SAndroid Build Coastguard Worker 
31*4bdc9457SAndroid Build Coastguard Worker     vacc0 = vbias + vacc0 * vmultiplier;
32*4bdc9457SAndroid Build Coastguard Worker     vacc1 = vbias + vacc1 * vmultiplier;
33*4bdc9457SAndroid Build Coastguard Worker     vacc2 = vbias + vacc2 * vmultiplier;
34*4bdc9457SAndroid Build Coastguard Worker     vacc3 = vbias + vacc3 * vmultiplier;
35*4bdc9457SAndroid Build Coastguard Worker 
36*4bdc9457SAndroid Build Coastguard Worker     int32_t vout0 = math_asr_s32(vacc0, 8);
37*4bdc9457SAndroid Build Coastguard Worker     int32_t vout1 = math_asr_s32(vacc1, 8);
38*4bdc9457SAndroid Build Coastguard Worker     int32_t vout2 = math_asr_s32(vacc2, 8);
39*4bdc9457SAndroid Build Coastguard Worker     int32_t vout3 = math_asr_s32(vacc3, 8);
40*4bdc9457SAndroid Build Coastguard Worker 
41*4bdc9457SAndroid Build Coastguard Worker     vout0 = math_max_s32(vout0, 0);
42*4bdc9457SAndroid Build Coastguard Worker     vout1 = math_max_s32(vout1, 0);
43*4bdc9457SAndroid Build Coastguard Worker     vout2 = math_max_s32(vout2, 0);
44*4bdc9457SAndroid Build Coastguard Worker     vout3 = math_max_s32(vout3, 0);
45*4bdc9457SAndroid Build Coastguard Worker 
46*4bdc9457SAndroid Build Coastguard Worker     vout0 = math_min_s32(vout0, 255);
47*4bdc9457SAndroid Build Coastguard Worker     vout1 = math_min_s32(vout1, 255);
48*4bdc9457SAndroid Build Coastguard Worker     vout2 = math_min_s32(vout2, 255);
49*4bdc9457SAndroid Build Coastguard Worker     vout3 = math_min_s32(vout3, 255);
50*4bdc9457SAndroid Build Coastguard Worker 
51*4bdc9457SAndroid Build Coastguard Worker     y[0] = (uint8_t) vout0;
52*4bdc9457SAndroid Build Coastguard Worker     y[1] = (uint8_t) vout1;
53*4bdc9457SAndroid Build Coastguard Worker     y[2] = (uint8_t) vout2;
54*4bdc9457SAndroid Build Coastguard Worker     y[3] = (uint8_t) vout3;
55*4bdc9457SAndroid Build Coastguard Worker     y += 4;
56*4bdc9457SAndroid Build Coastguard Worker   }
57*4bdc9457SAndroid Build Coastguard Worker   if XNN_UNLIKELY(n != 0) {
58*4bdc9457SAndroid Build Coastguard Worker     do {
59*4bdc9457SAndroid Build Coastguard Worker       int32_t vacc = *x++;
60*4bdc9457SAndroid Build Coastguard Worker       vacc = vbias + vacc * vmultiplier;
61*4bdc9457SAndroid Build Coastguard Worker 
62*4bdc9457SAndroid Build Coastguard Worker       int32_t vout = math_asr_s32(vacc, 8);
63*4bdc9457SAndroid Build Coastguard Worker       vout = math_max_s32(vout, 0);
64*4bdc9457SAndroid Build Coastguard Worker       vout = math_min_s32(vout, 255);
65*4bdc9457SAndroid Build Coastguard Worker       *y++ = (uint8_t) vout;
66*4bdc9457SAndroid Build Coastguard Worker 
67*4bdc9457SAndroid Build Coastguard Worker       n -= sizeof(uint8_t);
68*4bdc9457SAndroid Build Coastguard Worker     } while (n != 0);
69*4bdc9457SAndroid Build Coastguard Worker   }
70*4bdc9457SAndroid Build Coastguard Worker }
71