Come auto-vettorizzare le scritture stride con GCC?

Question

Quando compilato con GCC 5.2 utilizzando -std=c99, -O3, e -mavx2, il seguente esempio di codice auto-vettorizza (assembly here):

#include <stdint.h> 

void test(uint32_t *restrict a, 
      uint32_t *restrict b) { 
    uint32_t *a_aligned = __builtin_assume_aligned(a, 32); 
    uint32_t *b_aligned = __builtin_assume_aligned(b, 32); 

    for (int i = 0; i < (1L << 10); i += 2) { 
    a_aligned[i] = 42 * b_aligned[i]; 
    a_aligned[i+1] = 3 * a_aligned[i+1]; 
    } 
} 

Ma il seguente codice di esempio non auto-vectorize (assembly here):

#include <stdint.h> 

void test(uint32_t *restrict a, 
      uint32_t *restrict b) { 
    uint32_t *a_aligned = __builtin_assume_aligned(a, 32); 
    uint32_t *b_aligned = __builtin_assume_aligned(b, 32); 

    for (int i = 0; i < (1L << 10); i += 2) { 
    a_aligned[i] = 42 * b_aligned[i]; 
    a_aligned[i+1] = a_aligned[i+1]; 
    } 
} 

L'unica differenza tra i campioni è il fattore di scala su a_aligned[i+1].

Questo è stato anche il caso per GCC 4.8, 4.9 e 5.1. L'aggiunta della dichiarazione volatile a a_aligned inibisce completamente l'auto-vettorizzazione. Il primo campione viene eseguito costantemente più velocemente del secondo per noi, con un aumento della velocità più pronunciato per i tipi più piccoli (ad esempio uint8_t anziché uint32_t).

C'è un modo per rendere auto-vectorize il secondo codice con GCC?

Answer 1

I seguenti vectorises versione, ma questo è brutto, se mi chiedete ...

#include <stdint.h> 

void test(uint32_t *a, uint32_t *aa, 
      uint32_t *restrict b) { 
    #pragma omp simd aligned(a,aa,b:32) 
    for (int i = 0; i < (1L << 10); i += 2) { 
    a[i] = 2 * b[i]; 
    a[i+1] = aa[i+1]; 
    } 
} 

di compilare con -fopenmp e per chiamare con test(a, a, b).