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