OpenMP向量的SSE2约简

Question

我想用SSE2本质计算每个通道的平均图像(我们在这里忽略的3个感兴趣的通道+1个alpha通道)。我试过了：

  __m128 average = _mm_setzero_ps();

  #pragma omp parallel for reduction(+:average)
  for(size_t k = 0; k < roi_out->height * roi_out->width * ch; k += ch)
  {
    float *in = ((float *)temp) + k;
    average += _mm_load_ps(in);
  }

但我发现GCC的错误是：user-defined reduction not found for average。

这与SSE2是可能的吗？怎么了？

编辑

这样做是可行的：

float sum[4] = { 0.0f };

#pragma omp parallel for simd reduction(+:sum[:4])
for(size_t k = 0; k < roi_out->height * roi_out->width * ch; k += ch)
{
  float *in = ((float *)temp) + k;
  for (int i = 0; i < ch; ++i) sum[i] += in[i];
}

const __m128 average = _mm_load_ps(sum) / ((float)roi_out->height * roi_out->width);

Answer 1

您可以像这样定义一个自定义缩减：

#pragma omp declare reduction \
    (addps:__m128:omp_out+=omp_in) \
    initializer(omp_priv=_mm_setzero_ps())

然后用它就像：

#pragma omp parallel for reduction(addps:average)
for(size_t k = 0; k < size * ch; k += ch)
{
  average += _mm_loadu_ps(data+k);
}

我认为，最重要的是，openmp需要知道如何获得一个中性元素(此处为_mm_setzero_ps())。

完整工作示例：https://godbolt.org/z/Fpqttc

有趣的链接：http://pages.tacc.utexas.edu/~eijkhout/pcse/html/omp-reduction.html#User-definedreductions