// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * LoongArch SIMD XOR operations
 *
 * Copyright (C) 2023 WANG Xuerui <git@xen0n.name>
 */

#include <linux/export.h>
#include <linux/sched.h>
#include <asm/fpu.h>
#include <asm/xor_simd.h>
#include "xor_simd.h"

#define MAKE_XOR_GLUE_2(flavor)       \
void xor_##flavor##_2(unsigned long bytes, unsigned long * __restrict p1, \
        const unsigned long * __restrict p2)   \
{          \
 kernel_fpu_begin();       \
 __xor_##flavor##_2(bytes, p1, p2);     \
 kernel_fpu_end();       \
}          \
EXPORT_SYMBOL_GPL(xor_##flavor##_2)

#define MAKE_XOR_GLUE_3(flavor)       \
void xor_##flavor##_3(unsigned long bytes, unsigned long * __restrict p1, \
        const unsigned long * __restrict p2,   \
        const unsigned long * __restrict p3)   \
{          \
 kernel_fpu_begin();       \
 __xor_##flavor##_3(bytes, p1, p2, p3);     \
 kernel_fpu_end();       \
}          \
EXPORT_SYMBOL_GPL(xor_##flavor##_3)

#define MAKE_XOR_GLUE_4(flavor)       \
void xor_##flavor##_4(unsigned long bytes, unsigned long * __restrict p1, \
        const unsigned long * __restrict p2,   \
        const unsigned long * __restrict p3,   \
        const unsigned long * __restrict p4)   \
{          \
 kernel_fpu_begin();       \
 __xor_##flavor##_4(bytes, p1, p2, p3, p4);    \
 kernel_fpu_end();       \
}          \
EXPORT_SYMBOL_GPL(xor_##flavor##_4)

#define MAKE_XOR_GLUE_5(flavor)       \
void xor_##flavor##_5(unsigned long bytes, unsigned long * __restrict p1, \
        const unsigned long * __restrict p2,   \
        const unsigned long * __restrict p3,   \
        const unsigned long * __restrict p4,   \
        const unsigned long * __restrict p5)   \
{          \
 kernel_fpu_begin();       \
 __xor_##flavor##_5(bytes, p1, p2, p3, p4, p5);    \
 kernel_fpu_end();       \
}          \
EXPORT_SYMBOL_GPL(xor_##flavor##_5)

#define MAKE_XOR_GLUES(flavor)  \
 MAKE_XOR_GLUE_2(flavor); \
 MAKE_XOR_GLUE_3(flavor); \
 MAKE_XOR_GLUE_4(flavor); \
 MAKE_XOR_GLUE_5(flavor)

#ifdef CONFIG_CPU_HAS_LSX
MAKE_XOR_GLUES(lsx);
#endif

#ifdef CONFIG_CPU_HAS_LASX
MAKE_XOR_GLUES(lasx);
#endif