summaryrefslogtreecommitdiffstats
path: root/src/detection/cpu/cpu.c
blob: aadcdc452dce5252ce38f2032bfab0b3d129b64d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
#include "cpu.h"

const char* ffDetectCPUImpl(const FFCPUOptions* options, FFCPUResult* cpu);

const char* ffDetectCPU(const FFCPUOptions* options, FFCPUResult* cpu) {
    const char* error = ffDetectCPUImpl(options, cpu);
    if (error) {
        return error;
    }

    const char* removeStrings[] = {
        " CPU", " FPU", " APU", " Processor", " Dual-Core", " Quad-Core", " Six-Core", " Eight-Core", " Ten-Core", " 2-Core", " 4-Core", " 6-Core", " 8-Core", " 10-Core", " 12-Core", " 14-Core", " 16-Core"
    };
    ffStrbufRemoveStrings(&cpu->name, ARRAY_SIZE(removeStrings), removeStrings);
    uint32_t radeonGraphics = ffStrbufFirstIndexS(&cpu->name, " w/ Radeon "); // w/ Radeon 780M Graphics
    if (radeonGraphics >= cpu->name.length) {
        radeonGraphics = ffStrbufFirstIndexS(&cpu->name, " with Radeon ");
    }
    if (radeonGraphics < cpu->name.length) {
        ffStrbufSubstrBefore(&cpu->name, radeonGraphics);
    }
    ffStrbufSubstrBeforeFirstC(&cpu->name, '@'); // Cut the speed output in the name as we append our own
    ffStrbufTrimRight(&cpu->name, ' ');          // If we removed the @ in previous step there was most likely a space before it
    ffStrbufRemoveDupWhitespaces(&cpu->name);
    return NULL;
}

const char* ffCPUAppleCodeToName(uint32_t code) {
    // https://github.com/AsahiLinux/docs/wiki/Codenames
    switch (code) {
        case 8103:
            return "Apple M1";
        case 6000:
            return "Apple M1 Pro";
        case 6001:
            return "Apple M1 Max";
        case 6002:
            return "Apple M1 Ultra";
        case 8112:
            return "Apple M2";
        case 6020:
            return "Apple M2 Pro";
        case 6021:
            return "Apple M2 Max";
        case 6022:
            return "Apple M2 Ultra";
        case 8122:
            return "Apple M3";
        case 6030:
            return "Apple M3 Pro";
        case 6031:
        case 6034:
            return "Apple M3 Max";
        case 8132:
            return "Apple M4";
        case 6040:
            return "Apple M4 Pro";
        case 6041:
            return "Apple M4 Max";
        default:
            return NULL;
    }
}

const char* ffCPUQualcommCodeToName(uint32_t code) {
    // https://github.com/AsahiLinux/docs/wiki/Codenames
    switch (code) {
        case 7180:
            return "Qualcomm Snapdragon 7c";
        case 7280:
            return "Qualcomm Snapdragon 7c+ Gen 3";
        case 8180:
            return "Qualcomm Snapdragon 8cx Gen 2 5G";
        case 8280:
            return "Qualcomm Snapdragon 8cx Gen 3";
        default:
            return NULL;
    }
}

#if defined(__x86_64__) || defined(__i386__)

    #include <cpuid.h>

void ffCPUDetectByCpuid(FFCPUResult* cpu) {
    uint32_t eax = 0, ebx = 0, ecx = 0, edx = 0;
    if (__get_cpuid(0x16, &eax, &ebx, &ecx, &edx)) {
        // WARNING: CPUID may report frequencies of efficient cores
        // cpuid returns 0 MHz when hypervisor is enabled
        if (eax) {
            cpu->frequencyBase = eax;
        }
        if (ebx) {
            cpu->frequencyMax = ebx;
        }
    }

    if (__get_cpuid(1, &eax, &ebx, &ecx, &edx)) {
        // Feature tests (leaf1.ecx, leaf7.ebx)
        bool sse2 = (ecx & bit_SSE2) != 0;
        bool sse4_2 = (ecx & bit_SSE4_2) != 0;
        bool pclmul = (ecx & bit_PCLMUL) != 0;
        bool popcnt = (ecx & bit_POPCNT) != 0;
        bool fma = (ecx & bit_FMA) != 0;
        bool osxsave = (ecx & bit_OSXSAVE) != 0;

        unsigned int eax7 = 0, ebx7 = 0, ecx7 = 0, edx7 = 0;
        __get_cpuid_count(7, 0, &eax7, &ebx7, &ecx7, &edx7);

        bool avx2 = (ebx7 & bit_AVX2) != 0;
        bool bmi2 = (ebx7 & bit_BMI2) != 0;
        bool avx512f = (ebx7 & bit_AVX512F) != 0;
        bool avx512bw = (ebx7 & bit_AVX512BW) != 0;
        bool avx512dq = (ebx7 & bit_AVX512DQ) != 0;

        // OS support for AVX/AVX512: check XGETBV (requires OSXSAVE)
        bool avx_os = false;
        bool avx512_os = false;
        if (osxsave) {
            __asm__ __volatile__(
                "xgetbv"
                : "=a"(eax), "=d"(edx)
                : "c"(0)
                :);
            uint64_t xcr0 = ((uint64_t) edx << 32) | eax;

            // AVX requires XCR0[1:2] == 11b (XMM and YMM state)
            avx_os = (xcr0 & 0x6ULL) == 0x6ULL;
            // AVX512 requires XCR0[7,5,6] etc. common mask 0xE6 (bits 1,2,5,6,7)
            avx512_os = (xcr0 & 0xE6ULL) == 0xE6ULL;
        }

        cpu->march = "unknown";
        if (avx512f && avx512bw && avx512dq && avx512_os) {
            cpu->march = "x86_64-v4";
        } else if (avx2 && fma && bmi2 && avx_os) {
            cpu->march = "x86_64-v3";
        } else if (sse4_2 && popcnt && pclmul) {
            cpu->march = "x86_64-v2";
        } else if (sse2) {
            cpu->march = "x86_64-v1";
        }
    }
}

#elif defined(__aarch64__)

// This is not accurate because a lot of flags are optional from old versions
// https://developer.arm.com/documentation/109697/2025_06/Feature-descriptions?lang=en
// https://en.wikipedia.org/wiki/AArch64#ARM-A_(application_architecture)
// Worth noting: Apple M1 is marked as ARMv8.5-A on Wikipedia, but it lacks BTI (mandatory in v8.5)

    #ifdef __linux__
        #include "common/io.h"
        #include <elf.h>
// #include <asm/hwcap.h>

void ffCPUDetectByCpuid(FFCPUResult* cpu) {
    char buf[PROC_FILE_BUFFSIZ];
    ssize_t nRead = ffReadFileData("/proc/self/auxv", ARRAY_SIZE(buf), buf);

    if (nRead < (ssize_t) sizeof(Elf64_auxv_t)) {
        return;
    }

    uint64_t hwcap = 0, hwcap2 = 0;

    for (Elf64_auxv_t* auxv = (Elf64_auxv_t*) buf; (char*) auxv < buf + nRead; ++auxv) {
        if (auxv->a_type == AT_HWCAP) {
            hwcap = auxv->a_un.a_val;
        } else if (auxv->a_type == AT_HWCAP2) {
            hwcap2 = auxv->a_un.a_val;
        }
    }

    if (!hwcap) {
        return;
    }

    cpu->march = "unknown";

    // ARMv8-A
    bool has_fp = (hwcap & (1 << 0) /* HWCAP_FP */) != 0;
    bool has_asimd = (hwcap & (1 << 1) /* HWCAP_ASIMD */) != 0;

    // ARMv8.1-A
    bool has_atomics = (hwcap & (1 << 8) /* HWCAP_ATOMICS */) != 0;    // optional from v8.0
    bool has_crc32 = (hwcap & (1 << 7) /* HWCAP_CRC32 */) != 0;        // optional from v8.0
    bool has_asimdrdm = (hwcap & (1 << 12) /* HWCAP_ASIMDRDM */) != 0; // optional from v8.0

    // ARMv8.2-A
    bool has_fphp = (hwcap & (1 << 9) /* HWCAP_FPHP */) != 0;    // optional
    bool has_dcpop = (hwcap & (1 << 16) /* HWCAP_DCPOP */) != 0; // DC CVAP, optional from v8.1

    // ARMv8.3-A
    bool has_paca = (hwcap & (1 << 30) /* HWCAP_PACA */) != 0;   // optional from v8.2
    bool has_lrcpc = (hwcap & (1 << 15) /* HWCAP_LRCPC */) != 0; // optional from v8.2
    bool has_fcma = (hwcap & (1 << 14) /* HWCAP_FCMA */) != 0;   // optional from v8.2
    bool has_jscvt = (hwcap & (1 << 13) /* HWCAP_JSCVT */) != 0; // optional from v8.2

    // ARMv8.4-A
    bool has_dit = (hwcap & (1 << 24) /* HWCAP_DIT */) != 0;       // optional from v8.3
    bool has_flagm = (hwcap & (1 << 27) /* HWCAP_FLAGM */) != 0;   // optional from v8.1
    bool has_ilrcpc = (hwcap & (1 << 26) /* HWCAP_ILRCPC */) != 0; // optional from v8.2

    // ARMv8.5-A
    bool has_bti = (hwcap2 & (1 << 17) /* HWCAP2_BTI */) != 0;      // optional from v8.4
    bool has_sb = (hwcap & (1 << 29) /* HWCAP_SB */) != 0;          // optional from v8.0
    bool has_dcpodp = (hwcap2 & (1 << 0) /* HWCAP2_DCPODP */) != 0; // optional from v8.1
    bool has_flagm2 = (hwcap2 & (1 << 7) /* HWCAP2_FLAGM2 */) != 0; // optional from v8.4
    bool has_frint = (hwcap2 & (1 << 8) /* HWCAP2_FRINT */) != 0;   // optional from v8.4

    // ARMv9.0-A
    bool has_sve2 = (hwcap2 & (1 << 1) /* HWCAP2_SVE2 */) != 0;

    // ARMv9.1-A
    // ARMv8.6-A
    bool has_bf16 = (hwcap2 & (1 << 14) /* HWCAP2_BF16 */) != 0; // optional from v8.2
    bool has_i8mm = (hwcap2 & (1 << 13) /* HWCAP2_I8MM */) != 0; // optional from v8.1

    // ARMv8.7-A
    bool has_afp = (hwcap2 & (1 << 20) /* HWCAP2_AFP */) != 0; // optional from v8.6

    // ARMv9.2-A
    bool has_sme = (hwcap2 & (1 << 23) /* HWCAP2_SME */) != 0;

    // ARMv9.3-A
    bool has_sme2 = (hwcap2 & (1UL << 37) /* HWCAP2_SME2 */) != 0; // optional from v9.2

    // ARMv8.8-A
    bool has_mops = (hwcap2 & (1UL << 43) /* HWCAP2_MOPS */) != 0; // optional from v8.7

    // ARMv8.9-A
    bool has_cssc = (hwcap2 & (1UL << 34) /* HWCAP2_CSSC */) != 0; // optional from v8.7

    // ARMv9.4-A
    bool has_sme2p1 = (hwcap2 & (1UL << 38) /* HWCAP2_SME2P1 */) != 0; // optional from v9.2

    // ARMv9.5-A
    bool has_f8e4m3 = (hwcap2 & (1UL << 55) /* HWCAP2_F8E4M3 */) != 0; // optional from v9.2
    bool has_f8e5m2 = (hwcap2 & (1UL << 56) /* HWCAP2_F8E5M2 */) != 0; // optional from v9.2

    // ARMv9.6-A
    bool has_cmpbr = (hwcap & (1UL << 33) /* HWCAP_CMPBR */) != 0;   // optional from v9.5
    bool has_fprcvt = (hwcap & (1UL << 34) /* HWCAP_FPRCVT */) != 0; // optional from v9.5

    if (has_sve2 || has_sme) {
        // ARMv9
        if (has_cmpbr && has_fprcvt) {
            cpu->march = "ARMv9.6-A";
        } else if (has_f8e5m2 && has_f8e4m3) {
            cpu->march = "ARMv9.5-A";
        } else if (has_sme2p1) {
            cpu->march = "ARMv9.4-A";
        } else if (has_sme2) {
            cpu->march = "ARMv9.3-A";
        } else if (has_sme) {
            cpu->march = "ARMv9.2-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv9.1-A";
        } else {
            cpu->march = "ARMv9.0-A";
        }
    } else {
        // ARMv8
        if (has_cssc) {
            cpu->march = "ARMv8.9-A";
        } else if (has_mops) {
            cpu->march = "ARMv8.8-A";
        } else if (has_afp) {
            cpu->march = "ARMv8.7-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv8.6-A";
        } else if (has_bti && has_sb && has_dcpodp && has_flagm2 && has_frint) {
            cpu->march = "ARMv8.5-A";
        } else if (has_dit && has_flagm && has_ilrcpc) {
            cpu->march = "ARMv8.4-A";
        } else if (has_paca && has_lrcpc && has_fcma && has_jscvt) {
            cpu->march = "ARMv8.3-A";
        } else if (has_fphp && has_dcpop) {
            cpu->march = "ARMv8.2-A";
        } else if (has_atomics && has_crc32 && has_asimdrdm) {
            cpu->march = "ARMv8.1-A";
        } else if (has_asimd && has_fp) {
            cpu->march = "ARMv8-A";
        }
    }
}
    #elif __APPLE__
        #include <sys/sysctl.h>
// #include <arm/cpu_capabilities_public.h> // Not available in macOS 14-

void ffCPUDetectByCpuid(FFCPUResult* cpu) {
    uint64_t caps[2] = { 0 }; // 80-bit capability mask, split into two 64-bit values
    size_t size = sizeof(caps);

    if (sysctlbyname("hw.optional.arm.caps", caps, &size, NULL, 0) != 0) {
        return;
    }

        // Helper macro to test bit in 80-bit capability mask
        #define FF_HAS_CAP(bit) \
            (((bit) < 64) ? ((caps[0] >> (bit)) & 1ULL) : ((caps[1] >> ((bit) - 64U)) & 1ULL))

    cpu->march = "unknown";

    // ARMv8-A
    bool has_fp = FF_HAS_CAP(50); /* CAP_BIT_AdvSIMD_HPFPCvt */ // Full FP16 support (implies FP/ASIMD)
    bool has_asimd = FF_HAS_CAP(49); /* CAP_BIT_AdvSIMD */      // Advanced SIMD (NEON)

    // ARMv8.1-A
    bool has_lse = FF_HAS_CAP(6); /* CAP_BIT_FEAT_LSE */      // Large System Extensions, optional in v8.0
    bool has_crc32 = FF_HAS_CAP(51); /* CAP_BIT_FEAT_CRC32 */ // CRC32 instructions, optional in v8.0
    bool has_rdm = FF_HAS_CAP(5); /* CAP_BIT_FEAT_RDM */      // AdvSIMD rounding double multiply accumulate, optional in v8.0

    // ARMv8.2-A
    bool has_fp16 = FF_HAS_CAP(34); /* CAP_BIT_FEAT_FP16 */ // Half-precision FP support, optional
    bool has_dpb = FF_HAS_CAP(22); /* CAP_BIT_FEAT_DPB */   // DC CVAP, optional from v8.1

    // ARMv8.3-A
    bool has_pauth = FF_HAS_CAP(19); /* CAP_BIT_FEAT_PAuth */ // Pointer Authentication (PAC), optional from v8.2
    bool has_lrcpc = FF_HAS_CAP(15); /* CAP_BIT_FEAT_LRCPC */ // LDAPR/LR with RCPC semantics, optional from v8.2
    bool has_fcma = FF_HAS_CAP(17); /* CAP_BIT_FEAT_FCMA */   // Complex number multiply-add, optional from v8.2
    bool has_jscvt = FF_HAS_CAP(18); /* CAP_BIT_FEAT_JSCVT */ // JavaScript-style conversion (FJCVTZS), optional from v8.2

    // ARMv8.4-A
    bool has_lse2 = FF_HAS_CAP(30); /* CAP_BIT_FEAT_LSE2 */     // Large System Extensions version 2, optional from v8.2
    bool has_dit = FF_HAS_CAP(33); /* CAP_BIT_FEAT_DIT */       // Data Independent Timing, optional from v8.3
    bool has_flagm = FF_HAS_CAP(0); /* CAP_BIT_FEAT_FlagM */    // Flag manipulation (FMOV/FCVT), optional from v8.1
    bool has_lrcpc2 = FF_HAS_CAP(16); /* CAP_BIT_FEAT_LRCPC2 */ // Enhanced RCPC (LDAPUR/LDAPST), optional from v8.2

    // ARMv8.5-A
    bool has_bti = FF_HAS_CAP(36); /* CAP_BIT_FEAT_BTI */         // Branch Target Identification, optional from v8.4
    bool has_sb = FF_HAS_CAP(13); /* CAP_BIT_FEAT_SB */           // Speculative Barrier, optional from v8.0
    bool has_dpb2 = FF_HAS_CAP(23); /* CAP_BIT_FEAT_DPB2 */       // DC CVADP (DPB2), optional from v8.1
    bool has_flagm2 = FF_HAS_CAP(1); /* CAP_BIT_FEAT_FlagM2 */    // Enhanced FlagM, optional from v8.4
    bool has_frintts = FF_HAS_CAP(14); /* CAP_BIT_FEAT_FRINTTS */ // Floating-point to integer instructions, optional from v8.4

    // ARMv9.0-A
    bool has_sve2 = false; // Not exposed and not supported by Apple M4

    // ARMv9.1-A
    // ARMv8.6-A
    bool has_bf16 = FF_HAS_CAP(24); /* CAP_BIT_FEAT_BF16 */ // Brain float16, optional from v8.2
    bool has_i8mm = FF_HAS_CAP(25); /* CAP_BIT_FEAT_I8MM */ // Int8 Matrix Multiply, optional from v8.1

    // ARMv8.7-A
    bool has_afp = FF_HAS_CAP(29); /* CAP_BIT_FEAT_AFP */ // Alternate FP16 (FEXPA), optional from v8.6

    // ARMv9.2-A
    bool has_sme = FF_HAS_CAP(40); /* CAP_BIT_FEAT_SME */ // Scalable Matrix Extension, optional from v9.2

    // ARMv9.3-A
    bool has_sme2 = FF_HAS_CAP(41); /* CAP_BIT_FEAT_SME2 */ // SME2, optional from v9.2

    // ARMv8.8-A
    bool has_hbc = FF_HAS_CAP(64); /* CAP_BIT_FEAT_HBC */ // Hinted conditional branches, optional from v8.7

    // ARMv8.9-A
    bool has_cssc = FF_HAS_CAP(67); /* CAP_BIT_FEAT_CSSC */ // Common Short String Compare, optional from v8.7

    // ARMv9.4-A+ are not exposed yet

    if (has_sve2 || has_sme) {
        // ARMv9 family
        if (has_sme2) {
            cpu->march = "ARMv9.3-A";
        } else if (has_sme) {
            cpu->march = "ARMv9.2-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv9.1-A";
        } else {
            cpu->march = "ARMv9.0-A";
        }
    } else {
        // ARMv8 family
        if (has_cssc) {
            cpu->march = "ARMv8.9-A";
        } else if (has_hbc) {
            cpu->march = "ARMv8.8-A";
        } else if (has_afp) {
            cpu->march = "ARMv8.7-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv8.6-A";
        } else if (has_bti && has_sb && has_dpb2 && has_flagm2 && has_frintts) {
            cpu->march = "ARMv8.5-A";
        } else if (has_lse2 && has_dit && has_flagm && has_lrcpc2) {
            cpu->march = "ARMv8.4-A";
        } else if (has_pauth && has_lrcpc && has_fcma && has_jscvt) {
            cpu->march = "ARMv8.3-A";
        } else if (has_fp16 && has_dpb) {
            cpu->march = "ARMv8.2-A";
        } else if (has_lse && has_crc32 && has_rdm) {
            cpu->march = "ARMv8.1-A";
        } else if (has_asimd && has_fp) {
            cpu->march = "ARMv8-A";
        }
    }

        #undef HAS_CAP
}
    #elif _WIN32
        #include <processthreadsapi.h>

        // Missing from winnt.h of MinGW-w64
        #define PF_ARM_LSE2_AVAILABLE 62
        #define PF_RESERVED_FEATURE 63
        #define PF_ARM_SHA3_INSTRUCTIONS_AVAILABLE 64
        #define PF_ARM_SHA512_INSTRUCTIONS_AVAILABLE 65
        #define PF_ARM_V82_I8MM_INSTRUCTIONS_AVAILABLE 66
        #define PF_ARM_V82_FP16_INSTRUCTIONS_AVAILABLE 67
        #define PF_ARM_V86_BF16_INSTRUCTIONS_AVAILABLE 68
        #define PF_ARM_V86_EBF16_INSTRUCTIONS_AVAILABLE 69
        #define PF_ARM_SME_INSTRUCTIONS_AVAILABLE 70
        #define PF_ARM_SME2_INSTRUCTIONS_AVAILABLE 71
        #define PF_ARM_SME2_1_INSTRUCTIONS_AVAILABLE 72
        #define PF_ARM_SME2_2_INSTRUCTIONS_AVAILABLE 73
        #define PF_ARM_SME_AES_INSTRUCTIONS_AVAILABLE 74
        #define PF_ARM_SME_SBITPERM_INSTRUCTIONS_AVAILABLE 75
        #define PF_ARM_SME_SF8MM4_INSTRUCTIONS_AVAILABLE 76
        #define PF_ARM_SME_SF8MM8_INSTRUCTIONS_AVAILABLE 77
        #define PF_ARM_SME_SF8DP2_INSTRUCTIONS_AVAILABLE 78
        #define PF_ARM_SME_SF8DP4_INSTRUCTIONS_AVAILABLE 79
        #define PF_ARM_SME_SF8FMA_INSTRUCTIONS_AVAILABLE 80
        #define PF_ARM_SME_F8F32_INSTRUCTIONS_AVAILABLE 81
        #define PF_ARM_SME_F8F16_INSTRUCTIONS_AVAILABLE 82
        #define PF_ARM_SME_F16F16_INSTRUCTIONS_AVAILABLE 83
        #define PF_ARM_SME_B16B16_INSTRUCTIONS_AVAILABLE 84
        #define PF_ARM_SME_F64F64_INSTRUCTIONS_AVAILABLE 85
        #define PF_ARM_SME_I16I64_INSTRUCTIONS_AVAILABLE 86
        #define PF_ARM_SME_LUTv2_INSTRUCTIONS_AVAILABLE 87
        #define PF_ARM_SME_FA64_INSTRUCTIONS_AVAILABLE 88

void ffCPUDetectByCpuid(FFCPUResult* cpu) {
    // ARMv8-A
    bool has_vfp = IsProcessorFeaturePresent(PF_ARM_VFP_32_REGISTERS_AVAILABLE);   // Implies basic FP support
    bool has_neon = IsProcessorFeaturePresent(PF_ARM_NEON_INSTRUCTIONS_AVAILABLE); // NEON (ASIMD)

    // ARMv8.1-A
    bool has_atomics = IsProcessorFeaturePresent(PF_ARM_V81_ATOMIC_INSTRUCTIONS_AVAILABLE); // LSE atomics
    bool has_crc32 = IsProcessorFeaturePresent(PF_ARM_V8_CRC32_INSTRUCTIONS_AVAILABLE);     // CRC32

    // ARMv8.2-A
    bool has_fp16 = IsProcessorFeaturePresent(PF_ARM_V82_FP16_INSTRUCTIONS_AVAILABLE); // Half-precision FP

    // ARMv8.3-A
    bool has_lrcpc = IsProcessorFeaturePresent(PF_ARM_V83_LRCPC_INSTRUCTIONS_AVAILABLE); // LDAPR/LR with RCPC semantics
    bool has_jscvt = IsProcessorFeaturePresent(PF_ARM_V83_JSCVT_INSTRUCTIONS_AVAILABLE); // FJCVTZS

    // ARMv8.4-A
    // My CPU (Apple M1 Pro in VM) does support LSE2, but Windows doesn't detect it for some reason
    bool has_lse2 = IsProcessorFeaturePresent(PF_ARM_LSE2_AVAILABLE);              // Large System Extensions version 2, optional from v8.2
    bool has_dp = IsProcessorFeaturePresent(PF_ARM_V82_DP_INSTRUCTIONS_AVAILABLE); // DotProd, optional from v8.1 (*)

    // ARMv9.0-A
    bool has_sve2 = IsProcessorFeaturePresent(PF_ARM_SVE2_INSTRUCTIONS_AVAILABLE); // SVE2

    // ARMv9.1-A
    // ARMv8.6-A
    bool has_bf16 = IsProcessorFeaturePresent(PF_ARM_V86_BF16_INSTRUCTIONS_AVAILABLE); // BF16, optional from v8.2
    bool has_i8mm = IsProcessorFeaturePresent(PF_ARM_V82_I8MM_INSTRUCTIONS_AVAILABLE); // Int8 matrix multiply, optional from v8.2

    // ARMv8.7-A
    bool has_ebf16 = IsProcessorFeaturePresent(PF_ARM_V86_EBF16_INSTRUCTIONS_AVAILABLE); // Extended BFloat16 behaviors, optional from v8.2

    // ARMv9.2-A
    bool has_sme = IsProcessorFeaturePresent(PF_ARM_SME_INSTRUCTIONS_AVAILABLE); // SME

    // ARMv9.3-A
    bool has_sme2 = IsProcessorFeaturePresent(PF_ARM_SME2_INSTRUCTIONS_AVAILABLE); // SME2

    // ARMv9.4-A
    bool has_sme2p1 = IsProcessorFeaturePresent(PF_ARM_SME2_1_INSTRUCTIONS_AVAILABLE); // SME2.1

    if (has_sve2 || has_sme) {
        // ARMv9 family
        if (has_sme2p1) {
            cpu->march = "ARMv9.4-A";
        } else if (has_sme2) {
            cpu->march = "ARMv9.3-A";
        } else if (has_sme) {
            cpu->march = "ARMv9.2-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv9.1-A";
        } else {
            cpu->march = "ARMv9.0-A";
        }
    } else {
        // ARMv8 family
        if (has_ebf16) {
            cpu->march = "ARMv8.7-A";
        } else if (has_i8mm && has_bf16) {
            cpu->march = "ARMv8.6-A";
        } else if (has_dp && has_lse2) {
            cpu->march = "ARMv8.4-A";
        } else if (has_lrcpc && has_jscvt) {
            cpu->march = "ARMv8.3-A";
        } else if (has_fp16) {
            cpu->march = "ARMv8.2-A";
        } else if (has_atomics && has_crc32) {
            cpu->march = "ARMv8.1-A";
        } else if (has_neon && has_vfp) {
            cpu->march = "ARMv8-A";
        }
    }
}
    #else
void ffCPUDetectByCpuid(FF_A_UNUSED FFCPUResult* cpu) {
    // Unsupported system
}
    #endif

#else

void ffCPUDetectByCpuid(FF_A_UNUSED FFCPUResult* cpu) {
    // Unsupported architecture
}

#endif