📄 gemm_kernel_8x4_sse3.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 M %rdi#define N %rsi#define K %rdx#define A %rcx#define B %r8#define C %r9#define LDC %r10#define I %r11#define AO %r12#define BO %r13#define CO1 %r14#define CO2 %r15#define BB %rbp #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 ALPHA 0(%rsp)#define J 16(%rsp)#define OFFSET 24(%rsp)#define KK 32(%rsp)#define KKK 40(%rsp)#define BUFFER 128(%rsp)#define PREFETCH prefetcht0#define PREFETCHSIZE 320#define KERNEL1(address) \ mulps %xmm8, %xmm9; \ PREFETCH (PREFETCHSIZE + 0) * SIZE + (address) * SIZE(AO); \ addps %xmm9, %xmm0; \ movshdup 0 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm1; \ movsldup 4 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm2; \ movshdup 4 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ movaps 4 * SIZE + (address) * SIZE(AO), %xmm8; \ addps %xmm9, %xmm3; \ movsldup 0 * SIZE + (address) * SIZE(BO), %xmm9#define KERNEL2(address) \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm4; \ movshdup 0 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm5; \ movsldup 4 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm6; \ movshdup 4 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ movaps 8 * SIZE + (address) * SIZE(AO), %xmm8; \ addps %xmm9, %xmm7; \ movsldup 8 * SIZE + (address) * SIZE(BO), %xmm9#define KERNEL3(address) \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm0; \ movshdup 8 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm1; \ movsldup 12 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm2; \ movshdup 12 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ movaps 12 * SIZE + (address) * SIZE(AO), %xmm8; \ addps %xmm9, %xmm3; \ movsldup 8 * SIZE + (address) * SIZE(BO), %xmm9#define KERNEL4(address) \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm4; \ movshdup 8 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm5; \ movsldup 12 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ addps %xmm9, %xmm6; \ movshdup 12 * SIZE + (address) * SIZE(BO), %xmm9; \ mulps %xmm8, %xmm9; \ movaps 64 * SIZE + (address) * SIZE(AO), %xmm8; \ addps %xmm9, %xmm7; \ movsldup 64 * SIZE + (address) * SIZE(BO), %xmm9#define KERNEL5(address) \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm0; \ movshdup 16 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm1; \ movsldup 20 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm2; \ movshdup 20 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ movaps 20 * SIZE + (address) * SIZE(AO), %xmm10; \ addps %xmm11, %xmm3; \ movsldup 16 * SIZE + (address) * SIZE(BO), %xmm11#define KERNEL6(address) \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm4; \ movshdup 16 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm5; \ movsldup 20 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm6; \ movshdup 20 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ movaps 24 * SIZE + (address) * SIZE(AO), %xmm10; \ addps %xmm11, %xmm7; \ movsldup 24 * SIZE + (address) * SIZE(BO), %xmm11#define KERNEL7(address) \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm0; \ movshdup 24 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm1; \ movsldup 28 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm2; \ movshdup 28 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ movaps 28 * SIZE + (address) * SIZE(AO), %xmm10; \ addps %xmm11, %xmm3; \ movsldup 24 * SIZE + (address) * SIZE(BO), %xmm11#define KERNEL8(address) \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm4; \ movshdup 24 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm5; \ movsldup 28 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ addps %xmm11, %xmm6; \ movshdup 28 * SIZE + (address) * SIZE(BO), %xmm11; \ mulps %xmm10, %xmm11; \ movaps 80 * SIZE + (address) * SIZE(AO), %xmm10; \ addps %xmm11, %xmm7; \ movsldup 80 * SIZE + (address) * SIZE(BO), %xmm11#define KERNEL9(address) \ mulps %xmm12, %xmm13; \ PREFETCH (PREFETCHSIZE + 32) * SIZE + (address) * SIZE(AO); \ addps %xmm13, %xmm0; \ movshdup 32 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm1; \ movsldup 36 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm2; \ movshdup 36 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ movaps 36 * SIZE + (address) * SIZE(AO), %xmm12; \ addps %xmm13, %xmm3; \ movsldup 32 * SIZE + (address) * SIZE(BO), %xmm13#define KERNEL10(address) \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm4; \ movshdup 32 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm5; \ movsldup 36 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm6; \ movshdup 36 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ movaps 40 * SIZE + (address) * SIZE(AO), %xmm12; \ addps %xmm13, %xmm7; \ movsldup 40 * SIZE + (address) * SIZE(BO), %xmm13#define KERNEL11(address) \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm0; \ movshdup 40 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm1; \ movsldup 44 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm2; \ movshdup 44 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ movaps 44 * SIZE + (address) * SIZE(AO), %xmm12; \ addps %xmm13, %xmm3; \ movsldup 40 * SIZE + (address) * SIZE(BO), %xmm13#define KERNEL12(address) \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm4; \ movshdup 40 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm5; \ movsldup 44 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ addps %xmm13, %xmm6; \ movshdup 44 * SIZE + (address) * SIZE(BO), %xmm13; \ mulps %xmm12, %xmm13; \ movaps 96 * SIZE + (address) * SIZE(AO), %xmm12; \ addps %xmm13, %xmm7; \ movsldup 96 * SIZE + (address) * SIZE(BO), %xmm13#define KERNEL13(address) \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm0; \ movshdup 48 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm1; \ movsldup 52 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm2; \ movshdup 52 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ movaps 52 * SIZE + (address) * SIZE(AO), %xmm14; \ addps %xmm15, %xmm3; \ movsldup 48 * SIZE + (address) * SIZE(BO), %xmm15#define KERNEL14(address) \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm4; \ movshdup 48 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm5; \ movsldup 52 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm6; \ movshdup 52 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ movaps 56 * SIZE + (address) * SIZE(AO), %xmm14; \ addps %xmm15, %xmm7; \ movsldup 56 * SIZE + (address) * SIZE(BO), %xmm15#define KERNEL15(address) \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm0; \ movshdup 56 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm1; \ movsldup 60 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm2; \ movshdup 60 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ movaps 60 * SIZE + (address) * SIZE(AO), %xmm14; \ addps %xmm15, %xmm3; \ movsldup 56 * SIZE + (address) * SIZE(BO), %xmm15#define KERNEL16(address) \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm4; \ movshdup 56 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm5; \ movsldup 60 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ addps %xmm15, %xmm6; \ movshdup 60 * SIZE + (address) * SIZE(BO), %xmm15; \ mulps %xmm14, %xmm15; \ movaps 112 * SIZE + (address) * SIZE(AO), %xmm14; \ addps %xmm15, %xmm7; \ movsldup 112 * SIZE + (address) * SIZE(BO), %xmm15 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, M movq ARG2, N movq ARG3, K movq OLD_A, A movq OLD_B, B movq OLD_C, C movq OLD_LDC, LDC#ifdef TRMMKERNEL movsd OLD_OFFSET, %xmm4#endif movaps %xmm3, %xmm0#else movq OLD_LDC, LDC#ifdef TRMMKERNEL movsd OLD_OFFSET, %xmm4#endif#endif EMMS movq %rsp, %rbx # save old stack subq $128 + GEMM_Q * 32, %rsp andq $-4096, %rsp # align stack#ifdef WINDOWS_ABI#if GEMM_Q > 128 movq $0, 4096 * 1(%rsp)#endif movq $0, 4096 * 0(%rsp)#endif shufps $0, %xmm0, %xmm0 movaps %xmm0, ALPHA #ifdef TRMMKERNEL movsd %xmm4, OFFSET movsd %xmm4, KK#ifndef LEFT negq KK#endif #endif leaq (, LDC, SIZE), LDC movq N, J sarq $2, J # j = (n >> 2) jle .L50.L01:#if defined(TRMMKERNEL) && defined(LEFT) movq OFFSET, %rax movq %rax, KK#endif /* Copying to Sub Buffer */ leaq BUFFER, BO movq K, %rax sarq $2, %rax jle .L03 ALIGN_4 .L02: movddup 0 * SIZE(B), %xmm0 movddup 2 * SIZE(B), %xmm1 movddup 4 * SIZE(B), %xmm2 movddup 6 * SIZE(B), %xmm3 movddup 8 * SIZE(B), %xmm4 movddup 10 * SIZE(B), %xmm5 movddup 12 * SIZE(B), %xmm6 movddup 14 * SIZE(B), %xmm7 movaps %xmm0, 0 * SIZE(BO) movaps %xmm1, 4 * SIZE(BO) movaps %xmm2, 8 * SIZE(BO) movaps %xmm3, 12 * SIZE(BO) movaps %xmm4, 16 * SIZE(BO) movaps %xmm5, 20 * SIZE(BO) movaps %xmm6, 24 * SIZE(BO) movaps %xmm7, 28 * SIZE(BO) prefetcht1 128 * SIZE(BO) prefetcht0 112 * SIZE(B) addq $16 * SIZE, B addq $32 * SIZE, BO decq %rax jne .L02 ALIGN_4.L03: movq K, %rax andq $3, %rax BRANCH jle .L10 ALIGN_4.L04: movddup 0 * SIZE(B), %xmm0 movddup 2 * SIZE(B), %xmm1 movaps %xmm0, 0 * SIZE(BO) movaps %xmm1, 4 * SIZE(BO) addq $4 * SIZE, B addq $8 * SIZE, BO decq %rax jne .L04 ALIGN_4 .L10: movq C, CO1 # coffset1 = c leaq (C, LDC, 1), CO2 # coffset2 = c + ldc movq A, AO # aoffset = a leaq 112 * SIZE(B), BB movq M, I sarq $3, I # i = (m >> 3) jle .L20 ALIGN_4.L11: prefetcht0 0 * SIZE(BB) prefetcht0 8 * SIZE(BB) subq $-16 * SIZE, BB#if !defined(TRMMKERNEL) || \ (defined(TRMMKERNEL) && defined(LEFT) && defined(TRANSA)) || \ (defined(TRMMKERNEL) && !defined(LEFT) && !defined(TRANSA)) leaq BUFFER, BO#else leaq BUFFER, BO movq KK, %rax leaq (, %rax, 8), %rax leaq (AO, %rax, 4), AO leaq (BO, %rax, 4), BO#endif movaps 0 * SIZE(AO), %xmm8 movaps 16 * SIZE(AO), %xmm10 movaps 32 * SIZE(AO), %xmm12 movaps 48 * SIZE(AO), %xmm14 movsldup 0 * SIZE(BO), %xmm9 movsldup 16 * SIZE(BO), %xmm11 movsldup 32 * SIZE(BO), %xmm13 movsldup 48 * SIZE(BO), %xmm15 pxor %xmm0, %xmm0 pxor %xmm1, %xmm1 pxor %xmm2, %xmm2 pxor %xmm3, %xmm3 prefetchnta 8 * SIZE(CO1) pxor %xmm4, %xmm4 prefetchnta 8 * SIZE(CO2) pxor %xmm5, %xmm5 prefetchnta 8 * SIZE(CO1, LDC, 2) pxor %xmm6, %xmm6 prefetchnta 8 * SIZE(CO2, LDC, 2) pxor %xmm7, %xmm7#ifndef TRMMKERNEL movq K, %rax#elif (defined(LEFT) && !defined(TRANSA)) || (!defined(LEFT) && defined(TRANSA)) movq K, %rax subq KK, %rax movq %rax, KKK #else movq KK, %rax#ifdef LEFT addq $8, %rax#else addq $4, %rax#endif movq %rax, KKK#endif#if 1 andq $-8, %rax salq $4, %rax je .L15.L1X: KERNEL1 (64 * 0) KERNEL2 (64 * 0) KERNEL3 (64 * 0) KERNEL4 (64 * 0) KERNEL5 (64 * 0) KERNEL6 (64 * 0) KERNEL7 (64 * 0) KERNEL8 (64 * 0) KERNEL9 (64 * 0) KERNEL10(64 * 0) KERNEL11(64 * 0) KERNEL12(64 * 0) KERNEL13(64 * 0) KERNEL14(64 * 0) KERNEL15(64 * 0) KERNEL16(64 * 0) cmpq $128 * 1, %rax jle,pn .L12 KERNEL1 (64 * 1) KERNEL2 (64 * 1) KERNEL3 (64 * 1) KERNEL4 (64 * 1) KERNEL5 (64 * 1) KERNEL6 (64 * 1) KERNEL7 (64 * 1) KERNEL8 (64 * 1) KERNEL9 (64 * 1) KERNEL10(64 * 1) KERNEL11(64 * 1) KERNEL12(64 * 1) KERNEL13(64 * 1) KERNEL14(64 * 1) KERNEL15(64 * 1) KERNEL16(64 * 1) cmpq $128 * 2, %rax jle,pn .L12 KERNEL1 (64 * 2) KERNEL2 (64 * 2) KERNEL3 (64 * 2) KERNEL4 (64 * 2) KERNEL5 (64 * 2) KERNEL6 (64 * 2) KERNEL7 (64 * 2) KERNEL8 (64 * 2) KERNEL9 (64 * 2) KERNEL10(64 * 2) KERNEL11(64 * 2) KERNEL12(64 * 2) KERNEL13(64 * 2) KERNEL14(64 * 2) KERNEL15(64 * 2) KERNEL16(64 * 2) cmpq $128 * 3, %rax jle,pn .L12 KERNEL1 (64 * 3) KERNEL2 (64 * 3) KERNEL3 (64 * 3) KERNEL4 (64 * 3) KERNEL5 (64 * 3) KERNEL6 (64 * 3) KERNEL7 (64 * 3) KERNEL8 (64 * 3) KERNEL9 (64 * 3) KERNEL10(64 * 3) KERNEL11(64 * 3) KERNEL12(64 * 3) KERNEL13(64 * 3) KERNEL14(64 * 3) KERNEL15(64 * 3) KERNEL16(64 * 3) cmpq $128 * 4, %rax jle,pn .L12 KERNEL1 (64 * 4) KERNEL2 (64 * 4) KERNEL3 (64 * 4)
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -