📄 trsm_kernel_ln_4x4_penryn.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 OLD_M %rdi#define OLD_N %rsi#define OLD_K %rdx#define M %r13#define N %r14#define K %r15#define A %rcx#define B %r8#define C %r9#define LDC %r10 #define I %r11#define AO %rdi#define BO %rsi#define CO1 %rbx#define CO2 %rbp#define KK %rdx#define BB %r12#ifndef WINDOWS_ABI#define STACKSIZE 128#define OLD_LDC 8 + STACKSIZE(%rsp)#define OLD_OFFSET 16 + STACKSIZE(%rsp)#define OFFSET 48(%rsp)#define J 56(%rsp)#define KKK 64(%rsp)#define AORIG 72(%rsp)#else#define STACKSIZE 256#define OLD_A 40 + STACKSIZE(%rsp)#define OLD_B 48 + STACKSIZE(%rsp)#define OLD_C 56 + STACKSIZE(%rsp)#define OLD_LDC 64 + STACKSIZE(%rsp)#define OLD_OFFSET 72 + STACKSIZE(%rsp)#define OFFSET 224(%rsp)#define J 232(%rsp)#define KKK 240(%rsp)#define AORIG 248(%rsp)#endif#define PREFETCH_R (8 * 4 + 0)#define PREFETCHSIZE (8 * 21 + 6)#define PREFETCH prefetcht0 PROLOGUE PROFCODE subq $STACKSIZE, %rsp movq %rbx, 0(%rsp) movq %rbp, 8(%rsp) movq %r12, 16(%rsp) movq %r13, 24(%rsp) movq %r14, 32(%rsp) movq %r15, 40(%rsp)#ifdef WINDOWS_ABI movq %rdi, 48(%rsp) movq %rsi, 56(%rsp) movups %xmm6, 64(%rsp) movups %xmm7, 80(%rsp) movups %xmm8, 96(%rsp) movups %xmm9, 112(%rsp) movups %xmm10, 128(%rsp) movups %xmm11, 144(%rsp) movups %xmm12, 160(%rsp) movups %xmm13, 176(%rsp) movups %xmm14, 192(%rsp) movups %xmm15, 208(%rsp) movq ARG1, OLD_M movq ARG2, OLD_N movq ARG3, OLD_K movq OLD_A, A movq OLD_B, B movq OLD_C, C#endif movq OLD_M, M movq OLD_N, N movq OLD_K, K movq OLD_LDC, LDC movq OLD_OFFSET, KK subq $-16 * SIZE, A subq $-16 * SIZE, B leaq (, LDC, SIZE), LDC movq KK, OFFSET negq KK#ifdef LN leaq (, M, SIZE), %rax addq %rax, C imulq K, %rax addq %rax, A#endif#ifdef RT leaq (, N, SIZE), %rax imulq K, %rax addq %rax, B movq N, %rax imulq LDC, %rax addq %rax, C#endif#ifdef RT movq N, %rax subq OFFSET, %rax movq %rax, KK#endif movq N, J sarq $2, J jle,pn .L40 ALIGN_4.L01:#if defined(LT) || defined(RN) movq A, AO#else movq A, AORIG#endif#ifdef RT movq K, %rax salq $2 + BASE_SHIFT, %rax subq %rax, B leaq (, LDC, 4), %rax subq %rax, C#endif movq C, CO1 leaq (C, LDC, 1), CO2#ifndef RT leaq (C, LDC, 4), C#endif#ifdef LN movq OFFSET, %rax addq M, %rax movq %rax, KK#endif movq K, %rax salq $BASE_SHIFT + 2, %rax leaq (B, %rax), BB#ifdef LT movq OFFSET, %rax movq %rax, KK#endif testq $1, M jle,pt .L20 ALIGN_4#ifdef LN movq K, %rax salq $BASE_SHIFT, %rax subq %rax, AORIG#endif#if defined(LN) || defined(RT) movq KK, %rax leaq (, %rax, SIZE), %rax movq AORIG, AO leaq (AO, %rax, 1), AO leaq (B, %rax, 4), BO#else movq B, BO#endif movsd -16 * SIZE(AO), %xmm0 movaps -16 * SIZE(BO), %xmm2 movaps -14 * SIZE(BO), %xmm3 pxor %xmm8, %xmm8 pxor %xmm9, %xmm9 pxor %xmm10, %xmm10 pxor %xmm11, %xmm11#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif sarq $2, %rax jle,pn .L35 ALIGN_4.L32: PREFETCH (PREFETCHSIZE + 0) * SIZE(AO) shufps $0x44, %xmm0, %xmm0 mulpd %xmm0, %xmm2 mulpd %xmm0, %xmm3 movsd -15 * SIZE(AO), %xmm0 addpd %xmm2, %xmm8 movaps -12 * SIZE(BO), %xmm2 addpd %xmm3, %xmm9 movaps -10 * SIZE(BO), %xmm3 shufps $0x44, %xmm0, %xmm0 mulpd %xmm0, %xmm2 mulpd %xmm0, %xmm3 movsd -14 * SIZE(AO), %xmm0 addpd %xmm2, %xmm10 movaps -8 * SIZE(BO), %xmm2 addpd %xmm3, %xmm11 movaps -6 * SIZE(BO), %xmm3 shufps $0x44, %xmm0, %xmm0 mulpd %xmm0, %xmm2 mulpd %xmm0, %xmm3 movsd -13 * SIZE(AO), %xmm0 addpd %xmm2, %xmm8 movaps -4 * SIZE(BO), %xmm2 addpd %xmm3, %xmm9 movaps -2 * SIZE(BO), %xmm3 shufps $0x44, %xmm0, %xmm0 mulpd %xmm0, %xmm2 mulpd %xmm0, %xmm3 movsd -12 * SIZE(AO), %xmm0 addpd %xmm2, %xmm10 movaps 0 * SIZE(BO), %xmm2 addpd %xmm3, %xmm11 movaps 2 * SIZE(BO), %xmm3 subq $ -4 * SIZE, AO subq $-16 * SIZE, BO subq $1, %rax jg,pt .L32 ALIGN_4.L35:#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif andq $3, %rax # if (k & 1) je,pt .L38 ALIGN_4.L36: shufps $0x44, %xmm0, %xmm0 mulpd %xmm0, %xmm2 mulpd %xmm0, %xmm3 movsd -15 * SIZE(AO), %xmm0 addpd %xmm2, %xmm8 movaps -12 * SIZE(BO), %xmm2 addpd %xmm3, %xmm9 movaps -10 * SIZE(BO), %xmm3 addq $1 * SIZE, AO addq $4 * SIZE, BO subq $1, %rax jg,pt .L36 ALIGN_4.L38:#if defined(LN) || defined(RT) movq KK, %rax#ifdef LN subq $1, %rax#else subq $4, %rax#endif leaq (, %rax, SIZE), %rax movq AORIG, AO leaq (AO, %rax, 1), AO leaq (B, %rax, 4), BO#endif addpd %xmm10, %xmm8 addpd %xmm11, %xmm9#if defined(LN) || defined(LT) movaps -16 * SIZE(BO), %xmm12 movaps -14 * SIZE(BO), %xmm13#else movaps -16 * SIZE(AO), %xmm12 movaps -14 * SIZE(AO), %xmm13#endif subpd %xmm8, %xmm12 subpd %xmm9, %xmm13#if defined(RN) || defined(RT) movhlps %xmm13, %xmm15 movsd %xmm13, %xmm14 movhlps %xmm12, %xmm13 movsd %xmm12, %xmm12#endif#if defined(LN) || defined(LT) movddup -16 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm12 mulpd %xmm8, %xmm13#endif#ifdef RN mulsd -16 * SIZE(BO), %xmm12 movlpd -15 * SIZE(BO), %xmm9 mulsd %xmm12, %xmm9 subsd %xmm9, %xmm13 movlpd -14 * SIZE(BO), %xmm10 mulsd %xmm12, %xmm10 subsd %xmm10, %xmm14 movlpd -13 * SIZE(BO), %xmm11 mulsd %xmm12, %xmm11 subsd %xmm11, %xmm15 mulsd -11 * SIZE(BO), %xmm13 movlpd -10 * SIZE(BO), %xmm9 mulsd %xmm13, %xmm9 subsd %xmm9, %xmm14 movlpd -9 * SIZE(BO), %xmm10 mulsd %xmm13, %xmm10 subsd %xmm10, %xmm15 mulsd -6 * SIZE(BO), %xmm14 movlpd -5 * SIZE(BO), %xmm9 mulsd %xmm14, %xmm9 subsd %xmm9, %xmm15 mulsd -1 * SIZE(BO), %xmm15#endif#ifdef RT mulsd -1 * SIZE(BO), %xmm15 movlpd -2 * SIZE(BO), %xmm9 mulsd %xmm15, %xmm9 subsd %xmm9, %xmm14 movlpd -3 * SIZE(BO), %xmm10 mulsd %xmm15, %xmm10 subsd %xmm10, %xmm13 movlpd -4 * SIZE(BO), %xmm11 mulsd %xmm15, %xmm11 subsd %xmm11, %xmm12 mulsd -6 * SIZE(BO), %xmm14 movlpd -7 * SIZE(BO), %xmm9 mulsd %xmm14, %xmm9 subsd %xmm9, %xmm13 movlpd -8 * SIZE(BO), %xmm10 mulsd %xmm14, %xmm10 subsd %xmm10, %xmm12 mulsd -11 * SIZE(BO), %xmm13 movlpd -12 * SIZE(BO), %xmm9 mulsd %xmm13, %xmm9 subsd %xmm9, %xmm12 mulsd -16 * SIZE(BO), %xmm12#endif#ifdef LN subq $1 * SIZE, CO1 subq $1 * SIZE, CO2#endif#if defined(LN) || defined(LT) movsd %xmm12, 0 * SIZE(CO1) movhps %xmm12, 0 * SIZE(CO2) movsd %xmm13, 0 * SIZE(CO1, LDC, 2) movhps %xmm13, 0 * SIZE(CO2, LDC, 2) movaps %xmm12, -16 * SIZE(BO) movaps %xmm13, -14 * SIZE(BO)#else movsd %xmm12, 0 * SIZE(CO1) movsd %xmm13, 0 * SIZE(CO2) movsd %xmm14, 0 * SIZE(CO1, LDC, 2) movsd %xmm15, 0 * SIZE(CO2, LDC, 2) movsd %xmm12, -16 * SIZE(AO) movsd %xmm13, -15 * SIZE(AO) movsd %xmm14, -14 * SIZE(AO) movsd %xmm15, -13 * SIZE(AO)#endif#ifndef LN addq $1 * SIZE, CO1 addq $1 * SIZE, CO2#endif#if defined(LT) || defined(RN) movq K, %rax subq KK, %rax leaq (,%rax, SIZE), %rax leaq (AO, %rax, 1), AO leaq (BO, %rax, 4), BO#endif#ifdef LN subq $1, KK#endif#ifdef LT addq $1, KK#endif#ifdef RT movq K, %rax salq $BASE_SHIFT, %rax addq %rax, AORIG#endif ALIGN_4 .L20: testq $2, M jle,pt .L30 ALIGN_4#ifdef LN movq K, %rax salq $1 + BASE_SHIFT, %rax subq %rax, AORIG#endif#if defined(LN) || defined(RT) movq KK, %rax leaq (, %rax, SIZE), %rax movq AORIG, AO leaq (AO, %rax, 2), AO leaq (B, %rax, 4), BO#else movq B, BO#endif movaps -16 * SIZE(AO), %xmm0 pxor %xmm3, %xmm3 movaps -16 * SIZE(BO), %xmm2 pxor %xmm5, %xmm5 pxor %xmm8, %xmm8 pxor %xmm9, %xmm9 pxor %xmm10, %xmm10 pxor %xmm11, %xmm11#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif sarq $2, %rax jle,pn .L25 ALIGN_4.L22: addpd %xmm3, %xmm11 movaps -14 * SIZE(BO), %xmm3 pshufd $0x4e, %xmm2, %xmm7 PREFETCH (PREFETCHSIZE + 0) * SIZE(AO) mulpd %xmm0, %xmm2 addpd %xmm5, %xmm10 mulpd %xmm0, %xmm7 addpd %xmm2, %xmm9 movaps -12 * SIZE(BO), %xmm2 pshufd $0x4e, %xmm3, %xmm5 mulpd %xmm0, %xmm3 addpd %xmm7, %xmm8 mulpd %xmm0, %xmm5 movaps -14 * SIZE(AO), %xmm0 addpd %xmm3, %xmm11 movaps -10 * SIZE(BO), %xmm3 pshufd $0x4e, %xmm2, %xmm7 mulpd %xmm0, %xmm2 addpd %xmm5, %xmm10 mulpd %xmm0, %xmm7 addpd %xmm2, %xmm9 movaps -8 * SIZE(BO), %xmm2 pshufd $0x4e, %xmm3, %xmm5 mulpd %xmm0, %xmm3 addpd %xmm7, %xmm8 mulpd %xmm0, %xmm5 movaps -12 * SIZE(AO), %xmm0 addpd %xmm3, %xmm11 movaps -6 * SIZE(BO), %xmm3 pshufd $0x4e, %xmm2, %xmm7 mulpd %xmm0, %xmm2 addpd %xmm5, %xmm10 mulpd %xmm0, %xmm7 addpd %xmm2, %xmm9 movaps -4 * SIZE(BO), %xmm2 pshufd $0x4e, %xmm3, %xmm5 mulpd %xmm0, %xmm3 addpd %xmm7, %xmm8 mulpd %xmm0, %xmm5 movaps -10 * SIZE(AO), %xmm0 addpd %xmm3, %xmm11 movaps -2 * SIZE(BO), %xmm3 pshufd $0x4e, %xmm2, %xmm7 mulpd %xmm0, %xmm2 addpd %xmm5, %xmm10 mulpd %xmm0, %xmm7 subq $ -8 * SIZE, AO addpd %xmm2, %xmm9 movaps 0 * SIZE(BO), %xmm2 pshufd $0x4e, %xmm3, %xmm5 mulpd %xmm0, %xmm3 addpd %xmm7, %xmm8 mulpd %xmm0, %xmm5 movaps -16 * SIZE(AO), %xmm0 subq $-16 * SIZE, BO subq $1, %rax jg,pt .L22 ALIGN_4.L25:#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif andq $3, %rax # if (k & 1) je,pt .L28 ALIGN_4.L26: addpd %xmm3, %xmm11 movaps -14 * SIZE(BO), %xmm3 pshufd $0x4e, %xmm2, %xmm7 mulpd %xmm0, %xmm2 addpd %xmm5, %xmm10 mulpd %xmm0, %xmm7 addpd %xmm2, %xmm9 movaps -12 * SIZE(BO), %xmm2 pshufd $0x4e, %xmm3, %xmm5 mulpd %xmm0, %xmm3 addpd %xmm7, %xmm8 mulpd %xmm0, %xmm5 movaps -14 * SIZE(AO), %xmm0 addq $2 * SIZE, AO addq $4 * SIZE, BO subq $1, %rax jg,pt .L26 ALIGN_4.L28:#if defined(LN) || defined(RT) movq KK, %rax#ifdef LN subq $2, %rax#else subq $4, %rax#endif leaq (, %rax, SIZE), %rax movq AORIG, AO leaq (AO, %rax, 2), AO leaq (B, %rax, 4), BO#endif addpd %xmm3, %xmm11 addpd %xmm5, %xmm10 movapd %xmm8, %xmm0 movsd %xmm9, %xmm8 movsd %xmm0, %xmm9 movapd %xmm10, %xmm0 movsd %xmm11, %xmm10 movsd %xmm0, %xmm11#if defined(LN) || defined(LT) movapd %xmm8, %xmm0 unpcklpd %xmm9, %xmm8 unpckhpd %xmm9, %xmm0 movapd %xmm10, %xmm2 unpcklpd %xmm11, %xmm10 unpckhpd %xmm11, %xmm2 movapd -16 * SIZE(BO), %xmm9 movapd -14 * SIZE(BO), %xmm11 movapd -12 * SIZE(BO), %xmm13 movapd -10 * SIZE(BO), %xmm15 subpd %xmm8, %xmm9 subpd %xmm10, %xmm11 subpd %xmm0, %xmm13 subpd %xmm2, %xmm15#else movapd -16 * SIZE(AO), %xmm0 movapd -14 * SIZE(AO), %xmm2 movapd -12 * SIZE(AO), %xmm4 movapd -10 * SIZE(AO), %xmm6 subpd %xmm8, %xmm0 subpd %xmm9, %xmm2 subpd %xmm10, %xmm4 subpd %xmm11, %xmm6#endif#ifdef LN movddup -13 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm13 mulpd %xmm8, %xmm15 movddup -14 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm13, %xmm10 subpd %xmm10, %xmm9 mulpd %xmm15, %xmm12 subpd %xmm12, %xmm11 movddup -16 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm9 mulpd %xmm8, %xmm11#endif#ifdef LT movddup -16 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm9 mulpd %xmm8, %xmm11 movddup -15 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm9, %xmm10 subpd %xmm10, %xmm13 mulpd %xmm11, %xmm12 subpd %xmm12, %xmm15 movddup -13 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm13 mulpd %xmm8, %xmm15#endif#ifdef RN movddup -16 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm0 movddup -15 * SIZE(BO), %xmm9 mulpd %xmm0, %xmm9 subpd %xmm9, %xmm2 movddup -14 * SIZE(BO), %xmm10 mulpd %xmm0, %xmm10 subpd %xmm10, %xmm4 movddup -13 * SIZE(BO), %xmm11 mulpd %xmm0, %xmm11 subpd %xmm11, %xmm6 movddup -11 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm2 movddup -10 * SIZE(BO), %xmm9 mulpd %xmm2, %xmm9 subpd %xmm9, %xmm4 movddup -9 * SIZE(BO), %xmm10 mulpd %xmm2, %xmm10 subpd %xmm10, %xmm6 movddup -6 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm4 movddup -5 * SIZE(BO), %xmm9 mulpd %xmm4, %xmm9 subpd %xmm9, %xmm6 movddup -1 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm6#endif#ifdef RT movddup -1 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm6 movddup -2 * SIZE(BO), %xmm9 mulpd %xmm6, %xmm9 subpd %xmm9, %xmm4 movddup -3 * SIZE(BO), %xmm10 mulpd %xmm6, %xmm10 subpd %xmm10, %xmm2 movddup -4 * SIZE(BO), %xmm11 mulpd %xmm6, %xmm11 subpd %xmm11, %xmm0 movddup -6 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm4 movddup -7 * SIZE(BO), %xmm9 mulpd %xmm4, %xmm9 subpd %xmm9, %xmm2 movddup -8 * SIZE(BO), %xmm10 mulpd %xmm4, %xmm10 subpd %xmm10, %xmm0 movddup -11 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm2 movddup -12 * SIZE(BO), %xmm9 mulpd %xmm2, %xmm9 subpd %xmm9, %xmm0 movddup -16 * SIZE(BO), %xmm8 mulpd %xmm8, %xmm0#endif#ifdef LN subq $2 * SIZE, CO1 subq $2 * SIZE, CO2#endif#if defined(LN) || defined(LT) movsd %xmm9, 0 * SIZE(CO1) movsd %xmm13, 1 * SIZE(CO1) movhpd %xmm9, 0 * SIZE(CO2) movhpd %xmm13, 1 * SIZE(CO2) movsd %xmm11, 0 * SIZE(CO1, LDC, 2) movsd %xmm15, 1 * SIZE(CO1, LDC, 2) movhpd %xmm11, 0 * SIZE(CO2, LDC, 2) movhpd %xmm15, 1 * SIZE(CO2, LDC, 2)#else movsd %xmm0, 0 * SIZE(CO1) movhpd %xmm0, 1 * SIZE(CO1) movsd %xmm2, 0 * SIZE(CO2) movhpd %xmm2, 1 * SIZE(CO2) movsd %xmm4, 0 * SIZE(CO1, LDC, 2) movhpd %xmm4, 1 * SIZE(CO1, LDC, 2) movsd %xmm6, 0 * SIZE(CO2, LDC, 2) movhpd %xmm6, 1 * SIZE(CO2, LDC, 2)#endif#if defined(LN) || defined(LT) movapd %xmm9, -16 * SIZE(BO) movapd %xmm11, -14 * SIZE(BO) movapd %xmm13, -12 * SIZE(BO) movapd %xmm15, -10 * SIZE(BO)#else movapd %xmm0, -16 * SIZE(AO) movapd %xmm2, -14 * SIZE(AO) movapd %xmm4, -12 * SIZE(AO) movapd %xmm6, -10 * SIZE(AO)#endif#ifndef LN addq $2 * SIZE, CO1 addq $2 * SIZE, CO2#endif#if defined(LT) || defined(RN) movq K, %rax subq KK, %rax leaq (,%rax, SIZE), %rax leaq (AO, %rax, 2), AO leaq (BO, %rax, 4), BO#endif#ifdef LN subq $2, KK#endif#ifdef LT addq $2, KK#endif#ifdef RT movq K, %rax salq $1 + BASE_SHIFT, %rax addq %rax, AORIG#endif ALIGN_4 .L30: movq M, I sarq $2, I # i = (m >> 2) jle,pn .L39 ALIGN_4.L11:
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -