c++ - SSE浮点 vector 的减少

如何使用sse内在函数获取浮点 vector 的求和元素(归约)？

简单的序列号:

void(float *input, float &result, unsigned int NumElems)
{
     result = 0;
     for(auto i=0; i<NumElems; ++i)
         result += input[i];
}

最佳答案

通常，您在循环中生成4个部分和，然后在循环后仅对4个元素进行水平求和，例如

#include <cassert>
#include <cstdint>
#include <emmintrin.h>

float vsum(const float *a, int n)
{
    float sum;
    __m128 vsum = _mm_set1_ps(0.0f);
    assert((n & 3) == 0);
    assert(((uintptr_t)a & 15) == 0);
    for (int i = 0; i < n; i += 4)
    {
        __m128 v = _mm_load_ps(&a[i]);
        vsum = _mm_add_ps(vsum, v);
    }
    vsum = _mm_hadd_ps(vsum, vsum);
    vsum = _mm_hadd_ps(vsum, vsum);
    _mm_store_ss(&sum, vsum);
    return sum;
}

注意:对于上面的示例，a必须对齐16个字节，n必须为4的倍数。如果不能保证a对齐，请使用_mm_loadu_ps而不是_mm_load_ps。如果n不能保证为4的倍数，则在函数末尾添加标量循环以累加所有剩余元素。

关于c++ - SSE浮点 vector 的减少，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/17761154/