如何为Eigen向量的vop5a1函数启用SIMD优化？

问题背景 #

使用GCC 11、-O2优化级别编译以下x86/64架构的C++代码时，vop5a1函数未生成SIMD指令，而vop5a2（带__restrict__的指针版本）和vop5c（手动循环带重叠检测）均能生成SIMD或自适应代码：

static constexpr int N = 4;
typedef std::complex<float> cfloat;
typedef Eigen::Vector<float, N> VectorNf;
typedef Eigen::Vector<cfloat, N> VectorNz;

void vop5a1 (VectorNf & a, const VectorNz & b)
{
    a = b.array().abs2();       // 未生成SIMD指令
}
void vop5a2 (VectorNf * __restrict__ a, const VectorNz * __restrict__ b)
{
    *a = b->array().abs2();     // 加restrict时生成SIMD指令
}
void vop5c (VectorNf & a, const VectorNz & b)
{
    for (int i = 0; i < N; i++)
    {  // 检测无重叠则生成SIMD，否则生成标量x86代码
       a[i] = std::norm(b[i]);
    }
}

原因分析 #

vop5a1未生成SIMD的核心原因是：编译器无法通过普通引用参数确定a和b指向的内存区域是否存在重叠。为了避免内存访问冲突（比如写入a时覆盖b的未读取数据），编译器只能放弃SIMD优化，退化为标量代码。

而vop5a2的__restrict__指针明确告知编译器：两个指针指向的内存区域绝对无重叠，编译器可以安全地使用SIMD指令进行批量内存操作；vop5c的手动循环则触发了GCC的自动重叠检测逻辑，会在运行时判断内存是否重叠，再选择SIMD或标量路径。

解决方案 #

给vop5a1的引用参数添加__restrict__修饰，明确告知编译器a和b的内存区域无重叠，即可启用SIMD优化：

void vop5a1 (VectorNf & __restrict__ a, const VectorNz & __restrict__ b)
{
    a = b.array().abs2();       // 现在会生成SIMD指令
}

如果需要更好的跨编译器可移植性，也可以使用Eigen库提供的EIGEN_RESTRICT宏（该宏会根据不同编译器自动映射到对应的restrict关键字，比如GCC的__restrict__、MSVC的__restrict）：

void vop5a1 (VectorNf & EIGEN_RESTRICT a, const VectorNz & EIGEN_RESTRICT b)
{
    a = b.array().abs2();       // 跨编译器兼容的SIMD优化版本
}

内容的提问来源于stack exchange，提问作者Adhémar Patamob