📄 trsm_kernel_rt_4x2_sse2.s
字号:
/*********************************************************************//* *//* Optimized BLAS libraries *//* By Kazushige Goto <kgoto@tacc.utexas.edu> *//* *//* Copyright (c) The University of Texas, 2005. All rights reserved. *//* UNIVERSITY EXPRESSLY DISCLAIMS ANY AND ALL WARRANTIES CONCERNING *//* THIS SOFTWARE AND DOCUMENTATION, INCLUDING ANY WARRANTIES OF *//* MERCHANTABILITY, FITNESS FOR ANY PARTICULAR PURPOSE, *//* NON-INFRINGEMENT AND WARRANTIES OF PERFORMANCE, AND ANY WARRANTY *//* THAT MIGHT OTHERWISE ARISE FROM COURSE OF DEALING OR USAGE OF *//* TRADE. NO WARRANTY IS EITHER EXPRESS OR IMPLIED WITH RESPECT TO *//* THE USE OF THE SOFTWARE OR DOCUMENTATION. *//* Under no circumstances shall University be liable for incidental, *//* special, indirect, direct or consequential damages or loss of *//* profits, interruption of business, or related expenses which may *//* arise from use of Software or Documentation, including but not *//* limited to those resulting from defects in Software and/or *//* Documentation, or loss or inaccuracy of data of any kind. *//*********************************************************************/#define ASSEMBLER#include "common.h"#if !defined(HAVE_SSE2) || !defined(HAVE_MMX)#error You have to check your configuration.#endif#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 16 + STACK + ARGS(%esi)#define STACK_A 24 + STACK + ARGS(%esi)#define STACK_B 28 + STACK + ARGS(%esi)#define STACK_C 32 + STACK + ARGS(%esi)#define STACK_LDC 36 + STACK + ARGS(%esi)#define STACK_OFFT 40 + STACK + ARGS(%esi) #define ALPHA 0(%esp)#define K 16(%esp)#define N 20(%esp)#define M 24(%esp)#define A 28(%esp)#define C 32(%esp)#define J 36(%esp)#define OLD_STACK 40(%esp)#define OFFSET 44(%esp)#define KK 48(%esp)#define KKK 52(%esp)#define AORIG 56(%esp)#define BORIG 60(%esp)#define BUFFER 128(%esp)#define B %edi#define LDC %ebp#define STACK_ALIGN 4096#define STACK_OFFSET 1024#define LOCAL_BUFFER_SIZE GEMM_Q * GEMM_UNROLL_N * COMPSIZE * 16#define AA %edx#define BB %ecx#define PREFETCHSIZE (8 * 4)#define KERNEL1(address) \ movq (PREFETCHSIZE + 0) * SIZE + (address) * SIZE(AA), %mm2; \ mulpd %xmm0, %xmm2; \ mulpd 2 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm4; \ movapd 0 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm5; \ movapd 2 * SIZE + (address) * SIZE(AA), %xmm0; \ mulpd %xmm0, %xmm2; \ mulpd 2 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm6; \ movapd 4 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm7; \ movapd 4 * SIZE + (address) * SIZE(AA), %xmm0#define KERNEL2(address) \ mulpd %xmm0, %xmm2; \ mulpd 6 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm4; \ movapd 4 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm5; \ movapd 6 * SIZE + (address) * SIZE(AA), %xmm0; \ mulpd %xmm0, %xmm2; \ mulpd 6 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm6; \ movapd 16 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm7; \ movapd 16 * SIZE + (address) * SIZE(AA), %xmm0#define KERNEL3(address) \ movq (PREFETCHSIZE + 8) * SIZE + (address) * SIZE(AA), %mm2; \ mulpd %xmm1, %xmm3; \ mulpd 10 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm4; \ movapd 8 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm5; \ movapd 10 * SIZE + (address) * SIZE(AA), %xmm1; \ mulpd %xmm1, %xmm3; \ mulpd 10 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm6; \ movapd 12 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm7; \ movapd 12 * SIZE + (address) * SIZE(AA), %xmm1#define KERNEL4(address) \ mulpd %xmm1, %xmm3; \ mulpd 14 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm4; \ movapd 12 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm5; \ movapd 14 * SIZE + (address) * SIZE(AA), %xmm1; \ mulpd %xmm1, %xmm3; \ mulpd 14 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm6; \ movapd 24 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm7; \ movapd 24 * SIZE + (address) * SIZE(AA), %xmm1#define KERNEL5(address) \ movq (PREFETCHSIZE + 16) * SIZE + (address) * SIZE(AA), %mm2; \ mulpd %xmm0, %xmm2; \ mulpd 18 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm4; \ movapd 16 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm5; \ movapd 18 * SIZE + (address) * SIZE(AA), %xmm0; \ mulpd %xmm0, %xmm2; \ mulpd 18 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm6; \ movapd 20 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm7; \ movapd 20 * SIZE + (address) * SIZE(AA), %xmm0#define KERNEL6(address) \ mulpd %xmm0, %xmm2; \ mulpd 22 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm4; \ movapd 20 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm5; \ movapd 22 * SIZE + (address) * SIZE(AA), %xmm0; \ mulpd %xmm0, %xmm2; \ mulpd 22 * SIZE + (address) * SIZE(BB), %xmm0; \ addpd %xmm2, %xmm6; \ movapd 32 * SIZE + (address) * SIZE(BB), %xmm2; \ addpd %xmm0, %xmm7; \ movapd 32 * SIZE + (address) * SIZE(AA), %xmm0#define KERNEL7(address) \ movq (PREFETCHSIZE + 24) * SIZE + (address) * SIZE(AA), %mm2; \ mulpd %xmm1, %xmm3; \ mulpd 26 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm4; \ movapd 24 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm5; \ movapd 26 * SIZE + (address) * SIZE(AA), %xmm1; \ mulpd %xmm1, %xmm3; \ mulpd 26 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm6; \ movapd 28 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm7; \ movapd 28 * SIZE + (address) * SIZE(AA), %xmm1#define KERNEL8(address) \ mulpd %xmm1, %xmm3; \ mulpd 30 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm4; \ movapd 28 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm5; \ movapd 30 * SIZE + (address) * SIZE(AA), %xmm1; \ mulpd %xmm1, %xmm3; \ mulpd 30 * SIZE + (address) * SIZE(BB), %xmm1; \ addpd %xmm3, %xmm6; \ movapd 40 * SIZE + (address) * SIZE(BB), %xmm3; \ addpd %xmm1, %xmm7; \ movapd 40 * SIZE + (address) * 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 addl $STACK_OFFSET, %esp 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 movd STACK_OFFT, %mm4 movd %mm1, K movl %eax, N movd %mm0, M movd %mm2, A movd %mm3, C movl %esi, OLD_STACK movd %mm4, OFFSET movd %mm4, KK sall $BASE_SHIFT, LDC#ifdef LN movl M, %eax leal (, %eax, SIZE), %eax addl %eax, C imull K, %eax addl %eax, A#endif#ifdef RT movl N, %eax leal (, %eax, SIZE), %eax imull K, %eax addl %eax, B movl N, %eax imull LDC, %eax addl %eax, C#endif#ifdef RN negl KK#endif #ifdef RT movl N, %eax subl OFFSET, %eax movl %eax, KK#endif movl N, %eax testl $1, %eax jle .L100 ALIGN_2 .L101:/* Copying to Sub Buffer */#ifdef LN movl OFFSET, %eax addl M, %eax movl %eax, KK#endif leal BUFFER, %ecx#ifdef RT movl K, %eax sall $0 + BASE_SHIFT, %eax subl %eax, B#endif#if defined(LN) || defined(RT) movl KK, %eax movl B, BORIG leal (, %eax, SIZE), %eax leal (B, %eax, 1), B leal (BB, %eax, 2), BB#endif #ifdef LT movl OFFSET, %eax movl %eax, KK#endif#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif sarl $3, %eax jle .L103 ALIGN_4 .L102: movsd 0 * SIZE(B), %xmm0 movsd 1 * SIZE(B), %xmm1 movsd 2 * SIZE(B), %xmm2 movsd 3 * SIZE(B), %xmm3 movsd 4 * SIZE(B), %xmm4 movsd 5 * SIZE(B), %xmm5 movsd 6 * SIZE(B), %xmm6 movsd 7 * SIZE(B), %xmm7 unpcklpd %xmm0, %xmm0 unpcklpd %xmm1, %xmm1 unpcklpd %xmm2, %xmm2 unpcklpd %xmm3, %xmm3 unpcklpd %xmm4, %xmm4 unpcklpd %xmm5, %xmm5 unpcklpd %xmm6, %xmm6 unpcklpd %xmm7, %xmm7 movapd %xmm0, 0 * SIZE(%ecx) movapd %xmm1, 2 * SIZE(%ecx) movapd %xmm2, 4 * SIZE(%ecx) movapd %xmm3, 6 * SIZE(%ecx) movapd %xmm4, 8 * SIZE(%ecx) movapd %xmm5, 10 * SIZE(%ecx) movapd %xmm6, 12 * SIZE(%ecx) movapd %xmm7, 14 * SIZE(%ecx) prefetcht0 104 * SIZE(B) addl $ 8 * SIZE, B addl $16 * SIZE, %ecx decl %eax BRANCH jne .L102 ALIGN_2.L103:#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif andl $7, %eax BRANCH jle .L105 ALIGN_2.L104: movsd 0 * SIZE(B), %xmm0 unpcklpd %xmm0, %xmm0 movapd %xmm0, 0 * SIZE(%ecx) addl $1 * SIZE, B addl $2 * SIZE, %ecx decl %eax jne .L104 ALIGN_4 .L105:#if defined(LT) || defined(RN) movl A, AA#else movl A, %eax movl %eax, AORIG#endif#ifdef RT subl LDC, C#endif movl C, %esi # coffset = c#ifndef RT addl LDC, C#endif movl M, %ebx sarl $2, %ebx # i = (m >> 2) jle .L130 ALIGN_4.L110:#ifdef LN movl K, %eax sall $2 + BASE_SHIFT, %eax subl %eax, AORIG#endif#if defined(LN) || defined(RT) movl KK, %eax movl AORIG, AA leal (, %eax, SIZE), %eax leal (AA, %eax, 4), AA#endif leal BUFFER, BB#if defined(LN) || defined(RT) movl KK, %eax sall $0 + BASE_SHIFT, %eax leal (BB, %eax, 2), BB#endif movapd 0 * SIZE(BB), %xmm2 pxor %xmm4, %xmm4 movapd 0 * SIZE(AA), %xmm0 pxor %xmm5, %xmm5 movapd 8 * SIZE(BB), %xmm3 pxor %xmm6, %xmm6 movapd 8 * SIZE(AA), %xmm1 pxor %xmm7, %xmm7#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif sarl $3, %eax je .L112.L111: mulpd %xmm2, %xmm0 mulpd 2 * SIZE(AA), %xmm2 addpd %xmm0, %xmm4 movapd 4 * SIZE(AA), %xmm0 addpd %xmm2, %xmm6 movapd 2 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm0 mulpd 6 * SIZE(AA), %xmm2 addpd %xmm0, %xmm5 movapd 16 * SIZE(AA), %xmm0 addpd %xmm2, %xmm7 movapd 4 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm1 mulpd 10 * SIZE(AA), %xmm2 addpd %xmm1, %xmm4 movapd 12 * SIZE(AA), %xmm1 addpd %xmm2, %xmm6 movapd 6 * SIZE(BB), %xmm2 mulpd %xmm2, %xmm1 mulpd 14 * SIZE(AA), %xmm2 addpd %xmm1, %xmm5 movapd 24 * SIZE(AA), %xmm1 addpd %xmm2, %xmm7 movapd 16 * SIZE(BB), %xmm2 mulpd %xmm3, %xmm0 mulpd 18 * SIZE(AA), %xmm3 addpd %xmm0, %xmm4 movapd 20 * SIZE(AA), %xmm0 addpd %xmm3, %xmm6 movapd 10 * SIZE(BB), %xmm3 mulpd %xmm3, %xmm0 mulpd 22 * SIZE(AA), %xmm3 addpd %xmm0, %xmm5 movapd 32 * SIZE(AA), %xmm0 addpd %xmm3, %xmm7 movapd 12 * SIZE(BB), %xmm3 mulpd %xmm3, %xmm1 mulpd 26 * SIZE(AA), %xmm3 addpd %xmm1, %xmm4 movapd 28 * SIZE(AA), %xmm1 addpd %xmm3, %xmm6 movapd 14 * SIZE(BB), %xmm3 mulpd %xmm3, %xmm1 mulpd 30 * SIZE(AA), %xmm3 addpd %xmm1, %xmm5 movapd 40 * SIZE(AA), %xmm1 addpd %xmm3, %xmm7 movapd 24 * SIZE(BB), %xmm3 addl $32 * SIZE, AA addl $16 * SIZE, BB decl %eax jne .L111.L112:#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif andl $7, %eax # if (k & 1) BRANCH je .L114.L113: mulpd %xmm2, %xmm0 mulpd 2 * SIZE(AA), %xmm2 addpd %xmm0, %xmm4 movapd 4 * SIZE(AA), %xmm0 addpd %xmm2, %xmm6 movapd 2 * SIZE(BB), %xmm2 addl $4 * SIZE, AA # aoffset += 8 addl $2 * SIZE, BB # boffset1 += 8 subl $1, %eax jg .L113 ALIGN_4.L114: addpd %xmm5, %xmm4 addpd %xmm7, %xmm6#if defined(LN) || defined(RT) movl KK, %eax#ifdef LN subl $4, %eax#else subl $1, %eax#endif movl AORIG, AA movl BORIG, B leal BUFFER, BB leal (, %eax, SIZE), %eax leal (AA, %eax, 4), AA leal (B, %eax, 1), B leal (BB, %eax, 2), BB#endif#if defined(LN) || defined(LT) movapd 0 * SIZE(B), %xmm0 movapd 2 * SIZE(B), %xmm1#else movapd 0 * SIZE(AA), %xmm0 movapd 2 * SIZE(AA), %xmm1#endif subpd %xmm4, %xmm0 subpd %xmm6, %xmm1#ifdef LN movapd %xmm0, %xmm2 unpckhpd %xmm2, %xmm2 movapd %xmm1, %xmm3 unpckhpd %xmm3, %xmm3 movsd 15 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm3 movsd 14 * SIZE(AA), %xmm5 mulsd %xmm3, %xmm5 subsd %xmm5, %xmm1 movsd 13 * SIZE(AA), %xmm6 mulsd %xmm3, %xmm6 subsd %xmm6, %xmm2 movsd 12 * SIZE(AA), %xmm7 mulsd %xmm3, %xmm7 subsd %xmm7, %xmm0 movsd 10 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm1 movsd 9 * SIZE(AA), %xmm5 mulsd %xmm1, %xmm5 subsd %xmm5, %xmm2 movsd 8 * SIZE(AA), %xmm6 mulsd %xmm1, %xmm6 subsd %xmm6, %xmm0 movsd 5 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm2 movsd 4 * SIZE(AA), %xmm5 mulsd %xmm2, %xmm5 subsd %xmm5, %xmm0 movsd 0 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm0 unpcklpd %xmm2, %xmm0 unpcklpd %xmm3, %xmm1#endif#ifdef LT movapd %xmm0, %xmm2 unpckhpd %xmm2, %xmm2 movapd %xmm1, %xmm3 unpckhpd %xmm3, %xmm3 movsd 0 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm0 movsd 1 * SIZE(AA), %xmm5 mulsd %xmm0, %xmm5 subsd %xmm5, %xmm2 movsd 2 * SIZE(AA), %xmm6 mulsd %xmm0, %xmm6 subsd %xmm6, %xmm1 movsd 3 * SIZE(AA), %xmm7 mulsd %xmm0, %xmm7 subsd %xmm7, %xmm3 movsd 5 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm2 movsd 6 * SIZE(AA), %xmm5 mulsd %xmm2, %xmm5 subsd %xmm5, %xmm1 movsd 7 * SIZE(AA), %xmm6 mulsd %xmm2, %xmm6 subsd %xmm6, %xmm3 movsd 10 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm1 movsd 11 * SIZE(AA), %xmm5 mulsd %xmm1, %xmm5 subsd %xmm5, %xmm3 movsd 15 * SIZE(AA), %xmm4 mulsd %xmm4, %xmm3 unpcklpd %xmm2, %xmm0 unpcklpd %xmm3, %xmm1#endif#if defined(RN) || defined(RT) movsd 0 * SIZE(B), %xmm4 movhpd 0 * SIZE(B), %xmm4 mulpd %xmm4, %xmm0 mulpd %xmm4, %xmm1#endif#if defined(LN) || defined(LT) movapd %xmm0, 0 * SIZE(B) movapd %xmm1, 2 * SIZE(B) movsd %xmm0, 0 * SIZE(BB) movsd %xmm0, 1 * SIZE(BB) movhpd %xmm0, 2 * SIZE(BB) movhpd %xmm0, 3 * SIZE(BB) movsd %xmm1, 4 * SIZE(BB) movsd %xmm1, 5 * SIZE(BB) movhpd %xmm1, 6 * SIZE(BB) movhpd %xmm1, 7 * SIZE(BB)#else movapd %xmm0, 0 * SIZE(AA) movapd %xmm1, 2 * SIZE(AA)#endif#ifdef LN subl $4 * SIZE, %esi#endif movsd %xmm0, 0 * SIZE(%esi) movhpd %xmm0, 1 * SIZE(%esi) movsd %xmm1, 2 * SIZE(%esi) movhpd %xmm1, 3 * SIZE(%esi)#ifndef LN addl $4 * SIZE, %esi#endif#if defined(LT) || defined(RN) movl K, %eax subl KK, %eax leal (,%eax, SIZE), %eax leal (AA, %eax, 4), AA#ifdef LT addl $4 * SIZE, B#endif#endif#ifdef LN subl $4, KK movl BORIG, B#endif#ifdef LT addl $4, KK#endif#ifdef RT movl K, %eax movl BORIG, B sall $2 + BASE_SHIFT, %eax addl %eax, AORIG#endif BRANCH decl %ebx # i -- jg .L110 ALIGN_2 .L130: movl M, %ebx testl $2, %ebx jle .L150#ifdef LN movl K, %eax sall $1 + BASE_SHIFT, %eax subl %eax, AORIG#endif#if defined(LN) || defined(RT) movl KK, %eax movl AORIG, AA leal (, %eax, SIZE), %eax leal (AA, %eax, 2), AA#endif leal BUFFER, BB movapd 0 * SIZE(BB), %xmm2 pxor %xmm4, %xmm4 movapd 0 * SIZE(AA), %xmm0 pxor %xmm5, %xmm5 movapd 8 * SIZE(BB), %xmm3 pxor %xmm6, %xmm6 movapd 8 * SIZE(AA), %xmm1 pxor %xmm7, %xmm7#if defined(LN) || defined(RT) movl KK, %eax sall $0 + BASE_SHIFT, %eax leal (BB, %eax, 2), BB#endif #if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif sarl $3, %eax je .L132.L131: mulpd %xmm0, %xmm2 movapd 2 * SIZE(AA), %xmm0 addpd %xmm2, %xmm4 mulpd 2 * SIZE(BB), %xmm0 movapd 16 * SIZE(BB), %xmm2 addpd %xmm0, %xmm5 movapd 4 * SIZE(AA), %xmm0 mulpd 4 * SIZE(BB), %xmm0 addpd %xmm0, %xmm6 movapd 6 * SIZE(AA), %xmm0 mulpd 6 * SIZE(BB), %xmm0 addpd %xmm0, %xmm7 movapd 16 * SIZE(AA), %xmm0 mulpd %xmm1, %xmm3 movapd 10 * SIZE(AA), %xmm1 addpd %xmm3, %xmm4 mulpd 10 * SIZE(BB), %xmm1 movapd 24 * SIZE(BB), %xmm3 addpd %xmm1, %xmm5 movapd 12 * SIZE(AA), %xmm1 mulpd 12 * SIZE(BB), %xmm1 addpd %xmm1, %xmm6
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -