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()
16FUNC_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
25FUNC_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.
32FUNC_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)
441:	lwarx r4,0,r3
45	addi r4,r4,-1
46	stwcx. r4,0,r3
47	bne- 1b
48
492:	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
593:	POP_BASIC_STACK(512)
60	blr
61FUNC_END(preempt_vsx)
62