As operações escalares do AVX são muito mais rápidas
Eu testo a seguinte função simples
void mul(double *a, double *b) {
for (int i = 0; i<N; i++) a[i] *= b[i];
}
com matrizes muito grandes para que ele seja vinculado à largura de banda da memória. O código de teste que eu uso está abaixo. Quando eu compilar com-O2
leva 1,7 segundos. Quando eu compilar com-O2 -mavx
leva apenas 1,0 segundos. As operações escalares não codificadas por vex são 70% mais lentas!Por que é isso?
Aqui está a montagem para-O2
e-O2 -mavx
.
Sistema: [email protected] (Skylake) 32 GB mem, Ubuntu 16.10, GCC 6.3
Código de teste
//gcc -O2 -fopenmp test.c
//or
//gcc -O2 -mavx -fopenmp test.c
#include <string.h>
#include <stdio.h>
#include <x86intrin.h>
#include <omp.h>
#define N 1000000
#define R 1000
void mul(double *a, double *b) {
for (int i = 0; i<N; i++) a[i] *= b[i];
}
int main() {
double *a = (double*)_mm_malloc(sizeof *a * N, 32);
double *b = (double*)_mm_malloc(sizeof *b * N, 32);
//b must be initialized to get the correct bandwidth!!!
memset(a, 1, sizeof *a * N);
memset(b, 1, sizeof *b * N);
double dtime;
const double mem = 3*sizeof(double)*N*R/1024/1024/1024;
const double maxbw = 34.1;
dtime = -omp_get_wtime();
for(int i=0; i<R; i++) mul(a,b);
dtime += omp_get_wtime();
printf("time %.2f s, %.1f GB/s, efficency %.1f%%\n", dtime, mem/dtime, 100*mem/dtime/maxbw);
_mm_free(a), _mm_free(b);
}