Merge pull request #612 from Wunkolo/cpuinfo-concise
cpu_info: Add CPU Name, Family, compact output
This commit is contained in:
commit
fff4d9a4c7
1 changed files with 68 additions and 51 deletions
|
@ -7,76 +7,93 @@
|
|||
#include <utility>
|
||||
|
||||
#include <catch.hpp>
|
||||
#include <fmt/format.h>
|
||||
#include <xbyak_util.h>
|
||||
|
||||
TEST_CASE("Host CPU supports", "[a64]") {
|
||||
Xbyak::util::Cpu cpu_info;
|
||||
|
||||
std::array<uint32_t, 4> cpu_name;
|
||||
for (uint32_t i = 2; i < 5; ++i) {
|
||||
cpu_info.getCpuid(0x80000000 | i, cpu_name.data());
|
||||
std::printf("%.16s", reinterpret_cast<const char*>(cpu_name.data()));
|
||||
}
|
||||
std::putchar('\n');
|
||||
|
||||
cpu_info.putFamily();
|
||||
static constexpr std::array types{
|
||||
std::make_pair(Xbyak::util::Cpu::t3DN, "3DN"),
|
||||
std::make_pair(Xbyak::util::Cpu::tADX, "ADX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAESNI, "AESNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAMD, "AMD"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX, "AVX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX2, "AVX2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_4FMAPS, "AVX5124FMAPS"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_4VNNIW, "AVX5124VNNIW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_BITALG, "AVX512BITALG"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VBMI2, "AVX512VBMI2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VNNI, "AVX512VNNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VPOPCNTDQ, "AVX512VPOPCNTDQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512BW, "AVX512BW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512CD, "AVX512CD"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512DQ, "AVX512DQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512ER, "AVX512ER"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512F, "AVX512F"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512IFMA, "AVX512IFMA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512PF, "AVX512PF"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512VBMI, "AVX512VBMI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512VL, "AVX512VL"),
|
||||
std::make_pair(Xbyak::util::Cpu::tBMI1, "BMI1"),
|
||||
std::make_pair(Xbyak::util::Cpu::tBMI2, "BMI2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tCMOV, "CMOV"),
|
||||
std::make_pair(Xbyak::util::Cpu::tE3DN, "E3DN"),
|
||||
std::make_pair(Xbyak::util::Cpu::tENHANCED_REP, "ENHANCED_REP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tF16C, "F16C"),
|
||||
std::make_pair(Xbyak::util::Cpu::tFMA, "FMA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tGFNI, "GFNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tHLE, "HLE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tINTEL, "INTEL"),
|
||||
std::make_pair(Xbyak::util::Cpu::tLZCNT, "LZCNT"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMMX, "MMX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMMX2, "MMX2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tCMOV, "CMOV"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMOVBE, "MOVBE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMPX, "MPX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tOSXSAVE, "OSXSAVE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPCLMULQDQ, "PCLMULQDQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPOPCNT, "POPCNT"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPREFETCHW, "PREFETCHW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPREFETCHWT1, "PREFETCHWT1"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDRAND, "RDRAND"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDSEED, "RDSEED"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDTSCP, "RDTSCP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRTM, "RTM"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSHA, "SHA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSMAP, "SMAP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE, "SSE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE2, "SSE2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE3, "SSE3"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSSE3, "SSSE3"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE41, "SSE41"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE42, "SSE42"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPOPCNT, "POPCNT"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAESNI, "AESNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE5, "SSE5"),
|
||||
std::make_pair(Xbyak::util::Cpu::tOSXSAVE, "OSXSAVE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPCLMULQDQ, "PCLMULQDQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX, "AVX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tFMA, "FMA"),
|
||||
std::make_pair(Xbyak::util::Cpu::t3DN, "3DN"),
|
||||
std::make_pair(Xbyak::util::Cpu::tE3DN, "E3DN"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE4a, "SSE4a"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDTSCP, "RDTSCP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX2, "AVX2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tBMI1, "BMI1"),
|
||||
std::make_pair(Xbyak::util::Cpu::tBMI2, "BMI2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tLZCNT, "LZCNT"),
|
||||
std::make_pair(Xbyak::util::Cpu::tINTEL, "INTEL"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAMD, "AMD"),
|
||||
std::make_pair(Xbyak::util::Cpu::tENHANCED_REP, "ENHANCED_REP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDRAND, "RDRAND"),
|
||||
std::make_pair(Xbyak::util::Cpu::tADX, "ADX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRDSEED, "RDSEED"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSMAP, "SMAP"),
|
||||
std::make_pair(Xbyak::util::Cpu::tHLE, "HLE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tRTM, "RTM"),
|
||||
std::make_pair(Xbyak::util::Cpu::tF16C, "F16C"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMOVBE, "MOVBE"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512F, "AVX512F"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512DQ, "AVX512DQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_IFMA, "AVX512_IFMA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512IFMA, "AVX512IFMA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512PF, "AVX512PF"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512ER, "AVX512ER"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512CD, "AVX512CD"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512BW, "AVX512BW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512VL, "AVX512VL"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VBMI, "AVX512_VBMI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512VBMI, "AVX512VBMI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_4VNNIW, "AVX512_4VNNIW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_4FMAPS, "AVX512_4FMAPS"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPREFETCHWT1, "PREFETCHWT1"),
|
||||
std::make_pair(Xbyak::util::Cpu::tPREFETCHW, "PREFETCHW"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSHA, "SHA"),
|
||||
std::make_pair(Xbyak::util::Cpu::tMPX, "MPX"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VBMI2, "AVX512_VBMI2"),
|
||||
std::make_pair(Xbyak::util::Cpu::tGFNI, "GFNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSE5, "SSE5"),
|
||||
std::make_pair(Xbyak::util::Cpu::tSSSE3, "SSSE3"),
|
||||
std::make_pair(Xbyak::util::Cpu::tVAES, "VAES"),
|
||||
std::make_pair(Xbyak::util::Cpu::tVPCLMULQDQ, "VPCLMULQDQ"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VNNI, "AVX512_VNNI"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_BITALG, "AVX512_BITALG"),
|
||||
std::make_pair(Xbyak::util::Cpu::tAVX512_VPOPCNTDQ, "AVX512_VPOPCNTDQ"),
|
||||
};
|
||||
|
||||
constexpr size_t line_max = 80;
|
||||
size_t line_length = 0;
|
||||
for (const auto& [type, name] : types) {
|
||||
if (cpu_info.has(type)) {
|
||||
fmt::print("CPU has {}\n", name);
|
||||
const size_t name_length = std::strlen(name) + 1;
|
||||
if ((line_length + name_length) >= line_max) {
|
||||
line_length = name_length;
|
||||
std::putchar('\n');
|
||||
} else if (line_length) {
|
||||
std::putchar(' ');
|
||||
}
|
||||
std::fputs(name, stdout);
|
||||
line_length += name_length;
|
||||
}
|
||||
}
|
||||
std::putchar('\n');
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue