증분 루프의 경우 SSE를 사용하는 방법은 무엇입니까?

Question

다음과 같은 루프가 있습니다.

for(int i = 0;i < 28;i++) 
{ 
    a[i] = addr + flag; 
    flag = flag + b[i]; 
} 

여기서 각 값에 대한 플래그를 증가시킵니다. i. 따라서이 경우 SSE를 사용하여 루프 내부의 기능을 구현할 수 있습니다.

Answer 1

기본적으로 접두사 합계 (일명 누적 합계)입니다. 나는 SSE and AVX으로 이것을했다. 다음 코드는 SSE2를 사용하여 원하는 것을 수행해야합니다.

#include <x86intrin.h> 
#include <stdio.h> 

inline __m128i scan_SSE(__m128i x) { 
    x = _mm_add_epi32(x, _mm_slli_si128(x, 4)); 
    x = _mm_add_epi32(x, _mm_slli_si128(x, 8)); 
    return x; 
} 

void foo(int *a, int *b, const int n, const int addr, int flag) { 
    for(int i=0; i<n; i++) { 
     a[i] = addr + flag; 
     flag = flag + b[i]; 
    } 
    printf("flag %d\n", flag); 
} 

void foo_SSE(int *a, int *b, const int n, const int addr, int flag) { 
    __m128i offset = _mm_setzero_si128(); 
    __m128i k4 = _mm_set1_epi32(addr+flag-b[0]); 

    int i; 
    for (i = 0; i < n-3; i+=4) { 
     __m128i x = _mm_load_si128((__m128i*)&b[i]); 
     __m128i out = scan_SSE(x); 
     out = _mm_add_epi32(out, offset); 
     offset = _mm_shuffle_epi32(out, _MM_SHUFFLE(3, 3, 3, 3)); 
     out = _mm_add_epi32(out, k4); 
     _mm_store_si128((__m128i*)&a[i], out); 

    }  
    for(; i<n; i++) { 
     a[i] = addr + flag; 
     flag = flag + b[i]; 
    } 
    flag = a[n-1]-addr+b[0]; 
    printf("flag %d\n", flag); 
} 

int main() { 
    const int n = 24; 
    int a[n], b[n]; 

    for(int i=0; i<n; i++) a[i] = 1; 

    foo(b,a, n, 2,3); 
    for(int i=0; i<n; i++) printf("%d ", b[i]); printf("\n"); 

    foo_SSE(b,a, n, 2,3); 
    for(int i=0; i<n; i++) printf("%d ", b[i]); printf("\n"); 
} 

출력 :

flag 27 
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 
flag 27 
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32