MIPS: OCTEON: Fix FP context save.
[firefly-linux-kernel-4.4.55.git] / arch / mips / kernel / octeon_switch.S
1 /*
2  * This file is subject to the terms and conditions of the GNU General Public
3  * License.  See the file "COPYING" in the main directory of this archive
4  * for more details.
5  *
6  * Copyright (C) 1994, 1995, 1996, 1998, 1999, 2002, 2003 Ralf Baechle
7  * Copyright (C) 1996 David S. Miller (davem@davemloft.net)
8  * Copyright (C) 1994, 1995, 1996, by Andreas Busse
9  * Copyright (C) 1999 Silicon Graphics, Inc.
10  * Copyright (C) 2000 MIPS Technologies, Inc.
11  *    written by Carsten Langgaard, carstenl@mips.com
12  */
13
14 #define USE_ALTERNATE_RESUME_IMPL 1
15         .set push
16         .set arch=mips64r2
17 #include "r4k_switch.S"
18         .set pop
19 /*
20  * task_struct *resume(task_struct *prev, task_struct *next,
21  *                     struct thread_info *next_ti, int usedfpu)
22  */
23         .align  7
24         LEAF(resume)
25         .set arch=octeon
26         mfc0    t1, CP0_STATUS
27         LONG_S  t1, THREAD_STATUS(a0)
28         cpu_save_nonscratch a0
29         LONG_S  ra, THREAD_REG31(a0)
30
31         /*
32          * check if we need to save FPU registers
33          */
34         .set push
35         .set noreorder
36         beqz    a3, 1f
37          PTR_L  t3, TASK_THREAD_INFO(a0)
38         .set pop
39
40         /*
41          * clear saved user stack CU1 bit
42          */
43         LONG_L  t0, ST_OFF(t3)
44         li      t1, ~ST0_CU1
45         and     t0, t0, t1
46         LONG_S  t0, ST_OFF(t3)
47
48         .set push
49         .set arch=mips64r2
50         fpu_save_double a0 t0 t1                # c0_status passed in t0
51                                                 # clobbers t1
52         .set pop
53 1:
54
55         /* check if we need to save COP2 registers */
56         LONG_L  t0, ST_OFF(t3)
57         bbit0   t0, 30, 1f
58
59         /* Disable COP2 in the stored process state */
60         li      t1, ST0_CU2
61         xor     t0, t1
62         LONG_S  t0, ST_OFF(t3)
63
64         /* Enable COP2 so we can save it */
65         mfc0    t0, CP0_STATUS
66         or      t0, t1
67         mtc0    t0, CP0_STATUS
68
69         /* Save COP2 */
70         daddu   a0, THREAD_CP2
71         jal octeon_cop2_save
72         dsubu   a0, THREAD_CP2
73
74         /* Disable COP2 now that we are done */
75         mfc0    t0, CP0_STATUS
76         li      t1, ST0_CU2
77         xor     t0, t1
78         mtc0    t0, CP0_STATUS
79
80 1:
81 #if CONFIG_CAVIUM_OCTEON_CVMSEG_SIZE > 0
82         /* Check if we need to store CVMSEG state */
83         mfc0    t0, $11,7       /* CvmMemCtl */
84         bbit0   t0, 6, 3f       /* Is user access enabled? */
85
86         /* Store the CVMSEG state */
87         /* Extract the size of CVMSEG */
88         andi    t0, 0x3f
89         /* Multiply * (cache line size/sizeof(long)/2) */
90         sll     t0, 7-LONGLOG-1
91         li      t1, -32768      /* Base address of CVMSEG */
92         LONG_ADDI t2, a0, THREAD_CVMSEG /* Where to store CVMSEG to */
93         synciobdma
94 2:
95         .set noreorder
96         LONG_L  t8, 0(t1)       /* Load from CVMSEG */
97         subu    t0, 1           /* Decrement loop var */
98         LONG_L  t9, LONGSIZE(t1)/* Load from CVMSEG */
99         LONG_ADDU t1, LONGSIZE*2 /* Increment loc in CVMSEG */
100         LONG_S  t8, 0(t2)       /* Store CVMSEG to thread storage */
101         LONG_ADDU t2, LONGSIZE*2 /* Increment loc in thread storage */
102         bnez    t0, 2b          /* Loop until we've copied it all */
103          LONG_S t9, -LONGSIZE(t2)/* Store CVMSEG to thread storage */
104         .set reorder
105
106         /* Disable access to CVMSEG */
107         mfc0    t0, $11,7       /* CvmMemCtl */
108         xori    t0, t0, 0x40    /* Bit 6 is CVMSEG user enable */
109         mtc0    t0, $11,7       /* CvmMemCtl */
110 #endif
111 3:
112
113 #if defined(CONFIG_CC_STACKPROTECTOR) && !defined(CONFIG_SMP)
114         PTR_LA  t8, __stack_chk_guard
115         LONG_L  t9, TASK_STACK_CANARY(a1)
116         LONG_S  t9, 0(t8)
117 #endif
118
119         /*
120          * The order of restoring the registers takes care of the race
121          * updating $28, $29 and kernelsp without disabling ints.
122          */
123         move    $28, a2
124         cpu_restore_nonscratch a1
125
126         PTR_ADDU        t0, $28, _THREAD_SIZE - 32
127         set_saved_sp    t0, t1, t2
128
129         mfc0    t1, CP0_STATUS          /* Do we really need this? */
130         li      a3, 0xff01
131         and     t1, a3
132         LONG_L  a2, THREAD_STATUS(a1)
133         nor     a3, $0, a3
134         and     a2, a3
135         or      a2, t1
136         mtc0    a2, CP0_STATUS
137         move    v0, a0
138         jr      ra
139         END(resume)
140
141 /*
142  * void octeon_cop2_save(struct octeon_cop2_state *a0)
143  */
144         .align  7
145         LEAF(octeon_cop2_save)
146
147         dmfc0   t9, $9,7        /* CvmCtl register. */
148
149         /* Save the COP2 CRC state */
150         dmfc2   t0, 0x0201
151         dmfc2   t1, 0x0202
152         dmfc2   t2, 0x0200
153         sd      t0, OCTEON_CP2_CRC_IV(a0)
154         sd      t1, OCTEON_CP2_CRC_LENGTH(a0)
155         sd      t2, OCTEON_CP2_CRC_POLY(a0)
156         /* Skip next instructions if CvmCtl[NODFA_CP2] set */
157         bbit1   t9, 28, 1f
158
159         /* Save the LLM state */
160         dmfc2   t0, 0x0402
161         dmfc2   t1, 0x040A
162         sd      t0, OCTEON_CP2_LLM_DAT(a0)
163         sd      t1, OCTEON_CP2_LLM_DAT+8(a0)
164
165 1:      bbit1   t9, 26, 3f      /* done if CvmCtl[NOCRYPTO] set */
166
167         /* Save the COP2 crypto state */
168         /* this part is mostly common to both pass 1 and later revisions */
169         dmfc2   t0, 0x0084
170         dmfc2   t1, 0x0080
171         dmfc2   t2, 0x0081
172         dmfc2   t3, 0x0082
173         sd      t0, OCTEON_CP2_3DES_IV(a0)
174         dmfc2   t0, 0x0088
175         sd      t1, OCTEON_CP2_3DES_KEY(a0)
176         dmfc2   t1, 0x0111                      /* only necessary for pass 1 */
177         sd      t2, OCTEON_CP2_3DES_KEY+8(a0)
178         dmfc2   t2, 0x0102
179         sd      t3, OCTEON_CP2_3DES_KEY+16(a0)
180         dmfc2   t3, 0x0103
181         sd      t0, OCTEON_CP2_3DES_RESULT(a0)
182         dmfc2   t0, 0x0104
183         sd      t1, OCTEON_CP2_AES_INP0(a0)     /* only necessary for pass 1 */
184         dmfc2   t1, 0x0105
185         sd      t2, OCTEON_CP2_AES_IV(a0)
186         dmfc2   t2, 0x0106
187         sd      t3, OCTEON_CP2_AES_IV+8(a0)
188         dmfc2   t3, 0x0107
189         sd      t0, OCTEON_CP2_AES_KEY(a0)
190         dmfc2   t0, 0x0110
191         sd      t1, OCTEON_CP2_AES_KEY+8(a0)
192         dmfc2   t1, 0x0100
193         sd      t2, OCTEON_CP2_AES_KEY+16(a0)
194         dmfc2   t2, 0x0101
195         sd      t3, OCTEON_CP2_AES_KEY+24(a0)
196         mfc0    t3, $15,0       /* Get the processor ID register */
197         sd      t0, OCTEON_CP2_AES_KEYLEN(a0)
198         li      t0, 0x000d0000  /* This is the processor ID of Octeon Pass1 */
199         sd      t1, OCTEON_CP2_AES_RESULT(a0)
200         sd      t2, OCTEON_CP2_AES_RESULT+8(a0)
201         /* Skip to the Pass1 version of the remainder of the COP2 state */
202         beq     t3, t0, 2f
203
204         /* the non-pass1 state when !CvmCtl[NOCRYPTO] */
205         dmfc2   t1, 0x0240
206         dmfc2   t2, 0x0241
207         dmfc2   t3, 0x0242
208         dmfc2   t0, 0x0243
209         sd      t1, OCTEON_CP2_HSH_DATW(a0)
210         dmfc2   t1, 0x0244
211         sd      t2, OCTEON_CP2_HSH_DATW+8(a0)
212         dmfc2   t2, 0x0245
213         sd      t3, OCTEON_CP2_HSH_DATW+16(a0)
214         dmfc2   t3, 0x0246
215         sd      t0, OCTEON_CP2_HSH_DATW+24(a0)
216         dmfc2   t0, 0x0247
217         sd      t1, OCTEON_CP2_HSH_DATW+32(a0)
218         dmfc2   t1, 0x0248
219         sd      t2, OCTEON_CP2_HSH_DATW+40(a0)
220         dmfc2   t2, 0x0249
221         sd      t3, OCTEON_CP2_HSH_DATW+48(a0)
222         dmfc2   t3, 0x024A
223         sd      t0, OCTEON_CP2_HSH_DATW+56(a0)
224         dmfc2   t0, 0x024B
225         sd      t1, OCTEON_CP2_HSH_DATW+64(a0)
226         dmfc2   t1, 0x024C
227         sd      t2, OCTEON_CP2_HSH_DATW+72(a0)
228         dmfc2   t2, 0x024D
229         sd      t3, OCTEON_CP2_HSH_DATW+80(a0)
230         dmfc2   t3, 0x024E
231         sd      t0, OCTEON_CP2_HSH_DATW+88(a0)
232         dmfc2   t0, 0x0250
233         sd      t1, OCTEON_CP2_HSH_DATW+96(a0)
234         dmfc2   t1, 0x0251
235         sd      t2, OCTEON_CP2_HSH_DATW+104(a0)
236         dmfc2   t2, 0x0252
237         sd      t3, OCTEON_CP2_HSH_DATW+112(a0)
238         dmfc2   t3, 0x0253
239         sd      t0, OCTEON_CP2_HSH_IVW(a0)
240         dmfc2   t0, 0x0254
241         sd      t1, OCTEON_CP2_HSH_IVW+8(a0)
242         dmfc2   t1, 0x0255
243         sd      t2, OCTEON_CP2_HSH_IVW+16(a0)
244         dmfc2   t2, 0x0256
245         sd      t3, OCTEON_CP2_HSH_IVW+24(a0)
246         dmfc2   t3, 0x0257
247         sd      t0, OCTEON_CP2_HSH_IVW+32(a0)
248         dmfc2   t0, 0x0258
249         sd      t1, OCTEON_CP2_HSH_IVW+40(a0)
250         dmfc2   t1, 0x0259
251         sd      t2, OCTEON_CP2_HSH_IVW+48(a0)
252         dmfc2   t2, 0x025E
253         sd      t3, OCTEON_CP2_HSH_IVW+56(a0)
254         dmfc2   t3, 0x025A
255         sd      t0, OCTEON_CP2_GFM_MULT(a0)
256         dmfc2   t0, 0x025B
257         sd      t1, OCTEON_CP2_GFM_MULT+8(a0)
258         sd      t2, OCTEON_CP2_GFM_POLY(a0)
259         sd      t3, OCTEON_CP2_GFM_RESULT(a0)
260         sd      t0, OCTEON_CP2_GFM_RESULT+8(a0)
261         jr      ra
262
263 2:      /* pass 1 special stuff when !CvmCtl[NOCRYPTO] */
264         dmfc2   t3, 0x0040
265         dmfc2   t0, 0x0041
266         dmfc2   t1, 0x0042
267         dmfc2   t2, 0x0043
268         sd      t3, OCTEON_CP2_HSH_DATW(a0)
269         dmfc2   t3, 0x0044
270         sd      t0, OCTEON_CP2_HSH_DATW+8(a0)
271         dmfc2   t0, 0x0045
272         sd      t1, OCTEON_CP2_HSH_DATW+16(a0)
273         dmfc2   t1, 0x0046
274         sd      t2, OCTEON_CP2_HSH_DATW+24(a0)
275         dmfc2   t2, 0x0048
276         sd      t3, OCTEON_CP2_HSH_DATW+32(a0)
277         dmfc2   t3, 0x0049
278         sd      t0, OCTEON_CP2_HSH_DATW+40(a0)
279         dmfc2   t0, 0x004A
280         sd      t1, OCTEON_CP2_HSH_DATW+48(a0)
281         sd      t2, OCTEON_CP2_HSH_IVW(a0)
282         sd      t3, OCTEON_CP2_HSH_IVW+8(a0)
283         sd      t0, OCTEON_CP2_HSH_IVW+16(a0)
284
285 3:      /* pass 1 or CvmCtl[NOCRYPTO] set */
286         jr      ra
287         END(octeon_cop2_save)
288
289 /*
290  * void octeon_cop2_restore(struct octeon_cop2_state *a0)
291  */
292         .align  7
293         .set push
294         .set noreorder
295         LEAF(octeon_cop2_restore)
296         /* First cache line was prefetched before the call */
297         pref    4,  128(a0)
298         dmfc0   t9, $9,7        /* CvmCtl register. */
299
300         pref    4,  256(a0)
301         ld      t0, OCTEON_CP2_CRC_IV(a0)
302         pref    4,  384(a0)
303         ld      t1, OCTEON_CP2_CRC_LENGTH(a0)
304         ld      t2, OCTEON_CP2_CRC_POLY(a0)
305
306         /* Restore the COP2 CRC state */
307         dmtc2   t0, 0x0201
308         dmtc2   t1, 0x1202
309         bbit1   t9, 28, 2f      /* Skip LLM if CvmCtl[NODFA_CP2] is set */
310          dmtc2  t2, 0x4200
311
312         /* Restore the LLM state */
313         ld      t0, OCTEON_CP2_LLM_DAT(a0)
314         ld      t1, OCTEON_CP2_LLM_DAT+8(a0)
315         dmtc2   t0, 0x0402
316         dmtc2   t1, 0x040A
317
318 2:
319         bbit1   t9, 26, done_restore    /* done if CvmCtl[NOCRYPTO] set */
320          nop
321
322         /* Restore the COP2 crypto state common to pass 1 and pass 2 */
323         ld      t0, OCTEON_CP2_3DES_IV(a0)
324         ld      t1, OCTEON_CP2_3DES_KEY(a0)
325         ld      t2, OCTEON_CP2_3DES_KEY+8(a0)
326         dmtc2   t0, 0x0084
327         ld      t0, OCTEON_CP2_3DES_KEY+16(a0)
328         dmtc2   t1, 0x0080
329         ld      t1, OCTEON_CP2_3DES_RESULT(a0)
330         dmtc2   t2, 0x0081
331         ld      t2, OCTEON_CP2_AES_INP0(a0) /* only really needed for pass 1 */
332         dmtc2   t0, 0x0082
333         ld      t0, OCTEON_CP2_AES_IV(a0)
334         dmtc2   t1, 0x0098
335         ld      t1, OCTEON_CP2_AES_IV+8(a0)
336         dmtc2   t2, 0x010A                  /* only really needed for pass 1 */
337         ld      t2, OCTEON_CP2_AES_KEY(a0)
338         dmtc2   t0, 0x0102
339         ld      t0, OCTEON_CP2_AES_KEY+8(a0)
340         dmtc2   t1, 0x0103
341         ld      t1, OCTEON_CP2_AES_KEY+16(a0)
342         dmtc2   t2, 0x0104
343         ld      t2, OCTEON_CP2_AES_KEY+24(a0)
344         dmtc2   t0, 0x0105
345         ld      t0, OCTEON_CP2_AES_KEYLEN(a0)
346         dmtc2   t1, 0x0106
347         ld      t1, OCTEON_CP2_AES_RESULT(a0)
348         dmtc2   t2, 0x0107
349         ld      t2, OCTEON_CP2_AES_RESULT+8(a0)
350         mfc0    t3, $15,0       /* Get the processor ID register */
351         dmtc2   t0, 0x0110
352         li      t0, 0x000d0000  /* This is the processor ID of Octeon Pass1 */
353         dmtc2   t1, 0x0100
354         bne     t0, t3, 3f      /* Skip the next stuff for non-pass1 */
355          dmtc2  t2, 0x0101
356
357         /* this code is specific for pass 1 */
358         ld      t0, OCTEON_CP2_HSH_DATW(a0)
359         ld      t1, OCTEON_CP2_HSH_DATW+8(a0)
360         ld      t2, OCTEON_CP2_HSH_DATW+16(a0)
361         dmtc2   t0, 0x0040
362         ld      t0, OCTEON_CP2_HSH_DATW+24(a0)
363         dmtc2   t1, 0x0041
364         ld      t1, OCTEON_CP2_HSH_DATW+32(a0)
365         dmtc2   t2, 0x0042
366         ld      t2, OCTEON_CP2_HSH_DATW+40(a0)
367         dmtc2   t0, 0x0043
368         ld      t0, OCTEON_CP2_HSH_DATW+48(a0)
369         dmtc2   t1, 0x0044
370         ld      t1, OCTEON_CP2_HSH_IVW(a0)
371         dmtc2   t2, 0x0045
372         ld      t2, OCTEON_CP2_HSH_IVW+8(a0)
373         dmtc2   t0, 0x0046
374         ld      t0, OCTEON_CP2_HSH_IVW+16(a0)
375         dmtc2   t1, 0x0048
376         dmtc2   t2, 0x0049
377         b done_restore   /* unconditional branch */
378          dmtc2  t0, 0x004A
379
380 3:      /* this is post-pass1 code */
381         ld      t2, OCTEON_CP2_HSH_DATW(a0)
382         ld      t0, OCTEON_CP2_HSH_DATW+8(a0)
383         ld      t1, OCTEON_CP2_HSH_DATW+16(a0)
384         dmtc2   t2, 0x0240
385         ld      t2, OCTEON_CP2_HSH_DATW+24(a0)
386         dmtc2   t0, 0x0241
387         ld      t0, OCTEON_CP2_HSH_DATW+32(a0)
388         dmtc2   t1, 0x0242
389         ld      t1, OCTEON_CP2_HSH_DATW+40(a0)
390         dmtc2   t2, 0x0243
391         ld      t2, OCTEON_CP2_HSH_DATW+48(a0)
392         dmtc2   t0, 0x0244
393         ld      t0, OCTEON_CP2_HSH_DATW+56(a0)
394         dmtc2   t1, 0x0245
395         ld      t1, OCTEON_CP2_HSH_DATW+64(a0)
396         dmtc2   t2, 0x0246
397         ld      t2, OCTEON_CP2_HSH_DATW+72(a0)
398         dmtc2   t0, 0x0247
399         ld      t0, OCTEON_CP2_HSH_DATW+80(a0)
400         dmtc2   t1, 0x0248
401         ld      t1, OCTEON_CP2_HSH_DATW+88(a0)
402         dmtc2   t2, 0x0249
403         ld      t2, OCTEON_CP2_HSH_DATW+96(a0)
404         dmtc2   t0, 0x024A
405         ld      t0, OCTEON_CP2_HSH_DATW+104(a0)
406         dmtc2   t1, 0x024B
407         ld      t1, OCTEON_CP2_HSH_DATW+112(a0)
408         dmtc2   t2, 0x024C
409         ld      t2, OCTEON_CP2_HSH_IVW(a0)
410         dmtc2   t0, 0x024D
411         ld      t0, OCTEON_CP2_HSH_IVW+8(a0)
412         dmtc2   t1, 0x024E
413         ld      t1, OCTEON_CP2_HSH_IVW+16(a0)
414         dmtc2   t2, 0x0250
415         ld      t2, OCTEON_CP2_HSH_IVW+24(a0)
416         dmtc2   t0, 0x0251
417         ld      t0, OCTEON_CP2_HSH_IVW+32(a0)
418         dmtc2   t1, 0x0252
419         ld      t1, OCTEON_CP2_HSH_IVW+40(a0)
420         dmtc2   t2, 0x0253
421         ld      t2, OCTEON_CP2_HSH_IVW+48(a0)
422         dmtc2   t0, 0x0254
423         ld      t0, OCTEON_CP2_HSH_IVW+56(a0)
424         dmtc2   t1, 0x0255
425         ld      t1, OCTEON_CP2_GFM_MULT(a0)
426         dmtc2   t2, 0x0256
427         ld      t2, OCTEON_CP2_GFM_MULT+8(a0)
428         dmtc2   t0, 0x0257
429         ld      t0, OCTEON_CP2_GFM_POLY(a0)
430         dmtc2   t1, 0x0258
431         ld      t1, OCTEON_CP2_GFM_RESULT(a0)
432         dmtc2   t2, 0x0259
433         ld      t2, OCTEON_CP2_GFM_RESULT+8(a0)
434         dmtc2   t0, 0x025E
435         dmtc2   t1, 0x025A
436         dmtc2   t2, 0x025B
437
438 done_restore:
439         jr      ra
440          nop
441         END(octeon_cop2_restore)
442         .set pop
443
444 /*
445  * void octeon_mult_save()
446  * sp is assumed to point to a struct pt_regs
447  *
448  * NOTE: This is called in SAVE_TEMP in stackframe.h. It can
449  *       safely modify v1,k0, k1,$10-$15, and $24.  It will
450  *       be overwritten with a processor specific version of the code.
451  */
452         .p2align 7
453         .set push
454         .set noreorder
455         LEAF(octeon_mult_save)
456         jr      ra
457          nop
458         .space 30 * 4, 0
459 octeon_mult_save_end:
460         EXPORT(octeon_mult_save_end)
461         END(octeon_mult_save)
462
463         LEAF(octeon_mult_save2)
464         /* Save the multiplier state OCTEON II and earlier*/
465         v3mulu  k0, $0, $0
466         v3mulu  k1, $0, $0
467         sd      k0, PT_MTP(sp)        /* PT_MTP    has P0 */
468         v3mulu  k0, $0, $0
469         sd      k1, PT_MTP+8(sp)      /* PT_MTP+8  has P1 */
470         ori     k1, $0, 1
471         v3mulu  k1, k1, $0
472         sd      k0, PT_MTP+16(sp)     /* PT_MTP+16 has P2 */
473         v3mulu  k0, $0, $0
474         sd      k1, PT_MPL(sp)        /* PT_MPL    has MPL0 */
475         v3mulu  k1, $0, $0
476         sd      k0, PT_MPL+8(sp)      /* PT_MPL+8  has MPL1 */
477         jr      ra
478          sd     k1, PT_MPL+16(sp)     /* PT_MPL+16 has MPL2 */
479 octeon_mult_save2_end:
480         EXPORT(octeon_mult_save2_end)
481         END(octeon_mult_save2)
482
483         LEAF(octeon_mult_save3)
484         /* Save the multiplier state OCTEON III */
485         v3mulu  $10, $0, $0             /* read P0 */
486         v3mulu  $11, $0, $0             /* read P1 */
487         v3mulu  $12, $0, $0             /* read P2 */
488         sd      $10, PT_MTP+(0*8)(sp)   /* store P0 */
489         v3mulu  $10, $0, $0             /* read P3 */
490         sd      $11, PT_MTP+(1*8)(sp)   /*  store P1 */
491         v3mulu  $11, $0, $0             /* read P4 */
492         sd      $12, PT_MTP+(2*8)(sp)   /* store P2 */
493         ori     $13, $0, 1
494         v3mulu  $12, $0, $0             /* read P5 */
495         sd      $10, PT_MTP+(3*8)(sp)   /* store P3 */
496         v3mulu  $13, $13, $0            /* P4-P0 = MPL5-MPL1, $13 = MPL0 */
497         sd      $11, PT_MTP+(4*8)(sp)   /* store P4 */
498         v3mulu  $10, $0, $0             /* read MPL1 */
499         sd      $12, PT_MTP+(5*8)(sp)   /* store P5 */
500         v3mulu  $11, $0, $0             /* read MPL2 */
501         sd      $13, PT_MPL+(0*8)(sp)   /* store MPL0 */
502         v3mulu  $12, $0, $0             /* read MPL3 */
503         sd      $10, PT_MPL+(1*8)(sp)   /* store MPL1 */
504         v3mulu  $10, $0, $0             /* read MPL4 */
505         sd      $11, PT_MPL+(2*8)(sp)   /* store MPL2 */
506         v3mulu  $11, $0, $0             /* read MPL5 */
507         sd      $12, PT_MPL+(3*8)(sp)   /* store MPL3 */
508         sd      $10, PT_MPL+(4*8)(sp)   /* store MPL4 */
509         jr      ra
510          sd     $11, PT_MPL+(5*8)(sp)   /* store MPL5 */
511 octeon_mult_save3_end:
512         EXPORT(octeon_mult_save3_end)
513         END(octeon_mult_save3)
514         .set pop
515
516 /*
517  * void octeon_mult_restore()
518  * sp is assumed to point to a struct pt_regs
519  *
520  * NOTE: This is called in RESTORE_TEMP in stackframe.h.
521  */
522         .p2align 7
523         .set push
524         .set noreorder
525         LEAF(octeon_mult_restore)
526         jr      ra
527          nop
528         .space 30 * 4, 0
529 octeon_mult_restore_end:
530         EXPORT(octeon_mult_restore_end)
531         END(octeon_mult_restore)
532
533         LEAF(octeon_mult_restore2)
534         ld      v0, PT_MPL(sp)          /* MPL0 */
535         ld      v1, PT_MPL+8(sp)        /* MPL1 */
536         ld      k0, PT_MPL+16(sp)       /* MPL2 */
537         /* Restore the multiplier state */
538         ld      k1, PT_MTP+16(sp)       /* P2 */
539         mtm0    v0                      /* MPL0 */
540         ld      v0, PT_MTP+8(sp)        /* P1 */
541         mtm1    v1                      /* MPL1 */
542         ld      v1, PT_MTP(sp)          /* P0 */
543         mtm2    k0                      /* MPL2 */
544         mtp2    k1                      /* P2 */
545         mtp1    v0                      /* P1 */
546         jr      ra
547          mtp0   v1                      /* P0 */
548 octeon_mult_restore2_end:
549         EXPORT(octeon_mult_restore2_end)
550         END(octeon_mult_restore2)
551
552         LEAF(octeon_mult_restore3)
553         ld      $12, PT_MPL+(0*8)(sp)   /* read MPL0 */
554         ld      $13, PT_MPL+(3*8)(sp)   /* read MPL3 */
555         ld      $10, PT_MPL+(1*8)(sp)   /* read MPL1 */
556         ld      $11, PT_MPL+(4*8)(sp)   /* read MPL4 */
557         .word   0x718d0008
558         /* mtm0 $12, $13                   restore MPL0 and MPL3 */
559         ld      $12, PT_MPL+(2*8)(sp)   /* read MPL2 */
560         .word   0x714b000c
561         /* mtm1 $10, $11                   restore MPL1 and MPL4 */
562         ld      $13, PT_MPL+(5*8)(sp)   /* read MPL5 */
563         ld      $10, PT_MTP+(0*8)(sp)   /* read P0 */
564         ld      $11, PT_MTP+(3*8)(sp)   /* read P3 */
565         .word   0x718d000d
566         /* mtm2 $12, $13                   restore MPL2 and MPL5 */
567         ld      $12, PT_MTP+(1*8)(sp)   /* read P1 */
568         .word   0x714b0009
569         /* mtp0 $10, $11                   restore P0 and P3 */
570         ld      $13, PT_MTP+(4*8)(sp)   /* read P4 */
571         ld      $10, PT_MTP+(2*8)(sp)   /* read P2 */
572         ld      $11, PT_MTP+(5*8)(sp)   /* read P5 */
573         .word   0x718d000a
574         /* mtp1 $12, $13                   restore P1 and P4 */
575         jr      ra
576         .word   0x714b000b
577         /* mtp2 $10, $11                   restore P2 and P5 */
578
579 octeon_mult_restore3_end:
580         EXPORT(octeon_mult_restore3_end)
581         END(octeon_mult_restore3)
582         .set pop