xref: /aosp_15_r20/external/XNNPACK/src/x32-zip/x2-scalar.c (revision 4bdc94577ba0e567308109d787f7fec7b531ce36)
1*4bdc9457SAndroid Build Coastguard Worker // Copyright 2019 Google LLC
2*4bdc9457SAndroid Build Coastguard Worker //
3*4bdc9457SAndroid Build Coastguard Worker // This source code is licensed under the BSD-style license found in the
4*4bdc9457SAndroid Build Coastguard Worker // LICENSE file in the root directory of this source tree.
5*4bdc9457SAndroid Build Coastguard Worker 
6*4bdc9457SAndroid Build Coastguard Worker #include <assert.h>
7*4bdc9457SAndroid Build Coastguard Worker 
8*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/zip.h>
9*4bdc9457SAndroid Build Coastguard Worker 
10*4bdc9457SAndroid Build Coastguard Worker 
xnn_x32_zip_x2_ukernel__scalar(size_t n,const uint32_t * input,uint32_t * output)11*4bdc9457SAndroid Build Coastguard Worker void xnn_x32_zip_x2_ukernel__scalar(
12*4bdc9457SAndroid Build Coastguard Worker     size_t n,
13*4bdc9457SAndroid Build Coastguard Worker     const uint32_t* input,
14*4bdc9457SAndroid Build Coastguard Worker     uint32_t* output)
15*4bdc9457SAndroid Build Coastguard Worker {
16*4bdc9457SAndroid Build Coastguard Worker   assert(n != 0);
17*4bdc9457SAndroid Build Coastguard Worker   assert(n % 4 == 0);
18*4bdc9457SAndroid Build Coastguard Worker 
19*4bdc9457SAndroid Build Coastguard Worker   const uint32_t* x = input;
20*4bdc9457SAndroid Build Coastguard Worker   const uint32_t* y = (const uint32_t*) ((uintptr_t) x + n);
21*4bdc9457SAndroid Build Coastguard Worker 
22*4bdc9457SAndroid Build Coastguard Worker   do {
23*4bdc9457SAndroid Build Coastguard Worker     const uint32_t vx = *x++;
24*4bdc9457SAndroid Build Coastguard Worker     const uint32_t vy = *y++;
25*4bdc9457SAndroid Build Coastguard Worker     output[0] = vx;
26*4bdc9457SAndroid Build Coastguard Worker     output[1] = vy;
27*4bdc9457SAndroid Build Coastguard Worker     output += 2;
28*4bdc9457SAndroid Build Coastguard Worker 
29*4bdc9457SAndroid Build Coastguard Worker     n -= 4;
30*4bdc9457SAndroid Build Coastguard Worker   } while (n != 0);
31*4bdc9457SAndroid Build Coastguard Worker }
32