// SPDX-License-Identifier: MIT
// Copyright (c) 2024 nihui (https://github.com/nihui)
// Copyright (c) 2024 kernelbin (https://github.com/kernelbin)
//
// ruapu --- detect cpu isa features with single-file

#include <stdio.h>

#define RUAPU_IMPLEMENTATION
#include "ruapu.h"

int main()
{
    ruapu_init();

#define PRINT_ISA_SUPPORT(isa) printf("%s = %d\n", #isa, ruapu_supports(#isa));

#if defined(__i386__) || defined(__x86_64__) || defined(_M_IX86) || defined(_M_X64)
    PRINT_ISA_SUPPORT(mmx)
    PRINT_ISA_SUPPORT(sse)
    PRINT_ISA_SUPPORT(sse2)
    PRINT_ISA_SUPPORT(sse3)
    PRINT_ISA_SUPPORT(ssse3)
    PRINT_ISA_SUPPORT(sse41)
    PRINT_ISA_SUPPORT(sse42)
    PRINT_ISA_SUPPORT(sse4a)
    PRINT_ISA_SUPPORT(xop)
    PRINT_ISA_SUPPORT(avx)
    PRINT_ISA_SUPPORT(f16c)
    PRINT_ISA_SUPPORT(fma)
    PRINT_ISA_SUPPORT(fma4)
    PRINT_ISA_SUPPORT(avx2)
    PRINT_ISA_SUPPORT(avx512f)
    PRINT_ISA_SUPPORT(avx512bw)
    PRINT_ISA_SUPPORT(avx512cd)
    PRINT_ISA_SUPPORT(avx512dq)
    PRINT_ISA_SUPPORT(avx512vl)
    PRINT_ISA_SUPPORT(avx512vnni)
    PRINT_ISA_SUPPORT(avx512bf16)
    PRINT_ISA_SUPPORT(avx512ifma)
    PRINT_ISA_SUPPORT(avx512vbmi)
    PRINT_ISA_SUPPORT(avx512vbmi2)
    PRINT_ISA_SUPPORT(avx512fp16)
    PRINT_ISA_SUPPORT(avx512er)
    PRINT_ISA_SUPPORT(avx5124fmaps)
    PRINT_ISA_SUPPORT(avx5124vnniw)
    PRINT_ISA_SUPPORT(avxvnni)
    PRINT_ISA_SUPPORT(avxvnniint8)
    PRINT_ISA_SUPPORT(avxvnniint16)
    PRINT_ISA_SUPPORT(avxifma)
    PRINT_ISA_SUPPORT(amxfp16)
    PRINT_ISA_SUPPORT(amxbf16)
    PRINT_ISA_SUPPORT(amxint8)
    PRINT_ISA_SUPPORT(amxtile)
    PRINT_ISA_SUPPORT(bmi1)
    PRINT_ISA_SUPPORT(bmi2)
    PRINT_ISA_SUPPORT(gfni)
    PRINT_ISA_SUPPORT(aesni)
    PRINT_ISA_SUPPORT(vaes)
    PRINT_ISA_SUPPORT(sha)
    PRINT_ISA_SUPPORT(rdrand)
    PRINT_ISA_SUPPORT(rdseed)
    PRINT_ISA_SUPPORT(tsx)

#elif __aarch64__ || defined(_M_ARM64)
    PRINT_ISA_SUPPORT(neon)
    PRINT_ISA_SUPPORT(vfpv4)
    PRINT_ISA_SUPPORT(lse)
    PRINT_ISA_SUPPORT(cpuid)
    PRINT_ISA_SUPPORT(asimdrdm)
    PRINT_ISA_SUPPORT(asimdhp)
    PRINT_ISA_SUPPORT(asimddp)
    PRINT_ISA_SUPPORT(asimdfhm)
    PRINT_ISA_SUPPORT(bf16)
    PRINT_ISA_SUPPORT(i8mm)
    PRINT_ISA_SUPPORT(frint)
    PRINT_ISA_SUPPORT(jscvt)
    PRINT_ISA_SUPPORT(fcma)
    PRINT_ISA_SUPPORT(mte)
    PRINT_ISA_SUPPORT(mte2)
    PRINT_ISA_SUPPORT(sve)
    PRINT_ISA_SUPPORT(sve2)
    PRINT_ISA_SUPPORT(svebf16)
    PRINT_ISA_SUPPORT(svei8mm)
    PRINT_ISA_SUPPORT(svef32mm)
    PRINT_ISA_SUPPORT(svef64mm)
    PRINT_ISA_SUPPORT(sme)
    PRINT_ISA_SUPPORT(smef16f16)
    PRINT_ISA_SUPPORT(smef64f64)
    PRINT_ISA_SUPPORT(smei64i64)
    PRINT_ISA_SUPPORT(pmull)
    PRINT_ISA_SUPPORT(crc32)
    PRINT_ISA_SUPPORT(aes)
    PRINT_ISA_SUPPORT(sha1)
    PRINT_ISA_SUPPORT(sha2)
    PRINT_ISA_SUPPORT(sha3)
    PRINT_ISA_SUPPORT(sha512)
    PRINT_ISA_SUPPORT(sm3)
    PRINT_ISA_SUPPORT(sm4)
    PRINT_ISA_SUPPORT(svepmull)
    PRINT_ISA_SUPPORT(svebitperm)
    PRINT_ISA_SUPPORT(sveaes)
    PRINT_ISA_SUPPORT(svesha3)
    PRINT_ISA_SUPPORT(svesm4)
    PRINT_ISA_SUPPORT(amx)

#elif __arm__ || defined(_M_ARM)
    PRINT_ISA_SUPPORT(half)
    PRINT_ISA_SUPPORT(edsp)
    PRINT_ISA_SUPPORT(neon)
    PRINT_ISA_SUPPORT(vfpv4)
    PRINT_ISA_SUPPORT(idiv)

#elif __mips__
    PRINT_ISA_SUPPORT(msa)
    PRINT_ISA_SUPPORT(mmi)
    PRINT_ISA_SUPPORT(sx)
    PRINT_ISA_SUPPORT(asx)
    PRINT_ISA_SUPPORT(msa2)
    PRINT_ISA_SUPPORT(crypto)

#elif __powerpc__
    PRINT_ISA_SUPPORT(vsx)

#elif __s390x__
    PRINT_ISA_SUPPORT(zvector)

#elif __loongarch__
    PRINT_ISA_SUPPORT(lsx)
    PRINT_ISA_SUPPORT(lasx)

#elif __riscv
    PRINT_ISA_SUPPORT(i)
    PRINT_ISA_SUPPORT(m)
    PRINT_ISA_SUPPORT(a)
    PRINT_ISA_SUPPORT(f)
    PRINT_ISA_SUPPORT(d)
    PRINT_ISA_SUPPORT(c)
    PRINT_ISA_SUPPORT(v)
    PRINT_ISA_SUPPORT(zba)
    PRINT_ISA_SUPPORT(zbb)
    PRINT_ISA_SUPPORT(zbc)
    PRINT_ISA_SUPPORT(zbs)
    PRINT_ISA_SUPPORT(zbkb)
    PRINT_ISA_SUPPORT(zbkc)
    PRINT_ISA_SUPPORT(zbkx)
    PRINT_ISA_SUPPORT(zcb)
    PRINT_ISA_SUPPORT(zfa)
    PRINT_ISA_SUPPORT(zfbfmin)
    PRINT_ISA_SUPPORT(zfh)
    PRINT_ISA_SUPPORT(zfhmin)
    PRINT_ISA_SUPPORT(zicond)
    PRINT_ISA_SUPPORT(zicsr)
    PRINT_ISA_SUPPORT(zifencei)
    PRINT_ISA_SUPPORT(zmmul)
    PRINT_ISA_SUPPORT(zvbb)
    PRINT_ISA_SUPPORT(zvbc)
    PRINT_ISA_SUPPORT(zvfh)
    PRINT_ISA_SUPPORT(zvfhmin)
    PRINT_ISA_SUPPORT(zvfbfmin)
    PRINT_ISA_SUPPORT(zvfbfwma)
    PRINT_ISA_SUPPORT(zvkb)
    PRINT_ISA_SUPPORT(zvl32b)
    PRINT_ISA_SUPPORT(zvl64b)
    PRINT_ISA_SUPPORT(zvl128b)
    PRINT_ISA_SUPPORT(zvl256b)
    PRINT_ISA_SUPPORT(zvl512b)
    PRINT_ISA_SUPPORT(zvl1024b)

    PRINT_ISA_SUPPORT(xtheadba)
    PRINT_ISA_SUPPORT(xtheadbb)
    PRINT_ISA_SUPPORT(xtheadbs)
    PRINT_ISA_SUPPORT(xtheadcondmov)
    PRINT_ISA_SUPPORT(xtheadfmemidx)
    PRINT_ISA_SUPPORT(xtheadfmv)
    PRINT_ISA_SUPPORT(xtheadmac)
    PRINT_ISA_SUPPORT(xtheadmemidx)
    PRINT_ISA_SUPPORT(xtheadmempair)
    PRINT_ISA_SUPPORT(xtheadsync)
    PRINT_ISA_SUPPORT(xtheadvdot)

    PRINT_ISA_SUPPORT(spacemitvmadot)
    PRINT_ISA_SUPPORT(spacemitvmadotn)
    PRINT_ISA_SUPPORT(spacemitvfmadot)

#elif __openrisc__
    PRINT_ISA_SUPPORT(orbis32)
    PRINT_ISA_SUPPORT(orbis64)
    PRINT_ISA_SUPPORT(orfpx32)
    PRINT_ISA_SUPPORT(orfpx64)
    PRINT_ISA_SUPPORT(orvdx64)

#endif

    const char* const* supported = ruapu_rua();
    while (*supported)
    {
        printf("%s\n", *supported);
        supported++;
    }

    return 0;
}