📄 trsm_kernel_ln_4x4_sse.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"#define STACK 16 #define OLD_M 4 + STACK(%esi)#define OLD_N 8 + STACK(%esi)#define OLD_K 12 + STACK(%esi)#define OLD_A 20 + STACK(%esi)#define OLD_B 24 + STACK(%esi)#define OLD_C 28 + STACK(%esi)#define OLD_LDC 32 + STACK(%esi)#define STACK_OFFT 36 + STACK(%esi)#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 LOCAL_BUFFER_SIZE GEMM_Q * GEMM_UNROLL_N * COMPSIZE * 16#if defined(OPTERON) || defined(BARCELONA)#define PREFETCH prefetch#define PREFETCHW prefetchw#define PREFETCHSIZE (16 * 10 + 8)#endif#ifdef PENTIUM4#define PREFETCH prefetcht0#define PREFETCHW prefetcht0#define PREFETCHSIZE 96#endif#ifdef PENRYN#define PREFETCH prefetcht0#define PREFETCHW prefetcht0#define PREFETCHSIZE 96#endif#define B %edi#define AA %edx#define BB %ecx#define LDC %ebp#define CO1 %esi#define KERNEL1(address) \ mulps %xmm0, %xmm2; \ addps %xmm2, %xmm4; \ movaps 4 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm0, %xmm2; \ PREFETCH (PREFETCHSIZE + 0) * SIZE + (address) * 1 * SIZE(AA); \ addps %xmm2, %xmm5; \ movaps 8 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm0, %xmm2; \ mulps 12 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \ addps %xmm2, %xmm6; \ movaps 32 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ addps %xmm0, %xmm7; \ movaps 4 * SIZE + (address) * 1 * SIZE(AA), %xmm0#define KERNEL2(address) \ mulps %xmm0, %xmm3; \ addps %xmm3, %xmm4; \ movaps 20 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm0, %xmm3; \ addps %xmm3, %xmm5; \ movaps 24 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm0, %xmm3; \ mulps 28 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \ addps %xmm3, %xmm6; \ movaps 48 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ addps %xmm0, %xmm7; \ movaps 8 * SIZE + (address) * 1 * SIZE(AA), %xmm0#define KERNEL3(address) \ mulps %xmm0, %xmm2; \ addps %xmm2, %xmm4; \ movaps 36 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm0, %xmm2; \ addps %xmm2, %xmm5; \ movaps 40 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm0, %xmm2; \ mulps 44 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \ addps %xmm2, %xmm6; \ movaps 64 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ addps %xmm0, %xmm7; \ movaps 12 * SIZE + (address) * 1 * SIZE(AA), %xmm0#define KERNEL4(address) \ mulps %xmm0, %xmm3; \ addps %xmm3, %xmm4; \ movaps 52 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm0, %xmm3; \ addps %xmm3, %xmm5; \ movaps 56 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm0, %xmm3; \ mulps 60 * SIZE + (address) * 4 * SIZE(BB), %xmm0; \ addps %xmm3, %xmm6; \ movaps 80 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ addps %xmm0, %xmm7; \ movaps 32 * SIZE + (address) * 1 * SIZE(AA), %xmm0#define KERNEL5(address) \ mulps %xmm1, %xmm2; \ addps %xmm2, %xmm4; \ movaps 68 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm1, %xmm2; \ addps %xmm2, %xmm5; \ movaps 72 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm1, %xmm2; \ mulps 76 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \ addps %xmm2, %xmm6; \ movaps 96 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ addps %xmm1, %xmm7; \ movaps 20 * SIZE + (address) * 1 * SIZE(AA), %xmm1#define KERNEL6(address) \ mulps %xmm1, %xmm3; \ addps %xmm3, %xmm4; \ movaps 84 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm1, %xmm3; \ addps %xmm3, %xmm5; \ movaps 88 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm1, %xmm3; \ mulps 92 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \ addps %xmm3, %xmm6; \ movaps 112 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ addps %xmm1, %xmm7; \ movaps 24 * SIZE + (address) * 1 * SIZE(AA), %xmm1#define KERNEL7(address) \ mulps %xmm1, %xmm2; \ addps %xmm2, %xmm4; \ movaps 100 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm1, %xmm2; \ addps %xmm2, %xmm5; \ movaps 104 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ mulps %xmm1, %xmm2; \ mulps 108 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \ addps %xmm2, %xmm6; \ movaps 128 * SIZE + (address) * 4 * SIZE(BB), %xmm2; \ addps %xmm1, %xmm7; \ movaps 28 * SIZE + (address) * 1 * SIZE(AA), %xmm1#define KERNEL8(address) \ mulps %xmm1, %xmm3; \ addps %xmm3, %xmm4; \ movaps 116 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm1, %xmm3; \ addps %xmm3, %xmm5; \ movaps 120 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ mulps %xmm1, %xmm3; \ mulps 124 * SIZE + (address) * 4 * SIZE(BB), %xmm1; \ addps %xmm3, %xmm6; \ movaps 144 * SIZE + (address) * 4 * SIZE(BB), %xmm3; \ addps %xmm1, %xmm7; \ movaps 48 * SIZE + (address) * 1 * SIZE(AA), %xmm1; PROLOGUE pushl %ebp pushl %edi pushl %esi pushl %ebx PROFCODE EMMS movl %esp, %esi subl $128 + LOCAL_BUFFER_SIZE, %esp andl $-1024, %esp#ifdef WINDOWS_ABI#if GEMM_Q > 192 movl $0, 4096 * 3(%esp)#endif#if GEMM_Q > 128 movl $0, 4096 * 2(%esp)#endif#if GEMM_Q > 64 movl $0, 4096 * 1(%esp)#endif movl $0, 4096 * 0(%esp)#endif movl OLD_M, %ebx movl OLD_N, %eax movl OLD_K, %ecx movl OLD_A, %edx movl %ebx, M movl %eax, N movl %ecx, K movl %edx, A movl %esi, OLD_STACK movss STACK_OFFT, %xmm4 movl OLD_B, B movl OLD_C, %ebx movl %ebx, C movl OLD_LDC, LDC movss %xmm4, OFFSET movss %xmm4, KK leal (, LDC, SIZE), 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 sarl $2, %eax movl %eax, J jle .L40.L01:#ifdef LN movl OFFSET, %eax addl M, %eax movl %eax, KK#endif leal BUFFER, %ecx#ifdef RT movl K, %eax sall $2 + BASE_SHIFT, %eax subl %eax, B#endif#if defined(LN) || defined(RT) movl KK, %eax movl B, BORIG sall $2 + BASE_SHIFT, %eax leal (B, %eax, 1), B leal (BB, %eax, 4), 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 $1, %eax jle .L05 ALIGN_4 .L02: movaps 0 * SIZE(B), %xmm3 movaps 4 * SIZE(B), %xmm7 pshufd $0x00, %xmm3, %xmm0 pshufd $0x55, %xmm3, %xmm1 pshufd $0xaa, %xmm3, %xmm2 pshufd $0xff, %xmm3, %xmm3 pshufd $0x00, %xmm7, %xmm4 pshufd $0x55, %xmm7, %xmm5 pshufd $0xaa, %xmm7, %xmm6 pshufd $0xff, %xmm7, %xmm7 movaps %xmm0, 0 * SIZE(BB) movaps %xmm1, 4 * SIZE(BB) movaps %xmm2, 8 * SIZE(BB) movaps %xmm3, 12 * SIZE(BB) movaps %xmm4, 16 * SIZE(BB) movaps %xmm5, 20 * SIZE(BB) movaps %xmm6, 24 * SIZE(BB) movaps %xmm7, 28 * SIZE(BB) addl $ 8 * SIZE, B addl $32 * SIZE, %ecx decl %eax jne .L02 ALIGN_2.L05:#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif andl $1, %eax BRANCH jle .L10 movaps 0 * SIZE(B), %xmm3 pshufd $0x00, %xmm3, %xmm0 pshufd $0x55, %xmm3, %xmm1 pshufd $0xaa, %xmm3, %xmm2 pshufd $0xff, %xmm3, %xmm3 movaps %xmm0, 0 * SIZE(BB) movaps %xmm1, 4 * SIZE(BB) movaps %xmm2, 8 * SIZE(BB) movaps %xmm3, 12 * SIZE(BB) addl $4 * SIZE, B ALIGN_4 .L10:#if defined(LT) || defined(RN) movl A, AA#else movl A, %eax movl %eax, AORIG#endif leal (, LDC, 4), %eax#ifdef RT subl %eax, C#endif movl C, CO1#ifndef RT addl %eax, C#endif testl $1, M je .L20#ifdef LN movl K, %eax sall $BASE_SHIFT, %eax subl %eax, AORIG#endif#if defined(LN) || defined(RT) movl KK, %eax movl AORIG, AA leal (AA, %eax, SIZE), AA#endif leal BUFFER, BB#if defined(LN) || defined(RT) movl KK, %eax sall $2 + BASE_SHIFT, %eax leal (BB, %eax, 4), BB#endif movss 0 * SIZE(AA), %xmm0 pxor %xmm4, %xmm4 movss 4 * SIZE(AA), %xmm1 pxor %xmm5, %xmm5 movss 0 * SIZE(BB), %xmm2 pxor %xmm6, %xmm6 movss 16 * SIZE(BB), %xmm3 pxor %xmm7, %xmm7#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif sarl $3, %eax je .L35 ALIGN_4.L32: mulss %xmm0, %xmm2 addss %xmm2, %xmm4#if defined(OPTERON) || defined(BARCELONA) prefetcht0 (PREFETCHSIZE + 0) * SIZE(AA)#endif movss 4 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 addss %xmm2, %xmm5 movss 8 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 mulss 12 * SIZE(BB), %xmm0 addss %xmm2, %xmm6 movss 32 * SIZE(BB), %xmm2 addss %xmm0, %xmm7 movss 1 * SIZE(AA), %xmm0 mulss %xmm0, %xmm3 addss %xmm3, %xmm4 movss 20 * SIZE(BB), %xmm3 mulss %xmm0, %xmm3 addss %xmm3, %xmm5 movss 24 * SIZE(BB), %xmm3 mulss %xmm0, %xmm3 mulss 28 * SIZE(BB), %xmm0 addss %xmm3, %xmm6 movss 48 * SIZE(BB), %xmm3 addss %xmm0, %xmm7 movss 2 * SIZE(AA), %xmm0 mulss %xmm0, %xmm2 addss %xmm2, %xmm4 movss 36 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 addss %xmm2, %xmm5 movss 40 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 mulss 44 * SIZE(BB), %xmm0 addss %xmm2, %xmm6 movss 64 * SIZE(BB), %xmm2 addss %xmm0, %xmm7 movss 3 * SIZE(AA), %xmm0 mulss %xmm0, %xmm3 addss %xmm3, %xmm4 movss 52 * SIZE(BB), %xmm3 mulss %xmm0, %xmm3 addss %xmm3, %xmm5 movss 56 * SIZE(BB), %xmm3 mulss %xmm0, %xmm3 mulss 60 * SIZE(BB), %xmm0 addss %xmm3, %xmm6 movss 80 * SIZE(BB), %xmm3 addss %xmm0, %xmm7 movss 8 * SIZE(AA), %xmm0 mulss %xmm1, %xmm2 addss %xmm2, %xmm4 movss 68 * SIZE(BB), %xmm2 mulss %xmm1, %xmm2 addss %xmm2, %xmm5 movss 72 * SIZE(BB), %xmm2 mulss %xmm1, %xmm2 mulss 76 * SIZE(BB), %xmm1 addss %xmm2, %xmm6 movss 96 * SIZE(BB), %xmm2 addss %xmm1, %xmm7 movss 5 * SIZE(AA), %xmm1 mulss %xmm1, %xmm3 addss %xmm3, %xmm4 movss 84 * SIZE(BB), %xmm3 mulss %xmm1, %xmm3 addss %xmm3, %xmm5 movss 88 * SIZE(BB), %xmm3 mulss %xmm1, %xmm3 mulss 92 * SIZE(BB), %xmm1 addss %xmm3, %xmm6 movss 112 * SIZE(BB), %xmm3 addss %xmm1, %xmm7 movss 6 * SIZE(AA), %xmm1 mulss %xmm1, %xmm2 addss %xmm2, %xmm4 movss 100 * SIZE(BB), %xmm2 mulss %xmm1, %xmm2 addss %xmm2, %xmm5 movss 104 * SIZE(BB), %xmm2 mulss %xmm1, %xmm2 mulss 108 * SIZE(BB), %xmm1 addss %xmm2, %xmm6 movss 128 * SIZE(BB), %xmm2 addss %xmm1, %xmm7 movss 7 * SIZE(AA), %xmm1 mulss %xmm1, %xmm3 addss %xmm3, %xmm4 movss 116 * SIZE(BB), %xmm3 mulss %xmm1, %xmm3 addss %xmm3, %xmm5 movss 120 * SIZE(BB), %xmm3 mulss %xmm1, %xmm3 mulss 124 * SIZE(BB), %xmm1 addss %xmm3, %xmm6 movss 144 * SIZE(BB), %xmm3 addss %xmm1, %xmm7 movss 12 * SIZE(AA), %xmm1 addl $ 8 * SIZE, AA addl $128 * SIZE, BB decl %eax jne .L32 ALIGN_4.L35:#if defined(LT) || defined(RN) movl KK, %eax#else movl K, %eax subl KK, %eax#endif andl $7, %eax # if (k & 1) BRANCH je .L38 ALIGN_4.L36: mulss %xmm0, %xmm2 addss %xmm2, %xmm4 movss 4 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 addss %xmm2, %xmm5 movss 8 * SIZE(BB), %xmm2 mulss %xmm0, %xmm2 mulss 12 * SIZE(BB), %xmm0 addss %xmm2, %xmm6 movss 16 * SIZE(BB), %xmm2 addss %xmm0, %xmm7 movss 1 * SIZE(AA), %xmm0 addl $ 1 * SIZE, AA addl $16 * SIZE, BB decl %eax jg .L36 ALIGN_4.L38:#if defined(LN) || defined(RT) movl KK, %eax#ifdef LN subl $1, %eax#else subl $4, %eax#endif movl AORIG, AA movl BORIG, B leal BUFFER, BB leal (AA, %eax, SIZE), AA sall $2 + BASE_SHIFT, %eax leal (B, %eax, 1), B leal (BB, %eax, 4), BB#endif#if defined(LN) || defined(LT) unpcklps %xmm6, %xmm4 unpcklps %xmm7, %xmm5 unpcklps %xmm5, %xmm4 movaps 0 * SIZE(B), %xmm1 subps %xmm4, %xmm1#else movss 0 * SIZE(AA), %xmm0 movss 1 * SIZE(AA), %xmm1 movss 2 * SIZE(AA), %xmm2 movss 3 * SIZE(AA), %xmm3 subss %xmm4, %xmm0 subss %xmm5, %xmm1 subss %xmm6, %xmm2 subss %xmm7, %xmm3#endif#if defined(LN) || defined(LT) movss 0 * SIZE(AA), %xmm4 pshufd $0x00, %xmm4, %xmm6 mulps %xmm6, %xmm1#endif#ifdef RN movaps 0 * SIZE(B), %xmm6 pshufd $0x00, %xmm6, %xmm7 mulss %xmm7, %xmm0 pshufd $0x55, %xmm6, %xmm7 mulss %xmm0, %xmm7 subss %xmm7, %xmm1 pshufd $0xaa, %xmm6, %xmm7 mulss %xmm0, %xmm7 subss %xmm7, %xmm2 pshufd $0xff, %xmm6, %xmm7 mulss %xmm0, %xmm7 subss %xmm7, %xmm3 movaps 4 * SIZE(B), %xmm6 pshufd $0x55, %xmm6, %xmm7 mulss %xmm7, %xmm1 pshufd $0xaa, %xmm6, %xmm7 mulss %xmm1, %xmm7 subss %xmm7, %xmm2 pshufd $0xff, %xmm6, %xmm7 mulss %xmm1, %xmm7 subss %xmm7, %xmm3 movaps 8 * SIZE(B), %xmm6 pshufd $0xaa, %xmm6, %xmm7 mulss %xmm7, %xmm2 pshufd $0xff, %xmm6, %xmm7 mulss %xmm2, %xmm7 subss %xmm7, %xmm3 movaps 12 * SIZE(B), %xmm6 pshufd $0xff, %xmm6, %xmm7 mulss %xmm7, %xmm3#endif#ifdef RT movaps 12 * SIZE(B), %xmm6 pshufd $0xff, %xmm6, %xmm7 mulss %xmm7, %xmm3 pshufd $0xaa, %xmm6, %xmm7 mulss %xmm3, %xmm7 subss %xmm7, %xmm2 pshufd $0x55, %xmm6, %xmm7 mulss %xmm3, %xmm7 subss %xmm7, %xmm1 pshufd $0x00, %xmm6, %xmm7 mulss %xmm3, %xmm7 subss %xmm7, %xmm0 movaps 8 * SIZE(B), %xmm6 pshufd $0xaa, %xmm6, %xmm7 mulss %xmm7, %xmm2 pshufd $0x55, %xmm6, %xmm7 mulss %xmm2, %xmm7 subss %xmm7, %xmm1 pshufd $0x00, %xmm6, %xmm7 mulss %xmm2, %xmm7 subss %xmm7, %xmm0 movaps 4 * SIZE(B), %xmm6 pshufd $0x55, %xmm6, %xmm7 mulss %xmm7, %xmm1 pshufd $0x00, %xmm6, %xmm7 mulss %xmm1, %xmm7 subss %xmm7, %xmm0 movaps 0 * SIZE(B), %xmm6 pshufd $0x00, %xmm6, %xmm7 mulss %xmm7, %xmm0#endif#if defined(LN) || defined(LT) movaps %xmm1, 0 * SIZE(B) pshufd $0x00, %xmm1, %xmm0 pshufd $0x55, %xmm1, %xmm2 pshufd $0xaa, %xmm1, %xmm4 pshufd $0xff, %xmm1, %xmm6 movaps %xmm0, 0 * SIZE(BB) movaps %xmm2, 4 * SIZE(BB) movaps %xmm4, 8 * SIZE(BB) movaps %xmm6, 12 * SIZE(BB)#else movss %xmm0, 0 * SIZE(AA) movss %xmm1, 1 * SIZE(AA) movss %xmm2, 2 * SIZE(AA) movss %xmm3, 3 * SIZE(AA)#endif#ifdef LN subl $1 * SIZE, CO1#endif
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -