Browse Source
Also make the kernel_4x2 use intrinsics for readability and consistency
pull/1712/head
Arjan van de Ven
7 years ago
1 changed files with 17 additions and 61 deletions
Split View
Diff Options
+ 17
- 61
kernel/x86_64/dgemv_n_microk_haswell-4.c
View File
| @@ -70,76 +70,32 @@ static void dgemv_kernel_4x4( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, FLOAT | |||
| #define HAVE_KERNEL_4x2 | |||
| static void dgemv_kernel_4x2( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, FLOAT *alpha) __attribute__ ((noinline)); | |||
| static void dgemv_kernel_4x2( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, FLOAT *alpha) | |||
| { | |||
| BLASLONG register i = 0; | |||
| __asm__ __volatile__ | |||
| ( | |||
| "vbroadcastsd (%2), %%ymm12 \n\t" // x0 | |||
| "vbroadcastsd 8(%2), %%ymm13 \n\t" // x1 | |||
| "vmovups (%4,%0,8), %%ymm0 \n\t" | |||
| "vmovups (%5,%0,8), %%ymm1 \n\t" | |||
| "vbroadcastsd (%6), %%ymm6 \n\t" // alpha | |||
| "addq $4 , %0 \n\t" | |||
| "subq $4 , %1 \n\t" | |||
| "jz 2f \n\t" | |||
| "1: \n\t" | |||
| "vmulpd %%ymm0 , %%ymm12, %%ymm4 \n\t" | |||
| "vmulpd %%ymm1 , %%ymm13, %%ymm5 \n\t" | |||
| "vmovups (%4,%0,8), %%ymm0 \n\t" | |||
| "vmovups (%5,%0,8), %%ymm1 \n\t" | |||
| "vmovups -32(%3,%0,8), %%ymm8 \n\t" // 4 * y | |||
| "vaddpd %%ymm4 , %%ymm5 , %%ymm4 \n\t" | |||
| "vfmadd231pd %%ymm6 , %%ymm4 , %%ymm8 \n\t" | |||
| "vmovups %%ymm8, -32(%3,%0,8) \n\t" // 4 * y | |||
| "addq $4 , %0 \n\t" | |||
| "subq $4 , %1 \n\t" | |||
| "jnz 1b \n\t" | |||
| "2: \n\t" | |||
| int i = 0; | |||
| "vmulpd %%ymm0 , %%ymm12, %%ymm4 \n\t" | |||
| "vmulpd %%ymm1 , %%ymm13, %%ymm5 \n\t" | |||
| __m256d x0, x1; | |||
| __m256d __alpha; | |||
| x0 = _mm256_broadcastsd_pd(_mm_load_sd(&x[0])); | |||
| x1 = _mm256_broadcastsd_pd(_mm_load_sd(&x[1])); | |||
| "vmovups -32(%3,%0,8), %%ymm8 \n\t" // 4 * y | |||
| "vaddpd %%ymm4 , %%ymm5 , %%ymm4 \n\t" | |||
| "vfmadd231pd %%ymm6 , %%ymm4 , %%ymm8 \n\t" | |||
| __alpha = _mm256_broadcastsd_pd(_mm_load_sd(alpha)); | |||
| "vmovups %%ymm8, -32(%3,%0,8) \n\t" // 4 * y | |||
| for (i = 0; i < n; i+= 4) { | |||
| __m256d tempY; | |||
| __m256d sum; | |||
| "vzeroupper \n\t" | |||
| sum = _mm256_add_pd( | |||
| _mm256_mul_pd(_mm256_loadu_pd(&ap[0][i]), x0), | |||
| _mm256_mul_pd(_mm256_loadu_pd(&ap[1][i]), x1) | |||
| ); | |||
| tempY = _mm256_loadu_pd(&y[i]); | |||
| tempY = _mm256_add_pd(tempY, _mm256_mul_pd(sum, __alpha)); | |||
| _mm256_storeu_pd(&y[i], tempY); | |||
| } | |||
| : | |||
| "+r" (i), // 0 | |||
| "+r" (n) // 1 | |||
| : | |||
| "r" (x), // 2 | |||
| "r" (y), // 3 | |||
| "r" (ap[0]), // 4 | |||
| "r" (ap[1]), // 5 | |||
| "r" (alpha) // 6 | |||
| : "cc", | |||
| "%xmm0", "%xmm1", | |||
| "%xmm4", "%xmm5", | |||
| "%xmm6", | |||
| "%xmm8", | |||
| "%xmm12", "%xmm13", | |||
| "memory" | |||
| ); | |||
| } | |||