Mercurial > vec

comparison utils/gengeneric.c @ 45:7955bed1d169 default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
*: add preliminary floating point support no x86 intrinsics just yet, but I did add altivec since it's (arguably) the simplest :)
author Paper <paper@tflc.us>
date Wed, 30 Apr 2025 18:36:38 -0400
parents b0a3f0248ecc
children
comparison
equal deleted inserted replaced
44:b0a3f0248ecc 45:7955bed1d169
1 /**
2 * vec - a tiny SIMD vector library in C99
3 *
4 * Copyright (c) 2024-2025 Paper
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 **/
24
25 /* Use this file to generate include/vec/impl/generic.h !!
26 *
27 * `gcc -o gengeneric gengeneric.c` */
28
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <string.h>
32 #include <ctype.h>
33
34 #define ARRAY_SIZE(x) (sizeof(x)/sizeof((x)[0]))
35
36 /* XXX: would it be faster to unroll literally everything instead of defining everything,
37 * and then unpacking it all? */
38 static const char *header =
39 "/**\n"
40 " * vec - a tiny SIMD vector library in C99\n"
41 " * \n"
42 " * Copyright (c) 2024-2025 Paper\n"
43 " * \n"
44 " * Permission is hereby granted, free of charge, to any person obtaining a copy\n"
45 " * of this software and associated documentation files (the \"Software\"), to deal\n"
46 " * in the Software without restriction, including without limitation the rights\n"
47 " * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n"
48 " * copies of the Software, and to permit persons to whom the Software is\n"
49 " * furnished to do so, subject to the following conditions:\n"
50 " * \n"
51 " * The above copyright notice and this permission notice shall be included in all\n"
52 " * copies or substantial portions of the Software.\n"
53 " * \n"
54 " * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n"
55 " * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n"
56 " * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n"
57 " * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n"
58 " * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n"
59 " * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n"
60 " * SOFTWARE.\n"
61 "**/\n"
62 "\n"
63 "/* This file is automatically generated! Do not edit it directly!\n"
64 " * Edit the code that generates it in utils/gengeneric.c --paper */\n"
65 "\n"
66 "#ifndef VEC_IMPL_GENERIC_H_\n"
67 "#define VEC_IMPL_GENERIC_H_\n"
68 "\n"
69 "#define VEC_GENERIC_OPERATION(op, sign, bits, size) \\\n"
70 " do { \\\n"
71 " int i; \\\n"
72 " \\\n"
73 " for (i = 0; i < size; i++) \\\n"
74 " vec1.generic[i] = (op); \\\n"
75 " \\\n"
76 " return vec1; \\\n"
77 " } while (0)\n"
78 "\n"
79 "#define VEC_GENERIC_BUILTIN_OPERATION(op, sign, bits, size) \\\n"
80 " VEC_GENERIC_OPERATION(vec1.generic[i] op vec2.generic[i], sign, bits, size)\n"
81 "\n"
82 "#define VEC_GENERIC_CMP(op, sign, bits, size) \\\n"
83 " VEC_GENERIC_OPERATION((vec1.generic[i] op vec2.generic[i]) ? (vec_##sign##int##bits)VEC_MAX_OF_TYPE(vec_uint##bits) : 0, sign, bits, size)\n"
84 "\n"
85 "/* okay, now we can do this crap: */\n"
86 "\n"
87 "#define VEC_GENERIC_SPLAT(sign, bits, size) \\\n"
88 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_splat(vec_##sign##int##bits x) \\\n"
89 " { \\\n"
90 " v##sign##int##bits##x##size vec; \\\n"
91 " for (int i = 0; i < size; i++) \\\n"
92 " vec.generic[i] = x; \\\n"
93 " return vec; \\\n"
94 " }\n"
95 "\n"
96 "#define VEC_GENERIC_LOAD_EX(name, sign, bits, size) \\\n"
97 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_##name(const vec_##sign##int##bits in[size]) \\\n"
98 " { \\\n"
99 " v##sign##int##bits##x##size vec; \\\n"
100 " memcpy(&vec, in, sizeof(vec_##sign##int##bits) * size); \\\n"
101 " return vec; \\\n"
102 " }\n"
103 "\n"
104 "#define VEC_GENERIC_LOAD_ALIGNED(sign, bits, size) VEC_GENERIC_LOAD_EX(load_aligned, sign, bits, size)\n"
105 "#define VEC_GENERIC_LOAD(sign, bits, size) VEC_GENERIC_LOAD_EX(load, sign, bits, size)\n"
106 "\n"
107 "#define VEC_GENERIC_STORE_EX(name, sign, bits, size) \\\n"
108 " VEC_FUNC_IMPL void v##sign##int##bits##x##size##_##name(v##sign##int##bits##x##size vec, vec_##sign##int##bits out[size]) \\\n"
109 " { \\\n"
110 " memcpy(out, &vec, sizeof(vec_##sign##int##bits) * size); \\\n"
111 " }\n"
112 "\n"
113 "#define VEC_GENERIC_STORE_ALIGNED(sign, bits, size) VEC_GENERIC_STORE_EX(store_aligned, sign, bits, size)\n"
114 "#define VEC_GENERIC_STORE(sign, bits, size) VEC_GENERIC_STORE_EX(store, sign, bits, size)\n"
115 "\n"
116 "#define VEC_GENERIC_ADD(sign, bits, size) \\\n"
117 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_add(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
118 " { \\\n"
119 " VEC_GENERIC_BUILTIN_OPERATION(+, sign, bits, size); \\\n"
120 " }\n"
121 "\n"
122 "#define VEC_GENERIC_SUB(sign, bits, size) \\\n"
123 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_sub(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
124 " { \\\n"
125 " VEC_GENERIC_BUILTIN_OPERATION(-, sign, bits, size); \\\n"
126 " }\n"
127 "\n"
128 "#define VEC_GENERIC_MUL(sign, bits, size) \\\n"
129 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_mul(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
130 " { \\\n"
131 " VEC_GENERIC_BUILTIN_OPERATION(*, sign, bits, size); \\\n"
132 " }\n"
133 "\n"
134 "#define VEC_GENERIC_DIV(sign, bits, size) \\\n"
135 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_div(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
136 " { \\\n"
137 " VEC_GENERIC_OPERATION(vec2.generic[i] ? (vec1.generic[i] / vec2.generic[i]) : 0, sign, bits, size); \\\n"
138 " }\n"
139 "\n"
140 "#define VEC_GENERIC_MOD(sign, bits, size) \\\n"
141 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_mod(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
142 " { \\\n"
143 " VEC_GENERIC_OPERATION(vec2.generic[i] ? (vec1.generic[i] % vec2.generic[i]) : 0, sign, bits, size); \\\n"
144 " }\n"
145 "\n"
146 "#define VEC_GENERIC_AVG(sign, bits, size) \\\n"
147 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_avg(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
148 " { \\\n"
149 " for (int i = 0; i < size; i++) \\\n"
150 " vec1.generic[i] = vec_im##sign##avg(vec1.generic[i], vec2.generic[i]); \\\n"
151 " \\\n"
152 " return vec1; \\\n"
153 " }\n"
154 "\n"
155 "#define VEC_GENERIC_AND(sign, bits, size) \\\n"
156 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_and(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
157 " { \\\n"
158 " VEC_GENERIC_BUILTIN_OPERATION(&, sign, bits, size); \\\n"
159 " }\n"
160 "\n"
161 "#define VEC_GENERIC_OR(sign, bits, size) \\\n"
162 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_or(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
163 " { \\\n"
164 " VEC_GENERIC_BUILTIN_OPERATION(|, sign, bits, size); \\\n"
165 " }\n"
166 "\n"
167 "#define VEC_GENERIC_XOR(sign, bits, size) \\\n"
168 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_xor(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
169 " { \\\n"
170 " VEC_GENERIC_BUILTIN_OPERATION(^, sign, bits, size); \\\n"
171 " }\n"
172 "\n"
173 "#define VEC_GENERIC_NOT(sign, bits, size) \\\n"
174 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_not(v##sign##int##bits##x##size vec) \\\n"
175 " { \\\n"
176 " return v##sign##int##bits##x##size##_xor(vec, v##sign##int##bits##x##size##_splat((vec_##sign##int##bits)VEC_MAX_OF_TYPE(vec_uint##bits))); \\\n"
177 " }\n"
178 "\n"
179 "#define VEC_GENERIC_CMPLT(sign, bits, size) \\\n"
180 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_cmplt(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
181 " { \\\n"
182 " VEC_GENERIC_CMP(<, sign, bits, size); \\\n"
183 " }\n"
184 "\n"
185 "#define VEC_GENERIC_CMPLE(sign, bits, size) \\\n"
186 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_cmple(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
187 " { \\\n"
188 " return v##sign##int##bits##x##size##_not(v##sign##int##bits##x##size##_cmpgt(vec1, vec2)); \\\n"
189 " }\n"
190 "\n"
191 "#define VEC_GENERIC_CMPEQ(sign, bits, size) \\\n"
192 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_cmpeq(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
193 " { \\\n"
194 " VEC_GENERIC_CMP(==, sign, bits, size); \\\n"
195 " }\n"
196 "\n"
197 "#define VEC_GENERIC_CMPGE(sign, bits, size) \\\n"
198 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_cmpge(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
199 " { \\\n"
200 " return v##sign##int##bits##x##size##_not(v##sign##int##bits##x##size##_cmplt(vec1, vec2)); \\\n"
201 " }\n"
202 "\n"
203 "#define VEC_GENERIC_CMPGT(sign, bits, size) \\\n"
204 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_cmpgt(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
205 " { \\\n"
206 " VEC_GENERIC_CMP(>, sign, bits, size); \\\n"
207 " }\n"
208 "\n"
209 "#define VEC_GENERIC_LSHIFT(sign, bits, size) \\\n"
210 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_lshift(v##sign##int##bits##x##size vec1, vuint##bits##x##size vec2) \\\n"
211 " { \\\n"
212 " VEC_GENERIC_OPERATION(vec_##sign##lshift(vec1.generic[i], vec2.generic[i]), sign, bits, size); \\\n"
213 " }\n"
214 "\n"
215 "#define VEC_GENERIC_RSHIFT(sign, bits, size) \\\n"
216 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_rshift(v##sign##int##bits##x##size vec1, vuint##bits##x##size vec2) \\\n"
217 " { \\\n"
218 " VEC_GENERIC_OPERATION(vec_##sign##rshift(vec1.generic[i], vec2.generic[i]), sign, bits, size); \\\n"
219 " }\n"
220 "\n"
221 "#define VEC_GENERIC_LRSHIFT(sign, bits, size) \\\n"
222 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_lrshift(v##sign##int##bits##x##size vec1, vuint##bits##x##size vec2) \\\n"
223 " { \\\n"
224 " VEC_GENERIC_OPERATION(vec_urshift((vec_uint##bits)vec1.generic[i], vec2.generic[i]), sign, bits, size); \\\n"
225 " }\n"
226 "\n"
227 "#define VEC_GENERIC_MIN(sign, bits, size) \\\n"
228 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_min(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
229 " { \\\n"
230 " v##sign##int##bits##x##size cmplt = v##sign##int##bits##x##size##_cmplt(vec1, vec2); \\\n"
231 " \\\n"
232 " v##sign##int##bits##x##size a = v##sign##int##bits##x##size##_and(vec1, cmplt); \\\n"
233 " v##sign##int##bits##x##size b = v##sign##int##bits##x##size##_and(vec2, v##sign##int##bits##x##size##_not(cmplt)); \\\n"
234 " \\\n"
235 " return v##sign##int##bits##x##size##_or(a, b); \\\n"
236 " }\n"
237 "\n"
238 "#define VEC_GENERIC_MAX(sign, bits, size) \\\n"
239 " VEC_FUNC_IMPL v##sign##int##bits##x##size v##sign##int##bits##x##size##_max(v##sign##int##bits##x##size vec1, v##sign##int##bits##x##size vec2) \\\n"
240 " { \\\n"
241 " v##sign##int##bits##x##size cmplt = v##sign##int##bits##x##size##_cmpgt(vec1, vec2); \\\n"
242 " \\\n"
243 " v##sign##int##bits##x##size a = v##sign##int##bits##x##size##_and(vec1, cmplt); \\\n"
244 " v##sign##int##bits##x##size b = v##sign##int##bits##x##size##_and(vec2, v##sign##int##bits##x##size##_not(cmplt)); \\\n"
245 " \\\n"
246 " return v##sign##int##bits##x##size##_or(a, b); \\\n"
247 " }\n"
248 "\n"
249 "/* ------------------------------------------------------------------------ */\n"
250 "/* PREPROCESSOR HELL INCOMING */\n";
251
252 static const char *footer =
253 "#endif /* VEC_IMPL_GENERIC_H_ */\n";
254
255 /* ------------------------------------------------------------------------ */
256
257 static void print_generic_op(const char *op, int is_signed, int bits, int size)
258 {
259 printf(
260 "#ifndef V%sINT%dx%d_%s_DEFINED\n"
261 "VEC_GENERIC_%s(%s, %d, %d)\n"
262 "# define V%sINT%dx%d_%s_DEFINED\n"
263 "#endif\n",
264 (is_signed ? "" : "U"), bits, size, op, op, (is_signed ? "/* nothing */" : "u"), bits, size, (is_signed ? "" : "U"), bits, size, op);
265 }
266
267 typedef void (*print_op_spec)(const char *op, int is_signed, int bits, int size);
268
269 static inline void print_ops(int is_signed, int bits, int size, print_op_spec print_op)
270 {
271 /* all supported operations here */
272 static const char *ops[] = {
273 "SPLAT",
274 "LOAD_ALIGNED",
275 "LOAD",
276 "STORE_ALIGNED",
277 "STORE",
278 "ADD",
279 "SUB",
280 "MUL",
281 "DIV",
282 "MOD",
283 "AVG",
284 "AND",
285 "OR",
286 "XOR",
287 "NOT",
288 "CMPLT",
289 "CMPEQ",
290 "CMPGT",
291 "CMPLE", /* these two must be after CMPLT and CMPGT respectfully, */
292 "CMPGE", /* because their definitions call those functions */
293 "MIN",
294 "MAX",
295 "RSHIFT",
296 "LRSHIFT",
297 "LSHIFT",
298 NULL,
299 };
300 int i;
301
302 printf("\n\n/* v%sint%dx%d */\n\n", (is_signed ? "u" : ""), bits, size);
303
304 for (i = 0; ops[i]; i++)
305 print_op(ops[i], is_signed, bits, size);
306 }
307
308 int main(void)
309 {
310 static struct {
311 int bits, size;
312 print_op_spec print_op;
313 } defs[] = {
314 /* -- 8-bit */
315 {8, 2, print_generic_op},
316
317 /* -- 16-bit */
318 {16, 2, print_generic_op},
319
320 /* -- 32-bit */
321 {32, 2, print_generic_op},
322
323 /* -- 64-bit */
324 {64, 2, print_generic_op},
325 };
326 int i;
327
328 puts(header);
329
330 for (i = 0; i < ARRAY_SIZE(defs); i++) {
331 print_ops(1, defs[i].bits, defs[i].size, defs[i].print_op);
332 print_ops(0, defs[i].bits, defs[i].size, defs[i].print_op);
333 }
334
335 puts(footer);
336 }