如何编写编译器可以针对 SIMD 比较优化的代码？ [复制]

Question

std::array<int, 4> a = {1, 1, 1, 1};
std::array<int, 4> b = { 1, 2, 3, 4 };
std::array<int, 4> c;
bool res = false;
for (int i = 0; i < a.size(); i++) {
    a[i] = rand() % 10;
}

for (int i = 0; i < 4; i++) {
    c[i] = a[i] + b[i];
}

智能编译器可以很好地编译成SIMD。 但是如何编写下面的比较代码也可以很好地编译为 SIMD；

res = a[0] <= b[0] && a[1] <= b[1] && a[2] <= b[2] && a[3] <= b[3]; // not compile to SIMD

Answer 1

这样的事情怎么样：

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

?

如果您可以正确对齐输入（并在 openmp 编译指示中添加对齐子句），那么它应该很快。特别是如果您的输入真的超过 4 个元素。

res 将是 0-4 而不是 0 或 1，但这可能不是问题。 SIMD 指令倾向于处理水平加法而不是水平位与。