• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Auto-generated file. Do not edit!
2 //   Template: src/qs8-vmulc/scalar.c.in
3 //   Generator: tools/xngen
4 //
5 // Copyright 2021 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 <fp16.h>
13 
14 #include <xnnpack/math.h>
15 #include <xnnpack/vmul.h>
16 
17 
xnn_qu8_vmulc_minmax_fp32_ukernel__scalar_x2(size_t n,const uint8_t * input_a,const uint8_t * input_b,uint8_t * output,const union xnn_qu8_mul_minmax_params params[restrict XNN_MIN_ELEMENTS (1)])18 void xnn_qu8_vmulc_minmax_fp32_ukernel__scalar_x2(
19     size_t n,
20     const uint8_t* input_a,
21     const uint8_t* input_b,
22     uint8_t* output,
23     const union xnn_qu8_mul_minmax_params params[restrict XNN_MIN_ELEMENTS(1)])
24 {
25   const int32_t va_zero_point = params->fp32_scalar.a_zero_point;
26   const float vscale = params->fp32_scalar.scale;
27   const float voutput_min_less_zero_point = params->fp32_scalar.output_min_less_zero_point;
28   const float voutput_max_less_zero_point = params->fp32_scalar.output_max_less_zero_point;
29   const float vmagic_bias = params->fp32_scalar.magic_bias;
30   const int32_t vmagic_bias_less_output_zero_point = params->fp32_scalar.magic_bias_less_output_zero_point;
31 
32   const int32_t vb = (int32_t) *input_b - params->fp32_scalar.b_zero_point;
33   for (; n >= 2 * sizeof(uint8_t); n -= 2 * sizeof(uint8_t)) {
34     const int32_t va0 = input_a[0] - va_zero_point;
35     const int32_t va1 = input_a[1] - va_zero_point;
36     input_a += 2;
37 
38     const int32_t vacc0 = va0 * vb;
39     const int32_t vacc1 = va1 * vb;
40 
41     float vfpacc0 = (float) vacc0 * vscale;
42     float vfpacc1 = (float) vacc1 * vscale;
43 
44     vfpacc0 = math_max_f32(vfpacc0, voutput_min_less_zero_point);
45     vfpacc1 = math_max_f32(vfpacc1, voutput_min_less_zero_point);
46 
47     vfpacc0 = math_min_f32(vfpacc0, voutput_max_less_zero_point);
48     vfpacc1 = math_min_f32(vfpacc1, voutput_max_less_zero_point);
49 
50     vfpacc0 += vmagic_bias;
51     vfpacc1 += vmagic_bias;
52 
53     const int32_t vout0 = (int32_t) fp32_to_bits(vfpacc0) - vmagic_bias_less_output_zero_point;
54     const int32_t vout1 = (int32_t) fp32_to_bits(vfpacc1) - vmagic_bias_less_output_zero_point;
55 
56     output[0] = (uint8_t) vout0;
57     output[1] = (uint8_t) vout1;
58     output += 2;
59   }
60   if XNN_UNLIKELY(n != 0) {
61     const int32_t va = (int32_t) *input_a - va_zero_point;
62     const int32_t vacc = va * vb;
63 
64     float vfpacc = (float) vacc * vscale;
65     vfpacc = math_max_f32(vfpacc, voutput_min_less_zero_point);
66     vfpacc = math_min_f32(vfpacc, voutput_max_less_zero_point);
67     vfpacc += vmagic_bias;
68     const int32_t vout = (int32_t) fp32_to_bits(vfpacc) - vmagic_bias_less_output_zero_point;
69     *output = (uint8_t) vout;
70   }
71 }
72