📄 trsm_kernel_lt_4x4_core2.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 M %r13#define N %r14#define K %rdx#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 %r12#ifndef WINDOWS_ABI#define STACKSIZE 64#define OLD_LDC 8 + STACKSIZE(%rsp)#define OLD_OFFSET 16 + STACKSIZE(%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)#endif#define J 0(%rsp)#define OFFSET 8(%rsp)#define KKK 16(%rsp)#define AORIG 24(%rsp)#define BORIG 32(%rsp)#define BUFFER 128(%rsp) #define PREFETCH_R (8 * 4 + 0)#define PREFETCH_W (PREFETCH_R)#define PREFETCHSIZE (8 * 17 + 2)#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, K movq OLD_A, A movq OLD_B, B movq OLD_C, C#endif movq OLD_LDC, LDC movq OLD_OFFSET, KK movq %rsp, %r15 # save old stack subq $128 + GEMM_Q * 64, %rsp andq $-4096, %rsp # align stack#ifdef WINDOWS_ABI#if GEMM_Q > 192 movq $0, 4096 * 3(%rsp)#endif#if GEMM_Q > 128 movq $0, 4096 * 2(%rsp)#endif#if GEMM_Q > 64 movq $0, 4096 * 1(%rsp)#endif movq $0, 4096 * 0(%rsp)#endif movq KK, OFFSET movq OLD_M, M movq OLD_N, N subq $-16 * SIZE, A subq $-16 * SIZE, B leaq (, LDC, SIZE), LDC#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 RN negq KK#endif #ifdef RT movq N, %rax subq OFFSET, %rax movq %rax, KK#endif movq N, J sarq $2, J # j = (n >> 2) jle .L40.L01:/* Copying to Sub Buffer */#ifdef LN movq OFFSET, %rax addq M, %rax movq %rax, KK#endif leaq 16 * SIZE + BUFFER, BO#ifdef RT movq K, %rax salq $2 + BASE_SHIFT, %rax subq %rax, B#endif#if defined(LN) || defined(RT) movq KK, %rax movq B, BORIG leaq (, %rax, SIZE), %rax leaq (B, %rax, 4), B leaq (BO, %rax, 8), BO#endif #if defined(LT) movq OFFSET, %rax movq %rax, KK#endif#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif sarq $2, %rax jle .L03 ALIGN_4 .L02: prefetcht0 (PREFETCH_R + 0) * SIZE(B) movapd -16 * SIZE(B), %xmm0 movapd -14 * SIZE(B), %xmm1 movapd -12 * SIZE(B), %xmm2 movapd -10 * SIZE(B), %xmm3 movapd -8 * SIZE(B), %xmm4 movapd -6 * SIZE(B), %xmm5 movapd -4 * SIZE(B), %xmm6 movapd -2 * SIZE(B), %xmm7 prefetcht0 (PREFETCH_R + 8) * SIZE(B) movddup %xmm0, %xmm8 unpckhpd %xmm0, %xmm0 movddup %xmm1, %xmm9 unpckhpd %xmm1, %xmm1 movddup %xmm2, %xmm10 unpckhpd %xmm2, %xmm2 movddup %xmm3, %xmm11 unpckhpd %xmm3, %xmm3 movddup %xmm4, %xmm12 unpckhpd %xmm4, %xmm4 movddup %xmm5, %xmm13 unpckhpd %xmm5, %xmm5 movddup %xmm6, %xmm14 unpckhpd %xmm6, %xmm6 movddup %xmm7, %xmm15 unpckhpd %xmm7, %xmm7 prefetcht0 (PREFETCH_W + 0) * SIZE(BO) movapd %xmm8, -16 * SIZE(BO) movapd %xmm0, -14 * SIZE(BO) movapd %xmm9, -12 * SIZE(BO) movapd %xmm1, -10 * SIZE(BO) prefetcht0 (PREFETCH_W + 8) * SIZE(BO) movapd %xmm10, -8 * SIZE(BO) movapd %xmm2, -6 * SIZE(BO) movapd %xmm11, -4 * SIZE(BO) movapd %xmm3, -2 * SIZE(BO) prefetcht0 (PREFETCH_W + 16) * SIZE(BO) movapd %xmm12, 0 * SIZE(BO) movapd %xmm4, 2 * SIZE(BO) movapd %xmm13, 4 * SIZE(BO) movapd %xmm5, 6 * SIZE(BO) prefetcht0 (PREFETCH_W + 24) * SIZE(BO) movapd %xmm14, 8 * SIZE(BO) movapd %xmm6, 10 * SIZE(BO) movapd %xmm15, 12 * SIZE(BO) movapd %xmm7, 14 * SIZE(BO) subq $-16 * SIZE, B subq $-32 * SIZE, BO subq $1, %rax jne .L02 ALIGN_4.L03:#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif andq $3, %rax BRANCH jle .L10 ALIGN_4.L04: movapd -16 * SIZE(B), %xmm0 movapd -14 * SIZE(B), %xmm1 movddup %xmm0, %xmm8 unpckhpd %xmm0, %xmm0 movddup %xmm1, %xmm9 unpckhpd %xmm1, %xmm1 movapd %xmm8, -16 * SIZE(BO) movapd %xmm0, -14 * SIZE(BO) movapd %xmm9, -12 * SIZE(BO) movapd %xmm1, -10 * SIZE(BO) addq $4 * SIZE, B addq $8 * SIZE, BO subq $1, %rax jne .L04 ALIGN_4 .L10:#if defined(LT) || defined(RN) movq A, AO#else movq A, AORIG#endif#ifdef RT leaq (, LDC, 4), %rax subq %rax, C#endif movq C, CO1 # coffset1 = c leaq (C, LDC, 1), CO2 # coffset2 = c + ldc#ifndef RT leaq (C, LDC, 4), C#endif movq M, I sarq $2, I # i = (m >> 2) jle .L20 ALIGN_4.L11:#ifdef LN movq K, %rax salq $2 + BASE_SHIFT, %rax subq %rax, AORIG#endif#if defined(LN) || defined(RT) movq KK, %rax movq AORIG, AO leaq (, %rax, SIZE), %rax leaq (AO, %rax, 4), AO#endif leaq 16 * SIZE + BUFFER, BO#if defined(LN) || defined(RT) movq KK, %rax salq $2 + BASE_SHIFT, %rax leaq (BO, %rax, 2), BO#endif pxor %xmm8, %xmm8 pxor %xmm9, %xmm9 pxor %xmm10, %xmm10 pxor %xmm11, %xmm11 prefetcht2 3 * SIZE(CO1) pxor %xmm12, %xmm12 prefetcht2 3 * SIZE(CO2) pxor %xmm13, %xmm13 prefetcht2 3 * SIZE(CO1, LDC, 2) pxor %xmm14, %xmm14 prefetcht2 3 * SIZE(CO2, LDC, 2) pxor %xmm15, %xmm15 pxor %xmm2, %xmm2 pxor %xmm3, %xmm3 pxor %xmm4, %xmm4 pxor %xmm5, %xmm5#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif sarq $2, %rax jle .L15 ALIGN_4.L12: PREFETCH (PREFETCHSIZE + 0) * SIZE(AO) addpd %xmm2, %xmm10 movapd -16 * SIZE(AO), %xmm0 addpd %xmm3, %xmm14 movapd -16 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 movapd -14 * SIZE(AO), %xmm1 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm11 movapd -14 * SIZE(BO), %xmm4 addpd %xmm5, %xmm15 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm8 addpd %xmm3, %xmm12 movapd -12 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm9 movapd -10 * SIZE(BO), %xmm4 addpd %xmm5, %xmm13 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 movapd -12 * SIZE(AO), %xmm0 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm10 addpd %xmm3, %xmm14 movapd -8 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 movapd -10 * SIZE(AO), %xmm1 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm11 addpd %xmm5, %xmm15 movapd -6 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm8 addpd %xmm3, %xmm12 movapd -4 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm9 addpd %xmm5, %xmm13 movapd -2 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 PREFETCH (PREFETCHSIZE + 8) * SIZE(AO) movapd -8 * SIZE(AO), %xmm0 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm10 addpd %xmm3, %xmm14 movapd 0 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 movapd -6 * SIZE(AO), %xmm1 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm11 addpd %xmm5, %xmm15 movapd 2 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm8 addpd %xmm3, %xmm12 movapd 4 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm9 addpd %xmm5, %xmm13 movapd 6 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 movapd -4 * SIZE(AO), %xmm0 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm10 addpd %xmm3, %xmm14 movapd 8 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 movapd -2 * SIZE(AO), %xmm1 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm11 addpd %xmm5, %xmm15 movapd 10 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 addq $32 * SIZE, BO mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm8 addpd %xmm3, %xmm12 movapd -20 * SIZE(BO), %xmm2 movapd %xmm2, %xmm3 mulpd %xmm0, %xmm2 subq $-16 * SIZE, AO mulpd %xmm1, %xmm3 addpd %xmm4, %xmm9 addpd %xmm5, %xmm13 movapd -18 * SIZE(BO), %xmm4 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 subq $1, %rax mulpd %xmm1, %xmm5 jg,pt .L12 ALIGN_4.L15:#if defined(LT) || defined(RN) movq KK, %rax#else movq K, %rax subq KK, %rax#endif andq $3, %rax # if (k & 1) BRANCH je .L19 ALIGN_4.L16: movapd -16 * SIZE(AO), %xmm0 addpd %xmm2, %xmm10 movapd -16 * SIZE(BO), %xmm2 addpd %xmm3, %xmm14 movapd %xmm2, %xmm3 movapd -14 * SIZE(AO), %xmm1 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm11 movapd -14 * SIZE(BO), %xmm4 addpd %xmm5, %xmm15 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addpd %xmm2, %xmm8 movapd -12 * SIZE(BO), %xmm2 addpd %xmm3, %xmm12 movapd %xmm2, %xmm3 mulpd %xmm0, %xmm2 mulpd %xmm1, %xmm3 addpd %xmm4, %xmm9 movapd -10 * SIZE(BO), %xmm4 addpd %xmm5, %xmm13 movapd %xmm4, %xmm5 mulpd %xmm0, %xmm4 mulpd %xmm1, %xmm5 addq $4 * SIZE, AO # aoffset += 4 addq $8 * SIZE, BO # boffset1 += 8 subq $1, %rax jg,pt .L16 ALIGN_4.L19: addpd %xmm2, %xmm10 addpd %xmm3, %xmm14 addpd %xmm4, %xmm11 addpd %xmm5, %xmm15#if defined(LN) || defined(RT) movq KK, %rax#ifdef LN subq $4, %rax#else subq $4, %rax#endif movq AORIG, AO movq BORIG, B leaq 16 * SIZE + BUFFER, BO leaq (, %rax, SIZE), %rax leaq (AO, %rax, 4), AO leaq (B, %rax, 4), B leaq (BO, %rax, 8), BO#endif#if defined(LN) || defined(LT) movapd %xmm8, %xmm0 unpcklpd %xmm9, %xmm8 unpckhpd %xmm9, %xmm0 movapd %xmm10, %xmm2 unpcklpd %xmm11, %xmm10 unpckhpd %xmm11, %xmm2 movapd %xmm12, %xmm4 unpcklpd %xmm13, %xmm12 unpckhpd %xmm13, %xmm4 movapd %xmm14, %xmm6 unpcklpd %xmm15, %xmm14 unpckhpd %xmm15, %xmm6 movapd -16 * SIZE(B), %xmm9 movapd -14 * SIZE(B), %xmm11 movapd -12 * SIZE(B), %xmm13 movapd -10 * SIZE(B), %xmm15 movapd -8 * SIZE(B), %xmm1 movapd -6 * SIZE(B), %xmm3 movapd -4 * SIZE(B), %xmm5 movapd -2 * SIZE(B), %xmm7 subpd %xmm8, %xmm9 subpd %xmm10, %xmm11 subpd %xmm0, %xmm13 subpd %xmm2, %xmm15 subpd %xmm12, %xmm1 subpd %xmm14, %xmm3 subpd %xmm4, %xmm5 subpd %xmm6, %xmm7#else movapd -16 * SIZE(AO), %xmm0 movapd -14 * SIZE(AO), %xmm1 movapd -12 * SIZE(AO), %xmm2 movapd -10 * SIZE(AO), %xmm3 movapd -8 * SIZE(AO), %xmm4 movapd -6 * SIZE(AO), %xmm5 movapd -4 * SIZE(AO), %xmm6 movapd -2 * SIZE(AO), %xmm7 subpd %xmm8, %xmm0 subpd %xmm12, %xmm1 subpd %xmm9, %xmm2 subpd %xmm13, %xmm3 subpd %xmm10, %xmm4 subpd %xmm14, %xmm5 subpd %xmm11, %xmm6 subpd %xmm15, %xmm7#endif#ifdef LN movddup -1 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm5 mulpd %xmm8, %xmm7 movddup -2 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm5, %xmm10 subpd %xmm10, %xmm1 mulpd %xmm7, %xmm12 subpd %xmm12, %xmm3 movddup -3 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm5, %xmm10 subpd %xmm10, %xmm13 mulpd %xmm7, %xmm12 subpd %xmm12, %xmm15 movddup -4 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm5, %xmm10 subpd %xmm10, %xmm9 mulpd %xmm7, %xmm12 subpd %xmm12, %xmm11 movddup -6 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm1 mulpd %xmm8, %xmm3 movddup -7 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm1, %xmm10 subpd %xmm10, %xmm13 mulpd %xmm3, %xmm12 subpd %xmm12, %xmm15 movddup -8 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm1, %xmm10 subpd %xmm10, %xmm9 mulpd %xmm3, %xmm12 subpd %xmm12, %xmm11 movddup -11 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm13 mulpd %xmm8, %xmm15 movddup -12 * 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 -14 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm9, %xmm10 subpd %xmm10, %xmm1 mulpd %xmm11, %xmm12 subpd %xmm12, %xmm3 movddup -13 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm9, %xmm10 subpd %xmm10, %xmm5 mulpd %xmm11, %xmm12 subpd %xmm12, %xmm7 movddup -11 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm13 mulpd %xmm8, %xmm15 movddup -10 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm13, %xmm10 subpd %xmm10, %xmm1 mulpd %xmm15, %xmm12 subpd %xmm12, %xmm3 movddup -9 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm13, %xmm10 subpd %xmm10, %xmm5 mulpd %xmm15, %xmm12 subpd %xmm12, %xmm7 movddup -6 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm1 mulpd %xmm8, %xmm3 movddup -5 * SIZE(AO), %xmm10 movapd %xmm10, %xmm12 mulpd %xmm1, %xmm10 subpd %xmm10, %xmm5 mulpd %xmm3, %xmm12 subpd %xmm12, %xmm7 movddup -1 * SIZE(AO), %xmm8 mulpd %xmm8, %xmm5 mulpd %xmm8, %xmm7#endif#ifdef RN movddup -16 * SIZE(B), %xmm8
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -