|
|
@@ -0,0 +1,247 @@ |
|
|
|
/*************************************************************************** |
|
|
|
Copyright (c) 2014, The OpenBLAS Project |
|
|
|
All rights reserved. |
|
|
|
Redistribution and use in source and binary forms, with or without |
|
|
|
modification, are permitted provided that the following conditions are |
|
|
|
met: |
|
|
|
1. Redistributions of source code must retain the above copyright |
|
|
|
notice, this list of conditions and the following disclaimer. |
|
|
|
2. Redistributions in binary form must reproduce the above copyright |
|
|
|
notice, this list of conditions and the following disclaimer in |
|
|
|
the documentation and/or other materials provided with the |
|
|
|
distribution. |
|
|
|
3. Neither the name of the OpenBLAS project nor the names of |
|
|
|
its contributors may be used to endorse or promote products |
|
|
|
derived from this software without specific prior written permission. |
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE OPENBLAS PROJECT OR CONTRIBUTORS BE |
|
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
|
|
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
|
|
|
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
|
|
|
CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
|
|
|
OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE |
|
|
|
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
*****************************************************************************/ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define HAVE_KERNEL_4x8 1 |
|
|
|
static void dgemv_kernel_4x8( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, BLASLONG lda4, FLOAT *alpha) __attribute__ ((noinline)); |
|
|
|
|
|
|
|
static void dgemv_kernel_4x8( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, BLASLONG lda4, FLOAT *alpha) |
|
|
|
{ |
|
|
|
|
|
|
|
BLASLONG register i = 0; |
|
|
|
|
|
|
|
__asm__ __volatile__ |
|
|
|
( |
|
|
|
"vzeroupper \n\t" |
|
|
|
"vbroadcastsd (%2), %%ymm12 \n\t" // x0 |
|
|
|
"vbroadcastsd 8(%2), %%ymm13 \n\t" // x1 |
|
|
|
"vbroadcastsd 16(%2), %%ymm14 \n\t" // x2 |
|
|
|
"vbroadcastsd 24(%2), %%ymm15 \n\t" // x3 |
|
|
|
"vbroadcastsd 32(%2), %%ymm0 \n\t" // x4 |
|
|
|
"vbroadcastsd 40(%2), %%ymm1 \n\t" // x5 |
|
|
|
"vbroadcastsd 48(%2), %%ymm2 \n\t" // x6 |
|
|
|
"vbroadcastsd 56(%2), %%ymm3 \n\t" // x7 |
|
|
|
|
|
|
|
"vbroadcastsd (%9), %%ymm6 \n\t" // alpha |
|
|
|
|
|
|
|
"testq $0x04, %1 \n\t" |
|
|
|
"jz 2f \n\t" |
|
|
|
|
|
|
|
"vmovupd (%3,%0,8), %%ymm7 \n\t" // 4 * y |
|
|
|
"vxorpd %%ymm4 , %%ymm4, %%ymm4 \n\t" |
|
|
|
"vxorpd %%ymm5 , %%ymm5, %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%0,8), %%ymm12, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%5,%0,8), %%ymm13, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%0,8), %%ymm14, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%7,%0,8), %%ymm15, %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%8,8), %%ymm0 , %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%5,%8,8), %%ymm1 , %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%8,8), %%ymm2 , %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%7,%8,8), %%ymm3 , %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vaddpd %%ymm4 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
"vmulpd %%ymm6 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
"vaddpd %%ymm7 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
|
|
|
|
|
|
|
|
"vmovupd %%ymm5, (%3,%0,8) \n\t" // 4 * y |
|
|
|
|
|
|
|
"addq $4 , %8 \n\t" |
|
|
|
"addq $4 , %0 \n\t" |
|
|
|
"subq $4 , %1 \n\t" |
|
|
|
|
|
|
|
"2: \n\t" |
|
|
|
|
|
|
|
"cmpq $0, %1 \n\t" |
|
|
|
"je 3f \n\t" |
|
|
|
|
|
|
|
|
|
|
|
".align 16 \n\t" |
|
|
|
"1: \n\t" |
|
|
|
|
|
|
|
"vxorpd %%ymm4 , %%ymm4, %%ymm4 \n\t" |
|
|
|
"vxorpd %%ymm5 , %%ymm5, %%ymm5 \n\t" |
|
|
|
"vmovupd (%3,%0,8), %%ymm8 \n\t" // 4 * y |
|
|
|
"vmovupd 32(%3,%0,8), %%ymm9 \n\t" // 4 * y |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%0,8), %%ymm12, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%4,%0,8), %%ymm12, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%5,%0,8), %%ymm13, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%5,%0,8), %%ymm13, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%0,8), %%ymm14, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%6,%0,8), %%ymm14, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%7,%0,8), %%ymm15, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%7,%0,8), %%ymm15, %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%8,8), %%ymm0 , %%ymm4 \n\t" |
|
|
|
"addq $8 , %0 \n\t" |
|
|
|
"vfmadd231pd 32(%4,%8,8), %%ymm0 , %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%5,%8,8), %%ymm1 , %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%5,%8,8), %%ymm1 , %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%8,8), %%ymm2 , %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%6,%8,8), %%ymm2 , %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%7,%8,8), %%ymm3 , %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%7,%8,8), %%ymm3 , %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vfmadd231pd %%ymm6 , %%ymm4 , %%ymm8 \n\t" |
|
|
|
"vfmadd231pd %%ymm6 , %%ymm5 , %%ymm9 \n\t" |
|
|
|
|
|
|
|
"addq $8 , %8 \n\t" |
|
|
|
"vmovupd %%ymm8,-64(%3,%0,8) \n\t" // 4 * y |
|
|
|
"subq $8 , %1 \n\t" |
|
|
|
"vmovupd %%ymm9,-32(%3,%0,8) \n\t" // 4 * y |
|
|
|
|
|
|
|
"jnz 1b \n\t" |
|
|
|
|
|
|
|
"3: \n\t" |
|
|
|
"vzeroupper \n\t" |
|
|
|
|
|
|
|
: |
|
|
|
: |
|
|
|
"r" (i), // 0 |
|
|
|
"r" (n), // 1 |
|
|
|
"r" (x), // 2 |
|
|
|
"r" (y), // 3 |
|
|
|
"r" (ap[0]), // 4 |
|
|
|
"r" (ap[1]), // 5 |
|
|
|
"r" (ap[2]), // 6 |
|
|
|
"r" (ap[3]), // 7 |
|
|
|
"r" (lda4), // 8 |
|
|
|
"r" (alpha) // 9 |
|
|
|
: "cc", |
|
|
|
"%xmm0", "%xmm1", |
|
|
|
"%xmm2", "%xmm3", |
|
|
|
"%xmm4", "%xmm5", |
|
|
|
"%xmm6", "%xmm7", |
|
|
|
"%xmm8", "%xmm9", |
|
|
|
"%xmm12", "%xmm13", "%xmm14", "%xmm15", |
|
|
|
"memory" |
|
|
|
); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define HAVE_KERNEL_4x4 1 |
|
|
|
static void dgemv_kernel_4x4( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, FLOAT *alpha) __attribute__ ((noinline)); |
|
|
|
|
|
|
|
static void dgemv_kernel_4x4( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y, FLOAT *alpha) |
|
|
|
{ |
|
|
|
|
|
|
|
BLASLONG register i = 0; |
|
|
|
|
|
|
|
__asm__ __volatile__ |
|
|
|
( |
|
|
|
"vzeroupper \n\t" |
|
|
|
"vbroadcastsd (%2), %%ymm12 \n\t" // x0 |
|
|
|
"vbroadcastsd 8(%2), %%ymm13 \n\t" // x1 |
|
|
|
"vbroadcastsd 16(%2), %%ymm14 \n\t" // x2 |
|
|
|
"vbroadcastsd 24(%2), %%ymm15 \n\t" // x3 |
|
|
|
|
|
|
|
"vbroadcastsd (%8), %%ymm6 \n\t" // alpha |
|
|
|
|
|
|
|
"testq $0x04, %1 \n\t" |
|
|
|
"jz 2f \n\t" |
|
|
|
|
|
|
|
"vxorpd %%ymm4 , %%ymm4, %%ymm4 \n\t" |
|
|
|
"vxorpd %%ymm5 , %%ymm5, %%ymm5 \n\t" |
|
|
|
"vmovupd (%3,%0,8), %%ymm7 \n\t" // 4 * y |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%0,8), %%ymm12, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%5,%0,8), %%ymm13, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%0,8), %%ymm14, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd (%7,%0,8), %%ymm15, %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vaddpd %%ymm4 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
"vmulpd %%ymm6 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
"vaddpd %%ymm7 , %%ymm5 , %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vmovupd %%ymm5, (%3,%0,8) \n\t" // 4 * y |
|
|
|
|
|
|
|
"addq $4 , %0 \n\t" |
|
|
|
"subq $4 , %1 \n\t" |
|
|
|
|
|
|
|
"2: \n\t" |
|
|
|
|
|
|
|
"cmpq $0, %1 \n\t" |
|
|
|
"je 3f \n\t" |
|
|
|
|
|
|
|
|
|
|
|
".align 16 \n\t" |
|
|
|
"1: \n\t" |
|
|
|
"vxorpd %%ymm4 , %%ymm4, %%ymm4 \n\t" |
|
|
|
"vxorpd %%ymm5 , %%ymm5, %%ymm5 \n\t" |
|
|
|
"vmovupd (%3,%0,8), %%ymm8 \n\t" // 4 * y |
|
|
|
"vmovupd 32(%3,%0,8), %%ymm9 \n\t" // 4 * y |
|
|
|
|
|
|
|
"vfmadd231pd (%4,%0,8), %%ymm12, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%4,%0,8), %%ymm12, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%5,%0,8), %%ymm13, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%5,%0,8), %%ymm13, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%6,%0,8), %%ymm14, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%6,%0,8), %%ymm14, %%ymm5 \n\t" |
|
|
|
"vfmadd231pd (%7,%0,8), %%ymm15, %%ymm4 \n\t" |
|
|
|
"vfmadd231pd 32(%7,%0,8), %%ymm15, %%ymm5 \n\t" |
|
|
|
|
|
|
|
"vfmadd231pd %%ymm6 , %%ymm4 , %%ymm8 \n\t" |
|
|
|
"vfmadd231pd %%ymm6 , %%ymm5 , %%ymm9 \n\t" |
|
|
|
|
|
|
|
"vmovupd %%ymm8, (%3,%0,8) \n\t" // 4 * y |
|
|
|
"vmovupd %%ymm9, 32(%3,%0,8) \n\t" // 4 * y |
|
|
|
|
|
|
|
"addq $8 , %0 \n\t" |
|
|
|
"subq $8 , %1 \n\t" |
|
|
|
"jnz 1b \n\t" |
|
|
|
|
|
|
|
"3: \n\t" |
|
|
|
"vzeroupper \n\t" |
|
|
|
|
|
|
|
: |
|
|
|
: |
|
|
|
"r" (i), // 0 |
|
|
|
"r" (n), // 1 |
|
|
|
"r" (x), // 2 |
|
|
|
"r" (y), // 3 |
|
|
|
"r" (ap[0]), // 4 |
|
|
|
"r" (ap[1]), // 5 |
|
|
|
"r" (ap[2]), // 6 |
|
|
|
"r" (ap[3]), // 7 |
|
|
|
"r" (alpha) // 8 |
|
|
|
: "cc", |
|
|
|
"%xmm4", "%xmm5", |
|
|
|
"%xmm6", "%xmm7", |
|
|
|
"%xmm8", "%xmm9", |
|
|
|
"%xmm12", "%xmm13", "%xmm14", "%xmm15", |
|
|
|
"memory" |
|
|
|
); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|