xref: /aosp_15_r20/external/XNNPACK/src/s16-rmaxabs/gen/scalar-x2.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1 // Auto-generated file. Do not edit!
2 //   Template: src/s16-rmaxabs/scalar.c.in
3 //   Generator: tools/xngen
4 //
5 // Copyright 2022 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 #include <stddef.h>
12 #include <stdint.h>
13 
14 #include <xnnpack/math.h>
15 #include <xnnpack/rmaxabs.h>
16 
17 
xnn_s16_rmaxabs_ukernel__scalar_x2(size_t batch,const int16_t * input,uint16_t * output)18 void xnn_s16_rmaxabs_ukernel__scalar_x2(
19     size_t batch,
20     const int16_t* input,
21     uint16_t* output) {
22 
23   assert(batch > 0);
24   assert(input != NULL);
25   assert(output != NULL);
26 
27   uint32_t vmax0 = 0;
28   uint32_t vmax1 = 0;
29 
30   for (; batch >= 2; batch -= 2) {
31     const int32_t vi0 = (int32_t) input[0];
32     const int32_t vi1 = (int32_t) input[1];
33     input += 2;
34 
35     const uint32_t vabs0 = math_abs_s32(vi0);
36     const uint32_t vabs1 = math_abs_s32(vi1);
37 
38     vmax0 = math_max_u32(vmax0, vabs0);
39     vmax1 = math_max_u32(vmax1, vabs1);
40   }
41 
42   vmax0 = math_max_u32(vmax0, vmax1);
43 
44   if (batch != 0) {
45     do {
46       const int32_t vi = (int32_t) *input++;
47       const uint32_t vabs = math_abs_s32(vi);
48       vmax0 = math_max_u32(vmax0, vabs);
49     } while (--batch != 0);
50   }
51   *output = (uint16_t) vmax0;
52 }
53