/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2023 WANG Xuerui <git@xen0n.name>
 */
#ifndef _ASM_LOONGARCH_XOR_H
#define _ASM_LOONGARCH_XOR_H

#include <asm/cpu-features.h>
#include <asm/xor_simd.h>

#ifdef CONFIG_CPU_HAS_LSX
static struct xor_block_template xor_block_lsx = {
        .name = "lsx",
        .do_2 = xor_lsx_2,
        .do_3 = xor_lsx_3,
        .do_4 = xor_lsx_4,
        .do_5 = xor_lsx_5,
};

#define XOR_SPEED_LSX()                                 \
        do {                                            \
                if (cpu_has_lsx)                        \
                        xor_speed(&xor_block_lsx);      \
        } while (0)
#else /* CONFIG_CPU_HAS_LSX */
#define XOR_SPEED_LSX()
#endif /* CONFIG_CPU_HAS_LSX */

#ifdef CONFIG_CPU_HAS_LASX
static struct xor_block_template xor_block_lasx = {
        .name = "lasx",
        .do_2 = xor_lasx_2,
        .do_3 = xor_lasx_3,
        .do_4 = xor_lasx_4,
        .do_5 = xor_lasx_5,
};

#define XOR_SPEED_LASX()                                        \
        do {                                                    \
                if (cpu_has_lasx)                               \
                        xor_speed(&xor_block_lasx);             \
        } while (0)
#else /* CONFIG_CPU_HAS_LASX */
#define XOR_SPEED_LASX()
#endif /* CONFIG_CPU_HAS_LASX */

/*
 * For grins, also test the generic routines.
 *
 * More importantly: it cannot be ruled out at this point of time, that some
 * future (maybe reduced) models could run the vector algorithms slower than
 * the scalar ones, maybe for errata or micro-op reasons. It may be
 * appropriate to revisit this after one or two more uarch generations.
 */
#include <asm-generic/xor.h>

#undef XOR_TRY_TEMPLATES
#define XOR_TRY_TEMPLATES                               \
do {                                                    \
        xor_speed(&xor_block_8regs);                    \
        xor_speed(&xor_block_8regs_p);                  \
        xor_speed(&xor_block_32regs);                   \
        xor_speed(&xor_block_32regs_p);                 \
        XOR_SPEED_LSX();                                \
        XOR_SPEED_LASX();                               \
} while (0)

#endif /* _ASM_LOONGARCH_XOR_H */