xref: /aosp_15_r20/external/XNNPACK/src/f32-vbinary/gen/vmin-neon-x8.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker // Auto-generated file. Do not edit!
2*4bdc9457SAndroid Build Coastguard Worker //   Template: src/f32-vbinary/vop-neon.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/common.h>
15*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/vbinary.h>
16*4bdc9457SAndroid Build Coastguard Worker 
17*4bdc9457SAndroid Build Coastguard Worker 
xnn_f32_vmin_ukernel__neon_x8(size_t n,const float * a,const float * b,float * y,const union xnn_f32_default_params params[restrict XNN_MIN_ELEMENTS (1)])18*4bdc9457SAndroid Build Coastguard Worker void xnn_f32_vmin_ukernel__neon_x8(
19*4bdc9457SAndroid Build Coastguard Worker     size_t n,
20*4bdc9457SAndroid Build Coastguard Worker     const float* a,
21*4bdc9457SAndroid Build Coastguard Worker     const float* b,
22*4bdc9457SAndroid Build Coastguard Worker     float* y,
23*4bdc9457SAndroid Build Coastguard Worker     const union xnn_f32_default_params params[restrict XNN_MIN_ELEMENTS(1)]) XNN_OOB_READS
24*4bdc9457SAndroid Build Coastguard Worker {
25*4bdc9457SAndroid Build Coastguard Worker   assert(n != 0);
26*4bdc9457SAndroid Build Coastguard Worker   assert(n % sizeof(float) == 0);
27*4bdc9457SAndroid Build Coastguard Worker   assert(a != NULL);
28*4bdc9457SAndroid Build Coastguard Worker   assert(b != NULL);
29*4bdc9457SAndroid Build Coastguard Worker   assert(y != NULL);
30*4bdc9457SAndroid Build Coastguard Worker 
31*4bdc9457SAndroid Build Coastguard Worker 
32*4bdc9457SAndroid Build Coastguard Worker   for (; n >= 8 * sizeof(float); n -= 8 * sizeof(float)) {
33*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t va0123 = vld1q_f32(a); a += 4;
34*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t vb0123 = vld1q_f32(b); b += 4;
35*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t va4567 = vld1q_f32(a); a += 4;
36*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t vb4567 = vld1q_f32(b); b += 4;
37*4bdc9457SAndroid Build Coastguard Worker 
38*4bdc9457SAndroid Build Coastguard Worker     float32x4_t vy0123 = vminq_f32(va0123, vb0123);
39*4bdc9457SAndroid Build Coastguard Worker     float32x4_t vy4567 = vminq_f32(va4567, vb4567);
40*4bdc9457SAndroid Build Coastguard Worker 
41*4bdc9457SAndroid Build Coastguard Worker 
42*4bdc9457SAndroid Build Coastguard Worker 
43*4bdc9457SAndroid Build Coastguard Worker     vst1q_f32(y, vy0123); y += 4;
44*4bdc9457SAndroid Build Coastguard Worker     vst1q_f32(y, vy4567); y += 4;
45*4bdc9457SAndroid Build Coastguard Worker   }
46*4bdc9457SAndroid Build Coastguard Worker   for (; n >= 4 * sizeof(float); n -= 4 * sizeof(float)) {
47*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t va0123 = vld1q_f32(a); a += 4;
48*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t vb0123 = vld1q_f32(b); b += 4;
49*4bdc9457SAndroid Build Coastguard Worker 
50*4bdc9457SAndroid Build Coastguard Worker     float32x4_t vy0123 = vminq_f32(va0123, vb0123);
51*4bdc9457SAndroid Build Coastguard Worker     vst1q_f32(y, vy0123); y += 4;
52*4bdc9457SAndroid Build Coastguard Worker   }
53*4bdc9457SAndroid Build Coastguard Worker   if XNN_UNLIKELY(n != 0) {
54*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t va0123 = vld1q_f32(a);
55*4bdc9457SAndroid Build Coastguard Worker     const float32x4_t vb0123 = vld1q_f32(b);
56*4bdc9457SAndroid Build Coastguard Worker 
57*4bdc9457SAndroid Build Coastguard Worker     float32x4_t vy0123 = vminq_f32(va0123, vb0123);
58*4bdc9457SAndroid Build Coastguard Worker 
59*4bdc9457SAndroid Build Coastguard Worker     float32x2_t vy01 = vget_low_f32(vy0123);
60*4bdc9457SAndroid Build Coastguard Worker     if (n & (2 * sizeof(float))) {
61*4bdc9457SAndroid Build Coastguard Worker       vst1_f32(y, vy01); y += 2;
62*4bdc9457SAndroid Build Coastguard Worker       vy01 = vget_high_f32(vy0123);
63*4bdc9457SAndroid Build Coastguard Worker     }
64*4bdc9457SAndroid Build Coastguard Worker     if (n & (1 * sizeof(float))) {
65*4bdc9457SAndroid Build Coastguard Worker       vst1_lane_f32(y, vy01, 0);
66*4bdc9457SAndroid Build Coastguard Worker     }
67*4bdc9457SAndroid Build Coastguard Worker   }
68*4bdc9457SAndroid Build Coastguard Worker }
69