如何在Windows 10 64位系统中编程检测CPU是否支持FMA指令？

在Windows 10 64位环境下，你可以通过__cpuid intrinsic来检测CPU是否支持FMA3指令集，完美实现你的动态分支需求。下面是具体的实现方案：

1. 实现CheckFMA()函数 #

Windows下MSVC编译器提供了__cpuid intrinsic，可以直接读取CPU的功能特性信息。FMA3支持与否可以通过查询CPUID leaf 7的EBX寄存器第16位来判断：

#include <intrin.h>
#include <stdbool.h>

bool CheckFMA() {
    int cpuidInfo[4];
    
    // 先查询CPU支持的最大CPUID leaf
    __cpuid(cpuidInfo, 0);
    int maxSupportedLeaf = cpuidInfo[0];
    
    // 如果最大leaf小于7，说明不支持扩展功能查询，肯定没有FMA
    if (maxSupportedLeaf < 7) {
        return false;
    }
    
    // 查询leaf 7的功能信息
    __cpuid(cpuidInfo, 7);
    // EBX寄存器的第16位对应FMA3支持位
    return (cpuidInfo[1] & (1 << 16)) != 0;
}

2. 结合你的业务代码使用 #

因为你提到分支会放在函数外部，只需检测一次，所以可以在程序初始化阶段就获取FMA支持状态，后续直接使用该标志即可，避免重复检测带来的性能损耗：

#include <immintrin.h>

// 全局变量保存FMA支持状态，初始化时检测一次
static bool g_isFmaEnabled = CheckFMA();

__m256d ComputeMultiplyAdd(__m256d a, __m256d b, __m256d c) {
    if (g_isFmaEnabled) {
        return _mm256_fmadd_pd(a, b, c);
    } else {
        __m256d product = _mm256_mul_pd(a, b);
        return _mm256_add_pd(product, c);
    }
}

// 示例调用
void ProcessData() {
    __m256d a = _mm256_set_pd(1.0, 2.0, 3.0, 4.0);
    __m256d b = _mm256_set_pd(5.0, 6.0, 7.0, 8.0);
    __m256d c = _mm256_set_pd(9.0, 10.0, 11.0, 12.0);
    
    __m256d result = ComputeMultiplyAdd(a, b, c);
    // 处理结果...
}

关键注意事项 #

• 编译设置：MSVC编译时需要启用/arch:AVX2选项，这样编译器才允许生成_mm256_fmadd_pd对应的FMA指令。虽然启用了AVX2编译选项，但因为我们做了运行时检测，二进制仍然可以在不支持FMA的CPU上正常运行（只会走普通乘加的代码路径）。
• 兼容性：这个检测逻辑同时适用于Intel和AMD的现代CPU，两者都是用CPUID leaf7 EBX的第16位标识FMA3支持。
• 性能：由于只在初始化阶段做一次检测，后续调用不需要重复判断分支，完全不会因为检测逻辑带来性能损失。

内容的提问来源于stack exchange，提问作者Sylvestre