如何在C语言编译时检测CPU架构的AVX2/AVX/ARM指令集？

方案1：编译期预编译宏检测（最常用） #

这是你提到的#ifdef方案，主流编译器都会根据当前开启的编译架构、指令集选项预定义对应宏，你可以直接通过这些宏做编译期分支选择，对应宏定义如下：

• x86/x86_64架构相关宏
  • 架构识别：__x86_64__（GCC/Clang）、_M_X64（MSVC）
  • AVX2指令集：__AVX2__（GCC/Clang）、__AVX2（MSVC，开启/arch:AVX2编译选项时生效）
  • AVX512基础指令集：__AVX512F__，常用扩展集对应宏还有__AVX512BW__、__AVX512DQ__、__AVX512VL__，MSVC开启/arch:AVX512编译选项后会定义对应宏
• ARM架构相关宏
  • 32位ARM NEON支持：__ARM_NEON__
  • AArch64（ARM 64位）架构：__aarch64__（GCC/Clang）、_M_ARM64（MSVC），AArch64架构默认强制支持NEON，识别到架构即可直接使用NEON指令

参考使用示例：

#if defined(__x86_64__) || defined(_M_X64)
    #if defined(__AVX512F__)
        #define SIMD_IMPL avx512_func
    #elif defined(__AVX2__)
        #define SIMD_IMPL avx2_func
    #else
        #define SIMD_IMPL scalar_func
    #endif
#elif defined(__aarch64__) || defined(_M_ARM64) || defined(__ARM_NEON__)
    #define SIMD_IMPL neon_func
#else
    #define SIMD_IMPL scalar_func
#endif

// 业务逻辑中直接调用SIMD_IMPL即可

该方案优势是完全无运行时开销，编译阶段就确定了最终使用的实现，适合在目标运行设备架构明确的场景下使用；缺点是编译出的二进制只能适配当前编译选项对应的指令集，无法在更低指令集的设备上运行。

方案2：运行时动态检测（兼容性更强） #

如果需要编译出的单份二进制可以在所有同架构设备上运行，自动选择当前设备支持的最优SIMD实现，可以选择运行时检测方案：

• x86平台可以通过cpuid指令直接读取CPU支持的指令集特性，Windows下也可以调用IsProcessorFeaturePresent系统API，Linux下可以读取/proc/cpuinfo文件获取特性列表
• ARM平台可以通过读取/proc/cpuinfo或者调用系统提供的CPU特性检测接口判断NEON支持情况

该方案优势是单份二进制兼容性极强，只需要在程序启动时检测一次特性，后续直接调用对应实现即可，检测开销几乎可以忽略；缺点是需要为不同指令集的实现单独编译成独立编译单元，避免编译选项冲突。

注意事项 #

• 不同编译器的预定义宏存在少量差异，需要注意覆盖GCC、Clang、MSVC三大主流编译器的情况
• 使用对应SIMD指令集时需要引入对应头文件：AVX/AVX2/AVX512引入<immintrin.h>，NEON引入<arm_neon.h>

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