vector.S 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408
  1. #include <asm/processor.h>
  2. #include <asm/ppc_asm.h>
  3. #include <asm/reg.h>
  4. #include <asm/asm-offsets.h>
  5. #include <asm/cputable.h>
  6. #include <asm/thread_info.h>
  7. #include <asm/page.h>
  8. #include <asm/ptrace.h>
  9. /*
  10. * load_up_altivec(unused, unused, tsk)
  11. * Disable VMX for the task which had it previously,
  12. * and save its vector registers in its thread_struct.
  13. * Enables the VMX for use in the kernel on return.
  14. * On SMP we know the VMX is free, since we give it up every
  15. * switch (ie, no lazy save of the vector registers).
  16. */
  17. _GLOBAL(load_up_altivec)
  18. mfmsr r5 /* grab the current MSR */
  19. oris r5,r5,MSR_VEC@h
  20. MTMSRD(r5) /* enable use of AltiVec now */
  21. isync
  22. /*
  23. * For SMP, we don't do lazy VMX switching because it just gets too
  24. * horrendously complex, especially when a task switches from one CPU
  25. * to another. Instead we call giveup_altvec in switch_to.
  26. * VRSAVE isn't dealt with here, that is done in the normal context
  27. * switch code. Note that we could rely on vrsave value to eventually
  28. * avoid saving all of the VREGs here...
  29. */
  30. #ifndef CONFIG_SMP
  31. LOAD_REG_ADDRBASE(r3, last_task_used_altivec)
  32. toreal(r3)
  33. PPC_LL r4,ADDROFF(last_task_used_altivec)(r3)
  34. PPC_LCMPI 0,r4,0
  35. beq 1f
  36. /* Save VMX state to last_task_used_altivec's THREAD struct */
  37. toreal(r4)
  38. addi r4,r4,THREAD
  39. SAVE_32VRS(0,r5,r4)
  40. mfvscr vr0
  41. li r10,THREAD_VSCR
  42. stvx vr0,r10,r4
  43. /* Disable VMX for last_task_used_altivec */
  44. PPC_LL r5,PT_REGS(r4)
  45. toreal(r5)
  46. PPC_LL r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  47. lis r10,MSR_VEC@h
  48. andc r4,r4,r10
  49. PPC_STL r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  50. 1:
  51. #endif /* CONFIG_SMP */
  52. /* Hack: if we get an altivec unavailable trap with VRSAVE
  53. * set to all zeros, we assume this is a broken application
  54. * that fails to set it properly, and thus we switch it to
  55. * all 1's
  56. */
  57. mfspr r4,SPRN_VRSAVE
  58. cmpwi 0,r4,0
  59. bne+ 1f
  60. li r4,-1
  61. mtspr SPRN_VRSAVE,r4
  62. 1:
  63. /* enable use of VMX after return */
  64. #ifdef CONFIG_PPC32
  65. mfspr r5,SPRN_SPRG_THREAD /* current task's THREAD (phys) */
  66. oris r9,r9,MSR_VEC@h
  67. #else
  68. ld r4,PACACURRENT(r13)
  69. addi r5,r4,THREAD /* Get THREAD */
  70. oris r12,r12,MSR_VEC@h
  71. std r12,_MSR(r1)
  72. #endif
  73. li r4,1
  74. li r10,THREAD_VSCR
  75. stw r4,THREAD_USED_VR(r5)
  76. lvx vr0,r10,r5
  77. mtvscr vr0
  78. REST_32VRS(0,r4,r5)
  79. #ifndef CONFIG_SMP
  80. /* Update last_task_used_altivec to 'current' */
  81. subi r4,r5,THREAD /* Back to 'current' */
  82. fromreal(r4)
  83. PPC_STL r4,ADDROFF(last_task_used_altivec)(r3)
  84. #endif /* CONFIG_SMP */
  85. /* restore registers and return */
  86. blr
  87. /*
  88. * giveup_altivec(tsk)
  89. * Disable VMX for the task given as the argument,
  90. * and save the vector registers in its thread_struct.
  91. * Enables the VMX for use in the kernel on return.
  92. */
  93. _GLOBAL(giveup_altivec)
  94. mfmsr r5
  95. oris r5,r5,MSR_VEC@h
  96. SYNC
  97. MTMSRD(r5) /* enable use of VMX now */
  98. isync
  99. PPC_LCMPI 0,r3,0
  100. beqlr /* if no previous owner, done */
  101. addi r3,r3,THREAD /* want THREAD of task */
  102. PPC_LL r5,PT_REGS(r3)
  103. PPC_LCMPI 0,r5,0
  104. SAVE_32VRS(0,r4,r3)
  105. mfvscr vr0
  106. li r4,THREAD_VSCR
  107. stvx vr0,r4,r3
  108. beq 1f
  109. PPC_LL r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  110. #ifdef CONFIG_VSX
  111. BEGIN_FTR_SECTION
  112. lis r3,(MSR_VEC|MSR_VSX)@h
  113. FTR_SECTION_ELSE
  114. lis r3,MSR_VEC@h
  115. ALT_FTR_SECTION_END_IFSET(CPU_FTR_VSX)
  116. #else
  117. lis r3,MSR_VEC@h
  118. #endif
  119. andc r4,r4,r3 /* disable FP for previous task */
  120. PPC_STL r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  121. 1:
  122. #ifndef CONFIG_SMP
  123. li r5,0
  124. LOAD_REG_ADDRBASE(r4,last_task_used_altivec)
  125. PPC_STL r5,ADDROFF(last_task_used_altivec)(r4)
  126. #endif /* CONFIG_SMP */
  127. blr
  128. #ifdef CONFIG_VSX
  129. #ifdef CONFIG_PPC32
  130. #error This asm code isn't ready for 32-bit kernels
  131. #endif
  132. /*
  133. * load_up_vsx(unused, unused, tsk)
  134. * Disable VSX for the task which had it previously,
  135. * and save its vector registers in its thread_struct.
  136. * Reuse the fp and vsx saves, but first check to see if they have
  137. * been saved already.
  138. */
  139. _GLOBAL(load_up_vsx)
  140. /* Load FP and VSX registers if they haven't been done yet */
  141. andi. r5,r12,MSR_FP
  142. beql+ load_up_fpu /* skip if already loaded */
  143. andis. r5,r12,MSR_VEC@h
  144. beql+ load_up_altivec /* skip if already loaded */
  145. #ifndef CONFIG_SMP
  146. ld r3,last_task_used_vsx@got(r2)
  147. ld r4,0(r3)
  148. cmpdi 0,r4,0
  149. beq 1f
  150. /* Disable VSX for last_task_used_vsx */
  151. addi r4,r4,THREAD
  152. ld r5,PT_REGS(r4)
  153. ld r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  154. lis r6,MSR_VSX@h
  155. andc r6,r4,r6
  156. std r6,_MSR-STACK_FRAME_OVERHEAD(r5)
  157. 1:
  158. #endif /* CONFIG_SMP */
  159. ld r4,PACACURRENT(r13)
  160. addi r4,r4,THREAD /* Get THREAD */
  161. li r6,1
  162. stw r6,THREAD_USED_VSR(r4) /* ... also set thread used vsr */
  163. /* enable use of VSX after return */
  164. oris r12,r12,MSR_VSX@h
  165. std r12,_MSR(r1)
  166. #ifndef CONFIG_SMP
  167. /* Update last_task_used_vsx to 'current' */
  168. ld r4,PACACURRENT(r13)
  169. std r4,0(r3)
  170. #endif /* CONFIG_SMP */
  171. b fast_exception_return
  172. /*
  173. * __giveup_vsx(tsk)
  174. * Disable VSX for the task given as the argument.
  175. * Does NOT save vsx registers.
  176. * Enables the VSX for use in the kernel on return.
  177. */
  178. _GLOBAL(__giveup_vsx)
  179. mfmsr r5
  180. oris r5,r5,MSR_VSX@h
  181. mtmsrd r5 /* enable use of VSX now */
  182. isync
  183. cmpdi 0,r3,0
  184. beqlr- /* if no previous owner, done */
  185. addi r3,r3,THREAD /* want THREAD of task */
  186. ld r5,PT_REGS(r3)
  187. cmpdi 0,r5,0
  188. beq 1f
  189. ld r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  190. lis r3,MSR_VSX@h
  191. andc r4,r4,r3 /* disable VSX for previous task */
  192. std r4,_MSR-STACK_FRAME_OVERHEAD(r5)
  193. 1:
  194. #ifndef CONFIG_SMP
  195. li r5,0
  196. ld r4,last_task_used_vsx@got(r2)
  197. std r5,0(r4)
  198. #endif /* CONFIG_SMP */
  199. blr
  200. #endif /* CONFIG_VSX */
  201. /*
  202. * The routines below are in assembler so we can closely control the
  203. * usage of floating-point registers. These routines must be called
  204. * with preempt disabled.
  205. */
  206. #ifdef CONFIG_PPC32
  207. .data
  208. fpzero:
  209. .long 0
  210. fpone:
  211. .long 0x3f800000 /* 1.0 in single-precision FP */
  212. fphalf:
  213. .long 0x3f000000 /* 0.5 in single-precision FP */
  214. #define LDCONST(fr, name) \
  215. lis r11,name@ha; \
  216. lfs fr,name@l(r11)
  217. #else
  218. .section ".toc","aw"
  219. fpzero:
  220. .tc FD_0_0[TC],0
  221. fpone:
  222. .tc FD_3ff00000_0[TC],0x3ff0000000000000 /* 1.0 */
  223. fphalf:
  224. .tc FD_3fe00000_0[TC],0x3fe0000000000000 /* 0.5 */
  225. #define LDCONST(fr, name) \
  226. lfd fr,name@toc(r2)
  227. #endif
  228. .text
  229. /*
  230. * Internal routine to enable floating point and set FPSCR to 0.
  231. * Don't call it from C; it doesn't use the normal calling convention.
  232. */
  233. fpenable:
  234. #ifdef CONFIG_PPC32
  235. stwu r1,-64(r1)
  236. #else
  237. stdu r1,-64(r1)
  238. #endif
  239. mfmsr r10
  240. ori r11,r10,MSR_FP
  241. mtmsr r11
  242. isync
  243. stfd fr0,24(r1)
  244. stfd fr1,16(r1)
  245. stfd fr31,8(r1)
  246. LDCONST(fr1, fpzero)
  247. mffs fr31
  248. MTFSF_L(fr1)
  249. blr
  250. fpdisable:
  251. mtlr r12
  252. MTFSF_L(fr31)
  253. lfd fr31,8(r1)
  254. lfd fr1,16(r1)
  255. lfd fr0,24(r1)
  256. mtmsr r10
  257. isync
  258. addi r1,r1,64
  259. blr
  260. /*
  261. * Vector add, floating point.
  262. */
  263. _GLOBAL(vaddfp)
  264. mflr r12
  265. bl fpenable
  266. li r0,4
  267. mtctr r0
  268. li r6,0
  269. 1: lfsx fr0,r4,r6
  270. lfsx fr1,r5,r6
  271. fadds fr0,fr0,fr1
  272. stfsx fr0,r3,r6
  273. addi r6,r6,4
  274. bdnz 1b
  275. b fpdisable
  276. /*
  277. * Vector subtract, floating point.
  278. */
  279. _GLOBAL(vsubfp)
  280. mflr r12
  281. bl fpenable
  282. li r0,4
  283. mtctr r0
  284. li r6,0
  285. 1: lfsx fr0,r4,r6
  286. lfsx fr1,r5,r6
  287. fsubs fr0,fr0,fr1
  288. stfsx fr0,r3,r6
  289. addi r6,r6,4
  290. bdnz 1b
  291. b fpdisable
  292. /*
  293. * Vector multiply and add, floating point.
  294. */
  295. _GLOBAL(vmaddfp)
  296. mflr r12
  297. bl fpenable
  298. stfd fr2,32(r1)
  299. li r0,4
  300. mtctr r0
  301. li r7,0
  302. 1: lfsx fr0,r4,r7
  303. lfsx fr1,r5,r7
  304. lfsx fr2,r6,r7
  305. fmadds fr0,fr0,fr2,fr1
  306. stfsx fr0,r3,r7
  307. addi r7,r7,4
  308. bdnz 1b
  309. lfd fr2,32(r1)
  310. b fpdisable
  311. /*
  312. * Vector negative multiply and subtract, floating point.
  313. */
  314. _GLOBAL(vnmsubfp)
  315. mflr r12
  316. bl fpenable
  317. stfd fr2,32(r1)
  318. li r0,4
  319. mtctr r0
  320. li r7,0
  321. 1: lfsx fr0,r4,r7
  322. lfsx fr1,r5,r7
  323. lfsx fr2,r6,r7
  324. fnmsubs fr0,fr0,fr2,fr1
  325. stfsx fr0,r3,r7
  326. addi r7,r7,4
  327. bdnz 1b
  328. lfd fr2,32(r1)
  329. b fpdisable
  330. /*
  331. * Vector reciprocal estimate. We just compute 1.0/x.
  332. * r3 -> destination, r4 -> source.
  333. */
  334. _GLOBAL(vrefp)
  335. mflr r12
  336. bl fpenable
  337. li r0,4
  338. LDCONST(fr1, fpone)
  339. mtctr r0
  340. li r6,0
  341. 1: lfsx fr0,r4,r6
  342. fdivs fr0,fr1,fr0
  343. stfsx fr0,r3,r6
  344. addi r6,r6,4
  345. bdnz 1b
  346. b fpdisable
  347. /*
  348. * Vector reciprocal square-root estimate, floating point.
  349. * We use the frsqrte instruction for the initial estimate followed
  350. * by 2 iterations of Newton-Raphson to get sufficient accuracy.
  351. * r3 -> destination, r4 -> source.
  352. */
  353. _GLOBAL(vrsqrtefp)
  354. mflr r12
  355. bl fpenable
  356. stfd fr2,32(r1)
  357. stfd fr3,40(r1)
  358. stfd fr4,48(r1)
  359. stfd fr5,56(r1)
  360. li r0,4
  361. LDCONST(fr4, fpone)
  362. LDCONST(fr5, fphalf)
  363. mtctr r0
  364. li r6,0
  365. 1: lfsx fr0,r4,r6
  366. frsqrte fr1,fr0 /* r = frsqrte(s) */
  367. fmuls fr3,fr1,fr0 /* r * s */
  368. fmuls fr2,fr1,fr5 /* r * 0.5 */
  369. fnmsubs fr3,fr1,fr3,fr4 /* 1 - s * r * r */
  370. fmadds fr1,fr2,fr3,fr1 /* r = r + 0.5 * r * (1 - s * r * r) */
  371. fmuls fr3,fr1,fr0 /* r * s */
  372. fmuls fr2,fr1,fr5 /* r * 0.5 */
  373. fnmsubs fr3,fr1,fr3,fr4 /* 1 - s * r * r */
  374. fmadds fr1,fr2,fr3,fr1 /* r = r + 0.5 * r * (1 - s * r * r) */
  375. stfsx fr1,r3,r6
  376. addi r6,r6,4
  377. bdnz 1b
  378. lfd fr5,56(r1)
  379. lfd fr4,48(r1)
  380. lfd fr3,40(r1)
  381. lfd fr2,32(r1)
  382. b fpdisable