xref: /kernel/linux/linux-6.6/arch/loongarch/lib/xor_template.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2023 WANG Xuerui <git@xen0n.name>
 *
 * Template for XOR operations, instantiated in xor_simd.c.
 *
 * Expected preprocessor definitions:
 *
 * - LINE_WIDTH
 * - XOR_FUNC_NAME(nr)
 * - LD_INOUT_LINE(buf)
 * - LD_AND_XOR_LINE(buf)
 * - ST_LINE(buf)
 */

void XOR_FUNC_NAME(2)(unsigned long bytes,
		      unsigned long * __restrict v1,
		      const unsigned long * __restrict v2)
{
	unsigned long lines = bytes / LINE_WIDTH;

	do {
		__asm__ __volatile__ (
			LD_INOUT_LINE(v1)
			LD_AND_XOR_LINE(v2)
			ST_LINE(v1)
		: : [v1] "r"(v1), [v2] "r"(v2) : "memory"
		);

		v1 += LINE_WIDTH / sizeof(unsigned long);
		v2 += LINE_WIDTH / sizeof(unsigned long);
	} while (--lines > 0);
}

void XOR_FUNC_NAME(3)(unsigned long bytes,
		      unsigned long * __restrict v1,
		      const unsigned long * __restrict v2,
		      const unsigned long * __restrict v3)
{
	unsigned long lines = bytes / LINE_WIDTH;

	do {
		__asm__ __volatile__ (
			LD_INOUT_LINE(v1)
			LD_AND_XOR_LINE(v2)
			LD_AND_XOR_LINE(v3)
			ST_LINE(v1)
		: : [v1] "r"(v1), [v2] "r"(v2), [v3] "r"(v3) : "memory"
		);

		v1 += LINE_WIDTH / sizeof(unsigned long);
		v2 += LINE_WIDTH / sizeof(unsigned long);
		v3 += LINE_WIDTH / sizeof(unsigned long);
	} while (--lines > 0);
}

void XOR_FUNC_NAME(4)(unsigned long bytes,
		      unsigned long * __restrict v1,
		      const unsigned long * __restrict v2,
		      const unsigned long * __restrict v3,
		      const unsigned long * __restrict v4)
{
	unsigned long lines = bytes / LINE_WIDTH;

	do {
		__asm__ __volatile__ (
			LD_INOUT_LINE(v1)
			LD_AND_XOR_LINE(v2)
			LD_AND_XOR_LINE(v3)
			LD_AND_XOR_LINE(v4)
			ST_LINE(v1)
		: : [v1] "r"(v1), [v2] "r"(v2), [v3] "r"(v3), [v4] "r"(v4)
		: "memory"
		);

		v1 += LINE_WIDTH / sizeof(unsigned long);
		v2 += LINE_WIDTH / sizeof(unsigned long);
		v3 += LINE_WIDTH / sizeof(unsigned long);
		v4 += LINE_WIDTH / sizeof(unsigned long);
	} while (--lines > 0);
}

void XOR_FUNC_NAME(5)(unsigned long bytes,
		      unsigned long * __restrict v1,
		      const unsigned long * __restrict v2,
		      const unsigned long * __restrict v3,
		      const unsigned long * __restrict v4,
		      const unsigned long * __restrict v5)
{
	unsigned long lines = bytes / LINE_WIDTH;

	do {
		__asm__ __volatile__ (
			LD_INOUT_LINE(v1)
			LD_AND_XOR_LINE(v2)
			LD_AND_XOR_LINE(v3)
			LD_AND_XOR_LINE(v4)
			LD_AND_XOR_LINE(v5)
			ST_LINE(v1)
		: : [v1] "r"(v1), [v2] "r"(v2), [v3] "r"(v3), [v4] "r"(v4),
		    [v5] "r"(v5) : "memory"
		);

		v1 += LINE_WIDTH / sizeof(unsigned long);
		v2 += LINE_WIDTH / sizeof(unsigned long);
		v3 += LINE_WIDTH / sizeof(unsigned long);
		v4 += LINE_WIDTH / sizeof(unsigned long);
		v5 += LINE_WIDTH / sizeof(unsigned long);
	} while (--lines > 0);
}