// SPDX-License-Identifier: GPL-2.0-or-later
// Copyright (C) 2026 Andrea Mazzoleni

#include "internal.h"
#include "gf.h"
#include "cpu.h"

#ifdef CONFIG_X86_64

/*
 * GEN2 (RAID6 with powers of 2) AVX2 GFNI implementation
 */
void raid_gen2_avx2gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[0][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));

			asm volatile ("vpbroadcastb %0,%%ymm6" : : "m" (gfcauchy[1][d]));

			asm volatile ("vpxor     %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor     %ymm5,%ymm1,%ymm1");

			asm volatile ("vgf2p8mulb %ymm4,%ymm6,%ymm8");
			asm volatile ("vgf2p8mulb %ymm5,%ymm6,%ymm9");
			asm volatile ("vpxor    %ymm8,%ymm2,%ymm2");
			asm volatile ("vpxor    %ymm9,%ymm3,%ymm3");
		}

		asm volatile ("vmovntdq  %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%ymm1,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq  %%ymm2,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%ymm3,%0" : "=m" (q[i + 32]));
	}

	raid_avx_end();
}

/*
 * GEN2 (RAID6 with powers of 2) GFNI implementation
 */
void raid_gen2_avx512gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa64 %zmm0,%zmm1");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa64 %0,%%zmm6" : : "m" (v[d][i]));

			asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (gfcauchy[1][d]));

			asm volatile ("vgf2p8mulb %zmm6,%zmm7,%zmm12");

			asm volatile ("vpxorq    %zmm6,%zmm0,%zmm0");
			asm volatile ("vpxorq    %zmm12,%zmm1,%zmm1");
		}

		asm volatile ("vmovntdq  %%zmm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%zmm1,%0" : "=m" (q[i]));
	}

	raid_avx_end();
}

/*
 * GEN3 (triple parity with Cauchy matrix) AVX2 GFNI implementation
 */
void raid_gen3_avx2gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[0][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");
		asm volatile ("vmovdqa %ymm0,%ymm4");
		asm volatile ("vmovdqa %ymm1,%ymm5");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa %0,%%ymm6" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm7" : : "m" (v[d][i + 32]));

			asm volatile ("vpbroadcastb %0,%%ymm8" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%ymm9" : : "m" (gfcauchy[2][d]));

			asm volatile ("vpxor     %ymm6,%ymm0,%ymm0");
			asm volatile ("vpxor     %ymm7,%ymm1,%ymm1");

			asm volatile ("vgf2p8mulb %ymm6,%ymm8,%ymm10");
			asm volatile ("vgf2p8mulb %ymm7,%ymm8,%ymm11");
			asm volatile ("vpxor    %ymm10,%ymm2,%ymm2");
			asm volatile ("vpxor    %ymm11,%ymm3,%ymm3");

			asm volatile ("vgf2p8mulb %ymm6,%ymm9,%ymm10");
			asm volatile ("vgf2p8mulb %ymm7,%ymm9,%ymm11");
			asm volatile ("vpxor    %ymm10,%ymm4,%ymm4");
			asm volatile ("vpxor    %ymm11,%ymm5,%ymm5");
		}

		asm volatile ("vmovntdq  %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%ymm1,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq  %%ymm2,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%ymm3,%0" : "=m" (q[i + 32]));
		asm volatile ("vmovntdq  %%ymm4,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%ymm5,%0" : "=m" (r[i + 32]));
	}

	raid_avx_end();
}

/*
 * GEN3 (triple parity with Cauchy matrix) GFNI implementation
 */
void raid_gen3_avx512gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa64 %zmm0,%zmm1");
		asm volatile ("vmovdqa64 %zmm0,%zmm2");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa64 %0,%%zmm6" : : "m" (v[d][i]));

			asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (gfcauchy[2][d]));

			asm volatile ("vgf2p8mulb %zmm6,%zmm7,%zmm12");
			asm volatile ("vgf2p8mulb %zmm6,%zmm8,%zmm13");

			asm volatile ("vpxorq    %zmm6,%zmm0,%zmm0");
			asm volatile ("vpxorq    %zmm12,%zmm1,%zmm1");
			asm volatile ("vpxorq    %zmm13,%zmm2,%zmm2");
		}

		asm volatile ("vmovntdq  %%zmm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%zmm1,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%zmm2,%0" : "=m" (r[i]));
	}

	raid_avx_end();
}

/*
 * GEN4 (quad parity with Cauchy matrix) AVX2 GFNI implementation
 */
void raid_gen4_avx2gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[0][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");
		asm volatile ("vmovdqa %ymm0,%ymm4");
		asm volatile ("vmovdqa %ymm1,%ymm5");
		asm volatile ("vmovdqa %ymm0,%ymm6");
		asm volatile ("vmovdqa %ymm1,%ymm7");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa %0,%%ymm8" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm9" : : "m" (v[d][i + 32]));

			asm volatile ("vpbroadcastb %0,%%ymm10" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%ymm11" : : "m" (gfcauchy[2][d]));
			asm volatile ("vpbroadcastb %0,%%ymm12" : : "m" (gfcauchy[3][d]));

			asm volatile ("vpxor     %ymm8,%ymm0,%ymm0");
			asm volatile ("vpxor     %ymm9,%ymm1,%ymm1");

			asm volatile ("vgf2p8mulb %ymm8,%ymm10,%ymm13");
			asm volatile ("vgf2p8mulb %ymm9,%ymm10,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm2,%ymm2");
			asm volatile ("vpxor    %ymm14,%ymm3,%ymm3");

			asm volatile ("vgf2p8mulb %ymm8,%ymm11,%ymm13");
			asm volatile ("vgf2p8mulb %ymm9,%ymm11,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm4,%ymm4");
			asm volatile ("vpxor    %ymm14,%ymm5,%ymm5");

			asm volatile ("vgf2p8mulb %ymm8,%ymm12,%ymm13");
			asm volatile ("vgf2p8mulb %ymm9,%ymm12,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm6,%ymm6");
			asm volatile ("vpxor    %ymm14,%ymm7,%ymm7");
		}

		asm volatile ("vmovntdq  %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%ymm1,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq  %%ymm2,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%ymm3,%0" : "=m" (q[i + 32]));
		asm volatile ("vmovntdq  %%ymm4,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%ymm5,%0" : "=m" (r[i + 32]));
		asm volatile ("vmovntdq  %%ymm6,%0" : "=m" (s[i]));
		asm volatile ("vmovntdq  %%ymm7,%0" : "=m" (s[i + 32]));
	}

	raid_avx_end();
}

/*
 * GEN4 (quad parity with Cauchy matrix) GFNI implementation
 */
void raid_gen4_avx512gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa64 %zmm0,%zmm1");
		asm volatile ("vmovdqa64 %zmm0,%zmm2");
		asm volatile ("vmovdqa64 %zmm0,%zmm3");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa64 %0,%%zmm6" : : "m" (v[d][i]));

			asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (gfcauchy[2][d]));
			asm volatile ("vpbroadcastb %0,%%zmm9" : : "m" (gfcauchy[3][d]));

			asm volatile ("vgf2p8mulb %zmm6,%zmm7,%zmm12");
			asm volatile ("vgf2p8mulb %zmm6,%zmm8,%zmm13");
			asm volatile ("vgf2p8mulb %zmm6,%zmm9,%zmm14");

			asm volatile ("vpxorq    %zmm6,%zmm0,%zmm0");
			asm volatile ("vpxorq    %zmm12,%zmm1,%zmm1");
			asm volatile ("vpxorq    %zmm13,%zmm2,%zmm2");
			asm volatile ("vpxorq    %zmm14,%zmm3,%zmm3");
		}

		asm volatile ("vmovntdq  %%zmm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%zmm1,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%zmm2,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%zmm3,%0" : "=m" (s[i]));
	}

	raid_avx_end();
}

/*
 * GEN5 (penta parity with Cauchy matrix) AVX2 GFNI implementation
 */
void raid_gen5_avx2gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	uint8_t *t = v[nd + 4];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[0][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");
		asm volatile ("vmovdqa %ymm0,%ymm4");
		asm volatile ("vmovdqa %ymm1,%ymm5");
		asm volatile ("vmovdqa %ymm0,%ymm6");
		asm volatile ("vmovdqa %ymm1,%ymm7");
		asm volatile ("vmovdqa %ymm0,%ymm8");
		asm volatile ("vmovdqa %ymm1,%ymm9");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm11" : : "m" (v[d][i + 32]));

			asm volatile ("vpxor     %ymm10,%ymm0,%ymm0");
			asm volatile ("vpxor     %ymm11,%ymm1,%ymm1");

			asm volatile ("vpbroadcastb %0,%%ymm12" : : "m" (gfcauchy[1][d]));
			asm volatile ("vgf2p8mulb %ymm10,%ymm12,%ymm13");
			asm volatile ("vgf2p8mulb %ymm11,%ymm12,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm2,%ymm2");
			asm volatile ("vpxor    %ymm14,%ymm3,%ymm3");

			asm volatile ("vpbroadcastb %0,%%ymm12" : : "m" (gfcauchy[2][d]));
			asm volatile ("vgf2p8mulb %ymm10,%ymm12,%ymm13");
			asm volatile ("vgf2p8mulb %ymm11,%ymm12,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm4,%ymm4");
			asm volatile ("vpxor    %ymm14,%ymm5,%ymm5");

			asm volatile ("vpbroadcastb %0,%%ymm12" : : "m" (gfcauchy[3][d]));
			asm volatile ("vgf2p8mulb %ymm10,%ymm12,%ymm13");
			asm volatile ("vgf2p8mulb %ymm11,%ymm12,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm6,%ymm6");
			asm volatile ("vpxor    %ymm14,%ymm7,%ymm7");

			asm volatile ("vpbroadcastb %0,%%ymm12" : : "m" (gfcauchy[4][d]));
			asm volatile ("vgf2p8mulb %ymm10,%ymm12,%ymm13");
			asm volatile ("vgf2p8mulb %ymm11,%ymm12,%ymm14");
			asm volatile ("vpxor    %ymm13,%ymm8,%ymm8");
			asm volatile ("vpxor    %ymm14,%ymm9,%ymm9");
		}

		asm volatile ("vmovntdq  %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%ymm1,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq  %%ymm2,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%ymm3,%0" : "=m" (q[i + 32]));
		asm volatile ("vmovntdq  %%ymm4,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%ymm5,%0" : "=m" (r[i + 32]));
		asm volatile ("vmovntdq  %%ymm6,%0" : "=m" (s[i]));
		asm volatile ("vmovntdq  %%ymm7,%0" : "=m" (s[i + 32]));
		asm volatile ("vmovntdq  %%ymm8,%0" : "=m" (t[i]));
		asm volatile ("vmovntdq  %%ymm9,%0" : "=m" (t[i + 32]));
	}

	raid_avx_end();
}

/*
 * GEN5 (penta parity with Cauchy matrix) GFNI implementation
 */
void raid_gen5_avx512gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	uint8_t *t = v[nd + 4];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa64 %zmm0,%zmm1");
		asm volatile ("vmovdqa64 %zmm0,%zmm2");
		asm volatile ("vmovdqa64 %zmm0,%zmm3");
		asm volatile ("vmovdqa64 %zmm0,%zmm4");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa64 %0,%%zmm6" : : "m" (v[d][i]));

			asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (gfcauchy[2][d]));
			asm volatile ("vpbroadcastb %0,%%zmm9" : : "m" (gfcauchy[3][d]));
			asm volatile ("vpbroadcastb %0,%%zmm10" : : "m" (gfcauchy[4][d]));

			asm volatile ("vgf2p8mulb %zmm6,%zmm7,%zmm12");
			asm volatile ("vgf2p8mulb %zmm6,%zmm8,%zmm13");
			asm volatile ("vgf2p8mulb %zmm6,%zmm9,%zmm14");
			asm volatile ("vgf2p8mulb %zmm6,%zmm10,%zmm15");

			asm volatile ("vpxorq    %zmm6,%zmm0,%zmm0");
			asm volatile ("vpxorq    %zmm12,%zmm1,%zmm1");
			asm volatile ("vpxorq    %zmm13,%zmm2,%zmm2");
			asm volatile ("vpxorq    %zmm14,%zmm3,%zmm3");
			asm volatile ("vpxorq    %zmm15,%zmm4,%zmm4");
		}

		asm volatile ("vmovntdq  %%zmm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%zmm1,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%zmm2,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%zmm3,%0" : "=m" (s[i]));
		asm volatile ("vmovntdq  %%zmm4,%0" : "=m" (t[i]));
	}

	raid_avx_end();
}

/*
 * GEN6 (hexa parity with Cauchy matrix) AVX2 GFNI implementation
 */
void raid_gen6_avx2gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	uint8_t *t = v[nd + 4];
	uint8_t *u = v[nd + 5];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[0][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");
		asm volatile ("vmovdqa %ymm0,%ymm4");
		asm volatile ("vmovdqa %ymm1,%ymm5");
		asm volatile ("vmovdqa %ymm0,%ymm6");
		asm volatile ("vmovdqa %ymm1,%ymm7");
		asm volatile ("vmovdqa %ymm0,%ymm8");
		asm volatile ("vmovdqa %ymm1,%ymm9");
		asm volatile ("vmovdqa %ymm0,%ymm10");
		asm volatile ("vmovdqa %ymm1,%ymm11");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm13" : : "m" (v[d][i + 32]));

			asm volatile ("vpxor     %ymm12,%ymm0,%ymm0");
			asm volatile ("vpxor     %ymm13,%ymm1,%ymm1");

			asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfcauchy[1][d]));
			asm volatile ("vgf2p8mulb %ymm12,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm2,%ymm2");
			asm volatile ("vgf2p8mulb %ymm13,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm3,%ymm3");

			asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfcauchy[2][d]));
			asm volatile ("vgf2p8mulb %ymm12,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm4,%ymm4");
			asm volatile ("vgf2p8mulb %ymm13,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm5,%ymm5");

			asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfcauchy[3][d]));
			asm volatile ("vgf2p8mulb %ymm12,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm6,%ymm6");
			asm volatile ("vgf2p8mulb %ymm13,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm7,%ymm7");

			asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfcauchy[4][d]));
			asm volatile ("vgf2p8mulb %ymm12,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm8,%ymm8");
			asm volatile ("vgf2p8mulb %ymm13,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm9,%ymm9");

			asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfcauchy[5][d]));
			asm volatile ("vgf2p8mulb %ymm12,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm10,%ymm10");
			asm volatile ("vgf2p8mulb %ymm13,%ymm14,%ymm15");
			asm volatile ("vpxor    %ymm15,%ymm11,%ymm11");
		}

		asm volatile ("vmovntdq  %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%ymm1,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq  %%ymm2,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%ymm3,%0" : "=m" (q[i + 32]));
		asm volatile ("vmovntdq  %%ymm4,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%ymm5,%0" : "=m" (r[i + 32]));
		asm volatile ("vmovntdq  %%ymm6,%0" : "=m" (s[i]));
		asm volatile ("vmovntdq  %%ymm7,%0" : "=m" (s[i + 32]));
		asm volatile ("vmovntdq  %%ymm8,%0" : "=m" (t[i]));
		asm volatile ("vmovntdq  %%ymm9,%0" : "=m" (t[i + 32]));
		asm volatile ("vmovntdq  %%ymm10,%0" : "=m" (u[i]));
		asm volatile ("vmovntdq  %%ymm11,%0" : "=m" (u[i + 32]));
	}

	raid_avx_end();
}

/*
 * GEN6 (hexa parity with Cauchy matrix) GFNI implementation
 */
void raid_gen6_avx512gfni(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p = v[nd];
	uint8_t *q = v[nd + 1];
	uint8_t *r = v[nd + 2];
	uint8_t *s = v[nd + 3];
	uint8_t *t = v[nd + 4];
	uint8_t *u = v[nd + 5];
	int d, l = nd - 1;
	size_t i;

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
		asm volatile ("vmovdqa64 %zmm0,%zmm1");
		asm volatile ("vmovdqa64 %zmm0,%zmm2");
		asm volatile ("vmovdqa64 %zmm0,%zmm3");
		asm volatile ("vmovdqa64 %zmm0,%zmm4");
		asm volatile ("vmovdqa64 %zmm0,%zmm5");

		for (d = 1; d <= l; ++d) {
			asm volatile ("vmovdqa64 %0,%%zmm6" : : "m" (v[d][i]));

			asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (gfcauchy[1][d]));
			asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (gfcauchy[2][d]));
			asm volatile ("vpbroadcastb %0,%%zmm9" : : "m" (gfcauchy[3][d]));
			asm volatile ("vpbroadcastb %0,%%zmm10" : : "m" (gfcauchy[4][d]));
			asm volatile ("vpbroadcastb %0,%%zmm11" : : "m" (gfcauchy[5][d]));

			asm volatile ("vgf2p8mulb %zmm6,%zmm7,%zmm12");
			asm volatile ("vgf2p8mulb %zmm6,%zmm8,%zmm13");
			asm volatile ("vgf2p8mulb %zmm6,%zmm9,%zmm14");
			asm volatile ("vgf2p8mulb %zmm6,%zmm10,%zmm15");
			asm volatile ("vgf2p8mulb %zmm6,%zmm11,%zmm16");

			asm volatile ("vpxorq    %zmm6,%zmm0,%zmm0");
			asm volatile ("vpxorq    %zmm12,%zmm1,%zmm1");
			asm volatile ("vpxorq    %zmm13,%zmm2,%zmm2");
			asm volatile ("vpxorq    %zmm14,%zmm3,%zmm3");
			asm volatile ("vpxorq    %zmm15,%zmm4,%zmm4");
			asm volatile ("vpxorq    %zmm16,%zmm5,%zmm5");
		}

		asm volatile ("vmovntdq  %%zmm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq  %%zmm1,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq  %%zmm2,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq  %%zmm3,%0" : "=m" (s[i]));
		asm volatile ("vmovntdq  %%zmm4,%0" : "=m" (t[i]));
		asm volatile ("vmovntdq  %%zmm5,%0" : "=m" (u[i]));
	}

	raid_avx_end();
}

/*
 * RAID recovering for one disk AVX2 GFNI implementation
 */
void raid_rec1_avx2gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p, *pa;
	uint8_t G, V;
	size_t i;

	(void)nr;

	if (ip[0] == 0) {
		raid_rec1of1(id, nd, size, vv);
		return;
	}

	G = A(ip[0], id[0]);
	V = inv(G);

	raid_delta_gen(1, id, ip, nd, size, vv);

	p = v[nd + ip[0]];
	pa = v[id[0]];

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm4" : : "m" (V));

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (p[i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (p[i + 32]));

		asm volatile ("vmovdqa %0,%%ymm2" : : "m" (pa[i]));
		asm volatile ("vmovdqa %0,%%ymm3" : : "m" (pa[i + 32]));

		asm volatile ("vpxor    %ymm2,%ymm0,%ymm0");
		asm volatile ("vpxor    %ymm3,%ymm1,%ymm1");

		asm volatile ("vgf2p8mulb %ymm4,%ymm0,%ymm0");
		asm volatile ("vgf2p8mulb %ymm4,%ymm1,%ymm1");

		asm volatile ("vmovdqa %%ymm0,%0" : "=m" (pa[i]));
		asm volatile ("vmovdqa %%ymm1,%0" : "=m" (pa[i + 32]));
	}

	raid_avx_end();
}

/*
 * RAID recovering for one disk GFNI implementation
 */
void raid_rec1_avx512gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p, *pa;
	uint8_t G, V;
	size_t i;

	(void)nr;

	if (ip[0] == 0) {
		raid_rec1of1(id, nd, size, vv);
		return;
	}

	G = A(ip[0], id[0]);
	V = inv(G);

	raid_delta_gen(1, id, ip, nd, size, vv);

	p = v[nd + ip[0]];
	pa = v[id[0]];

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%zmm4" : : "m" (V));

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (p[i]));
		asm volatile ("vmovdqa64 %0,%%zmm1" : : "m" (pa[i]));
		asm volatile ("vpxorq    %zmm1,%zmm0,%zmm0");
		asm volatile ("vgf2p8mulb %zmm4,%zmm0,%zmm0");
		asm volatile ("vmovdqa64 %%zmm0,%0" : "=m" (pa[i]));
	}

	raid_avx_end();
}

/*
 * RAID recovering for two disks AVX2 GFNI implementation
 */
void raid_rec2_avx2gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	const int N = 2;
	uint8_t *p[N], *pa[N];
	uint8_t G[N * N], V[N * N];
	size_t i;
	int j, k;

	(void)nr;

	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	raid_invert(G, V, N);
	raid_delta_gen(N, id, ip, nd, size, vv);

	for (j = 0; j < N; ++j) {
		p[j] = v[nd + ip[j]];
		pa[j] = v[id[j]];
	}

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (p[0][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (p[0][i + 32]));

		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (pa[0][i]));
		asm volatile ("vmovdqa %0,%%ymm5" : : "m" (pa[0][i + 32]));

		asm volatile ("vpxor    %ymm4,%ymm0,%ymm0");
		asm volatile ("vpxor    %ymm5,%ymm1,%ymm1");

		asm volatile ("vmovdqa %0,%%ymm2" : : "m" (p[1][i]));
		asm volatile ("vmovdqa %0,%%ymm3" : : "m" (p[1][i + 32]));

		asm volatile ("vmovdqa %0,%%ymm6" : : "m" (pa[1][i]));
		asm volatile ("vmovdqa %0,%%ymm7" : : "m" (pa[1][i + 32]));

		asm volatile ("vpxor    %ymm6,%ymm2,%ymm2");
		asm volatile ("vpxor    %ymm7,%ymm3,%ymm3");

		asm volatile ("vpbroadcastb %0,%%ymm8" : : "m" (V[0]));
		asm volatile ("vgf2p8mulb %ymm0,%ymm8,%ymm4");
		asm volatile ("vgf2p8mulb %ymm1,%ymm8,%ymm5");

		asm volatile ("vpbroadcastb %0,%%ymm8" : : "m" (V[1]));
		asm volatile ("vgf2p8mulb %ymm2,%ymm8,%ymm9");
		asm volatile ("vpxor    %ymm9,%ymm4,%ymm4");
		asm volatile ("vgf2p8mulb %ymm3,%ymm8,%ymm9");
		asm volatile ("vpxor    %ymm9,%ymm5,%ymm5");

		asm volatile ("vmovdqa %%ymm4,%0" : "=m" (pa[0][i]));
		asm volatile ("vmovdqa %%ymm5,%0" : "=m" (pa[0][i + 32]));

		asm volatile ("vpbroadcastb %0,%%ymm8" : : "m" (V[2]));
		asm volatile ("vgf2p8mulb %ymm0,%ymm8,%ymm6");
		asm volatile ("vgf2p8mulb %ymm1,%ymm8,%ymm7");

		asm volatile ("vpbroadcastb %0,%%ymm8" : : "m" (V[3]));
		asm volatile ("vgf2p8mulb %ymm2,%ymm8,%ymm9");
		asm volatile ("vpxor    %ymm9,%ymm6,%ymm6");
		asm volatile ("vgf2p8mulb %ymm3,%ymm8,%ymm9");
		asm volatile ("vpxor    %ymm9,%ymm7,%ymm7");

		asm volatile ("vmovdqa %%ymm6,%0" : "=m" (pa[1][i]));
		asm volatile ("vmovdqa %%ymm7,%0" : "=m" (pa[1][i + 32]));
	}

	raid_avx_end();
}

/*
 * RAID recovering for two disks GFNI implementation
 */
void raid_rec2_avx512gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	const int N = 2;
	uint8_t *p[N], *pa[N];
	uint8_t G[N * N], V[N * N];
	size_t i;
	int j, k;

	(void)nr;

	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	raid_invert(G, V, N);
	raid_delta_gen(N, id, ip, nd, size, vv);

	for (j = 0; j < N; ++j) {
		p[j] = v[nd + ip[j]];
		pa[j] = v[id[j]];
	}

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (p[0][i]));
		asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (pa[0][i]));
		asm volatile ("vmovdqa64 %0,%%zmm1" : : "m" (p[1][i]));
		asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (pa[1][i]));
		asm volatile ("vpxorq    %zmm2,%zmm0,%zmm0");
		asm volatile ("vpxorq    %zmm3,%zmm1,%zmm1");

		asm volatile ("vpxorq    %zmm6,%zmm6,%zmm6");

		asm volatile ("vpbroadcastb %0,%%zmm2" : : "m" (V[0]));
		asm volatile ("vgf2p8mulb %zmm0,%zmm2,%zmm2");
		asm volatile ("vpxorq    %zmm2,%zmm6,%zmm6");

		asm volatile ("vpbroadcastb %0,%%zmm3" : : "m" (V[1]));
		asm volatile ("vgf2p8mulb %zmm1,%zmm3,%zmm3");
		asm volatile ("vpxorq    %zmm3,%zmm6,%zmm6");

		asm volatile ("vmovdqa64 %%zmm6,%0" : "=m" (pa[0][i]));

		asm volatile ("vpxorq    %zmm6,%zmm6,%zmm6");

		asm volatile ("vpbroadcastb %0,%%zmm2" : : "m" (V[2]));
		asm volatile ("vgf2p8mulb %zmm0,%zmm2,%zmm2");
		asm volatile ("vpxorq    %zmm2,%zmm6,%zmm6");

		asm volatile ("vpbroadcastb %0,%%zmm3" : : "m" (V[3]));
		asm volatile ("vgf2p8mulb %zmm1,%zmm3,%zmm3");
		asm volatile ("vpxorq    %zmm3,%zmm6,%zmm6");

		asm volatile ("vmovdqa64 %%zmm6,%0" : "=m" (pa[1][i]));
	}

	raid_avx_end();
}

/*
 * RAID recovering AVX2 GFNI implementation
 */
void raid_recX_avx2gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	int N = nr;
	uint8_t *p[RAID_PARITY_MAX];
	uint8_t *pa[RAID_PARITY_MAX];
	uint8_t G[RAID_PARITY_MAX * RAID_PARITY_MAX];
	uint8_t V[RAID_PARITY_MAX * RAID_PARITY_MAX];
	uint8_t buffer[RAID_PARITY_MAX*64+64];
	uint8_t *pd = __align_ptr(buffer, 64);
	size_t i;
	int j, k;

	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	raid_invert(G, V, N);
	raid_delta_gen(N, id, ip, nd, size, vv);

	for (j = 0; j < N; ++j) {
		p[j] = v[nd + ip[j]];
		pa[j] = v[id[j]];
	}

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		/* delta */
		for (j = 0; j < N; ++j) {
			asm volatile ("vmovdqa %0,%%ymm0" : : "m" (p[j][i]));
			asm volatile ("vmovdqa %0,%%ymm1" : : "m" (p[j][i + 32]));

			asm volatile ("vmovdqa %0,%%ymm2" : : "m" (pa[j][i]));
			asm volatile ("vmovdqa %0,%%ymm3" : : "m" (pa[j][i + 32]));

			asm volatile ("vpxor    %ymm2,%ymm0,%ymm0");
			asm volatile ("vpxor    %ymm3,%ymm1,%ymm1");

			asm volatile ("vmovdqa %%ymm0,%0" : "=m" (pd[j * 64]));
			asm volatile ("vmovdqa %%ymm1,%0" : "=m" (pd[j * 64 + 32]));
		}

		/* reconstruct */
		for (j = 0; j < N; ++j) {
			asm volatile ("vpxor %ymm0,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm1,%ymm1,%ymm1");
			for (k = 0; k < N; ++k) {
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (pd[k * 64]));
				asm volatile ("vmovdqa %0,%%ymm5" : : "m" (pd[k * 64 + 32]));

				asm volatile ("vpbroadcastb %0,%%ymm2" : : "m" (V[j * N + k]));

				asm volatile ("vgf2p8mulb %ymm4,%ymm2,%ymm3");
				asm volatile ("vpxor    %ymm3,%ymm0,%ymm0");

				asm volatile ("vgf2p8mulb %ymm5,%ymm2,%ymm3");
				asm volatile ("vpxor    %ymm3,%ymm1,%ymm1");
			}
			asm volatile ("vmovdqa %%ymm0,%0" : "=m" (pa[j][i]));
			asm volatile ("vmovdqa %%ymm1,%0" : "=m" (pa[j][i + 32]));
		}
	}

	raid_avx_end();
}

/*
 * RAID recovering GFNI implementation
 */
void raid_recX_avx512gfni(int nr, int *id, int *ip, int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	int N = nr;
	uint8_t *p[RAID_PARITY_MAX];
	uint8_t *pa[RAID_PARITY_MAX];
	uint8_t G[RAID_PARITY_MAX * RAID_PARITY_MAX];
	uint8_t V[RAID_PARITY_MAX * RAID_PARITY_MAX];
	uint8_t buffer[RAID_PARITY_MAX*64+64];
	uint8_t *pd = __align_ptr(buffer, 64);
	size_t i;
	int j, k;

	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	raid_invert(G, V, N);
	raid_delta_gen(N, id, ip, nd, size, vv);

	for (j = 0; j < N; ++j) {
		p[j] = v[nd + ip[j]];
		pa[j] = v[id[j]];
	}

	raid_avx_begin();

	for (i = 0; i < size; i += 64) {
		/* delta */
		for (j = 0; j < N; ++j) {
			asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (p[j][i]));
			asm volatile ("vmovdqa64 %0,%%zmm1" : : "m" (pa[j][i]));
			asm volatile ("vpxorq    %zmm1,%zmm0,%zmm0");
			asm volatile ("vmovdqa64 %%zmm0,%0" : "=m" (pd[j * 64]));
		}

		/* reconstruct */
		for (j = 0; j < N; ++j) {
			asm volatile ("vpxorq %zmm0,%zmm0,%zmm0");
			for (k = 0; k < N; ++k) {
				asm volatile ("vmovdqa64 %0,%%zmm4" : : "m" (pd[k * 64]));
				asm volatile ("vpbroadcastb %0,%%zmm2" : : "m" (V[j * N + k]));
				asm volatile ("vgf2p8mulb %zmm4,%zmm2,%zmm2");
				asm volatile ("vpxorq    %zmm2,%zmm0,%zmm0");
			}
			asm volatile ("vmovdqa64 %%zmm0,%0" : "=m" (pa[j][i]));
		}
	}

	raid_avx_end();
}

void raid_register_avx2gfni(void)
{
#ifdef USE_RAID_AES
	if (raid_cpu_has_avx2gfni()) {
		raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "gfni", raid_gen2_avx2gfni);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR3, "gfni", raid_gen3_avx2gfni);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR4, "gfni", raid_gen4_avx2gfni);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR5, "gfni", raid_gen5_avx2gfni);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR6, "gfni", raid_gen6_avx2gfni);

		raid_rec_register(RAID_ALGO_CAUCHY_PAR1, "gfni", raid_rec1_avx2gfni);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR2, "gfni", raid_rec2_avx2gfni);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR3, "gfni", raid_recX_avx2gfni);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR4, "gfni", raid_recX_avx2gfni);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR5, "gfni", raid_recX_avx2gfni);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR6, "gfni", raid_recX_avx2gfni);
	}
#endif
}

void raid_register_avx512gfni(void)
{
#ifdef USE_RAID_AES
	if (raid_cpu_has_avx512gfni()) {
		if (!raid_cpu_has_slow_avx512()) {
			raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "gfni512", raid_gen2_avx512gfni);
			raid_gen_register(RAID_ALGO_CAUCHY_PAR3, "gfni512", raid_gen3_avx512gfni);
			raid_gen_register(RAID_ALGO_CAUCHY_PAR4, "gfni512", raid_gen4_avx512gfni);
			raid_gen_register(RAID_ALGO_CAUCHY_PAR5, "gfni512", raid_gen5_avx512gfni);
			raid_gen_register(RAID_ALGO_CAUCHY_PAR6, "gfni512", raid_gen6_avx512gfni);

			raid_rec_register(RAID_ALGO_CAUCHY_PAR1, "gfni512", raid_rec1_avx512gfni);
			raid_rec_register(RAID_ALGO_CAUCHY_PAR2, "gfni512", raid_rec2_avx512gfni);
			raid_rec_register(RAID_ALGO_CAUCHY_PAR3, "gfni512", raid_recX_avx512gfni);
			raid_rec_register(RAID_ALGO_CAUCHY_PAR4, "gfni512", raid_recX_avx512gfni);
			raid_rec_register(RAID_ALGO_CAUCHY_PAR5, "gfni512", raid_recX_avx512gfni);
			raid_rec_register(RAID_ALGO_CAUCHY_PAR6, "gfni512", raid_recX_avx512gfni);
		}
	}
#endif
}

#endif /* CONFIG_X86_64 */