1 // Auto-generated file. Do not edit!
2 // Template: src/cs16-bfly4/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/fft.h>
16
17
xnn_cs16_bfly4_ukernel__scalar_x2(size_t samples,int16_t * data,const size_t stride,const int16_t * twiddle)18 void xnn_cs16_bfly4_ukernel__scalar_x2(
19 size_t samples,
20 int16_t* data,
21 const size_t stride,
22 const int16_t* twiddle)
23 {
24 const int16_t* tw1 = twiddle;
25 const int16_t* tw2 = twiddle;
26 const int16_t* tw3 = twiddle;
27 int16_t* data0 = data;
28 int16_t* data1 = data + samples * 2;
29 int16_t* data2 = data + samples * 4;
30 int16_t* data3 = data + samples * 6;
31
32 assert(samples != 0);
33 assert(data != NULL);
34 assert(stride != 0);
35 assert(twiddle != NULL);
36
37 for (; samples >= 2; samples -= 2) {
38 int32_t vout0r0 = (int32_t) data0[0];
39 int32_t vout0i0 = (int32_t) data0[1];
40 int32_t vout0r1 = (int32_t) data0[2];
41 int32_t vout0i1 = (int32_t) data0[3];
42 int32_t vout1r0 = (int32_t) data1[0];
43 int32_t vout1i0 = (int32_t) data1[1];
44 int32_t vout1r1 = (int32_t) data1[2];
45 int32_t vout1i1 = (int32_t) data1[3];
46 int32_t vout2r0 = (int32_t) data2[0];
47 int32_t vout2i0 = (int32_t) data2[1];
48 int32_t vout2r1 = (int32_t) data2[2];
49 int32_t vout2i1 = (int32_t) data2[3];
50 int32_t vout3r0 = (int32_t) data3[0];
51 int32_t vout3i0 = (int32_t) data3[1];
52 int32_t vout3r1 = (int32_t) data3[2];
53 int32_t vout3i1 = (int32_t) data3[3];
54
55 const int32_t vtw1r0 = (const int32_t) tw1[0];
56 const int32_t vtw1i0 = (const int32_t) tw1[1];
57 tw1 += stride * 2;
58 const int32_t vtw1r1 = (const int32_t) tw1[0];
59 const int32_t vtw1i1 = (const int32_t) tw1[1];
60 tw1 += stride * 2;
61 const int32_t vtw2r0 = (const int32_t) tw2[0];
62 const int32_t vtw2i0 = (const int32_t) tw2[1];
63 tw2 += stride * 4;
64 const int32_t vtw2r1 = (const int32_t) tw2[0];
65 const int32_t vtw2i1 = (const int32_t) tw2[1];
66 tw2 += stride * 4;
67 const int32_t vtw3r0 = (const int32_t) tw3[0];
68 const int32_t vtw3i0 = (const int32_t) tw3[1];
69 tw3 += stride * 6;
70 const int32_t vtw3r1 = (const int32_t) tw3[0];
71 const int32_t vtw3i1 = (const int32_t) tw3[1];
72 tw3 += stride * 6;
73
74 // Note 32767 / 4 = 8191. Should be 8192.
75 vout0r0 = math_asr_s32(vout0r0 * 8191 + 16384, 15);
76 vout0r1 = math_asr_s32(vout0r1 * 8191 + 16384, 15);
77 vout0i0 = math_asr_s32(vout0i0 * 8191 + 16384, 15);
78 vout0i1 = math_asr_s32(vout0i1 * 8191 + 16384, 15);
79 vout1r0 = math_asr_s32(vout1r0 * 8191 + 16384, 15);
80 vout1r1 = math_asr_s32(vout1r1 * 8191 + 16384, 15);
81 vout1i0 = math_asr_s32(vout1i0 * 8191 + 16384, 15);
82 vout1i1 = math_asr_s32(vout1i1 * 8191 + 16384, 15);
83 vout2r0 = math_asr_s32(vout2r0 * 8191 + 16384, 15);
84 vout2r1 = math_asr_s32(vout2r1 * 8191 + 16384, 15);
85 vout2i0 = math_asr_s32(vout2i0 * 8191 + 16384, 15);
86 vout2i1 = math_asr_s32(vout2i1 * 8191 + 16384, 15);
87 vout3r0 = math_asr_s32(vout3r0 * 8191 + 16384, 15);
88 vout3r1 = math_asr_s32(vout3r1 * 8191 + 16384, 15);
89 vout3i0 = math_asr_s32(vout3i0 * 8191 + 16384, 15);
90 vout3i1 = math_asr_s32(vout3i1 * 8191 + 16384, 15);
91
92 const int32_t vtmp0r0 = math_asr_s32(vout1r0 * vtw1r0 - vout1i0 * vtw1i0 + 16384, 15);
93 const int32_t vtmp0r1 = math_asr_s32(vout1r1 * vtw1r1 - vout1i1 * vtw1i1 + 16384, 15);
94 const int32_t vtmp0i0 = math_asr_s32(vout1r0 * vtw1i0 + vout1i0 * vtw1r0 + 16384, 15);
95 const int32_t vtmp0i1 = math_asr_s32(vout1r1 * vtw1i1 + vout1i1 * vtw1r1 + 16384, 15);
96 const int32_t vtmp1r0 = math_asr_s32(vout2r0 * vtw2r0 - vout2i0 * vtw2i0 + 16384, 15);
97 const int32_t vtmp1r1 = math_asr_s32(vout2r1 * vtw2r1 - vout2i1 * vtw2i1 + 16384, 15);
98 const int32_t vtmp1i0 = math_asr_s32(vout2r0 * vtw2i0 + vout2i0 * vtw2r0 + 16384, 15);
99 const int32_t vtmp1i1 = math_asr_s32(vout2r1 * vtw2i1 + vout2i1 * vtw2r1 + 16384, 15);
100 const int32_t vtmp2r0 = math_asr_s32(vout3r0 * vtw3r0 - vout3i0 * vtw3i0 + 16384, 15);
101 const int32_t vtmp2r1 = math_asr_s32(vout3r1 * vtw3r1 - vout3i1 * vtw3i1 + 16384, 15);
102 const int32_t vtmp2i0 = math_asr_s32(vout3r0 * vtw3i0 + vout3i0 * vtw3r0 + 16384, 15);
103 const int32_t vtmp2i1 = math_asr_s32(vout3r1 * vtw3i1 + vout3i1 * vtw3r1 + 16384, 15);
104
105 const int32_t vtmp5r0 = vout0r0 - vtmp1r0;
106 const int32_t vtmp5r1 = vout0r1 - vtmp1r1;
107 const int32_t vtmp5i0 = vout0i0 - vtmp1i0;
108 const int32_t vtmp5i1 = vout0i1 - vtmp1i1;
109 vout0r0 += vtmp1r0;
110 vout0r1 += vtmp1r1;
111 vout0i0 += vtmp1i0;
112 vout0i1 += vtmp1i1;
113 const int32_t vtmp3r0 = vtmp0r0 + vtmp2r0;
114 const int32_t vtmp3r1 = vtmp0r1 + vtmp2r1;
115 const int32_t vtmp3i0 = vtmp0i0 + vtmp2i0;
116 const int32_t vtmp3i1 = vtmp0i1 + vtmp2i1;
117 const int32_t vtmp4r0 = vtmp0r0 - vtmp2r0;
118 const int32_t vtmp4r1 = vtmp0r1 - vtmp2r1;
119 const int32_t vtmp4i0 = vtmp0i0 - vtmp2i0;
120 const int32_t vtmp4i1 = vtmp0i1 - vtmp2i1;
121 vout2r0 = vout0r0 - vtmp3r0;
122 vout2r1 = vout0r1 - vtmp3r1;
123 vout2i0 = vout0i0 - vtmp3i0;
124 vout2i1 = vout0i1 - vtmp3i1;
125 vout0r0 += vtmp3r0;
126 vout0r1 += vtmp3r1;
127 vout0i0 += vtmp3i0;
128 vout0i1 += vtmp3i1;
129 vout1r0 = vtmp5r0 + vtmp4i0;
130 vout1r1 = vtmp5r1 + vtmp4i1;
131 vout1i0 = vtmp5i0 - vtmp4r0;
132 vout1i1 = vtmp5i1 - vtmp4r1;
133 vout3r0 = vtmp5r0 - vtmp4i0;
134 vout3r1 = vtmp5r1 - vtmp4i1;
135 vout3i0 = vtmp5i0 + vtmp4r0;
136 vout3i1 = vtmp5i1 + vtmp4r1;
137
138 data0[0] = (int16_t) vout0r0;
139 data0[1] = (int16_t) vout0i0;
140 data0[2] = (int16_t) vout0r1;
141 data0[3] = (int16_t) vout0i1;
142 data0 += 2 * 2;
143 data1[0] = (int16_t) vout1r0;
144 data1[1] = (int16_t) vout1i0;
145 data1[2] = (int16_t) vout1r1;
146 data1[3] = (int16_t) vout1i1;
147 data1 += 2 * 2;
148 data2[0] = (int16_t) vout2r0;
149 data2[1] = (int16_t) vout2i0;
150 data2[2] = (int16_t) vout2r1;
151 data2[3] = (int16_t) vout2i1;
152 data2 += 2 * 2;
153 data3[0] = (int16_t) vout3r0;
154 data3[1] = (int16_t) vout3i0;
155 data3[2] = (int16_t) vout3r1;
156 data3[3] = (int16_t) vout3i1;
157 data3 += 2 * 2;
158 }
159
160 if XNN_UNLIKELY(samples != 0) {
161 do {
162 int32_t vout0r = (int32_t) data0[0];
163 int32_t vout0i = (int32_t) data0[1];
164 int32_t vout1r = (int32_t) data1[0];
165 int32_t vout1i = (int32_t) data1[1];
166 int32_t vout2r = (int32_t) data2[0];
167 int32_t vout2i = (int32_t) data2[1];
168 int32_t vout3r = (int32_t) data3[0];
169 int32_t vout3i = (int32_t) data3[1];
170
171 const int32_t vtw1r = (const int32_t) tw1[0];
172 const int32_t vtw1i = (const int32_t) tw1[1];
173 const int32_t vtw2r = (const int32_t) tw2[0];
174 const int32_t vtw2i = (const int32_t) tw2[1];
175 const int32_t vtw3r = (const int32_t) tw3[0];
176 const int32_t vtw3i = (const int32_t) tw3[1];
177 tw1 += stride * 2;
178 tw2 += stride * 4;
179 tw3 += stride * 6;
180
181 // Note 32767 / 4 = 8191. Should be 8192.
182 vout0r = math_asr_s32(vout0r * 8191 + 16384, 15);
183 vout0i = math_asr_s32(vout0i * 8191 + 16384, 15);
184 vout1r = math_asr_s32(vout1r * 8191 + 16384, 15);
185 vout1i = math_asr_s32(vout1i * 8191 + 16384, 15);
186 vout2r = math_asr_s32(vout2r * 8191 + 16384, 15);
187 vout2i = math_asr_s32(vout2i * 8191 + 16384, 15);
188 vout3r = math_asr_s32(vout3r * 8191 + 16384, 15);
189 vout3i = math_asr_s32(vout3i * 8191 + 16384, 15);
190
191 const int32_t vtmp0r = math_asr_s32(vout1r * vtw1r - vout1i * vtw1i + 16384, 15);
192 const int32_t vtmp0i = math_asr_s32(vout1r * vtw1i + vout1i * vtw1r + 16384, 15);
193 const int32_t vtmp1r = math_asr_s32(vout2r * vtw2r - vout2i * vtw2i + 16384, 15);
194 const int32_t vtmp1i = math_asr_s32(vout2r * vtw2i + vout2i * vtw2r + 16384, 15);
195 const int32_t vtmp2r = math_asr_s32(vout3r * vtw3r - vout3i * vtw3i + 16384, 15);
196 const int32_t vtmp2i = math_asr_s32(vout3r * vtw3i + vout3i * vtw3r + 16384, 15);
197
198 const int32_t vtmp5r = vout0r - vtmp1r;
199 const int32_t vtmp5i = vout0i - vtmp1i;
200 vout0r += vtmp1r;
201 vout0i += vtmp1i;
202 const int32_t vtmp3r = vtmp0r + vtmp2r;
203 const int32_t vtmp3i = vtmp0i + vtmp2i;
204 const int32_t vtmp4r = vtmp0r - vtmp2r;
205 const int32_t vtmp4i = vtmp0i - vtmp2i;
206 vout2r = vout0r - vtmp3r;
207 vout2i = vout0i - vtmp3i;
208
209 vout0r += vtmp3r;
210 vout0i += vtmp3i;
211
212 vout1r = vtmp5r + vtmp4i;
213 vout1i = vtmp5i - vtmp4r;
214 vout3r = vtmp5r - vtmp4i;
215 vout3i = vtmp5i + vtmp4r;
216
217 data0[0] = (int16_t) vout0r;
218 data0[1] = (int16_t) vout0i;
219 data1[0] = (int16_t) vout1r;
220 data1[1] = (int16_t) vout1i;
221 data2[0] = (int16_t) vout2r;
222 data2[1] = (int16_t) vout2i;
223 data3[0] = (int16_t) vout3r;
224 data3[1] = (int16_t) vout3i;
225 data0 += 2;
226 data1 += 2;
227 data2 += 2;
228 data3 += 2;
229 } while(--samples != 0);
230 }
231 }
232