SIMD-able代码?

时间:2013-01-10 12:40:11

标签: c++ x86 sse simd avx

什么代码可以使用SIMD指令集的严格定义是什么?是否可以并行运行计算?

所以,如果我有:

for(int i=0; i<100; i++){
    sum += array[i];
}

这可以利用SIMD,因为我们可以运行:

for(int i=0; i<100;i=i+4){
    sum0 += array[i];
    sum1 += array[i+1];
    sum2 += array[i+2];
    sum3 += array[i+3];
}

sum = sum0 + sum1 + sum2 + sum3;

它必须是float类型,还是double和integer?

1 个答案:

答案 0 :(得分:4)

假设你在谈论x86(SSE ),那么支持的算术类型是8,16,32和64位整数,以及单精度和双精度浮点数。但请注意,并非所有数据类型都支持所有算术运算 - SSE在这方面缺乏正交性。

假设32位整数和适当对齐的数组(16字节对齐),那么您可以将上面的循环示例实现为:

#include <emmintrin.h>                     // SSE2 intrinsics

int32_t a[100] __attribute__ ((aligned(16)));
                                           // suitably aligned array

__m128i vsum = _mm_set1_epi32(0);          // init vsum = { 0, 0, 0, 0 }
for (int i = 0; i < 100; i += 4)
{
    __m128i v = _mm_load_si128(&a[i]);     // load 4 ints from a[i]..a[i+3]
    vsum = _mm_add_epi32(vsum, v);         // accumulate 4 partial sums
}
// final horizontal sum of partial sums
vsum = _mm_add_epi32(vsum, _mm_srli_si128(vsum, 8));
int32_t sum = _mm_cvtsi128_si32(vsum);     // sum = scalar sum of a[]