C++利用cpuid实现硬件特性检测

在C++中，我们可以使用__cpuid和__cpuidex内部函数来实现硬件特性检测

#include <iostream>#include <bitset>#include <vector>#include <string>#include <intrin.h> // 包含CPUID相关的头文件// 定义一个结构体来存储CPU信息struct CPUInfo {    int nIds;    int nExIds;    std::string vendor;    std::string brand;    bool isIntel;    bool isAMD;    std::bitset<32> f_1_ECX;    std::bitset<32> f_1_EDX;    std::bitset<32> f_7_EBX;    std::bitset<32> f_7_ECX;    std::bitset<32> f_81_ECX;    std::bitset<32> f_81_EDX;};// 获取CPU信息的函数void getCPUInfo(CPUInfo& cpuinfo) {    std::vector<std::array<int, 4>> data;    std::array<int, 4> cpui;    // 调用__cpuid获取基本信息    __cpuid(cpui.data(), 0);    cpuinfo.nIds = cpui[0];    for (int i = 0; i <= cpuinfo.nIds; ++i) {        __cpuid(cpui.data(), i);        data.push_back(cpui);    }    // 获取CPU厂商信息    char vendor[0x20];    mEMSet(vendor, 0, sizeof(vendor));    *reinterpret_cast<int*>(vendor) = data[0][1];    *reinterpret_cast<int*>(vendor + 4) = data[0][3];    *reinterpret_cast<int*>(vendor + 8) = data[0][2];    cpuinfo.vendor = vendor;    cpuinfo.isIntel = (cpuinfo.vendor == "GenuineIntel");    cpuinfo.isAMD = (cpuinfo.vendor == "AuthenticAMD");    // 获取扩展信息    __cpuid(cpui.data(), 0x80000000);    cpuinfo.nExIds = cpui[0];    for (int i = 0x80000000; i <= cpuinfo.nExIds; ++i) {        __cpuid(cpui.data(), i);        data.push_back(cpui);    }    // 获取CPU品牌信息    char brand[0x40];    memset(brand, 0, sizeof(brand));    memcpy(brand, data[2].data(), sizeof(cpui));    memcpy(brand + 16, data[3].data(), sizeof(cpui));    memcpy(brand + 32, data[4].data(), sizeof(cpui));    cpuinfo.brand = brand;    // 获取功能位信息    cpuinfo.f_1_ECX = data[1][2];    cpuinfo.f_1_EDX = data[1][3];    cpuinfo.f_7_EBX = data[7][1];    cpuinfo.f_7_ECX = data[7][2];    cpuinfo.f_81_ECX = data[8][2];    cpuinfo.f_81_EDX = data[8][3];}int main() {    CPUInfo cpuinfo;    getCPUInfo(cpuinfo);    std::cout << "CPU Vendor: " << cpuinfo.vendor << std::endl;    std::cout << "CPU Brand: " << cpuinfo.brand << std::endl;    // 检测SSE4.1支持    if (cpuinfo.f_1_ECX[19]) {        std::cout << "SSE4.1 supported" << std::endl;    } else {        std::cout << "SSE4.1 not supported" << std::endl;    }    // 检测AVX支持    if (cpuinfo.f_1_ECX[28]) {        std::cout << "AVX supported" << std::endl;    } else {        std::cout << "AVX not supported" << std::endl;    }    return 0;}

这个示例程序首先获取CPU的基本信息，然后根据获取到的信息判断CPU是否支持特定的硬件特性。在这个例子中，我们检查了SSE4.1和AVX的支持情况。你可以根据需要检查其他硬件特性。