[mirror_ubuntu-jammy-kernel.git] / tools / testing / selftests / powerpc / math / vsx_asm.S

/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright 2015, Cyril Bur, IBM Corp.
 */

#include "basic_asm.h"
#include "vsx_asm.h"

#long check_vsx(vector int *r3);
#This function wraps storeing VSX regs to the end of an array and a
#call to a comparison function in C which boils down to a memcmp()
FUNC_START(check_vsx)
	PUSH_BASIC_STACK(32)
	std	r3,STACK_FRAME_PARAM(0)(sp)
	addi r3, r3, 16 * 12 #Second half of array
	bl store_vsx
	ld r3,STACK_FRAME_PARAM(0)(sp)
	bl vsx_memcmp
	POP_BASIC_STACK(32)
	blr
FUNC_END(check_vsx)

# int preempt_vmx(vector int *varray, int *threads_starting,
#                 int *running);
# On starting will (atomically) decrement threads_starting as a signal
# that the VMX have been loaded with varray. Will proceed to check the
# validity of the VMX registers while running is not zero.
FUNC_START(preempt_vsx)
	PUSH_BASIC_STACK(512)
	std r3,STACK_FRAME_PARAM(0)(sp) # vector int *varray
	std r4,STACK_FRAME_PARAM(1)(sp) # int *threads_starting
	std r5,STACK_FRAME_PARAM(2)(sp) # int *running

	bl load_vsx
	nop

	sync
	# Atomic DEC
	ld r3,STACK_FRAME_PARAM(1)(sp)
1:	lwarx r4,0,r3
	addi r4,r4,-1
	stwcx. r4,0,r3
	bne- 1b

2:	ld r3,STACK_FRAME_PARAM(0)(sp)
	bl check_vsx
	nop
	cmpdi r3,0
	bne 3f
	ld r4,STACK_FRAME_PARAM(2)(sp)
	ld r5,0(r4)
	cmpwi r5,0
	bne 2b

3:	POP_BASIC_STACK(512)
	blr
FUNC_END(preempt_vsx)
Commit	Line	Data
2874c5fd	1	/* SPDX-License-Identifier: GPL-2.0-or-later */
65ca668f CB	2	/*
65ca668f CB	3	* Copyright 2015, Cyril Bur, IBM Corp.
65ca668f CB	4	*/
65ca668f CB	5
15ec3997 SG	6	#include "basic_asm.h"
15ec3997 SG	7	#include "vsx_asm.h"
65ca668f CB	8
	9	#long check_vsx(vector int *r3);
	10	#This function wraps storeing VSX regs to the end of an array and a
	11	#call to a comparison function in C which boils down to a memcmp()
	12	FUNC_START(check_vsx)
	13	PUSH_BASIC_STACK(32)
	14	std r3,STACK_FRAME_PARAM(0)(sp)
	15	addi r3, r3, 16 * 12 #Second half of array
	16	bl store_vsx
	17	ld r3,STACK_FRAME_PARAM(0)(sp)
	18	bl vsx_memcmp
	19	POP_BASIC_STACK(32)
	20	blr
	21	FUNC_END(check_vsx)
	22
	23	# int preempt_vmx(vector int varray, int threads_starting,
	24	# int *running);
	25	# On starting will (atomically) decrement threads_starting as a signal
	26	# that the VMX have been loaded with varray. Will proceed to check the
	27	# validity of the VMX registers while running is not zero.
	28	FUNC_START(preempt_vsx)
	29	PUSH_BASIC_STACK(512)
	30	std r3,STACK_FRAME_PARAM(0)(sp) # vector int *varray
	31	std r4,STACK_FRAME_PARAM(1)(sp) # int *threads_starting
	32	std r5,STACK_FRAME_PARAM(2)(sp) # int *running
	33
	34	bl load_vsx
	35	nop
	36
	37	sync
	38	# Atomic DEC
	39	ld r3,STACK_FRAME_PARAM(1)(sp)
	40	1: lwarx r4,0,r3
	41	addi r4,r4,-1
	42	stwcx. r4,0,r3
	43	bne- 1b
	44
	45	2: ld r3,STACK_FRAME_PARAM(0)(sp)
	46	bl check_vsx
	47	nop
	48	cmpdi r3,0
	49	bne 3f
	50	ld r4,STACK_FRAME_PARAM(2)(sp)
	51	ld r5,0(r4)
	52	cmpwi r5,0
	53	bne 2b
	54
	55	3: POP_BASIC_STACK(512)
	56	blr
	57	FUNC_END(preempt_vsx)