ARK-OS/arkos/kernel/lib/raid6/rvv.h

/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright 2024 Institute of Software, CAS.
 *
 * raid6/rvv.h
 *
 * Definitions for RISC-V RAID-6 code
 */

#ifdef __KERNEL__
#include <asm/vector.h>
#else
#define kernel_vector_begin()
#define kernel_vector_end()
#include <sys/auxv.h>
#include <asm/hwcap.h>
#define has_vector() (getauxval(AT_HWCAP) & COMPAT_HWCAP_ISA_V)
#endif

#include <linux/raid/pq.h>

static int rvv_has_vector(void)
{
	return has_vector();
}

#define RAID6_RVV_WRAPPER(_n)						\
	static void raid6_rvv ## _n ## _gen_syndrome(int disks,		\
					size_t bytes, void **ptrs)	\
	{								\
		void raid6_rvv ## _n  ## _gen_syndrome_real(int d,	\
					unsigned long b, void **p);	\
		kernel_vector_begin();					\
		raid6_rvv ## _n ## _gen_syndrome_real(disks,		\
				(unsigned long)bytes, ptrs);		\
		kernel_vector_end();					\
	}								\
	static void raid6_rvv ## _n ## _xor_syndrome(int disks,		\
					int start, int stop,		\
					size_t bytes, void **ptrs)	\
	{								\
		void raid6_rvv ## _n  ## _xor_syndrome_real(int d,	\
					int s1, int s2,			\
					unsigned long b, void **p);	\
		kernel_vector_begin();					\
		raid6_rvv ## _n ## _xor_syndrome_real(disks,		\
			start, stop, (unsigned long)bytes, ptrs);	\
		kernel_vector_end();					\
	}								\
	struct raid6_calls const raid6_rvvx ## _n = {			\
		raid6_rvv ## _n ## _gen_syndrome,			\
		raid6_rvv ## _n ## _xor_syndrome,			\
		rvv_has_vector,						\
		"rvvx" #_n,						\
		0							\
	}