|
- /***************************************************************************
- 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_4x4 1
- static void zgemv_kernel_4x4( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y) __attribute__ ((noinline));
-
- static void zgemv_kernel_4x4( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y)
- {
-
- BLASLONG register i = 0;
-
- __asm__ __volatile__
- (
- "vzeroupper \n\t"
-
- "vbroadcastsd (%2), %%ymm0 \n\t" // real part x0
- "vbroadcastsd 8(%2), %%ymm1 \n\t" // imag part x0
- "vbroadcastsd 16(%2), %%ymm2 \n\t" // real part x1
- "vbroadcastsd 24(%2), %%ymm3 \n\t" // imag part x1
- "vbroadcastsd 32(%2), %%ymm4 \n\t" // real part x2
- "vbroadcastsd 40(%2), %%ymm5 \n\t" // imag part x2
- "vbroadcastsd 48(%2), %%ymm6 \n\t" // real part x3
- "vbroadcastsd 56(%2), %%ymm7 \n\t" // imag part x3
-
-
- ".p2align 4 \n\t"
- "1: \n\t"
-
- //"prefetcht0 256(%4,%0,8) \n\t"
- "vmovups (%4,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vmovups 32(%4,%0,8), %%ymm9 \n\t" // 2 complex values form a0
-
- "vmulpd %%ymm8 , %%ymm0 , %%ymm12 \n\t"
- "vmulpd %%ymm8 , %%ymm1 , %%ymm13 \n\t"
- //"prefetcht0 256(%5,%0,8) \n\t"
- "vmulpd %%ymm9 , %%ymm0 , %%ymm14 \n\t"
- "vmovups (%5,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vmulpd %%ymm9 , %%ymm1 , %%ymm15 \n\t"
- "vmovups 32(%5,%0,8), %%ymm9 \n\t" // 2 complex values form a0
-
- "vmulpd %%ymm8 , %%ymm2 , %%ymm10 \n\t"
- "vmulpd %%ymm8 , %%ymm3 , %%ymm11 \n\t"
- "vaddpd %%ymm12, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm13, %%ymm11, %%ymm13 \n\t"
- //"prefetcht0 256(%6,%0,8) \n\t"
- "vmulpd %%ymm9 , %%ymm2 , %%ymm10 \n\t"
- "vmovups (%6,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vaddpd %%ymm14, %%ymm10, %%ymm14 \n\t"
- "vmulpd %%ymm9 , %%ymm3 , %%ymm11 \n\t"
-
- "vmovups 32(%6,%0,8), %%ymm9 \n\t" // 2 complex values form a0
- "vaddpd %%ymm15, %%ymm11, %%ymm15 \n\t"
-
- "vmulpd %%ymm8 , %%ymm4 , %%ymm10 \n\t"
- "vmulpd %%ymm8 , %%ymm5 , %%ymm11 \n\t"
- "vaddpd %%ymm12, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm13, %%ymm11, %%ymm13 \n\t"
- // "prefetcht0 256(%7,%0,8) \n\t"
- "vmulpd %%ymm9 , %%ymm4 , %%ymm10 \n\t"
- "vmovups (%7,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vaddpd %%ymm14, %%ymm10, %%ymm14 \n\t"
- "vmulpd %%ymm9 , %%ymm5 , %%ymm11 \n\t"
-
- "vmovups 32(%7,%0,8), %%ymm9 \n\t" // 2 complex values form a0
- "vaddpd %%ymm15, %%ymm11, %%ymm15 \n\t"
-
- "vmulpd %%ymm8 , %%ymm6 , %%ymm10 \n\t"
- "vmulpd %%ymm8 , %%ymm7 , %%ymm11 \n\t"
- "vaddpd %%ymm12, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm13, %%ymm11, %%ymm13 \n\t"
- "vmulpd %%ymm9 , %%ymm6 , %%ymm10 \n\t"
- "vmulpd %%ymm9 , %%ymm7 , %%ymm11 \n\t"
- "vaddpd %%ymm14, %%ymm10, %%ymm14 \n\t"
- "vaddpd %%ymm15, %%ymm11, %%ymm15 \n\t"
-
- // "prefetcht0 256(%3,%0,8) \n\t"
- "vmovups (%3,%0,8), %%ymm10 \n\t"
- "vmovups 32(%3,%0,8), %%ymm11 \n\t"
-
- #if ( !defined(CONJ) && !defined(XCONJ) ) || ( defined(CONJ) && defined(XCONJ) )
- "vpermilpd $0x5 , %%ymm13, %%ymm13 \n\t"
- "vpermilpd $0x5 , %%ymm15, %%ymm15 \n\t"
- "vaddsubpd %%ymm13, %%ymm12, %%ymm8 \n\t"
- "vaddsubpd %%ymm15, %%ymm14, %%ymm9 \n\t"
- #else
- "vpermilpd $0x5 , %%ymm12, %%ymm12 \n\t"
- "vpermilpd $0x5 , %%ymm14, %%ymm14 \n\t"
- "vaddsubpd %%ymm12, %%ymm13, %%ymm8 \n\t"
- "vaddsubpd %%ymm14, %%ymm15, %%ymm9 \n\t"
- "vpermilpd $0x5 , %%ymm8 , %%ymm8 \n\t"
- "vpermilpd $0x5 , %%ymm9 , %%ymm9 \n\t"
- #endif
-
- "vaddpd %%ymm8, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm9, %%ymm11, %%ymm13 \n\t"
-
- "vmovups %%ymm12, (%3,%0,8) \n\t" // 2 complex values to y
- "vmovups %%ymm13, 32(%3,%0,8) \n\t"
-
- "addq $8 , %0 \n\t"
- "subq $4 , %1 \n\t"
- "jnz 1b \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
- : "cc",
- "%xmm0", "%xmm1", "%xmm2", "%xmm3",
- "%xmm4", "%xmm5", "%xmm6", "%xmm7",
- "%xmm8", "%xmm9", "%xmm10", "%xmm11",
- "%xmm12", "%xmm13", "%xmm14", "%xmm15",
- "memory"
- );
-
- }
-
-
- #define HAVE_KERNEL_4x2 1
- static void zgemv_kernel_4x2( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y) __attribute__ ((noinline));
-
- static void zgemv_kernel_4x2( BLASLONG n, FLOAT **ap, FLOAT *x, FLOAT *y)
- {
-
- BLASLONG register i = 0;
-
- __asm__ __volatile__
- (
- "vzeroupper \n\t"
-
- "vbroadcastsd (%2), %%ymm0 \n\t" // real part x0
- "vbroadcastsd 8(%2), %%ymm1 \n\t" // imag part x0
- "vbroadcastsd 16(%2), %%ymm2 \n\t" // real part x1
- "vbroadcastsd 24(%2), %%ymm3 \n\t" // imag part x1
-
- ".p2align 4 \n\t"
- "1: \n\t"
-
- // "prefetcht0 256(%4,%0,8) \n\t"
- "vmovups (%4,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vmovups 32(%4,%0,8), %%ymm9 \n\t" // 2 complex values form a0
- "vmulpd %%ymm8 , %%ymm0 , %%ymm12 \n\t"
- "vmulpd %%ymm8 , %%ymm1 , %%ymm13 \n\t"
-
- // "prefetcht0 256(%5,%0,8) \n\t"
- "vmulpd %%ymm9 , %%ymm0 , %%ymm14 \n\t"
- "vmovups (%5,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vmulpd %%ymm9 , %%ymm1 , %%ymm15 \n\t"
- "vmovups 32(%5,%0,8), %%ymm9 \n\t" // 2 complex values form a0
-
- "vmulpd %%ymm8 , %%ymm2 , %%ymm10 \n\t"
- "vmulpd %%ymm8 , %%ymm3 , %%ymm11 \n\t"
- "vaddpd %%ymm12, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm13, %%ymm11, %%ymm13 \n\t"
-
- "vmulpd %%ymm9 , %%ymm2 , %%ymm10 \n\t"
- "vmulpd %%ymm9 , %%ymm3 , %%ymm11 \n\t"
- "vaddpd %%ymm14, %%ymm10, %%ymm14 \n\t"
- "vaddpd %%ymm15, %%ymm11, %%ymm15 \n\t"
-
-
- // "prefetcht0 256(%3,%0,8) \n\t"
- "vmovups (%3,%0,8), %%ymm10 \n\t"
- "vmovups 32(%3,%0,8), %%ymm11 \n\t"
-
- #if ( !defined(CONJ) && !defined(XCONJ) ) || ( defined(CONJ) && defined(XCONJ) )
- "vpermilpd $0x5 , %%ymm13, %%ymm13 \n\t"
- "vpermilpd $0x5 , %%ymm15, %%ymm15 \n\t"
- "vaddsubpd %%ymm13, %%ymm12, %%ymm8 \n\t"
- "vaddsubpd %%ymm15, %%ymm14, %%ymm9 \n\t"
- #else
- "vpermilpd $0x5 , %%ymm12, %%ymm12 \n\t"
- "vpermilpd $0x5 , %%ymm14, %%ymm14 \n\t"
- "vaddsubpd %%ymm12, %%ymm13, %%ymm8 \n\t"
- "vaddsubpd %%ymm14, %%ymm15, %%ymm9 \n\t"
- "vpermilpd $0x5 , %%ymm8 , %%ymm8 \n\t"
- "vpermilpd $0x5 , %%ymm9 , %%ymm9 \n\t"
- #endif
-
- "vaddpd %%ymm8, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm9, %%ymm11, %%ymm13 \n\t"
-
- "vmovups %%ymm12, (%3,%0,8) \n\t" // 2 complex values to y
- "vmovups %%ymm13, 32(%3,%0,8) \n\t"
-
- "addq $8 , %0 \n\t"
- "subq $4 , %1 \n\t"
- "jnz 1b \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
- : "cc",
- "%xmm0", "%xmm1", "%xmm2", "%xmm3",
- "%xmm4", "%xmm5", "%xmm6", "%xmm7",
- "%xmm8", "%xmm9", "%xmm10", "%xmm11",
- "%xmm12", "%xmm13", "%xmm14", "%xmm15",
- "memory"
- );
-
- }
-
-
- #define HAVE_KERNEL_4x1 1
- static void zgemv_kernel_4x1( BLASLONG n, FLOAT *ap, FLOAT *x, FLOAT *y) __attribute__ ((noinline));
-
- static void zgemv_kernel_4x1( BLASLONG n, FLOAT *ap, FLOAT *x, FLOAT *y)
- {
-
- BLASLONG register i = 0;
-
- __asm__ __volatile__
- (
- "vzeroupper \n\t"
-
- "vbroadcastsd (%2), %%ymm0 \n\t" // real part x0
- "vbroadcastsd 8(%2), %%ymm1 \n\t" // imag part x0
-
- ".p2align 4 \n\t"
- "1: \n\t"
-
- // "prefetcht0 256(%4,%0,8) \n\t"
- "vmovups (%4,%0,8), %%ymm8 \n\t" // 2 complex values form a0
- "vmovups 32(%4,%0,8), %%ymm9 \n\t" // 2 complex values form a0
- "vmulpd %%ymm8 , %%ymm0 , %%ymm12 \n\t"
- "vmulpd %%ymm8 , %%ymm1 , %%ymm13 \n\t"
-
- "vmulpd %%ymm9 , %%ymm0 , %%ymm14 \n\t"
- "vmulpd %%ymm9 , %%ymm1 , %%ymm15 \n\t"
-
- // "prefetcht0 256(%3,%0,8) \n\t"
- "vmovups (%3,%0,8), %%ymm10 \n\t"
- "vmovups 32(%3,%0,8), %%ymm11 \n\t"
-
- #if ( !defined(CONJ) && !defined(XCONJ) ) || ( defined(CONJ) && defined(XCONJ) )
- "vpermilpd $0x5 , %%ymm13, %%ymm13 \n\t"
- "vpermilpd $0x5 , %%ymm15, %%ymm15 \n\t"
- "vaddsubpd %%ymm13, %%ymm12, %%ymm8 \n\t"
- "vaddsubpd %%ymm15, %%ymm14, %%ymm9 \n\t"
- #else
- "vpermilpd $0x5 , %%ymm12, %%ymm12 \n\t"
- "vpermilpd $0x5 , %%ymm14, %%ymm14 \n\t"
- "vaddsubpd %%ymm12, %%ymm13, %%ymm8 \n\t"
- "vaddsubpd %%ymm14, %%ymm15, %%ymm9 \n\t"
- "vpermilpd $0x5 , %%ymm8 , %%ymm8 \n\t"
- "vpermilpd $0x5 , %%ymm9 , %%ymm9 \n\t"
- #endif
-
- "vaddpd %%ymm8, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm9, %%ymm11, %%ymm13 \n\t"
-
- "vmovups %%ymm12, (%3,%0,8) \n\t" // 2 complex values to y
- "vmovups %%ymm13, 32(%3,%0,8) \n\t"
-
- "addq $8 , %0 \n\t"
- "subq $4 , %1 \n\t"
- "jnz 1b \n\t"
- "vzeroupper \n\t"
-
- :
- "+r" (i), // 0
- "+r" (n) // 1
- :
- "r" (x), // 2
- "r" (y), // 3
- "r" (ap) // 4
- : "cc",
- "%xmm0", "%xmm1", "%xmm2", "%xmm3",
- "%xmm4", "%xmm5", "%xmm6", "%xmm7",
- "%xmm8", "%xmm9", "%xmm10", "%xmm11",
- "%xmm12", "%xmm13", "%xmm14", "%xmm15",
- "memory"
- );
-
- }
-
-
-
- #define HAVE_KERNEL_ADDY 1
-
- static void add_y(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_dest,FLOAT alpha_r, FLOAT alpha_i) __attribute__ ((noinline));
-
- static void add_y(BLASLONG n, FLOAT *src, FLOAT *dest, BLASLONG inc_dest,FLOAT alpha_r, FLOAT alpha_i)
- {
- BLASLONG i;
-
- if ( inc_dest != 2 )
- {
-
- FLOAT temp_r;
- FLOAT temp_i;
- for ( i=0; i<n; i++ )
- {
- #if !defined(XCONJ)
- temp_r = alpha_r * src[0] - alpha_i * src[1];
- temp_i = alpha_r * src[1] + alpha_i * src[0];
- #else
- temp_r = alpha_r * src[0] + alpha_i * src[1];
- temp_i = -alpha_r * src[1] + alpha_i * src[0];
- #endif
-
- *dest += temp_r;
- *(dest+1) += temp_i;
-
- src+=2;
- dest += inc_dest;
- }
- return;
- }
-
- i=0;
-
- __asm__ __volatile__
- (
-
- "vzeroupper \n\t"
-
- "vbroadcastsd (%4), %%ymm0 \n\t" // alpha_r
- "vbroadcastsd (%5), %%ymm1 \n\t" // alpha_i
-
- ".p2align 4 \n\t"
- "1: \n\t"
- // "prefetcht0 192(%2,%0,8) \n\t"
- "vmovups (%2,%0,8), %%ymm8 \n\t" // 2 complex values from src
- "vmovups 32(%2,%0,8), %%ymm9 \n\t"
-
- "vmulpd %%ymm8 , %%ymm0, %%ymm12 \n\t" // a_r[0] * x_r , a_i[0] * x_r, a_r[1] * x_r, a_i[1] * x_r
- "vmulpd %%ymm8 , %%ymm1, %%ymm13 \n\t" // a_r[0] * x_i , a_i[0] * x_i, a_r[1] * x_i, a_i[1] * x_i
- "vmulpd %%ymm9 , %%ymm0, %%ymm14 \n\t" // a_r[2] * x_r , a_i[2] * x_r, a_r[3] * x_r, a_i[3] * x_r
- "vmulpd %%ymm9 , %%ymm1, %%ymm15 \n\t" // a_r[2] * x_i , a_i[2] * x_i, a_r[3] * x_i, a_i[3] * x_i
-
- // "prefetcht0 192(%3,%0,8) \n\t"
- "vmovups (%3,%0,8), %%ymm10 \n\t" // 2 complex values from dest
- "vmovups 32(%3,%0,8), %%ymm11 \n\t"
-
- #if !defined(XCONJ)
- "vpermilpd $0x5 , %%ymm13, %%ymm13 \n\t"
- "vpermilpd $0x5 , %%ymm15, %%ymm15 \n\t"
- "vaddsubpd %%ymm13, %%ymm12, %%ymm8 \n\t"
- "vaddsubpd %%ymm15, %%ymm14, %%ymm9 \n\t"
- #else
- "vpermilpd $0x5 , %%ymm12, %%ymm12 \n\t"
- "vpermilpd $0x5 , %%ymm14, %%ymm14 \n\t"
- "vaddsubpd %%ymm12, %%ymm13, %%ymm8 \n\t"
- "vaddsubpd %%ymm14, %%ymm15, %%ymm9 \n\t"
- "vpermilpd $0x5 , %%ymm8 , %%ymm8 \n\t"
- "vpermilpd $0x5 , %%ymm9 , %%ymm9 \n\t"
- #endif
-
- "vaddpd %%ymm8, %%ymm10, %%ymm12 \n\t"
- "vaddpd %%ymm9, %%ymm11, %%ymm13 \n\t"
-
- "vmovups %%ymm12, (%3,%0,8) \n\t" // 2 complex values to y
- "vmovups %%ymm13, 32(%3,%0,8) \n\t"
-
- "addq $8 , %0 \n\t"
- "subq $4 , %1 \n\t"
- "jnz 1b \n\t"
- "vzeroupper \n\t"
-
- :
- "+r" (i), // 0
- "+r" (n) // 1
- :
- "r" (src), // 2
- "r" (dest), // 3
- "r" (&alpha_r), // 4
- "r" (&alpha_i) // 5
- : "cc",
- "%xmm0", "%xmm1", "%xmm2", "%xmm3",
- "%xmm4", "%xmm5", "%xmm6", "%xmm7",
- "%xmm8", "%xmm9", "%xmm10", "%xmm11",
- "%xmm12", "%xmm13", "%xmm14", "%xmm15",
- "memory"
- );
-
- return;
-
- }
|