如何让OpenMP生成AVX2指令？AVX512对应的编译选项是什么？

我编写了如下求和函数：

编译命令：

g++ -g -O3 -march=core-avx2 sum.cpp

C++代码：

#include <omp.h>

float sum(const float* a, int n) {
    float total_sum = 0.0f;

#pragma omp simd reduction(+:total_sum)
    for (int i = 0; i < n; ++i) {
        total_sum += a[i];
    }

    return total_sum;
}

编译后生成的代码使用xmm寄存器而非ymm寄存器，请问如何让其生成使用ymm寄存器的AVX2代码？另外，生成AVX512代码的编译选项是什么？

一、生成使用YMM寄存器的AVX2代码 #

出现仅使用XMM寄存器的情况，通常是编译器未充分利用AVX2的256位向量能力，或代码存在限制导致优化未生效，可尝试以下方法：

• 显式指定向量宽度并确保OpenMP SIMD生效：
  编译时在-march=core-avx2基础上，添加-fopenmp-simd确保OpenMP SIMD特性启用，同时用-mprefer-vector-width=256强制编译器优先使用256位向量（对应YMM寄存器）：

  g++ -g -O3 -march=core-avx2 -fopenmp-simd -mprefer-vector-width=256 sum.cpp
  

• 优化数据对齐与指针属性：
  如果数组a未按32字节（256位浮点数向量的对齐要求）对齐，编译器可能退回到128位向量。可在代码中显式声明对齐内存：

  // 示例：分配32字节对齐的内存
  float* a = (float*)aligned_alloc(32, n * sizeof(float));
  

  或在函数参数中添加对齐与__restrict__属性，消除指针别名疑虑以帮助编译器优化：

  float sum(const float* __restrict__ __attribute__((aligned(32))) a, int n)
  

• 升级编译器版本：
  旧版GCC对OpenMP SIMD的AVX2优化支持不完善，升级到GCC 8及以上版本通常能更好地生成256位向量代码。

二、生成AVX512代码的编译选项 #

需根据目标CPU支持的AVX512子集选择对应选项：

• 针对全AVX512支持的CPU（如Skylake-X、Cascade Lake）：
  使用-march=skylake-avx512，该选项会启用AVX512F、AVX512VL、AVX512BW、AVX512DQ等常用子集：

  g++ -g -O3 -march=skylake-avx512 -fopenmp-simd sum.cpp
  

• 针对仅支持基础AVX512子集的CPU：
  若仅需AVX512基础指令集（AVX512F），可使用-mavx512f，但更推荐使用对应CPU架构的-march选项以启用全部支持的指令：

  g++ -g -O3 -mavx512f -fopenmp-simd sum.cpp
  

• 强制使用512位向量宽度：
  类似AVX2的处理，添加-mprefer-vector-width=512确保编译器优先使用512位向量（对应ZMM寄存器）：

  g++ -g -O3 -march=skylake-avx512 -fopenmp-simd -mprefer-vector-width=512 sum.cpp
  

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