⭐ 欢迎来到虫虫下载站! | 📦 资源下载 📁 资源专辑 ℹ️ 关于我们
⭐ 虫虫下载站

📄 gemm_kernel_8x4_sse3.s

📁 Optimized GotoBLAS libraries
💻 S
📖 第 1 页 / 共 5 页
字号:
/*********************************************************************//*                                                                   *//*             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 + -