Merge tag 'powerpc-4.9-1' of git://git.kernel.org/pub/scm/linux/kernel/git/powerpc...
[cascardo/linux.git] / tools / testing / selftests / powerpc / math / vsx_asm.S
1 /*
2  * Copyright 2015, Cyril Bur, IBM Corp.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version
7  * 2 of the License, or (at your option) any later version.
8  */
9
10 #include "../basic_asm.h"
11 #include "../vsx_asm.h"
12
13 #long check_vsx(vector int *r3);
14 #This function wraps storeing VSX regs to the end of an array and a
15 #call to a comparison function in C which boils down to a memcmp()
16 FUNC_START(check_vsx)
17         PUSH_BASIC_STACK(32)
18         std     r3,STACK_FRAME_PARAM(0)(sp)
19         addi r3, r3, 16 * 12 #Second half of array
20         bl store_vsx
21         ld r3,STACK_FRAME_PARAM(0)(sp)
22         bl vsx_memcmp
23         POP_BASIC_STACK(32)
24         blr
25 FUNC_END(check_vsx)
26
27 # int preempt_vmx(vector int *varray, int *threads_starting,
28 #                 int *running);
29 # On starting will (atomically) decrement threads_starting as a signal
30 # that the VMX have been loaded with varray. Will proceed to check the
31 # validity of the VMX registers while running is not zero.
32 FUNC_START(preempt_vsx)
33         PUSH_BASIC_STACK(512)
34         std r3,STACK_FRAME_PARAM(0)(sp) # vector int *varray
35         std r4,STACK_FRAME_PARAM(1)(sp) # int *threads_starting
36         std r5,STACK_FRAME_PARAM(2)(sp) # int *running
37
38         bl load_vsx
39         nop
40
41         sync
42         # Atomic DEC
43         ld r3,STACK_FRAME_PARAM(1)(sp)
44 1:      lwarx r4,0,r3
45         addi r4,r4,-1
46         stwcx. r4,0,r3
47         bne- 1b
48
49 2:      ld r3,STACK_FRAME_PARAM(0)(sp)
50         bl check_vsx
51         nop
52         cmpdi r3,0
53         bne 3f
54         ld r4,STACK_FRAME_PARAM(2)(sp)
55         ld r5,0(r4)
56         cmpwi r5,0
57         bne 2b
58
59 3:      POP_BASIC_STACK(512)
60         blr
61 FUNC_END(preempt_vsx)