xref: /OK3568_Linux_fs/kernel/arch/ia64/lib/memcpy_mck.S (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun/* SPDX-License-Identifier: GPL-2.0 */
2*4882a593Smuzhiyun/*
3*4882a593Smuzhiyun * Itanium 2-optimized version of memcpy and copy_user function
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Inputs:
6*4882a593Smuzhiyun * 	in0:	destination address
7*4882a593Smuzhiyun *	in1:	source address
8*4882a593Smuzhiyun *	in2:	number of bytes to copy
9*4882a593Smuzhiyun * Output:
10*4882a593Smuzhiyun *	for memcpy:    return dest
11*4882a593Smuzhiyun * 	for copy_user: return 0 if success,
12*4882a593Smuzhiyun *		       or number of byte NOT copied if error occurred.
13*4882a593Smuzhiyun *
14*4882a593Smuzhiyun * Copyright (C) 2002 Intel Corp.
15*4882a593Smuzhiyun * Copyright (C) 2002 Ken Chen <kenneth.w.chen@intel.com>
16*4882a593Smuzhiyun */
17*4882a593Smuzhiyun#include <asm/asmmacro.h>
18*4882a593Smuzhiyun#include <asm/page.h>
19*4882a593Smuzhiyun#include <asm/export.h>
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun#define EK(y...) EX(y)
22*4882a593Smuzhiyun
23*4882a593Smuzhiyun/* McKinley specific optimization */
24*4882a593Smuzhiyun
25*4882a593Smuzhiyun#define retval		r8
26*4882a593Smuzhiyun#define saved_pfs	r31
27*4882a593Smuzhiyun#define saved_lc	r10
28*4882a593Smuzhiyun#define saved_pr	r11
29*4882a593Smuzhiyun#define saved_in0	r14
30*4882a593Smuzhiyun#define saved_in1	r15
31*4882a593Smuzhiyun#define saved_in2	r16
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun#define src0		r2
34*4882a593Smuzhiyun#define src1		r3
35*4882a593Smuzhiyun#define dst0		r17
36*4882a593Smuzhiyun#define dst1		r18
37*4882a593Smuzhiyun#define cnt		r9
38*4882a593Smuzhiyun
39*4882a593Smuzhiyun/* r19-r30 are temp for each code section */
40*4882a593Smuzhiyun#define PREFETCH_DIST	8
41*4882a593Smuzhiyun#define src_pre_mem	r19
42*4882a593Smuzhiyun#define dst_pre_mem	r20
43*4882a593Smuzhiyun#define src_pre_l2	r21
44*4882a593Smuzhiyun#define dst_pre_l2	r22
45*4882a593Smuzhiyun#define t1		r23
46*4882a593Smuzhiyun#define t2		r24
47*4882a593Smuzhiyun#define t3		r25
48*4882a593Smuzhiyun#define t4		r26
49*4882a593Smuzhiyun#define t5		t1	// alias!
50*4882a593Smuzhiyun#define t6		t2	// alias!
51*4882a593Smuzhiyun#define t7		t3	// alias!
52*4882a593Smuzhiyun#define n8		r27
53*4882a593Smuzhiyun#define t9		t5	// alias!
54*4882a593Smuzhiyun#define t10		t4	// alias!
55*4882a593Smuzhiyun#define t11		t7	// alias!
56*4882a593Smuzhiyun#define t12		t6	// alias!
57*4882a593Smuzhiyun#define t14		t10	// alias!
58*4882a593Smuzhiyun#define t13		r28
59*4882a593Smuzhiyun#define t15		r29
60*4882a593Smuzhiyun#define tmp		r30
61*4882a593Smuzhiyun
62*4882a593Smuzhiyun/* defines for long_copy block */
63*4882a593Smuzhiyun#define	A	0
64*4882a593Smuzhiyun#define B	(PREFETCH_DIST)
65*4882a593Smuzhiyun#define C	(B + PREFETCH_DIST)
66*4882a593Smuzhiyun#define D	(C + 1)
67*4882a593Smuzhiyun#define N	(D + 1)
68*4882a593Smuzhiyun#define Nrot	((N + 7) & ~7)
69*4882a593Smuzhiyun
70*4882a593Smuzhiyun/* alias */
71*4882a593Smuzhiyun#define in0		r32
72*4882a593Smuzhiyun#define in1		r33
73*4882a593Smuzhiyun#define in2		r34
74*4882a593Smuzhiyun
75*4882a593SmuzhiyunGLOBAL_ENTRY(memcpy)
76*4882a593Smuzhiyun	and	r28=0x7,in0
77*4882a593Smuzhiyun	and	r29=0x7,in1
78*4882a593Smuzhiyun	mov	f6=f0
79*4882a593Smuzhiyun	mov	retval=in0
80*4882a593Smuzhiyun	br.cond.sptk .common_code
81*4882a593Smuzhiyun	;;
82*4882a593SmuzhiyunEND(memcpy)
83*4882a593SmuzhiyunEXPORT_SYMBOL(memcpy)
84*4882a593SmuzhiyunGLOBAL_ENTRY(__copy_user)
85*4882a593Smuzhiyun	.prologue
86*4882a593Smuzhiyun// check dest alignment
87*4882a593Smuzhiyun	and	r28=0x7,in0
88*4882a593Smuzhiyun	and	r29=0x7,in1
89*4882a593Smuzhiyun	mov	f6=f1
90*4882a593Smuzhiyun	mov	saved_in0=in0	// save dest pointer
91*4882a593Smuzhiyun	mov	saved_in1=in1	// save src pointer
92*4882a593Smuzhiyun	mov	retval=r0	// initialize return value
93*4882a593Smuzhiyun	;;
94*4882a593Smuzhiyun.common_code:
95*4882a593Smuzhiyun	cmp.gt	p15,p0=8,in2	// check for small size
96*4882a593Smuzhiyun	cmp.ne	p13,p0=0,r28	// check dest alignment
97*4882a593Smuzhiyun	cmp.ne	p14,p0=0,r29	// check src alignment
98*4882a593Smuzhiyun	add	src0=0,in1
99*4882a593Smuzhiyun	sub	r30=8,r28	// for .align_dest
100*4882a593Smuzhiyun	mov	saved_in2=in2	// save len
101*4882a593Smuzhiyun	;;
102*4882a593Smuzhiyun	add	dst0=0,in0
103*4882a593Smuzhiyun	add	dst1=1,in0	// dest odd index
104*4882a593Smuzhiyun	cmp.le	p6,p0 = 1,r30	// for .align_dest
105*4882a593Smuzhiyun(p15)	br.cond.dpnt .memcpy_short
106*4882a593Smuzhiyun(p13)	br.cond.dpnt .align_dest
107*4882a593Smuzhiyun(p14)	br.cond.dpnt .unaligned_src
108*4882a593Smuzhiyun	;;
109*4882a593Smuzhiyun
110*4882a593Smuzhiyun// both dest and src are aligned on 8-byte boundary
111*4882a593Smuzhiyun.aligned_src:
112*4882a593Smuzhiyun	.save ar.pfs, saved_pfs
113*4882a593Smuzhiyun	alloc	saved_pfs=ar.pfs,3,Nrot-3,0,Nrot
114*4882a593Smuzhiyun	.save pr, saved_pr
115*4882a593Smuzhiyun	mov	saved_pr=pr
116*4882a593Smuzhiyun
117*4882a593Smuzhiyun	shr.u	cnt=in2,7	// this much cache line
118*4882a593Smuzhiyun	;;
119*4882a593Smuzhiyun	cmp.lt	p6,p0=2*PREFETCH_DIST,cnt
120*4882a593Smuzhiyun	cmp.lt	p7,p8=1,cnt
121*4882a593Smuzhiyun	.save ar.lc, saved_lc
122*4882a593Smuzhiyun	mov	saved_lc=ar.lc
123*4882a593Smuzhiyun	.body
124*4882a593Smuzhiyun	add	cnt=-1,cnt
125*4882a593Smuzhiyun	add	src_pre_mem=0,in1	// prefetch src pointer
126*4882a593Smuzhiyun	add	dst_pre_mem=0,in0	// prefetch dest pointer
127*4882a593Smuzhiyun	;;
128*4882a593Smuzhiyun(p7)	mov	ar.lc=cnt	// prefetch count
129*4882a593Smuzhiyun(p8)	mov	ar.lc=r0
130*4882a593Smuzhiyun(p6)	br.cond.dpnt .long_copy
131*4882a593Smuzhiyun	;;
132*4882a593Smuzhiyun
133*4882a593Smuzhiyun.prefetch:
134*4882a593Smuzhiyun	lfetch.fault	  [src_pre_mem], 128
135*4882a593Smuzhiyun	lfetch.fault.excl [dst_pre_mem], 128
136*4882a593Smuzhiyun	br.cloop.dptk.few .prefetch
137*4882a593Smuzhiyun	;;
138*4882a593Smuzhiyun
139*4882a593Smuzhiyun.medium_copy:
140*4882a593Smuzhiyun	and	tmp=31,in2	// copy length after iteration
141*4882a593Smuzhiyun	shr.u	r29=in2,5	// number of 32-byte iteration
142*4882a593Smuzhiyun	add	dst1=8,dst0	// 2nd dest pointer
143*4882a593Smuzhiyun	;;
144*4882a593Smuzhiyun	add	cnt=-1,r29	// ctop iteration adjustment
145*4882a593Smuzhiyun	cmp.eq	p10,p0=r29,r0	// do we really need to loop?
146*4882a593Smuzhiyun	add	src1=8,src0	// 2nd src pointer
147*4882a593Smuzhiyun	cmp.le	p6,p0=8,tmp
148*4882a593Smuzhiyun	;;
149*4882a593Smuzhiyun	cmp.le	p7,p0=16,tmp
150*4882a593Smuzhiyun	mov	ar.lc=cnt	// loop setup
151*4882a593Smuzhiyun	cmp.eq	p16,p17 = r0,r0
152*4882a593Smuzhiyun	mov	ar.ec=2
153*4882a593Smuzhiyun(p10)	br.dpnt.few .aligned_src_tail
154*4882a593Smuzhiyun	;;
155*4882a593Smuzhiyun	TEXT_ALIGN(32)
156*4882a593Smuzhiyun1:
157*4882a593SmuzhiyunEX(.ex_handler, (p16)	ld8	r34=[src0],16)
158*4882a593SmuzhiyunEK(.ex_handler, (p16)	ld8	r38=[src1],16)
159*4882a593SmuzhiyunEX(.ex_handler, (p17)	st8	[dst0]=r33,16)
160*4882a593SmuzhiyunEK(.ex_handler, (p17)	st8	[dst1]=r37,16)
161*4882a593Smuzhiyun	;;
162*4882a593SmuzhiyunEX(.ex_handler, (p16)	ld8	r32=[src0],16)
163*4882a593SmuzhiyunEK(.ex_handler, (p16)	ld8	r36=[src1],16)
164*4882a593SmuzhiyunEX(.ex_handler, (p16)	st8	[dst0]=r34,16)
165*4882a593SmuzhiyunEK(.ex_handler, (p16)	st8	[dst1]=r38,16)
166*4882a593Smuzhiyun	br.ctop.dptk.few 1b
167*4882a593Smuzhiyun	;;
168*4882a593Smuzhiyun
169*4882a593Smuzhiyun.aligned_src_tail:
170*4882a593SmuzhiyunEX(.ex_handler, (p6)	ld8	t1=[src0])
171*4882a593Smuzhiyun	mov	ar.lc=saved_lc
172*4882a593Smuzhiyun	mov	ar.pfs=saved_pfs
173*4882a593SmuzhiyunEX(.ex_hndlr_s, (p7)	ld8	t2=[src1],8)
174*4882a593Smuzhiyun	cmp.le	p8,p0=24,tmp
175*4882a593Smuzhiyun	and	r21=-8,tmp
176*4882a593Smuzhiyun	;;
177*4882a593SmuzhiyunEX(.ex_hndlr_s, (p8)	ld8	t3=[src1])
178*4882a593SmuzhiyunEX(.ex_handler, (p6)	st8	[dst0]=t1)	// store byte 1
179*4882a593Smuzhiyun	and	in2=7,tmp	// remaining length
180*4882a593SmuzhiyunEX(.ex_hndlr_d, (p7)	st8	[dst1]=t2,8)	// store byte 2
181*4882a593Smuzhiyun	add	src0=src0,r21	// setting up src pointer
182*4882a593Smuzhiyun	add	dst0=dst0,r21	// setting up dest pointer
183*4882a593Smuzhiyun	;;
184*4882a593SmuzhiyunEX(.ex_handler, (p8)	st8	[dst1]=t3)	// store byte 3
185*4882a593Smuzhiyun	mov	pr=saved_pr,-1
186*4882a593Smuzhiyun	br.dptk.many .memcpy_short
187*4882a593Smuzhiyun	;;
188*4882a593Smuzhiyun
189*4882a593Smuzhiyun/* code taken from copy_page_mck */
190*4882a593Smuzhiyun.long_copy:
191*4882a593Smuzhiyun	.rotr v[2*PREFETCH_DIST]
192*4882a593Smuzhiyun	.rotp p[N]
193*4882a593Smuzhiyun
194*4882a593Smuzhiyun	mov src_pre_mem = src0
195*4882a593Smuzhiyun	mov pr.rot = 0x10000
196*4882a593Smuzhiyun	mov ar.ec = 1				// special unrolled loop
197*4882a593Smuzhiyun
198*4882a593Smuzhiyun	mov dst_pre_mem = dst0
199*4882a593Smuzhiyun
200*4882a593Smuzhiyun	add src_pre_l2 = 8*8, src0
201*4882a593Smuzhiyun	add dst_pre_l2 = 8*8, dst0
202*4882a593Smuzhiyun	;;
203*4882a593Smuzhiyun	add src0 = 8, src_pre_mem		// first t1 src
204*4882a593Smuzhiyun	mov ar.lc = 2*PREFETCH_DIST - 1
205*4882a593Smuzhiyun	shr.u cnt=in2,7				// number of lines
206*4882a593Smuzhiyun	add src1 = 3*8, src_pre_mem		// first t3 src
207*4882a593Smuzhiyun	add dst0 = 8, dst_pre_mem		// first t1 dst
208*4882a593Smuzhiyun	add dst1 = 3*8, dst_pre_mem		// first t3 dst
209*4882a593Smuzhiyun	;;
210*4882a593Smuzhiyun	and tmp=127,in2				// remaining bytes after this block
211*4882a593Smuzhiyun	add cnt = -(2*PREFETCH_DIST) - 1, cnt
212*4882a593Smuzhiyun	// same as .line_copy loop, but with all predicated-off instructions removed:
213*4882a593Smuzhiyun.prefetch_loop:
214*4882a593SmuzhiyunEX(.ex_hndlr_lcpy_1, (p[A])	ld8 v[A] = [src_pre_mem], 128)		// M0
215*4882a593SmuzhiyunEK(.ex_hndlr_lcpy_1, (p[B])	st8 [dst_pre_mem] = v[B], 128)		// M2
216*4882a593Smuzhiyun	br.ctop.sptk .prefetch_loop
217*4882a593Smuzhiyun	;;
218*4882a593Smuzhiyun	cmp.eq p16, p0 = r0, r0			// reset p16 to 1
219*4882a593Smuzhiyun	mov ar.lc = cnt
220*4882a593Smuzhiyun	mov ar.ec = N				// # of stages in pipeline
221*4882a593Smuzhiyun	;;
222*4882a593Smuzhiyun.line_copy:
223*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8 t2 = [src0], 3*8)			// M0
224*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8 t4 = [src1], 3*8)			// M1
225*4882a593SmuzhiyunEX(.ex_handler_lcpy,	(p[B])	st8 [dst_pre_mem] = v[B], 128)		// M2 prefetch dst from memory
226*4882a593SmuzhiyunEK(.ex_handler_lcpy,	(p[D])	st8 [dst_pre_l2] = n8, 128)		// M3 prefetch dst from L2
227*4882a593Smuzhiyun	;;
228*4882a593SmuzhiyunEX(.ex_handler_lcpy,	(p[A])	ld8 v[A] = [src_pre_mem], 128)		// M0 prefetch src from memory
229*4882a593SmuzhiyunEK(.ex_handler_lcpy,	(p[C])	ld8 n8 = [src_pre_l2], 128)		// M1 prefetch src from L2
230*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] =  t1, 8)			// M2
231*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] =  t3, 8)			// M3
232*4882a593Smuzhiyun	;;
233*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8  t5 = [src0], 8)
234*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8  t7 = [src1], 3*8)
235*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] =  t2, 3*8)
236*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] =  t4, 3*8)
237*4882a593Smuzhiyun	;;
238*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8  t6 = [src0], 3*8)
239*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8 t10 = [src1], 8)
240*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] =  t5, 8)
241*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] =  t7, 3*8)
242*4882a593Smuzhiyun	;;
243*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8  t9 = [src0], 3*8)
244*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8 t11 = [src1], 3*8)
245*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] =  t6, 3*8)
246*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] = t10, 8)
247*4882a593Smuzhiyun	;;
248*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8 t12 = [src0], 8)
249*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8 t14 = [src1], 8)
250*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] =  t9, 3*8)
251*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] = t11, 3*8)
252*4882a593Smuzhiyun	;;
253*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	ld8 t13 = [src0], 4*8)
254*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	ld8 t15 = [src1], 4*8)
255*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] = t12, 8)
256*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] = t14, 8)
257*4882a593Smuzhiyun	;;
258*4882a593SmuzhiyunEX(.ex_handler,	(p[C])	ld8  t1 = [src0], 8)
259*4882a593SmuzhiyunEK(.ex_handler,	(p[C])	ld8  t3 = [src1], 8)
260*4882a593SmuzhiyunEX(.ex_handler,	(p[D])	st8 [dst0] = t13, 4*8)
261*4882a593SmuzhiyunEK(.ex_handler,	(p[D])	st8 [dst1] = t15, 4*8)
262*4882a593Smuzhiyun	br.ctop.sptk .line_copy
263*4882a593Smuzhiyun	;;
264*4882a593Smuzhiyun
265*4882a593Smuzhiyun	add dst0=-8,dst0
266*4882a593Smuzhiyun	add src0=-8,src0
267*4882a593Smuzhiyun	mov in2=tmp
268*4882a593Smuzhiyun	.restore sp
269*4882a593Smuzhiyun	br.sptk.many .medium_copy
270*4882a593Smuzhiyun	;;
271*4882a593Smuzhiyun
272*4882a593Smuzhiyun#define BLOCK_SIZE	128*32
273*4882a593Smuzhiyun#define blocksize	r23
274*4882a593Smuzhiyun#define curlen		r24
275*4882a593Smuzhiyun
276*4882a593Smuzhiyun// dest is on 8-byte boundary, src is not. We need to do
277*4882a593Smuzhiyun// ld8-ld8, shrp, then st8.  Max 8 byte copy per cycle.
278*4882a593Smuzhiyun.unaligned_src:
279*4882a593Smuzhiyun	.prologue
280*4882a593Smuzhiyun	.save ar.pfs, saved_pfs
281*4882a593Smuzhiyun	alloc	saved_pfs=ar.pfs,3,5,0,8
282*4882a593Smuzhiyun	.save ar.lc, saved_lc
283*4882a593Smuzhiyun	mov	saved_lc=ar.lc
284*4882a593Smuzhiyun	.save pr, saved_pr
285*4882a593Smuzhiyun	mov	saved_pr=pr
286*4882a593Smuzhiyun	.body
287*4882a593Smuzhiyun.4k_block:
288*4882a593Smuzhiyun	mov	saved_in0=dst0	// need to save all input arguments
289*4882a593Smuzhiyun	mov	saved_in2=in2
290*4882a593Smuzhiyun	mov	blocksize=BLOCK_SIZE
291*4882a593Smuzhiyun	;;
292*4882a593Smuzhiyun	cmp.lt	p6,p7=blocksize,in2
293*4882a593Smuzhiyun	mov	saved_in1=src0
294*4882a593Smuzhiyun	;;
295*4882a593Smuzhiyun(p6)	mov	in2=blocksize
296*4882a593Smuzhiyun	;;
297*4882a593Smuzhiyun	shr.u	r21=in2,7	// this much cache line
298*4882a593Smuzhiyun	shr.u	r22=in2,4	// number of 16-byte iteration
299*4882a593Smuzhiyun	and	curlen=15,in2	// copy length after iteration
300*4882a593Smuzhiyun	and	r30=7,src0	// source alignment
301*4882a593Smuzhiyun	;;
302*4882a593Smuzhiyun	cmp.lt	p7,p8=1,r21
303*4882a593Smuzhiyun	add	cnt=-1,r21
304*4882a593Smuzhiyun	;;
305*4882a593Smuzhiyun
306*4882a593Smuzhiyun	add	src_pre_mem=0,src0	// prefetch src pointer
307*4882a593Smuzhiyun	add	dst_pre_mem=0,dst0	// prefetch dest pointer
308*4882a593Smuzhiyun	and	src0=-8,src0		// 1st src pointer
309*4882a593Smuzhiyun(p7)	mov	ar.lc = cnt
310*4882a593Smuzhiyun(p8)	mov	ar.lc = r0
311*4882a593Smuzhiyun	;;
312*4882a593Smuzhiyun	TEXT_ALIGN(32)
313*4882a593Smuzhiyun1:	lfetch.fault	  [src_pre_mem], 128
314*4882a593Smuzhiyun	lfetch.fault.excl [dst_pre_mem], 128
315*4882a593Smuzhiyun	br.cloop.dptk.few 1b
316*4882a593Smuzhiyun	;;
317*4882a593Smuzhiyun
318*4882a593Smuzhiyun	shladd	dst1=r22,3,dst0	// 2nd dest pointer
319*4882a593Smuzhiyun	shladd	src1=r22,3,src0	// 2nd src pointer
320*4882a593Smuzhiyun	cmp.eq	p8,p9=r22,r0	// do we really need to loop?
321*4882a593Smuzhiyun	cmp.le	p6,p7=8,curlen;	// have at least 8 byte remaining?
322*4882a593Smuzhiyun	add	cnt=-1,r22	// ctop iteration adjustment
323*4882a593Smuzhiyun	;;
324*4882a593SmuzhiyunEX(.ex_handler, (p9)	ld8	r33=[src0],8)	// loop primer
325*4882a593SmuzhiyunEK(.ex_handler, (p9)	ld8	r37=[src1],8)
326*4882a593Smuzhiyun(p8)	br.dpnt.few .noloop
327*4882a593Smuzhiyun	;;
328*4882a593Smuzhiyun
329*4882a593Smuzhiyun// The jump address is calculated based on src alignment. The COPYU
330*4882a593Smuzhiyun// macro below need to confine its size to power of two, so an entry
331*4882a593Smuzhiyun// can be caulated using shl instead of an expensive multiply. The
332*4882a593Smuzhiyun// size is then hard coded by the following #define to match the
333*4882a593Smuzhiyun// actual size.  This make it somewhat tedious when COPYU macro gets
334*4882a593Smuzhiyun// changed and this need to be adjusted to match.
335*4882a593Smuzhiyun#define LOOP_SIZE 6
336*4882a593Smuzhiyun1:
337*4882a593Smuzhiyun	mov	r29=ip		// jmp_table thread
338*4882a593Smuzhiyun	mov	ar.lc=cnt
339*4882a593Smuzhiyun	;;
340*4882a593Smuzhiyun	add	r29=.jump_table - 1b - (.jmp1-.jump_table), r29
341*4882a593Smuzhiyun	shl	r28=r30, LOOP_SIZE	// jmp_table thread
342*4882a593Smuzhiyun	mov	ar.ec=2		// loop setup
343*4882a593Smuzhiyun	;;
344*4882a593Smuzhiyun	add	r29=r29,r28		// jmp_table thread
345*4882a593Smuzhiyun	cmp.eq	p16,p17=r0,r0
346*4882a593Smuzhiyun	;;
347*4882a593Smuzhiyun	mov	b6=r29			// jmp_table thread
348*4882a593Smuzhiyun	;;
349*4882a593Smuzhiyun	br.cond.sptk.few b6
350*4882a593Smuzhiyun
351*4882a593Smuzhiyun// for 8-15 byte case
352*4882a593Smuzhiyun// We will skip the loop, but need to replicate the side effect
353*4882a593Smuzhiyun// that the loop produces.
354*4882a593Smuzhiyun.noloop:
355*4882a593SmuzhiyunEX(.ex_handler, (p6)	ld8	r37=[src1],8)
356*4882a593Smuzhiyun	add	src0=8,src0
357*4882a593Smuzhiyun(p6)	shl	r25=r30,3
358*4882a593Smuzhiyun	;;
359*4882a593SmuzhiyunEX(.ex_handler, (p6)	ld8	r27=[src1])
360*4882a593Smuzhiyun(p6)	shr.u	r28=r37,r25
361*4882a593Smuzhiyun(p6)	sub	r26=64,r25
362*4882a593Smuzhiyun	;;
363*4882a593Smuzhiyun(p6)	shl	r27=r27,r26
364*4882a593Smuzhiyun	;;
365*4882a593Smuzhiyun(p6)	or	r21=r28,r27
366*4882a593Smuzhiyun
367*4882a593Smuzhiyun.unaligned_src_tail:
368*4882a593Smuzhiyun/* check if we have more than blocksize to copy, if so go back */
369*4882a593Smuzhiyun	cmp.gt	p8,p0=saved_in2,blocksize
370*4882a593Smuzhiyun	;;
371*4882a593Smuzhiyun(p8)	add	dst0=saved_in0,blocksize
372*4882a593Smuzhiyun(p8)	add	src0=saved_in1,blocksize
373*4882a593Smuzhiyun(p8)	sub	in2=saved_in2,blocksize
374*4882a593Smuzhiyun(p8)	br.dpnt	.4k_block
375*4882a593Smuzhiyun	;;
376*4882a593Smuzhiyun
377*4882a593Smuzhiyun/* we have up to 15 byte to copy in the tail.
378*4882a593Smuzhiyun * part of work is already done in the jump table code
379*4882a593Smuzhiyun * we are at the following state.
380*4882a593Smuzhiyun * src side:
381*4882a593Smuzhiyun *
382*4882a593Smuzhiyun *   xxxxxx xx                   <----- r21 has xxxxxxxx already
383*4882a593Smuzhiyun * -------- -------- --------
384*4882a593Smuzhiyun * 0        8        16
385*4882a593Smuzhiyun *          ^
386*4882a593Smuzhiyun *          |
387*4882a593Smuzhiyun *          src1
388*4882a593Smuzhiyun *
389*4882a593Smuzhiyun * dst
390*4882a593Smuzhiyun * -------- -------- --------
391*4882a593Smuzhiyun * ^
392*4882a593Smuzhiyun * |
393*4882a593Smuzhiyun * dst1
394*4882a593Smuzhiyun */
395*4882a593SmuzhiyunEX(.ex_handler, (p6)	st8	[dst1]=r21,8)	// more than 8 byte to copy
396*4882a593Smuzhiyun(p6)	add	curlen=-8,curlen	// update length
397*4882a593Smuzhiyun	mov	ar.pfs=saved_pfs
398*4882a593Smuzhiyun	;;
399*4882a593Smuzhiyun	mov	ar.lc=saved_lc
400*4882a593Smuzhiyun	mov	pr=saved_pr,-1
401*4882a593Smuzhiyun	mov	in2=curlen	// remaining length
402*4882a593Smuzhiyun	mov	dst0=dst1	// dest pointer
403*4882a593Smuzhiyun	add	src0=src1,r30	// forward by src alignment
404*4882a593Smuzhiyun	;;
405*4882a593Smuzhiyun
406*4882a593Smuzhiyun// 7 byte or smaller.
407*4882a593Smuzhiyun.memcpy_short:
408*4882a593Smuzhiyun	cmp.le	p8,p9   = 1,in2
409*4882a593Smuzhiyun	cmp.le	p10,p11 = 2,in2
410*4882a593Smuzhiyun	cmp.le	p12,p13 = 3,in2
411*4882a593Smuzhiyun	cmp.le	p14,p15 = 4,in2
412*4882a593Smuzhiyun	add	src1=1,src0	// second src pointer
413*4882a593Smuzhiyun	add	dst1=1,dst0	// second dest pointer
414*4882a593Smuzhiyun	;;
415*4882a593Smuzhiyun
416*4882a593SmuzhiyunEX(.ex_handler_short, (p8)	ld1	t1=[src0],2)
417*4882a593SmuzhiyunEK(.ex_handler_short, (p10)	ld1	t2=[src1],2)
418*4882a593Smuzhiyun(p9)	br.ret.dpnt rp		// 0 byte copy
419*4882a593Smuzhiyun	;;
420*4882a593Smuzhiyun
421*4882a593SmuzhiyunEX(.ex_handler_short, (p8)	st1	[dst0]=t1,2)
422*4882a593SmuzhiyunEK(.ex_handler_short, (p10)	st1	[dst1]=t2,2)
423*4882a593Smuzhiyun(p11)	br.ret.dpnt rp		// 1 byte copy
424*4882a593Smuzhiyun
425*4882a593SmuzhiyunEX(.ex_handler_short, (p12)	ld1	t3=[src0],2)
426*4882a593SmuzhiyunEK(.ex_handler_short, (p14)	ld1	t4=[src1],2)
427*4882a593Smuzhiyun(p13)	br.ret.dpnt rp		// 2 byte copy
428*4882a593Smuzhiyun	;;
429*4882a593Smuzhiyun
430*4882a593Smuzhiyun	cmp.le	p6,p7   = 5,in2
431*4882a593Smuzhiyun	cmp.le	p8,p9   = 6,in2
432*4882a593Smuzhiyun	cmp.le	p10,p11 = 7,in2
433*4882a593Smuzhiyun
434*4882a593SmuzhiyunEX(.ex_handler_short, (p12)	st1	[dst0]=t3,2)
435*4882a593SmuzhiyunEK(.ex_handler_short, (p14)	st1	[dst1]=t4,2)
436*4882a593Smuzhiyun(p15)	br.ret.dpnt rp		// 3 byte copy
437*4882a593Smuzhiyun	;;
438*4882a593Smuzhiyun
439*4882a593SmuzhiyunEX(.ex_handler_short, (p6)	ld1	t5=[src0],2)
440*4882a593SmuzhiyunEK(.ex_handler_short, (p8)	ld1	t6=[src1],2)
441*4882a593Smuzhiyun(p7)	br.ret.dpnt rp		// 4 byte copy
442*4882a593Smuzhiyun	;;
443*4882a593Smuzhiyun
444*4882a593SmuzhiyunEX(.ex_handler_short, (p6)	st1	[dst0]=t5,2)
445*4882a593SmuzhiyunEK(.ex_handler_short, (p8)	st1	[dst1]=t6,2)
446*4882a593Smuzhiyun(p9)	br.ret.dptk rp		// 5 byte copy
447*4882a593Smuzhiyun
448*4882a593SmuzhiyunEX(.ex_handler_short, (p10)	ld1	t7=[src0],2)
449*4882a593Smuzhiyun(p11)	br.ret.dptk rp		// 6 byte copy
450*4882a593Smuzhiyun	;;
451*4882a593Smuzhiyun
452*4882a593SmuzhiyunEX(.ex_handler_short, (p10)	st1	[dst0]=t7,2)
453*4882a593Smuzhiyun	br.ret.dptk rp		// done all cases
454*4882a593Smuzhiyun
455*4882a593Smuzhiyun
456*4882a593Smuzhiyun/* Align dest to nearest 8-byte boundary. We know we have at
457*4882a593Smuzhiyun * least 7 bytes to copy, enough to crawl to 8-byte boundary.
458*4882a593Smuzhiyun * Actual number of byte to crawl depend on the dest alignment.
459*4882a593Smuzhiyun * 7 byte or less is taken care at .memcpy_short
460*4882a593Smuzhiyun
461*4882a593Smuzhiyun * src0 - source even index
462*4882a593Smuzhiyun * src1 - source  odd index
463*4882a593Smuzhiyun * dst0 - dest even index
464*4882a593Smuzhiyun * dst1 - dest  odd index
465*4882a593Smuzhiyun * r30  - distance to 8-byte boundary
466*4882a593Smuzhiyun */
467*4882a593Smuzhiyun
468*4882a593Smuzhiyun.align_dest:
469*4882a593Smuzhiyun	add	src1=1,in1	// source odd index
470*4882a593Smuzhiyun	cmp.le	p7,p0 = 2,r30	// for .align_dest
471*4882a593Smuzhiyun	cmp.le	p8,p0 = 3,r30	// for .align_dest
472*4882a593SmuzhiyunEX(.ex_handler_short, (p6)	ld1	t1=[src0],2)
473*4882a593Smuzhiyun	cmp.le	p9,p0 = 4,r30	// for .align_dest
474*4882a593Smuzhiyun	cmp.le	p10,p0 = 5,r30
475*4882a593Smuzhiyun	;;
476*4882a593SmuzhiyunEX(.ex_handler_short, (p7)	ld1	t2=[src1],2)
477*4882a593SmuzhiyunEK(.ex_handler_short, (p8)	ld1	t3=[src0],2)
478*4882a593Smuzhiyun	cmp.le	p11,p0 = 6,r30
479*4882a593SmuzhiyunEX(.ex_handler_short, (p6)	st1	[dst0] = t1,2)
480*4882a593Smuzhiyun	cmp.le	p12,p0 = 7,r30
481*4882a593Smuzhiyun	;;
482*4882a593SmuzhiyunEX(.ex_handler_short, (p9)	ld1	t4=[src1],2)
483*4882a593SmuzhiyunEK(.ex_handler_short, (p10)	ld1	t5=[src0],2)
484*4882a593SmuzhiyunEX(.ex_handler_short, (p7)	st1	[dst1] = t2,2)
485*4882a593SmuzhiyunEK(.ex_handler_short, (p8)	st1	[dst0] = t3,2)
486*4882a593Smuzhiyun	;;
487*4882a593SmuzhiyunEX(.ex_handler_short, (p11)	ld1	t6=[src1],2)
488*4882a593SmuzhiyunEK(.ex_handler_short, (p12)	ld1	t7=[src0],2)
489*4882a593Smuzhiyun	cmp.eq	p6,p7=r28,r29
490*4882a593SmuzhiyunEX(.ex_handler_short, (p9)	st1	[dst1] = t4,2)
491*4882a593SmuzhiyunEK(.ex_handler_short, (p10)	st1	[dst0] = t5,2)
492*4882a593Smuzhiyun	sub	in2=in2,r30
493*4882a593Smuzhiyun	;;
494*4882a593SmuzhiyunEX(.ex_handler_short, (p11)	st1	[dst1] = t6,2)
495*4882a593SmuzhiyunEK(.ex_handler_short, (p12)	st1	[dst0] = t7)
496*4882a593Smuzhiyun	add	dst0=in0,r30	// setup arguments
497*4882a593Smuzhiyun	add	src0=in1,r30
498*4882a593Smuzhiyun(p6)	br.cond.dptk .aligned_src
499*4882a593Smuzhiyun(p7)	br.cond.dpnt .unaligned_src
500*4882a593Smuzhiyun	;;
501*4882a593Smuzhiyun
502*4882a593Smuzhiyun/* main loop body in jump table format */
503*4882a593Smuzhiyun#define COPYU(shift)									\
504*4882a593Smuzhiyun1:											\
505*4882a593SmuzhiyunEX(.ex_handler,  (p16)	ld8	r32=[src0],8);		/* 1 */				\
506*4882a593SmuzhiyunEK(.ex_handler,  (p16)	ld8	r36=[src1],8);						\
507*4882a593Smuzhiyun		 (p17)	shrp	r35=r33,r34,shift;;	/* 1 */				\
508*4882a593SmuzhiyunEX(.ex_handler,  (p6)	ld8	r22=[src1]);	/* common, prime for tail section */	\
509*4882a593Smuzhiyun		 nop.m	0;								\
510*4882a593Smuzhiyun		 (p16)	shrp	r38=r36,r37,shift;					\
511*4882a593SmuzhiyunEX(.ex_handler,  (p17)	st8	[dst0]=r35,8);		/* 1 */				\
512*4882a593SmuzhiyunEK(.ex_handler,  (p17)	st8	[dst1]=r39,8);						\
513*4882a593Smuzhiyun		 br.ctop.dptk.few 1b;;							\
514*4882a593Smuzhiyun		 (p7)	add	src1=-8,src1;	/* back out for <8 byte case */		\
515*4882a593Smuzhiyun		 shrp	r21=r22,r38,shift;	/* speculative work */			\
516*4882a593Smuzhiyun		 br.sptk.few .unaligned_src_tail /* branch out of jump table */		\
517*4882a593Smuzhiyun		 ;;
518*4882a593Smuzhiyun	TEXT_ALIGN(32)
519*4882a593Smuzhiyun.jump_table:
520*4882a593Smuzhiyun	COPYU(8)	// unaligned cases
521*4882a593Smuzhiyun.jmp1:
522*4882a593Smuzhiyun	COPYU(16)
523*4882a593Smuzhiyun	COPYU(24)
524*4882a593Smuzhiyun	COPYU(32)
525*4882a593Smuzhiyun	COPYU(40)
526*4882a593Smuzhiyun	COPYU(48)
527*4882a593Smuzhiyun	COPYU(56)
528*4882a593Smuzhiyun
529*4882a593Smuzhiyun#undef A
530*4882a593Smuzhiyun#undef B
531*4882a593Smuzhiyun#undef C
532*4882a593Smuzhiyun#undef D
533*4882a593Smuzhiyun
534*4882a593Smuzhiyun/*
535*4882a593Smuzhiyun * Due to lack of local tag support in gcc 2.x assembler, it is not clear which
536*4882a593Smuzhiyun * instruction failed in the bundle.  The exception algorithm is that we
537*4882a593Smuzhiyun * first figure out the faulting address, then detect if there is any
538*4882a593Smuzhiyun * progress made on the copy, if so, redo the copy from last known copied
539*4882a593Smuzhiyun * location up to the faulting address (exclusive). In the copy_from_user
540*4882a593Smuzhiyun * case, remaining byte in kernel buffer will be zeroed.
541*4882a593Smuzhiyun *
542*4882a593Smuzhiyun * Take copy_from_user as an example, in the code there are multiple loads
543*4882a593Smuzhiyun * in a bundle and those multiple loads could span over two pages, the
544*4882a593Smuzhiyun * faulting address is calculated as page_round_down(max(src0, src1)).
545*4882a593Smuzhiyun * This is based on knowledge that if we can access one byte in a page, we
546*4882a593Smuzhiyun * can access any byte in that page.
547*4882a593Smuzhiyun *
548*4882a593Smuzhiyun * predicate used in the exception handler:
549*4882a593Smuzhiyun * p6-p7: direction
550*4882a593Smuzhiyun * p10-p11: src faulting addr calculation
551*4882a593Smuzhiyun * p12-p13: dst faulting addr calculation
552*4882a593Smuzhiyun */
553*4882a593Smuzhiyun
554*4882a593Smuzhiyun#define A	r19
555*4882a593Smuzhiyun#define B	r20
556*4882a593Smuzhiyun#define C	r21
557*4882a593Smuzhiyun#define D	r22
558*4882a593Smuzhiyun#define F	r28
559*4882a593Smuzhiyun
560*4882a593Smuzhiyun#define saved_retval	loc0
561*4882a593Smuzhiyun#define saved_rtlink	loc1
562*4882a593Smuzhiyun#define saved_pfs_stack	loc2
563*4882a593Smuzhiyun
564*4882a593Smuzhiyun.ex_hndlr_s:
565*4882a593Smuzhiyun	add	src0=8,src0
566*4882a593Smuzhiyun	br.sptk .ex_handler
567*4882a593Smuzhiyun	;;
568*4882a593Smuzhiyun.ex_hndlr_d:
569*4882a593Smuzhiyun	add	dst0=8,dst0
570*4882a593Smuzhiyun	br.sptk .ex_handler
571*4882a593Smuzhiyun	;;
572*4882a593Smuzhiyun.ex_hndlr_lcpy_1:
573*4882a593Smuzhiyun	mov	src1=src_pre_mem
574*4882a593Smuzhiyun	mov	dst1=dst_pre_mem
575*4882a593Smuzhiyun	cmp.gtu	p10,p11=src_pre_mem,saved_in1
576*4882a593Smuzhiyun	cmp.gtu	p12,p13=dst_pre_mem,saved_in0
577*4882a593Smuzhiyun	;;
578*4882a593Smuzhiyun(p10)	add	src0=8,saved_in1
579*4882a593Smuzhiyun(p11)	mov	src0=saved_in1
580*4882a593Smuzhiyun(p12)	add	dst0=8,saved_in0
581*4882a593Smuzhiyun(p13)	mov	dst0=saved_in0
582*4882a593Smuzhiyun	br.sptk	.ex_handler
583*4882a593Smuzhiyun.ex_handler_lcpy:
584*4882a593Smuzhiyun	// in line_copy block, the preload addresses should always ahead
585*4882a593Smuzhiyun	// of the other two src/dst pointers.  Furthermore, src1/dst1 should
586*4882a593Smuzhiyun	// always ahead of src0/dst0.
587*4882a593Smuzhiyun	mov	src1=src_pre_mem
588*4882a593Smuzhiyun	mov	dst1=dst_pre_mem
589*4882a593Smuzhiyun.ex_handler:
590*4882a593Smuzhiyun	mov	pr=saved_pr,-1		// first restore pr, lc, and pfs
591*4882a593Smuzhiyun	mov	ar.lc=saved_lc
592*4882a593Smuzhiyun	mov	ar.pfs=saved_pfs
593*4882a593Smuzhiyun	;;
594*4882a593Smuzhiyun.ex_handler_short: // fault occurred in these sections didn't change pr, lc, pfs
595*4882a593Smuzhiyun	cmp.ltu	p6,p7=saved_in0, saved_in1	// get the copy direction
596*4882a593Smuzhiyun	cmp.ltu	p10,p11=src0,src1
597*4882a593Smuzhiyun	cmp.ltu	p12,p13=dst0,dst1
598*4882a593Smuzhiyun	fcmp.eq	p8,p0=f6,f0		// is it memcpy?
599*4882a593Smuzhiyun	mov	tmp = dst0
600*4882a593Smuzhiyun	;;
601*4882a593Smuzhiyun(p11)	mov	src1 = src0		// pick the larger of the two
602*4882a593Smuzhiyun(p13)	mov	dst0 = dst1		// make dst0 the smaller one
603*4882a593Smuzhiyun(p13)	mov	dst1 = tmp		// and dst1 the larger one
604*4882a593Smuzhiyun	;;
605*4882a593Smuzhiyun(p6)	dep	F = r0,dst1,0,PAGE_SHIFT // usr dst round down to page boundary
606*4882a593Smuzhiyun(p7)	dep	F = r0,src1,0,PAGE_SHIFT // usr src round down to page boundary
607*4882a593Smuzhiyun	;;
608*4882a593Smuzhiyun(p6)	cmp.le	p14,p0=dst0,saved_in0	// no progress has been made on store
609*4882a593Smuzhiyun(p7)	cmp.le	p14,p0=src0,saved_in1	// no progress has been made on load
610*4882a593Smuzhiyun	mov	retval=saved_in2
611*4882a593Smuzhiyun(p8)	ld1	tmp=[src1]		// force an oops for memcpy call
612*4882a593Smuzhiyun(p8)	st1	[dst1]=r0		// force an oops for memcpy call
613*4882a593Smuzhiyun(p14)	br.ret.sptk.many rp
614*4882a593Smuzhiyun
615*4882a593Smuzhiyun/*
616*4882a593Smuzhiyun * The remaining byte to copy is calculated as:
617*4882a593Smuzhiyun *
618*4882a593Smuzhiyun * A =	(faulting_addr - orig_src)	-> len to faulting ld address
619*4882a593Smuzhiyun *	or
620*4882a593Smuzhiyun * 	(faulting_addr - orig_dst)	-> len to faulting st address
621*4882a593Smuzhiyun * B =	(cur_dst - orig_dst)		-> len copied so far
622*4882a593Smuzhiyun * C =	A - B				-> len need to be copied
623*4882a593Smuzhiyun * D =	orig_len - A			-> len need to be left along
624*4882a593Smuzhiyun */
625*4882a593Smuzhiyun(p6)	sub	A = F, saved_in0
626*4882a593Smuzhiyun(p7)	sub	A = F, saved_in1
627*4882a593Smuzhiyun	clrrrb
628*4882a593Smuzhiyun	;;
629*4882a593Smuzhiyun	alloc	saved_pfs_stack=ar.pfs,3,3,3,0
630*4882a593Smuzhiyun	cmp.lt	p8,p0=A,r0
631*4882a593Smuzhiyun	sub	B = dst0, saved_in0	// how many byte copied so far
632*4882a593Smuzhiyun	;;
633*4882a593Smuzhiyun(p8)	mov	A = 0;			// A shouldn't be negative, cap it
634*4882a593Smuzhiyun	;;
635*4882a593Smuzhiyun	sub	C = A, B
636*4882a593Smuzhiyun	sub	D = saved_in2, A
637*4882a593Smuzhiyun	;;
638*4882a593Smuzhiyun	cmp.gt	p8,p0=C,r0		// more than 1 byte?
639*4882a593Smuzhiyun	mov	r8=0
640*4882a593Smuzhiyun	mov	saved_retval = D
641*4882a593Smuzhiyun	mov	saved_rtlink = b0
642*4882a593Smuzhiyun
643*4882a593Smuzhiyun	add	out0=saved_in0, B
644*4882a593Smuzhiyun	add	out1=saved_in1, B
645*4882a593Smuzhiyun	mov	out2=C
646*4882a593Smuzhiyun(p8)	br.call.sptk.few b0=__copy_user	// recursive call
647*4882a593Smuzhiyun	;;
648*4882a593Smuzhiyun
649*4882a593Smuzhiyun	add	saved_retval=saved_retval,r8	// above might return non-zero value
650*4882a593Smuzhiyun	;;
651*4882a593Smuzhiyun
652*4882a593Smuzhiyun	mov	retval=saved_retval
653*4882a593Smuzhiyun	mov	ar.pfs=saved_pfs_stack
654*4882a593Smuzhiyun	mov	b0=saved_rtlink
655*4882a593Smuzhiyun	br.ret.sptk.many rp
656*4882a593Smuzhiyun
657*4882a593Smuzhiyun/* end of McKinley specific optimization */
658*4882a593SmuzhiyunEND(__copy_user)
659*4882a593SmuzhiyunEXPORT_SYMBOL(__copy_user)
660