cpu_id.hpp

 
#include "TTauri/Foundation/cpu_id.hpp"
#include "TTauri/Foundation/os_detect.hpp"
#include <array>
 
#if TT_COMPILER == TT_CC_MSVC
#include <intrin.h>
#elif TT_COMPILER == TT_CC_GCC || TT_COMPILER == TT_CC_CLANG
#include <cpuid.h>
#else
#error "Unsuported compiler for x64 cpu_id"
#endif
 
namespace tt {
 
#if TT_COMPILER == TT_CC_MSVC
std::array<uint32_t,4> cpu_id_x64_result cpu_id_x64(uint32_t cpu_id_leaf)
{
    std::array<int,4> info;
    __cpuid(info.data(), static_cast<int>(cpu_id_leaf));
 
    std::array<uint32_t,4> r;
    r[0] = static_cast<uint32_t>info[0];
    r[1] = static_cast<uint32_t>info[1];
    r[2] = static_cast<uint32_t>info[2];
    r[3] = static_cast<uint32_t>info[3];
    return r;
}
 
#elif TT_COMPILER == TT_CC_GCC || TT_COMPILER == TT_CC_CLANG
std::array<uint32_t,4> cpu_id_x64_result cpu_id_x64(uint32_t cpu_id_leaf)
{
    std::array<uint32_t,4> r;
    __cpuid(cpu_id_leaf, r[0], r[1], r[2], r[3]);
    return r;
}
 
#else
#error "Unsuported compiler for x64 cpu_id"
#endif
 
inline std::array<uint32_t,4> cpu_id_leaf1 = cpu_id_x64(1);
inline std::array<uint32_t,4> cpu_id_leaf7 = cpu_id_x64(7);
 
template<int Bit>
tt_force_inline bool cpu_id_leaf1_ecx() {
    constexpr uint32_t mask = 1 << Bit;
    return (cpu_id_leaf1[2] & mask) != 0;
}
 
template<int Bit>
tt_force_inline bool cpu_id_leaf1_edx() {
    constexpr uint32_t mask = 1 << Bit;
    return (cpu_id_leaf1[3] & mask) != 0;
}
 
template<int Bit>
tt_force_inline bool cpu_id_leaf7_ebx() {
    constexpr uint32_t mask = 1 << Bit;
    return (cpu_id_leaf7[1] & mask) != 0;
}
 
template<int Bit>
tt_force_inline bool cpu_id_leaf7_ecx() {
    constexpr uint32_t mask = 1 << Bit;
    return (cpu_id_leaf7[2] & mask) != 0;
}
 
template<int Bit>
tt_force_inline bool cpu_id_leaf7_edx() {
    constexpr uint32_t mask = 1 << Bit;
    return (cpu_id_leaf7[3] & mask) != 0;
}
 
// LEAF1.0: EDX
tt_force_inline bool cpu_has_fpu() { return cpu_id_leaf1_edx<0>(); }
tt_force_inline bool cpu_has_vme() { return cpu_id_leaf1_edx<1>(); }
tt_force_inline bool cpu_has_de() { return cpu_id_leaf1_edx<2>(); }
tt_force_inline bool cpu_has_pse() { return cpu_id_leaf1_edx<3>(); }
tt_force_inline bool cpu_has_tsc() { return cpu_id_leaf1_edx<4>(); }
tt_force_inline bool cpu_has_msr() { return cpu_id_leaf1_edx<5>(); }
tt_force_inline bool cpu_has_pae() { return cpu_id_leaf1_edx<6>(); }
tt_force_inline bool cpu_has_mce() { return cpu_id_leaf1_edx<7>(); }
tt_force_inline bool cpu_has_cx8() { return cpu_id_leaf1_edx<8>(); }
tt_force_inline bool cpu_has_apic() { return cpu_id_leaf1_edx<9>(); }
// reserved
tt_force_inline bool cpu_has_sep() { return cpu_id_leaf1_edx<11>(); }
tt_force_inline bool cpu_has_mtrr() { return cpu_id_leaf1_edx<12>(); }
tt_force_inline bool cpu_has_pge() { return cpu_id_leaf1_edx<13>(); }
tt_force_inline bool cpu_has_mca() { return cpu_id_leaf1_edx<14>(); }
tt_force_inline bool cpu_has_cmov() { return cpu_id_leaf1_edx<15>(); }
tt_force_inline bool cpu_has_pat() { return cpu_id_leaf1_edx<16>(); }
tt_force_inline bool cpu_has_pse_36() { return cpu_id_leaf1_edx<17>(); }
tt_force_inline bool cpu_has_psn() { return cpu_id_leaf1_edx<18>(); }
tt_force_inline bool cpu_has_clfsh() { return cpu_id_leaf1_edx<19>(); }
// reserved
tt_force_inline bool cpu_has_ds() { return cpu_id_leaf1_edx<21>(); }
tt_force_inline bool cpu_has_acpi() { return cpu_id_leaf1_edx<22>(); }
tt_force_inline bool cpu_has_mmx() { return cpu_id_leaf1_edx<23>(); }
tt_force_inline bool cpu_has_fxsr() { return cpu_id_leaf1_edx<24>(); }
tt_force_inline bool cpu_has_sse() { return cpu_id_leaf1_edx<25>(); }
tt_force_inline bool cpu_has_sse2() { return cpu_id_leaf1_edx<26>(); }
tt_force_inline bool cpu_has_ss() { return cpu_id_leaf1_edx<27>(); }
tt_force_inline bool cpu_has_htt() { return cpu_id_leaf1_edx<28>(); }
tt_force_inline bool cpu_has_tm() { return cpu_id_leaf1_edx<29>(); }
tt_force_inline bool cpu_has_ia64() { return cpu_id_leaf1_edx<30>(); }
tt_force_inline bool cpu_has_pbe() { return cpu_id_leaf1_edx<31>(); }
 
// LEAF1.0: ECX
tt_force_inline bool cpu_has_sse3() { return cpu_id_leaf1_ecx<0>(); }
tt_force_inline bool cpu_has_pclmulqdq() { return cpu_id_leaf1_ecx<1>(); }
tt_force_inline bool cpu_has_dtes64() { return cpu_id_leaf1_ecx<2>(); }
tt_force_inline bool cpu_has_monitor() { return cpu_id_leaf1_ecx<3>(); }
tt_force_inline bool cpu_has_ds_cpl() { return cpu_id_leaf1_ecx<4>(); }
tt_force_inline bool cpu_has_vmx() { return cpu_id_leaf1_ecx<5>(); }
tt_force_inline bool cpu_has_smx() { return cpu_id_leaf1_ecx<6>(); }
tt_force_inline bool cpu_has_est() { return cpu_id_leaf1_ecx<7>(); }
tt_force_inline bool cpu_has_tm2() { return cpu_id_leaf1_ecx<8>(); }
tt_force_inline bool cpu_has_ssse3() { return cpu_id_leaf1_ecx<9>(); }
tt_force_inline bool cpu_has_cnxt_id() { return cpu_id_leaf1_ecx<10>(); }
tt_force_inline bool cpu_has_sdbg() { return cpu_id_leaf1_ecx<11>(); }
tt_force_inline bool cpu_has_fma() { return cpu_id_leaf1_ecx<12>(); }
tt_force_inline bool cpu_has_cx16() { return cpu_id_leaf1_ecx<13>(); }
tt_force_inline bool cpu_has_xtpr() { return cpu_id_leaf1_ecx<14>(); }
tt_force_inline bool cpu_has_pdcm() { return cpu_id_leaf1_ecx<15>(); }
// reserved
tt_force_inline bool cpu_has_pcid() { return cpu_id_leaf1_ecx<17>(); }
tt_force_inline bool cpu_has_dca() { return cpu_id_leaf1_ecx<18>(); }
tt_force_inline bool cpu_has_sse4_1() { return cpu_id_leaf1_ecx<19>(); }
tt_force_inline bool cpu_has_sse4_2() { return cpu_id_leaf1_ecx<20>(); }
tt_force_inline bool cpu_has_x2apic() { return cpu_id_leaf1_ecx<21>(); }
tt_force_inline bool cpu_has_movbe() { return cpu_id_leaf1_ecx<22>(); }
tt_force_inline bool cpu_has_popcnt() { return cpu_id_leaf1_ecx<23>(); }
tt_force_inline bool cpu_has_tsc_deadline() { return cpu_id_leaf1_ecx<24>(); }
tt_force_inline bool cpu_has_aes() { return cpu_id_leaf1_ecx<25>(); }
tt_force_inline bool cpu_has_xsave() { return cpu_id_leaf1_ecx<26>(); }
tt_force_inline bool cpu_has_osxsave() { return cpu_id_leaf1_ecx<27>(); }
tt_force_inline bool cpu_has_avx() { return cpu_id_leaf1_ecx<28>(); }
tt_force_inline bool cpu_has_f16c() { return cpu_id_leaf1_ecx<29>(); }
tt_force_inline bool cpu_has_rdrnd() { return cpu_id_leaf1_ecx<30>(); }
tt_force_inline bool cpu_has_hypervisor() { return cpu_id_leaf1_ecx<31>(); }
 
// LEAF1.0: EBX
 
 
// LEAF1.0: EAX
tt_force_inline bool cpu_stepping() { return cpu_id_leaf1[0] & 0xf; }
tt_force_inline bool cpu_model_id() {
    uint32_t family_id = (cpu_id_leaf1[0] >> 8) & 0xf;
    uint32_t model_id = (cpu_id_leaf1[0] >> 4) & 0xf;
    if (family_id == 6 || family_id == 15) {
        uint32_t extended_model_id = (cpu_id_leaf1[0] >> 16) & 0xf;
        return (extended_model_id << 4) | model_id;
    } else {
        return model_id;
    }
}
tt_force_inline bool cpu_family_id() {
    uint32_t family_id = (cpu_id_leaf1[0] >> 8) & 0xf;
    if (family_id == 15) {
        uint32_t extended_family_id = (cpu_id_leaf1[0] >> 20) & 0xff;
        return family_id + extended_family_id;
    } else {
        return extended_family_id;
    }
}
 
// LEAF7.0: EBX
tt_force_inline bool cpu_has_fsgsbase() { return cpu_id_leaf7_ebx<0>(); }
tt_force_inline bool cpu_has_tsc_adjust() { return cpu_id_leaf7_ebx<1>(); }
tt_force_inline bool cpu_has_sgx() { return cpu_id_leaf7_ebx<2>(); }
tt_force_inline bool cpu_has_bmi1() { return cpu_id_leaf7_ebx<3>(); }
tt_force_inline bool cpu_has_hle() { return cpu_id_leaf7_ebx<4>(); }
tt_force_inline bool cpu_has_avx2() { return cpu_id_leaf7_ebx<5>(); }
// reserved
tt_force_inline bool cpu_has_smep() { return cpu_id_leaf7_ebx<7>(); }
tt_force_inline bool cpu_has_bmi2() { return cpu_id_leaf7_ebx<8>(); }
tt_force_inline bool cpu_has_erms() { return cpu_id_leaf7_ebx<9>(); }
tt_force_inline bool cpu_has_invpcid() { return cpu_id_leaf7_ebx<10>(); }
tt_force_inline bool cpu_has_rtm() { return cpu_id_leaf7_ebx<11>(); }
tt_force_inline bool cpu_has_pqm() { return cpu_id_leaf7_ebx<12>(); }
tt_force_inline bool cpu_has_deprecated_fpu_cs_ds() { return cpu_id_leaf7_ebx<13>(); }
tt_force_inline bool cpu_has_mpx() { return cpu_id_leaf7_ebx<14>(); }
tt_force_inline bool cpu_has_pqe() { return cpu_id_leaf7_ebx<15>(); }
tt_force_inline bool cpu_has_avx512_f() { return cpu_id_leaf7_ebx<16>(); }
tt_force_inline bool cpu_has_avx512_dq() { return cpu_id_leaf7_ebx<17>(); }
tt_force_inline bool cpu_has_rdseed() { return cpu_id_leaf7_ebx<18>(); }
tt_force_inline bool cpu_has_adx() { return cpu_id_leaf7_ebx<19>(); }
tt_force_inline bool cpu_has_smap() { return cpu_id_leaf7_ebx<20>(); }
tt_force_inline bool cpu_has_avx512_ifma() { return cpu_id_leaf7_ebx<21>(); }
tt_force_inline bool cpu_has_pcommit() { return cpu_id_leaf7_ebx<22>(); }
tt_force_inline bool cpu_has_clflushopt() { return cpu_id_leaf7_ebx<23>(); }
tt_force_inline bool cpu_has_clwb() { return cpu_id_leaf7_ebx<24>(); }
tt_force_inline bool cpu_has_intelpt() { return cpu_id_leaf7_ebx<25>(); }
tt_force_inline bool cpu_has_avx512_pf() { return cpu_id_leaf7_ebx<26>(); }
tt_force_inline bool cpu_has_avx512_er() { return cpu_id_leaf7_ebx<27>(); }
tt_force_inline bool cpu_has_avx512_cd() { return cpu_id_leaf7_ebx<28>(); }
tt_force_inline bool cpu_has_sha() { return cpu_id_leaf7_ebx<29>(); }
tt_force_inline bool cpu_has_avx512_bw() { return cpu_id_leaf7_ebx<30>(); }
tt_force_inline bool cpu_has_avx512_vl() { return cpu_id_leaf7_ebx<31>(); }
 
 
 
 
// LEAF7.0: ECX
tt_force_inline bool cpu_has_prefetchwt1() { return cpu_id_leaf7_ecx<0>(); }
tt_force_inline bool cpu_has_avx512_vbmi() { return cpu_id_leaf7_ecx<1>(); }
tt_force_inline bool cpu_has_umip() { return cpu_id_leaf7_ecx<2>(); }
tt_force_inline bool cpu_has_pku() { return cpu_id_leaf7_ecx<3>(); }
tt_force_inline bool cpu_has_ospke() { return cpu_id_leaf7_ecx<4>(); }
tt_force_inline bool cpu_has_waitpkg() { return cpu_id_leaf7_ecx<5>(); }
tt_force_inline bool cpu_has_avx512_vmbi2() { return cpu_id_leaf7_ecx<6>(); }
tt_force_inline bool cpu_has_shstk() { return cpu_id_leaf7_ecx<7>(); }
tt_force_inline bool cpu_has_gfni() { return cpu_id_leaf7_ecx<8>(); }
tt_force_inline bool cpu_has_vaes() { return cpu_id_leaf7_ecx<9>(); }
tt_force_inline bool cpu_has_vpclmulqdq() { return cpu_id_leaf7_ecx<10>(); }
tt_force_inline bool cpu_has_avx512_vnni() { return cpu_id_leaf7_ecx<11>(); }
tt_force_inline bool cpu_has_avx512_bitalg() { return cpu_id_leaf7_ecx<12>(); }
// reserved
tt_force_inline bool cpu_has_avx512_vpopcntdq() { return cpu_id_leaf7_ecx<14>(); }
// reserved
tt_force_inline bool cpu_has_5level_paging() { return cpu_id_leaf7_ecx<16>(); }
tt_force_inline int cpu_has_mawau() { }
tt_force_inline bool cpu_has_rdpid() { return cpu_id_leaf7_ecx<22>(); }
// reserved
// reserved
tt_force_inline bool cpu_has_cldemote() { return cpu_id_leaf7_ecx<25>(); }
// reserved
tt_force_inline bool cpu_has_movdir() { return cpu_id_leaf7_ecx<27>(); }
tt_force_inline bool cpu_has_movdir64b() { return cpu_id_leaf7_ecx<28>(); }
// reserved
tt_force_inline bool cpu_has_sgx_lc() { return cpu_id_leaf7_ecx<30>(); }
// reserved
 
// LEAF7.0: EDX
// reserved
// reserved
tt_force_inline bool cpu_has_avx512_4vnniw() { return cpu_id_leaf7_edx<2>(); }
tt_force_inline bool cpu_has_avx512_4fmaps() { return cpu_id_leaf7_edx<3>(); }
tt_force_inline bool cpu_has_fsrm() { return cpu_id_leaf7_edx<4>(); }
tt_force_inline bool cpu_has_pconfig() { return cpu_id_leaf7_edx<18>(); }
// reserved
tt_force_inline bool cpu_has_ibt() { return cpu_id_leaf7_edx<20>(); }
// reserved 5
tt_force_inline bool cpu_has_spec_ctrl() { return cpu_id_leaf7_edx<26>(); }
tt_force_inline bool cpu_has_stibp() { return cpu_id_leaf7_edx<27>(); }
// reserved
tt_force_inline bool cpu_has_capabilities() { return cpu_id_leaf7_edx<29>(); }
// reserved
tt_force_inline bool cpu_has_ssbd() { return cpu_id_leaf7_edx<31>(); }
}