2012-08-08 6 views
5

Come si SIMIDize il seguente codice in C (utilizzando naturalmente SIMD intrinseca)? Sto avendo difficoltà a capire intrinseche SIMD e questo sarebbe di grande aiuto:SIMD il seguente codice

int sum_naive(int n, int *a) 
{ 
    int sum = 0; 
    for(int i = 0; i < n; i++) 
     sum += a[i]; 
    return sum; 
} 
+0

Quale SIMD avevi in ​​mente? SSE2? – harold

+0

SSE I seguenti intrinseci possono essere usati .__ _mm_setzero_si128 m128i() __m128i _mm_loadu_si128 (__m128i * p) __m128i _mm_add_epi32 (__m128i a, __m128i b) (a0 + b0, a1 + b1, a2 + b2, a3 + b3) vuoto _mm_storeu_si128 (__m128i * p, __m128i a) – user1585869

+0

Ok, quindi SSE2. Che cosa hai provato? – harold

risposta

8

Ecco un'implementazione abbastanza semplice (attenzione: codice non testato):

int32_t sum_array(const int32_t a[], const int n) 
{ 
    __m128i vsum = _mm_set1_epi32(0);  // initialise vector of four partial 32 bit sums 
    int32_t sum; 
    int i; 

    for (i = 0; i < n; i += 4) 
    { 
     __m128i v = _mm_load_si128(&a[i]); // load vector of 4 x 32 bit values 
     vsum = _mm_add_epi32(vsum, v);  // accumulate to 32 bit partial sum vector 
    } 
    // horizontal add of four 32 bit partial sums and return result 
    vsum = _mm_add_epi32(vsum, _mm_srli_si128(vsum, 8)); 
    vsum = _mm_add_epi32(vsum, _mm_srli_si128(vsum, 4)); 
    sum = _mm_cvtsi128_si32(vsum); 
    return sum; 
} 

Si noti che la matrice di ingresso, a[], ha la necessità di essere allineato a 16 byte e n deve essere un multiplo di 4.