/*********************************************************************/ /* Copyright 2009, 2010 The University of Texas at Austin. */ /* 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. */ /* */ /* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */ /* AUSTIN ``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 UNIVERSITY OF TEXAS AT */ /* AUSTIN 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. */ /* */ /* The views and conclusions contained in the software and */ /* documentation are those of the authors and should not be */ /* interpreted as representing official policies, either expressed */ /* or implied, of The University of Texas at Austin. */ /*********************************************************************/ #define ASSEMBLER #include "common.h" #define STACK 16 #define ARGS 0 #define STACK_M 4 + STACK + ARGS(%esi) #define STACK_N 8 + STACK + ARGS(%esi) #define STACK_K 12 + STACK + ARGS(%esi) #define STACK_ALPHA_R 16 + STACK + ARGS(%esi) #define STACK_ALPHA_I 24 + STACK + ARGS(%esi) #define STACK_A 32 + STACK + ARGS(%esi) #define STACK_B 36 + STACK + ARGS(%esi) #define STACK_C 40 + STACK + ARGS(%esi) #define STACK_LDC 44 + STACK + ARGS(%esi) #define STACK_OFFT 48 + STACK + ARGS(%esi) #define ALPHA_R 16(%esp) #define ALPHA_I 32(%esp) #define K 48(%esp) #define N 52(%esp) #define M 56(%esp) #define A 60(%esp) #define C 64(%esp) #define J 68(%esp) #define BX 72(%esp) #define OLD_STACK 76(%esp) #define OFFSET 80(%esp) #define KK 84(%esp) #define KKK 88(%esp) #define BUFFER 128(%esp) #define STACK_ALIGN 4096 #define STACK_OFFSET 1024 #define PREFETCH_R (8 * 16 + 0) #define PREFETCH_W (PREFETCH_R * 2) #define PREFETCHSIZE (8 * 16 + 4) #define PREFETCH prefetcht0 #define B %edi #define LDC %ebp #define AA %edx #define BB %ecx #define C1 %esi #if defined(NN) || defined(NT) || defined(TN) || defined(TT) #define ADD1 addpd #define ADD2 addpd #elif defined(NR) || defined(NC) || defined(TR) || defined(TC) #define ADD1 addpd #define ADD2 subpd #elif defined(RN) || defined(RT) || defined(CN) || defined(CT) #define ADD1 subpd #define ADD2 addpd #else #define ADD1 subpd #define ADD2 subpd #endif PROLOGUE pushl %ebp pushl %edi pushl %esi pushl %ebx PROFCODE movl %esp, %esi # save old stack subl $128 + LOCAL_BUFFER_SIZE + STACK_OFFSET, %esp andl $-STACK_ALIGN, %esp # align stack addl $STACK_OFFSET, %esp STACK_TOUCHING movd STACK_M, %mm0 movl STACK_N, %eax movd STACK_K, %mm1 movd STACK_A, %mm2 movl STACK_B, B movd STACK_C, %mm3 movl STACK_LDC, LDC #ifdef TRMMKERNEL movd STACK_OFFT, %mm4 #endif movsd STACK_ALPHA_R, %xmm0 movsd STACK_ALPHA_I, %xmm1 movddup %xmm0, %xmm0 movddup %xmm1, %xmm1 movapd %xmm0, ALPHA_R movapd %xmm1, ALPHA_I movd %mm1, K movl %eax, N movd %mm0, M movd %mm2, A movd %mm3, C movl %esi, OLD_STACK #ifdef TRMMKERNEL movd %mm4, OFFSET movd %mm4, KK #ifndef LEFT negl KK #endif #endif subl $-16 * SIZE, A subl $-16 * SIZE, B sall $ZBASE_SHIFT, LDC movl %eax, J # j = n testl %eax, %eax jle .L999 ALIGN_2 .L01: leal 16 * SIZE + BUFFER, BB #if defined(TRMMKERNEL) && defined(LEFT) movl OFFSET, %eax movl %eax, KK #endif movl K, %eax sarl $2, %eax jle .L03 ALIGN_2 .L02: prefetcht0 (PREFETCH_R + 0) * SIZE(B) movddup -16 * SIZE(B), %xmm0 movddup -15 * SIZE(B), %xmm1 movddup -14 * SIZE(B), %xmm2 movddup -13 * SIZE(B), %xmm3 movddup -12 * SIZE(B), %xmm4 movddup -11 * SIZE(B), %xmm5 movddup -10 * SIZE(B), %xmm6 movddup -9 * SIZE(B), %xmm7 movapd %xmm0, -16 * SIZE(BB) movapd %xmm1, -14 * SIZE(BB) movapd %xmm2, -12 * SIZE(BB) movapd %xmm3, -10 * SIZE(BB) movapd %xmm4, -8 * SIZE(BB) movapd %xmm5, -6 * SIZE(BB) movapd %xmm6, -4 * SIZE(BB) movapd %xmm7, -2 * SIZE(BB) addl $ 8 * SIZE, B subl $-16 * SIZE, BB decl %eax jne .L02 ALIGN_2 .L03: movl K, %eax andl $3, %eax BRANCH jle .L05 ALIGN_2 .L04: movddup -16 * SIZE(B), %xmm0 movddup -15 * SIZE(B), %xmm1 movapd %xmm0, -16 * SIZE(BB) movapd %xmm1, -14 * SIZE(BB) addl $ 2 * SIZE, B addl $ 4 * SIZE, BB decl %eax jne .L04 ALIGN_4 .L05: movl B, BX movl C, C1 # coffset = c movl A, AA # aoffset = a movl M, %ebx sarl $1, %ebx # i = (m >> 2) jle .L20 ALIGN_4 .L10: #if !defined(TRMMKERNEL) || \ (defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \ (defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA)) leal 16 * SIZE + BUFFER, BB #else leal 16 * SIZE + BUFFER, BB movl KK, %eax leal (, %eax, SIZE), %eax leal (AA, %eax, 4), AA leal (BB, %eax, 4), BB /* because it's doubled */ #endif movapd -16 * SIZE(AA), %xmm0 pxor %xmm4, %xmm4 movapd -16 * SIZE(BB), %xmm1 pxor %xmm5, %xmm5 movapd -8 * SIZE(AA), %xmm3 pxor %xmm6, %xmm6 prefetcht0 3 * SIZE(C1) pxor %xmm7, %xmm7 movapd %xmm1, %xmm2 movl BX, %eax prefetcht0 (%eax) subl $-8 * SIZE, %eax movl %eax, BX #ifndef TRMMKERNEL movl K, %eax #elif (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA)) movl K, %eax subl KK, %eax movl %eax, KKK #else movl KK, %eax #ifdef LEFT addl $2, %eax #else addl $1, %eax #endif movl %eax, KKK #endif sarl $3, %eax je .L15 ALIGN_4 .L12: mulpd %xmm0, %xmm1 ADD1 %xmm1, %xmm4 movapd -14 * SIZE(BB), %xmm1 mulpd %xmm1, %xmm0 ADD2 %xmm0, %xmm5 movapd -14 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm2 ADD1 %xmm2, %xmm6 movapd -12 * SIZE(BB), %xmm2 mulpd %xmm0, %xmm1 movapd -12 * SIZE(AA), %xmm0 ADD2 %xmm1, %xmm7 PADDING; movapd %xmm2, %xmm1 mulpd %xmm0, %xmm2 ADD1 %xmm2, %xmm4 movapd -10 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm0 ADD2 %xmm0, %xmm5 movapd -10 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm1 ADD1 %xmm1, %xmm6 movapd -8 * SIZE(BB), %xmm1 mulpd %xmm0, %xmm2 PADDING; movapd 0 * SIZE(AA), %xmm0 ADD2 %xmm2, %xmm7 PADDING; movapd %xmm1, %xmm2 mulpd %xmm3, %xmm1 ADD1 %xmm1, %xmm4 movapd -6 * SIZE(BB), %xmm1 mulpd %xmm1, %xmm3 ADD2 %xmm3, %xmm5 movapd -6 * SIZE(AA), %xmm3 mulpd %xmm3, %xmm2 ADD1 %xmm2, %xmm6 movapd -4 * SIZE(BB), %xmm2 mulpd %xmm3, %xmm1 movapd -4 * SIZE(AA), %xmm3 ADD2 %xmm1, %xmm7 PADDING; movapd %xmm2, %xmm1 mulpd %xmm3, %xmm2 ADD1 %xmm2, %xmm4 movapd -2 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm3 ADD2 %xmm3, %xmm5 movapd -2 * SIZE(AA), %xmm3 mulpd %xmm3, %xmm1 ADD1 %xmm1, %xmm6 PADDING; movapd 0 * SIZE(BB), %xmm1 mulpd %xmm3, %xmm2 movapd 8 * SIZE(AA), %xmm3 ADD2 %xmm2, %xmm7 PADDING; movapd %xmm1, %xmm2 mulpd %xmm0, %xmm1 ADD1 %xmm1, %xmm4 movapd 2 * SIZE(BB), %xmm1 mulpd %xmm1, %xmm0 ADD2 %xmm0, %xmm5 movapd 2 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm2 ADD1 %xmm2, %xmm6 movapd 4 * SIZE(BB), %xmm2 mulpd %xmm0, %xmm1 movapd 4 * SIZE(AA), %xmm0 ADD2 %xmm1, %xmm7 PADDING; movapd %xmm2, %xmm1 mulpd %xmm0, %xmm2 ADD1 %xmm2, %xmm4 movapd 6 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm0 ADD2 %xmm0, %xmm5 movapd 6 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm1 ADD1 %xmm1, %xmm6 movapd 8 * SIZE(BB), %xmm1 mulpd %xmm0, %xmm2 movapd 16 * SIZE(AA), %xmm0 ADD2 %xmm2, %xmm7 PADDING; movapd %xmm1, %xmm2 mulpd %xmm3, %xmm1 ADD1 %xmm1, %xmm4 movapd 10 * SIZE(BB), %xmm1 mulpd %xmm1, %xmm3 ADD2 %xmm3, %xmm5 movapd 10 * SIZE(AA), %xmm3 mulpd %xmm3, %xmm2 ADD1 %xmm2, %xmm6 movapd 12 * SIZE(BB), %xmm2 mulpd %xmm3, %xmm1 movapd 12 * SIZE(AA), %xmm3 ADD2 %xmm1, %xmm7 PADDING; movapd %xmm2, %xmm1 mulpd %xmm3, %xmm2 ADD1 %xmm2, %xmm4 movapd 14 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm3 subl $-32 * SIZE, BB ADD2 %xmm3, %xmm5 movapd 14 * SIZE(AA), %xmm3 mulpd %xmm3, %xmm1 ADD1 %xmm1, %xmm6 movapd -16 * SIZE(BB), %xmm1 mulpd %xmm3, %xmm2 movapd 24 * SIZE(AA), %xmm3 ADD2 %xmm2, %xmm7 PADDING; movapd %xmm1, %xmm2 subl $-32 * SIZE, AA decl %eax BRANCH jne .L12 ALIGN_4 .L15: #ifndef TRMMKERNEL movl K, %eax #else movl KKK, %eax #endif andl $7, %eax BRANCH je .L18 ALIGN_4 .L16: mulpd %xmm0, %xmm1 ADD1 %xmm1, %xmm4 movapd -14 * SIZE(BB), %xmm1 movapd %xmm1, %xmm3 mulpd %xmm0, %xmm1 movapd -14 * SIZE(AA), %xmm0 ADD2 %xmm1, %xmm5 movapd -12 * SIZE(BB), %xmm1 mulpd %xmm0, %xmm2 ADD1 %xmm2, %xmm6 mulpd %xmm0, %xmm3 movapd -12 * SIZE(AA), %xmm0 ADD2 %xmm3, %xmm7 movapd %xmm1, %xmm2 addl $4 * SIZE, AA addl $4 * SIZE, BB decl %eax jg .L16 ALIGN_4 .L18: movapd ALPHA_R, %xmm2 movapd ALPHA_I, %xmm3 SHUFPD_1 %xmm5, %xmm5 SHUFPD_1 %xmm7, %xmm7 #if defined(NN) || defined(NT) || defined(TN) || defined(TT) || \ defined(NR) || defined(NC) || defined(TR) || defined(TC) addsubpd %xmm5, %xmm4 addsubpd %xmm7, %xmm6 movapd %xmm4, %xmm5 movapd %xmm6, %xmm7 #else addsubpd %xmm4, %xmm5 addsubpd %xmm6, %xmm7 movapd %xmm5, %xmm4 movapd %xmm7, %xmm6 #endif #ifndef TRMMKERNEL movsd 0 * SIZE(C1), %xmm0 movhpd 1 * SIZE(C1), %xmm0 movsd 2 * SIZE(C1), %xmm1 movhpd 3 * SIZE(C1), %xmm1 #endif SHUFPD_1 %xmm5, %xmm5 SHUFPD_1 %xmm7, %xmm7 mulpd %xmm2, %xmm4 mulpd %xmm2, %xmm6 mulpd %xmm3, %xmm5 mulpd %xmm3, %xmm7 addsubpd %xmm5, %xmm4 addsubpd %xmm7, %xmm6 #if! defined(TRMMKERNEL) && !defined(BETAZERO) addpd %xmm0, %xmm4 addpd %xmm1, %xmm6 #endif movsd %xmm4, 0 * SIZE(C1) movhpd %xmm4, 1 * SIZE(C1) movsd %xmm6, 2 * SIZE(C1) movhpd %xmm6, 3 * SIZE(C1) #if (defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \ (defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA)) movl K, %eax subl KKK, %eax leal (,%eax, SIZE), %eax leal (AA, %eax, 4), AA leal (BB, %eax, 4), BB #endif #if defined(TRMMKERNEL) && defined(LEFT) addl $2, KK #endif addl $4 * SIZE, C1 # coffset += 4 decl %ebx # i -- jg .L10 .L20: movl M, %ebx testl $1, %ebx je .L29 #if !defined(TRMMKERNEL) || \ (defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \ (defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA)) leal 16 * SIZE + BUFFER, %ecx #else leal 16 * SIZE + BUFFER, BB movl KK, %eax leal (, %eax, SIZE), %eax leal (AA, %eax, 2), AA leal (BB, %eax, 4), BB /* because it's doubled */ #endif movapd -16 * SIZE(AA), %xmm0 pxor %xmm4, %xmm4 movapd -16 * SIZE(BB), %xmm1 pxor %xmm5, %xmm5 movapd -8 * SIZE(AA), %xmm2 pxor %xmm6, %xmm6 movapd -8 * SIZE(BB), %xmm3 pxor %xmm7, %xmm7 #ifndef TRMMKERNEL movl K, %eax #elif (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA)) movl K, %eax subl KK, %eax movl %eax, KKK #else movl KK, %eax addl $1, %eax movl %eax, KKK #endif sarl $3, %eax jle .L22 .L21: mulpd %xmm0, %xmm1 mulpd -14 * SIZE(BB), %xmm0 ADD1 %xmm1, %xmm4 movapd -12 * SIZE(BB), %xmm1 ADD2 %xmm0, %xmm5 movapd -14 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm1 mulpd -10 * SIZE(BB), %xmm0 ADD1 %xmm1, %xmm6 movapd 0 * SIZE(BB), %xmm1 ADD2 %xmm0, %xmm7 movapd -12 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm3 mulpd -6 * SIZE(BB), %xmm0 ADD1 %xmm3, %xmm4 movapd -4 * SIZE(BB), %xmm3 ADD2 %xmm0, %xmm5 movapd -10 * SIZE(AA), %xmm0 mulpd %xmm0, %xmm3 mulpd -2 * SIZE(BB), %xmm0 ADD1 %xmm3, %xmm6 movapd 8 * SIZE(BB), %xmm3 ADD2 %xmm0, %xmm7 movapd 0 * SIZE(AA), %xmm0 mulpd %xmm2, %xmm1 mulpd 2 * SIZE(BB), %xmm2 ADD1 %xmm1, %xmm4 movapd 4 * SIZE(BB), %xmm1 ADD2 %xmm2, %xmm5 movapd -6 * SIZE(AA), %xmm2 mulpd %xmm2, %xmm1 mulpd 6 * SIZE(BB), %xmm2 ADD1 %xmm1, %xmm6 movapd 16 * SIZE(BB), %xmm1 ADD2 %xmm2, %xmm7 movapd -4 * SIZE(AA), %xmm2 mulpd %xmm2, %xmm3 mulpd 10 * SIZE(BB), %xmm2 ADD1 %xmm3, %xmm4 movapd 12 * SIZE(BB), %xmm3 ADD2 %xmm2, %xmm5 movapd -2 * SIZE(AA), %xmm2 mulpd %xmm2, %xmm3 mulpd 14 * SIZE(BB), %xmm2 ADD1 %xmm3, %xmm6 movapd 24 * SIZE(BB), %xmm3 ADD2 %xmm2, %xmm7 movapd 8 * SIZE(AA), %xmm2 subl $-16 * SIZE, AA addl $ 32 * SIZE, BB decl %eax # l-- jg .L21 ALIGN_2 .L22: #ifndef TRMMKERNEL movl K, %eax #else movl KKK, %eax #endif andl $7, %eax # l = (k & 3) jle .L24 ALIGN_2 .L23: mulpd %xmm0, %xmm1 mulpd -14 * SIZE(BB), %xmm0 ADD1 %xmm1, %xmm4 movapd -12 * SIZE(BB), %xmm1 ADD2 %xmm0, %xmm5 movapd -14 * SIZE(AA), %xmm0 addl $2 * SIZE, AA addl $4 * SIZE, BB decl %eax # l-- jg .L23 .L24: addpd %xmm6, %xmm4 addpd %xmm7, %xmm5 movapd ALPHA_R, %xmm2 movapd ALPHA_I, %xmm3 SHUFPD_1 %xmm5, %xmm5 #if defined(NN) || defined(NT) || defined(TN) || defined(TT) || \ defined(NR) || defined(NC) || defined(TR) || defined(TC) addsubpd %xmm5, %xmm4 movapd %xmm4, %xmm5 #else addsubpd %xmm4, %xmm5 movapd %xmm5, %xmm4 #endif #if! defined(TRMMKERNEL) && !defined(BETAZERO) movsd 0 * SIZE(C1), %xmm0 movhpd 1 * SIZE(C1), %xmm0 #endif SHUFPD_1 %xmm5, %xmm5 mulpd %xmm2, %xmm4 mulpd %xmm3, %xmm5 addsubpd %xmm5, %xmm4 #if! defined(TRMMKERNEL) && !defined(BETAZERO) addpd %xmm0, %xmm4 #endif movsd %xmm4, 0 * SIZE(C1) movhpd %xmm4, 1 * SIZE(C1) ALIGN_2 .L29: #if defined(TRMMKERNEL) && !defined(LEFT) addl $1, KK #endif addl LDC, C # c += ldc decl J # j -- jg .L01 .L999: movl OLD_STACK, %esp emms popl %ebx popl %esi popl %edi popl %ebp ret ALIGN_2 EPILOGUE