Import GotoBLAS2 1.13 BSD version codes.
This commit is contained in:
909
kernel/x86/zgemm_kernel_1x2_sse2.S
Normal file
909
kernel/x86/zgemm_kernel_1x2_sse2.S
Normal file
@@ -0,0 +1,909 @@
|
||||
/*********************************************************************/
|
||||
/* 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 POSINV 0(%esp)
|
||||
#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 OLD_STACK 72(%esp)
|
||||
#define OFFSET 76(%esp)
|
||||
#define KK 80(%esp)
|
||||
#define KKK 84(%esp)
|
||||
#define BUFFER 128(%esp)
|
||||
|
||||
#define STACK_ALIGN 4096
|
||||
#define STACK_OFFSET 1024
|
||||
|
||||
#if defined(OPTERON) || defined(BARCELONA)
|
||||
#define PREFETCH prefetch
|
||||
#endif
|
||||
|
||||
#define PREFETCHSIZE (8 * 10 + 4)
|
||||
|
||||
#define AA %edx
|
||||
#define BB %ecx
|
||||
#define LDC %ebp
|
||||
#define B %edi
|
||||
|
||||
|
||||
#define KERNEL1(address) \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
addpd %xmm2, %xmm4; \
|
||||
PREFETCH (PREFETCHSIZE + 0) * SIZE + (address) * 1 * SIZE(AA); \
|
||||
movapd 2 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
addpd %xmm2, %xmm5; \
|
||||
movapd 4 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
mulpd 6 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \
|
||||
addpd %xmm2, %xmm6; \
|
||||
movapd 16 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
addpd %xmm0, %xmm7; \
|
||||
movapd 2 * SIZE + (address) * 1 * SIZE(AA), %xmm0
|
||||
|
||||
#define KERNEL2(address) \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
addpd %xmm3, %xmm4; \
|
||||
movapd 10 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
addpd %xmm3, %xmm5; \
|
||||
movapd 12 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
mulpd 14 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \
|
||||
addpd %xmm3, %xmm6; \
|
||||
movapd 24 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
addpd %xmm0, %xmm7; \
|
||||
movapd 4 * SIZE + (address) * 1 * SIZE(AA), %xmm0
|
||||
|
||||
#define KERNEL3(address) \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
addpd %xmm2, %xmm4; \
|
||||
movapd 18 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
addpd %xmm2, %xmm5; \
|
||||
movapd 20 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm0, %xmm2; \
|
||||
mulpd 22 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \
|
||||
addpd %xmm2, %xmm6; \
|
||||
movapd 32 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
addpd %xmm0, %xmm7; \
|
||||
movapd 6 * SIZE + (address) * 1 * SIZE(AA), %xmm0
|
||||
|
||||
#define KERNEL4(address) \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
addpd %xmm3, %xmm4; \
|
||||
movapd 26 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
addpd %xmm3, %xmm5; \
|
||||
movapd 28 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm0, %xmm3; \
|
||||
mulpd 30 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \
|
||||
addpd %xmm3, %xmm6; \
|
||||
movapd 40 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
addpd %xmm0, %xmm7; \
|
||||
movapd 16 * SIZE + (address) * 1 * SIZE(AA), %xmm0
|
||||
|
||||
#define KERNEL5(address) \
|
||||
PREFETCH (PREFETCHSIZE + 8) * SIZE + (address) * 1 * SIZE(AA); \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
addpd %xmm2, %xmm4; \
|
||||
movapd 34 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
addpd %xmm2, %xmm5; \
|
||||
movapd 36 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
mulpd 38 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \
|
||||
addpd %xmm2, %xmm6; \
|
||||
movapd 48 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
addpd %xmm1, %xmm7; \
|
||||
movapd 10 * SIZE + (address) * 1 * SIZE(AA), %xmm1
|
||||
|
||||
#define KERNEL6(address) \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
addpd %xmm3, %xmm4; \
|
||||
movapd 42 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
addpd %xmm3, %xmm5; \
|
||||
movapd 44 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
mulpd 46 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \
|
||||
addpd %xmm3, %xmm6; \
|
||||
movapd 56 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
addpd %xmm1, %xmm7; \
|
||||
movapd 12 * SIZE + (address) * 1 * SIZE(AA), %xmm1
|
||||
|
||||
#define KERNEL7(address) \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
addpd %xmm2, %xmm4; \
|
||||
movapd 50 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
addpd %xmm2, %xmm5; \
|
||||
movapd 52 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
mulpd %xmm1, %xmm2; \
|
||||
mulpd 54 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \
|
||||
addpd %xmm2, %xmm6; \
|
||||
movapd 64 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \
|
||||
addpd %xmm1, %xmm7; \
|
||||
movapd 14 * SIZE + (address) * 1 * SIZE(AA), %xmm1
|
||||
|
||||
#define KERNEL8(address) \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
addpd %xmm3, %xmm4; \
|
||||
movapd 58 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
addpd %xmm3, %xmm5; \
|
||||
movapd 60 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
mulpd %xmm1, %xmm3; \
|
||||
mulpd 62 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \
|
||||
addpd %xmm3, %xmm6; \
|
||||
movapd 72 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \
|
||||
addpd %xmm1, %xmm7; \
|
||||
movapd 24 * SIZE + (address) * 1 * SIZE(AA), %xmm1
|
||||
|
||||
PROLOGUE
|
||||
|
||||
pushl %ebp
|
||||
pushl %edi
|
||||
pushl %esi
|
||||
pushl %ebx
|
||||
|
||||
PROFCODE
|
||||
|
||||
EMMS
|
||||
|
||||
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
|
||||
|
||||
movl STACK_M, %ebx
|
||||
movl STACK_N, %eax
|
||||
movl STACK_K, %ecx
|
||||
movl STACK_A, %edx
|
||||
|
||||
movl %ebx, M
|
||||
movl %eax, N
|
||||
movl %ecx, K
|
||||
movl %edx, A
|
||||
movl %esi, OLD_STACK
|
||||
|
||||
movl STACK_B, B
|
||||
movl STACK_C, %ebx
|
||||
#ifdef TRMMKERNEL
|
||||
movss STACK_OFFT, %xmm4
|
||||
#endif
|
||||
|
||||
movlpd STACK_ALPHA_R, %xmm0
|
||||
movlpd STACK_ALPHA_I, %xmm1
|
||||
|
||||
pcmpeqb %xmm7, %xmm7
|
||||
psllq $63, %xmm7 # Generate mask
|
||||
pxor %xmm2, %xmm2
|
||||
|
||||
movlpd %xmm0, 0 + ALPHA_R
|
||||
movlpd %xmm0, 8 + ALPHA_R
|
||||
|
||||
movlpd %xmm1, 8 + ALPHA_I
|
||||
xorpd %xmm7, %xmm1
|
||||
movlpd %xmm1, 0 + ALPHA_I
|
||||
|
||||
movlpd %xmm2, 0 + POSINV
|
||||
movlpd %xmm7, 8 + POSINV
|
||||
|
||||
movl %ebx, C
|
||||
movl STACK_LDC, LDC
|
||||
|
||||
#ifdef TRMMKERNEL
|
||||
movss %xmm4, OFFSET
|
||||
movss %xmm4, KK
|
||||
#ifndef LEFT
|
||||
negl KK
|
||||
#endif
|
||||
#endif
|
||||
|
||||
sall $ZBASE_SHIFT, LDC
|
||||
|
||||
sarl $1, %eax
|
||||
movl %eax, J # j = n
|
||||
jle .L100
|
||||
ALIGN_4
|
||||
|
||||
.L01:
|
||||
#if defined(TRMMKERNEL) && defined(LEFT)
|
||||
movl OFFSET, %eax
|
||||
movl %eax, KK
|
||||
#endif
|
||||
|
||||
leal BUFFER, %ecx
|
||||
|
||||
movapd POSINV, %xmm7
|
||||
|
||||
movl K, %eax
|
||||
sarl $1, %eax
|
||||
jle .L03
|
||||
ALIGN_4
|
||||
|
||||
.L02:
|
||||
prefetchnta 56 * SIZE(B)
|
||||
|
||||
movlpd 0 * SIZE(B), %xmm0
|
||||
movlpd 1 * SIZE(B), %xmm1
|
||||
movlpd 2 * SIZE(B), %xmm2
|
||||
movlpd 3 * SIZE(B), %xmm3
|
||||
movlpd 4 * SIZE(B), %xmm4
|
||||
movlpd 5 * SIZE(B), %xmm5
|
||||
movlpd 6 * SIZE(B), %xmm6
|
||||
movlpd 7 * SIZE(B), %xmm7
|
||||
|
||||
movlpd %xmm0, 0 * SIZE(BB)
|
||||
movlpd %xmm0, 1 * SIZE(BB)
|
||||
movlpd %xmm1, 2 * SIZE(BB)
|
||||
movlpd %xmm1, 3 * SIZE(BB)
|
||||
movlpd %xmm2, 4 * SIZE(BB)
|
||||
movlpd %xmm2, 5 * SIZE(BB)
|
||||
movlpd %xmm3, 6 * SIZE(BB)
|
||||
movlpd %xmm3, 7 * SIZE(BB)
|
||||
movlpd %xmm4, 8 * SIZE(BB)
|
||||
movlpd %xmm4, 9 * SIZE(BB)
|
||||
movlpd %xmm5, 10 * SIZE(BB)
|
||||
movlpd %xmm5, 11 * SIZE(BB)
|
||||
movlpd %xmm6, 12 * SIZE(BB)
|
||||
movlpd %xmm6, 13 * SIZE(BB)
|
||||
movlpd %xmm7, 14 * SIZE(BB)
|
||||
movlpd %xmm7, 15 * SIZE(BB)
|
||||
|
||||
addl $ 8 * SIZE, B
|
||||
subl $-16 * SIZE, BB
|
||||
|
||||
decl %eax
|
||||
jne .L02
|
||||
ALIGN_4
|
||||
|
||||
.L03:
|
||||
movl K, %eax
|
||||
andl $1, %eax
|
||||
BRANCH
|
||||
jle .L05
|
||||
|
||||
movlpd 0 * SIZE(B), %xmm0
|
||||
movlpd 1 * SIZE(B), %xmm1
|
||||
movlpd 2 * SIZE(B), %xmm2
|
||||
movlpd 3 * SIZE(B), %xmm3
|
||||
|
||||
movlpd %xmm0, 0 * SIZE(BB)
|
||||
movlpd %xmm0, 1 * SIZE(BB)
|
||||
movlpd %xmm1, 2 * SIZE(BB)
|
||||
movlpd %xmm1, 3 * SIZE(BB)
|
||||
movlpd %xmm2, 4 * SIZE(BB)
|
||||
movlpd %xmm2, 5 * SIZE(BB)
|
||||
movlpd %xmm3, 6 * SIZE(BB)
|
||||
movlpd %xmm3, 7 * SIZE(BB)
|
||||
|
||||
addl $4 * SIZE, B
|
||||
ALIGN_4
|
||||
|
||||
.L05:
|
||||
movl C, %esi # coffset = c
|
||||
movl A, AA # aoffset = a
|
||||
movl M, %ebx
|
||||
testl %ebx, %ebx
|
||||
jle .L100
|
||||
ALIGN_4
|
||||
|
||||
.L10:
|
||||
#if !defined(TRMMKERNEL) || \
|
||||
(defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \
|
||||
(defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA))
|
||||
|
||||
leal BUFFER, BB # boffset1 = boffset
|
||||
#else
|
||||
leal BUFFER, BB # boffset1 = boffset
|
||||
movl KK, %eax
|
||||
leal (, %eax, SIZE), %eax
|
||||
leal (AA, %eax, 2), AA
|
||||
leal (BB, %eax, 8), BB
|
||||
#endif
|
||||
|
||||
movapd 0 * SIZE(AA), %xmm0
|
||||
pxor %xmm4, %xmm4
|
||||
movapd 8 * SIZE(AA), %xmm1
|
||||
pxor %xmm5, %xmm5
|
||||
movapd 0 * SIZE(BB), %xmm2
|
||||
pxor %xmm6, %xmm6
|
||||
movapd 8 * SIZE(BB), %xmm3
|
||||
pxor %xmm7, %xmm7
|
||||
|
||||
prefetchw 2 * SIZE(%esi)
|
||||
prefetchw 2 * SIZE(%esi, LDC)
|
||||
|
||||
#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 $1, %eax
|
||||
#else
|
||||
addl $2, %eax
|
||||
#endif
|
||||
movl %eax, KKK
|
||||
#endif
|
||||
|
||||
#if 1
|
||||
andl $-8, %eax
|
||||
sall $4, %eax
|
||||
je .L15
|
||||
.L1X:
|
||||
KERNEL1(16 * 0)
|
||||
KERNEL2(16 * 0)
|
||||
KERNEL3(16 * 0)
|
||||
KERNEL4(16 * 0)
|
||||
KERNEL5(16 * 0)
|
||||
KERNEL6(16 * 0)
|
||||
KERNEL7(16 * 0)
|
||||
KERNEL8(16 * 0)
|
||||
cmpl $128 * 1, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 1)
|
||||
KERNEL2(16 * 1)
|
||||
KERNEL3(16 * 1)
|
||||
KERNEL4(16 * 1)
|
||||
KERNEL5(16 * 1)
|
||||
KERNEL6(16 * 1)
|
||||
KERNEL7(16 * 1)
|
||||
KERNEL8(16 * 1)
|
||||
cmpl $128 * 2, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 2)
|
||||
KERNEL2(16 * 2)
|
||||
KERNEL3(16 * 2)
|
||||
KERNEL4(16 * 2)
|
||||
KERNEL5(16 * 2)
|
||||
KERNEL6(16 * 2)
|
||||
KERNEL7(16 * 2)
|
||||
KERNEL8(16 * 2)
|
||||
cmpl $128 * 3, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 3)
|
||||
KERNEL2(16 * 3)
|
||||
KERNEL3(16 * 3)
|
||||
KERNEL4(16 * 3)
|
||||
KERNEL5(16 * 3)
|
||||
KERNEL6(16 * 3)
|
||||
KERNEL7(16 * 3)
|
||||
KERNEL8(16 * 3)
|
||||
cmpl $128 * 4, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 4)
|
||||
KERNEL2(16 * 4)
|
||||
KERNEL3(16 * 4)
|
||||
KERNEL4(16 * 4)
|
||||
KERNEL5(16 * 4)
|
||||
KERNEL6(16 * 4)
|
||||
KERNEL7(16 * 4)
|
||||
KERNEL8(16 * 4)
|
||||
cmpl $128 * 5, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 5)
|
||||
KERNEL2(16 * 5)
|
||||
KERNEL3(16 * 5)
|
||||
KERNEL4(16 * 5)
|
||||
KERNEL5(16 * 5)
|
||||
KERNEL6(16 * 5)
|
||||
KERNEL7(16 * 5)
|
||||
KERNEL8(16 * 5)
|
||||
cmpl $128 * 6, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 6)
|
||||
KERNEL2(16 * 6)
|
||||
KERNEL3(16 * 6)
|
||||
KERNEL4(16 * 6)
|
||||
KERNEL5(16 * 6)
|
||||
KERNEL6(16 * 6)
|
||||
KERNEL7(16 * 6)
|
||||
KERNEL8(16 * 6)
|
||||
cmpl $128 * 7, %eax
|
||||
jle .L12
|
||||
KERNEL1(16 * 7)
|
||||
KERNEL2(16 * 7)
|
||||
KERNEL3(16 * 7)
|
||||
KERNEL4(16 * 7)
|
||||
KERNEL5(16 * 7)
|
||||
KERNEL6(16 * 7)
|
||||
KERNEL7(16 * 7)
|
||||
KERNEL8(16 * 7)
|
||||
|
||||
addl $128 * 4 * SIZE, BB
|
||||
addl $128 * 1 * SIZE, AA
|
||||
subl $128 * 8, %eax
|
||||
jg .L1X
|
||||
jmp .L15
|
||||
|
||||
.L12:
|
||||
leal (AA, %eax, 1), AA
|
||||
leal (BB, %eax, 4), BB
|
||||
ALIGN_4
|
||||
#else
|
||||
|
||||
sarl $3, %eax
|
||||
je .L15
|
||||
ALIGN_4
|
||||
|
||||
.L12:
|
||||
KERNEL1(16 * 0)
|
||||
KERNEL2(16 * 0)
|
||||
KERNEL3(16 * 0)
|
||||
KERNEL4(16 * 0)
|
||||
KERNEL5(16 * 0)
|
||||
KERNEL6(16 * 0)
|
||||
KERNEL7(16 * 0)
|
||||
KERNEL8(16 * 0)
|
||||
|
||||
addl $64 * SIZE, BB
|
||||
addl $16 * SIZE, AA
|
||||
decl %eax
|
||||
jne .L11
|
||||
ALIGN_4
|
||||
#endif
|
||||
|
||||
.L15:
|
||||
#ifndef TRMMKERNEL
|
||||
movl K, %eax
|
||||
#else
|
||||
movl KKK, %eax
|
||||
#endif
|
||||
andl $7, %eax # if (k & 1)
|
||||
BRANCH
|
||||
je .L14
|
||||
ALIGN_4
|
||||
|
||||
.L13:
|
||||
mulpd %xmm0, %xmm2
|
||||
addpd %xmm2, %xmm4
|
||||
movapd 2 * SIZE(BB), %xmm2
|
||||
mulpd %xmm0, %xmm2
|
||||
addpd %xmm2, %xmm5
|
||||
movapd 4 * SIZE(BB), %xmm2
|
||||
mulpd %xmm0, %xmm2
|
||||
mulpd 6 * SIZE(BB), %xmm0
|
||||
addpd %xmm2, %xmm6
|
||||
movapd 8 * SIZE(BB), %xmm2
|
||||
addpd %xmm0, %xmm7
|
||||
movapd 2 * SIZE(AA), %xmm0
|
||||
|
||||
addl $2 * SIZE, AA
|
||||
addl $8 * SIZE, BB
|
||||
decl %eax
|
||||
jg .L13
|
||||
ALIGN_4
|
||||
|
||||
.L14:
|
||||
movapd POSINV, %xmm1
|
||||
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)
|
||||
xorpd %xmm1, %xmm5
|
||||
xorpd %xmm1, %xmm7
|
||||
#else
|
||||
xorpd %xmm1, %xmm4
|
||||
xorpd %xmm1, %xmm6
|
||||
#endif
|
||||
|
||||
#ifndef TRMMKERNEL
|
||||
movlpd 0 * SIZE(%esi), %xmm0
|
||||
movhpd 1 * SIZE(%esi), %xmm0
|
||||
movlpd 0 * SIZE(%esi, LDC), %xmm1
|
||||
movhpd 1 * SIZE(%esi, LDC), %xmm1
|
||||
#endif
|
||||
|
||||
#if defined(NN) || defined(NT) || defined(TN) || defined(TT) || \
|
||||
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
||||
subpd %xmm5, %xmm4
|
||||
subpd %xmm7, %xmm6
|
||||
#else
|
||||
addpd %xmm5, %xmm4
|
||||
addpd %xmm7, %xmm6
|
||||
#endif
|
||||
|
||||
pshufd $0x4e, %xmm4, %xmm5
|
||||
pshufd $0x4e, %xmm6, %xmm7
|
||||
|
||||
mulpd %xmm2, %xmm4
|
||||
mulpd %xmm3, %xmm5
|
||||
mulpd %xmm2, %xmm6
|
||||
mulpd %xmm3, %xmm7
|
||||
|
||||
addpd %xmm5, %xmm4
|
||||
addpd %xmm7, %xmm6
|
||||
|
||||
#ifndef TRMMKERNEL
|
||||
addpd %xmm0, %xmm4
|
||||
addpd %xmm1, %xmm6
|
||||
#endif
|
||||
|
||||
movlpd %xmm4, 0 * SIZE(%esi)
|
||||
movhpd %xmm4, 1 * SIZE(%esi)
|
||||
movlpd %xmm6, 0 * SIZE(%esi, LDC)
|
||||
movhpd %xmm6, 1 * SIZE(%esi, LDC)
|
||||
|
||||
#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, 2), AA
|
||||
leal (BB, %eax, 8), BB
|
||||
#endif
|
||||
|
||||
#if defined(TRMMKERNEL) && defined(LEFT)
|
||||
addl $1, KK
|
||||
#endif
|
||||
|
||||
addl $2 * SIZE, %esi # coffset += 4
|
||||
decl %ebx # i --
|
||||
jg .L10
|
||||
ALIGN_4
|
||||
|
||||
.L99:
|
||||
#if defined(TRMMKERNEL) && !defined(LEFT)
|
||||
addl $2, KK
|
||||
#endif
|
||||
|
||||
leal (, LDC, 2), %eax
|
||||
addl %eax, C # c += ldc
|
||||
decl J # j --
|
||||
jg .L01
|
||||
ALIGN_4
|
||||
|
||||
.L100:
|
||||
movl N, %eax
|
||||
andl $1, %eax
|
||||
jle .L500
|
||||
ALIGN_4
|
||||
|
||||
.L101:
|
||||
#if defined(TRMMKERNEL) && defined(LEFT)
|
||||
movl OFFSET, %eax
|
||||
movl %eax, KK
|
||||
#endif
|
||||
|
||||
leal BUFFER, %ecx
|
||||
movapd POSINV, %xmm7
|
||||
|
||||
movl K, %eax
|
||||
sarl $2, %eax
|
||||
jle .L103
|
||||
ALIGN_4
|
||||
|
||||
.L102:
|
||||
prefetchnta 56 * SIZE(B)
|
||||
|
||||
movlpd 0 * SIZE(B), %xmm0
|
||||
movlpd 1 * SIZE(B), %xmm1
|
||||
movlpd 2 * SIZE(B), %xmm2
|
||||
movlpd 3 * SIZE(B), %xmm3
|
||||
movlpd 4 * SIZE(B), %xmm4
|
||||
movlpd 5 * SIZE(B), %xmm5
|
||||
movlpd 6 * SIZE(B), %xmm6
|
||||
movlpd 7 * SIZE(B), %xmm7
|
||||
|
||||
movlpd %xmm0, 0 * SIZE(BB)
|
||||
movlpd %xmm0, 1 * SIZE(BB)
|
||||
movlpd %xmm1, 2 * SIZE(BB)
|
||||
movlpd %xmm1, 3 * SIZE(BB)
|
||||
movlpd %xmm2, 4 * SIZE(BB)
|
||||
movlpd %xmm2, 5 * SIZE(BB)
|
||||
movlpd %xmm3, 6 * SIZE(BB)
|
||||
movlpd %xmm3, 7 * SIZE(BB)
|
||||
movlpd %xmm4, 8 * SIZE(BB)
|
||||
movlpd %xmm4, 9 * SIZE(BB)
|
||||
movlpd %xmm5, 10 * SIZE(BB)
|
||||
movlpd %xmm5, 11 * SIZE(BB)
|
||||
movlpd %xmm6, 12 * SIZE(BB)
|
||||
movlpd %xmm6, 13 * SIZE(BB)
|
||||
movlpd %xmm7, 14 * SIZE(BB)
|
||||
movlpd %xmm7, 15 * SIZE(BB)
|
||||
|
||||
addl $ 8 * SIZE, B
|
||||
subl $-16 * SIZE, %ecx
|
||||
decl %eax
|
||||
jne .L102
|
||||
ALIGN_4
|
||||
|
||||
.L103:
|
||||
movl K, %eax
|
||||
andl $3, %eax
|
||||
BRANCH
|
||||
jle .L105
|
||||
ALIGN_4
|
||||
|
||||
.L104:
|
||||
movlpd 0 * SIZE(B), %xmm0
|
||||
movlpd 1 * SIZE(B), %xmm1
|
||||
|
||||
movlpd %xmm0, 0 * SIZE(BB)
|
||||
movlpd %xmm0, 1 * SIZE(BB)
|
||||
movlpd %xmm1, 2 * SIZE(BB)
|
||||
movlpd %xmm1, 3 * SIZE(BB)
|
||||
|
||||
addl $2 * SIZE, B
|
||||
addl $4 * SIZE, %ecx
|
||||
decl %eax
|
||||
jne .L104
|
||||
ALIGN_4
|
||||
|
||||
.L105:
|
||||
movl C, %esi # coffset = c
|
||||
movl A, AA # aoffset = a
|
||||
movl M, %ebx
|
||||
testl %ebx, %ebx
|
||||
jle .L500
|
||||
ALIGN_4
|
||||
|
||||
.L110:
|
||||
#if !defined(TRMMKERNEL) || \
|
||||
(defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \
|
||||
(defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA))
|
||||
|
||||
leal BUFFER, BB
|
||||
#else
|
||||
leal BUFFER, BB
|
||||
movl KK, %eax
|
||||
leal (, %eax, SIZE), %eax
|
||||
leal (AA, %eax, 2), AA
|
||||
leal (BB, %eax, 4), BB
|
||||
#endif
|
||||
|
||||
pxor %xmm4, %xmm4
|
||||
pxor %xmm5, %xmm5
|
||||
pxor %xmm6, %xmm6
|
||||
pxor %xmm7, %xmm7
|
||||
|
||||
movapd 0 * SIZE(AA), %xmm0
|
||||
movapd 8 * SIZE(AA), %xmm1
|
||||
movapd 0 * SIZE(BB), %xmm2
|
||||
movapd 8 * SIZE(BB), %xmm3
|
||||
|
||||
#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 $1, %eax
|
||||
#else
|
||||
addl $1, %eax
|
||||
#endif
|
||||
movl %eax, KKK
|
||||
#endif
|
||||
sarl $3, %eax
|
||||
je .L112
|
||||
ALIGN_4
|
||||
|
||||
.L111:
|
||||
PREFETCH (PREFETCHSIZE + 0) * SIZE(AA)
|
||||
mulpd %xmm0, %xmm2
|
||||
mulpd 2 * SIZE(BB), %xmm0
|
||||
addpd %xmm2, %xmm4
|
||||
movapd 4 * SIZE(BB), %xmm2
|
||||
addpd %xmm0, %xmm5
|
||||
movapd 2 * SIZE(AA), %xmm0
|
||||
mulpd %xmm0, %xmm2
|
||||
mulpd 6 * SIZE(BB), %xmm0
|
||||
addpd %xmm2, %xmm6
|
||||
movapd 16 * SIZE(BB), %xmm2
|
||||
addpd %xmm0, %xmm7
|
||||
movapd 4 * SIZE(AA), %xmm0
|
||||
mulpd %xmm0, %xmm3
|
||||
mulpd 10 * SIZE(BB), %xmm0
|
||||
addpd %xmm3, %xmm4
|
||||
movapd 12 * SIZE(BB), %xmm3
|
||||
addpd %xmm0, %xmm5
|
||||
movapd 6 * SIZE(AA), %xmm0
|
||||
mulpd %xmm0, %xmm3
|
||||
mulpd 14 * SIZE(BB), %xmm0
|
||||
addpd %xmm3, %xmm6
|
||||
movapd 24 * SIZE(BB), %xmm3
|
||||
addpd %xmm0, %xmm7
|
||||
movapd 16 * SIZE(AA), %xmm0
|
||||
mulpd %xmm1, %xmm2
|
||||
mulpd 18 * SIZE(BB), %xmm1
|
||||
addpd %xmm2, %xmm4
|
||||
movapd 20 * SIZE(BB), %xmm2
|
||||
addpd %xmm1, %xmm5
|
||||
movapd 10 * SIZE(AA), %xmm1
|
||||
mulpd %xmm1, %xmm2
|
||||
mulpd 22 * SIZE(BB), %xmm1
|
||||
addpd %xmm2, %xmm6
|
||||
movapd 32 * SIZE(BB), %xmm2
|
||||
addpd %xmm1, %xmm7
|
||||
movapd 12 * SIZE(AA), %xmm1
|
||||
mulpd %xmm1, %xmm3
|
||||
mulpd 26 * SIZE(BB), %xmm1
|
||||
addpd %xmm3, %xmm4
|
||||
movapd 28 * SIZE(BB), %xmm3
|
||||
addpd %xmm1, %xmm5
|
||||
movapd 14 * SIZE(AA), %xmm1
|
||||
mulpd %xmm1, %xmm3
|
||||
mulpd 30 * SIZE(BB), %xmm1
|
||||
addpd %xmm3, %xmm6
|
||||
movapd 40 * SIZE(BB), %xmm3
|
||||
addpd %xmm1, %xmm7
|
||||
movapd 24 * SIZE(AA), %xmm1
|
||||
|
||||
addl $16 * SIZE, AA
|
||||
addl $32 * SIZE, BB
|
||||
decl %eax
|
||||
jne .L111
|
||||
ALIGN_4
|
||||
|
||||
.L112:
|
||||
#ifndef TRMMKERNEL
|
||||
movl K, %eax
|
||||
#else
|
||||
movl KKK, %eax
|
||||
#endif
|
||||
andl $7, %eax # if (k & 1)
|
||||
BRANCH
|
||||
je .L114
|
||||
ALIGN_4
|
||||
|
||||
.L113:
|
||||
mulpd %xmm0, %xmm2
|
||||
mulpd 2 * SIZE(BB), %xmm0
|
||||
addpd %xmm2, %xmm4
|
||||
movapd 4 * SIZE(BB), %xmm2
|
||||
addpd %xmm0, %xmm5
|
||||
movapd 2 * SIZE(AA), %xmm0
|
||||
|
||||
addl $2 * SIZE, AA
|
||||
addl $4 * SIZE, BB
|
||||
decl %eax
|
||||
jg .L113
|
||||
ALIGN_4
|
||||
|
||||
.L114:
|
||||
movapd POSINV, %xmm1
|
||||
movapd ALPHA_R, %xmm2
|
||||
movapd ALPHA_I, %xmm3
|
||||
|
||||
addpd %xmm6, %xmm4
|
||||
addpd %xmm7, %xmm5
|
||||
|
||||
SHUFPD_1 %xmm5, %xmm5
|
||||
|
||||
#if defined(NN) || defined(NT) || defined(TN) || defined(TT) || \
|
||||
defined(NR) || defined(NC) || defined(TR) || defined(TC)
|
||||
xorpd %xmm1, %xmm5
|
||||
#else
|
||||
xorpd %xmm1, %xmm4
|
||||
#endif
|
||||
|
||||
#ifndef TRMMKERNEL
|
||||
movlpd 0 * SIZE(%esi), %xmm0
|
||||
movhpd 1 * SIZE(%esi), %xmm0
|
||||
#endif
|
||||
|
||||
#if defined(NN) || defined(NT) || defined(TN) || defined(TT) || \
|
||||
defined(RR) || defined(RC) || defined(CR) || defined(CC)
|
||||
subpd %xmm5, %xmm4
|
||||
#else
|
||||
addpd %xmm5, %xmm4
|
||||
#endif
|
||||
|
||||
pshufd $0x4e, %xmm4, %xmm5
|
||||
|
||||
mulpd %xmm2, %xmm4
|
||||
mulpd %xmm3, %xmm5
|
||||
|
||||
addpd %xmm5, %xmm4
|
||||
|
||||
#ifndef TRMMKERNEL
|
||||
addpd %xmm0, %xmm4
|
||||
#endif
|
||||
|
||||
movlpd %xmm4, 0 * SIZE(%esi)
|
||||
movhpd %xmm4, 1 * SIZE(%esi)
|
||||
|
||||
#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, 2), AA
|
||||
leal (BB, %eax, 4), BB
|
||||
#endif
|
||||
|
||||
#if defined(TRMMKERNEL) && defined(LEFT)
|
||||
addl $1, KK
|
||||
#endif
|
||||
|
||||
addl $2 * SIZE, %esi # coffset += 4
|
||||
decl %ebx # i --
|
||||
jg .L110
|
||||
ALIGN_4
|
||||
|
||||
.L500:
|
||||
movl OLD_STACK, %esp
|
||||
|
||||
EMMS
|
||||
|
||||
popl %ebx
|
||||
popl %esi
|
||||
popl %edi
|
||||
popl %ebp
|
||||
ret
|
||||
|
||||
EPILOGUE
|
||||
Reference in New Issue
Block a user