// 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

/*
 * GEN1 (RAID5 with xor) AVX2 implementation
 */
void raid_gen1_avx2(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];

	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]));

		for (d = 1; d <= l; ++d) {
			asm volatile ("vpxor %0,%%ymm0,%%ymm0" : : "m" (v[d][i]));
			asm volatile ("vpxor %0,%%ymm1,%%ymm1" : : "m" (v[d][i + 32]));
		}

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

	raid_avx_end();
}

/*
 * GEN2 Cauchy AVX2 implementation using only Q *= 2.
 */
static __always_inline void raid_gen2_avx2_x2(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpxor %ymm6,%ymm6,%ymm6");

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

		for (d = l - 1; d >= 0; --d) {
			/* scale Q before adding the current disk */
			asm volatile ("vpcmpgtb %ymm2,%ymm6,%ymm4");
			asm volatile ("vpcmpgtb %ymm3,%ymm6,%ymm5");
			asm volatile ("vpaddb %ymm2,%ymm2,%ymm2");
			asm volatile ("vpaddb %ymm3,%ymm3,%ymm3");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%ymm3,%ymm3");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%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 Cauchy AVX2 implementation using the AES G23 Q recurrence.
 */
static __always_inline void raid_gen2_avx2_g23(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];

	int g23_boundary = raid_g23_boundary(l - 1);

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpxor %ymm6,%ymm6,%ymm6");

	for (i = 0; i < size; i += 64) {
		int boundary = g23_boundary;

		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[l][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");

		d = l - 1;

		for (;;) {
			/*
			 * All transitions above 'boundary' are x2.
			 */
			for (; d > boundary; --d) {
				/* scale Q before adding the current disk */
				asm volatile ("vpcmpgtb %ymm2,%ymm6,%ymm4");
				asm volatile ("vpcmpgtb %ymm3,%ymm6,%ymm5");
				asm volatile ("vpaddb %ymm2,%ymm2,%ymm2");
				asm volatile ("vpaddb %ymm3,%ymm3,%ymm3");
				asm volatile ("vpand %ymm7,%ymm4,%ymm4");
				asm volatile ("vpand %ymm7,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm5,%ymm3,%ymm3");

				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
				asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
				asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
				asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm5,%ymm3,%ymm3");
			}

			/*
			 * boundary == -1 identifies the final x2-only segment.
			 * At this point all remaining disks have been consumed.
			 */
			if (boundary < 0)
				break;

			/*
			 * d == boundary.
			 *
			 * Perform the fixed G23 x3 transition and then add
			 * the current disk to P and Q.
			 */
			asm volatile ("vmovdqa %ymm2,%ymm4");
			asm volatile ("vmovdqa %ymm3,%ymm5");
			asm volatile ("vpaddb %ymm2,%ymm2,%ymm2");
			asm volatile ("vpaddb %ymm3,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%ymm3,%ymm3");
			asm volatile ("vpcmpgtb %ymm4,%ymm6,%ymm4");
			asm volatile ("vpcmpgtb %ymm5,%ymm6,%ymm5");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%ymm3,%ymm3");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%ymm3,%ymm3");

			--d;

			/*
			 * G23 boundaries are exactly 51 transitions apart.
			 * After d = 50 there are no more x3 transitions.
			 */
			boundary -= 51;
			if (boundary < 50)
				boundary = -1;
		}

		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();
}

void raid_gen2_avx2_raid(int nd, size_t size, void **vv)
{
	raid_gen2_avx2_x2(nd, size, vv);
}

void raid_gen2_avx2_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_gen2_avx2_x2(nd, size, vv);
	else
		raid_gen2_avx2_g23(nd, size, vv);
}

#ifdef CONFIG_X86_64
/*
 * GEN2 Cauchy AVX2 implementation using only Q *= 2.
 *
 * Process two data disks at a time and process the two 32-byte lanes
 * sequentially.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen2_avx2ext_x2(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];

	raid_avx_begin();

	asm volatile ("vpxor %ymm14,%ymm14,%ymm14");
	asm volatile ("vpbroadcastb %0,%%ymm15" : : "m" (gfconst16.poly[0]));

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

		/* process two disks per iteration */
		for (d = l - 1; d >= 1; d -= 2) {
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
			asm volatile ("vmovdqa %0,%%ymm6" : : "m" (v[d - 1][i]));
			asm volatile ("vmovdqa %0,%%ymm7" : : "m" (v[d - 1][i + 32]));

			/*
			 * First 32-byte lane.
			 *
			 * Compute the two consecutive x2 Horner transitions
			 * while updating P once with D[d] ^ D[d - 1].
			 */
			asm volatile ("vpcmpgtb %ymm2,%ymm14,%ymm10");
			asm volatile ("vpaddb %ymm2,%ymm2,%ymm8");
			asm volatile ("vpxor %ymm6,%ymm4,%ymm12");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpxor %ymm10,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm4,%ymm8,%ymm8");

			asm volatile ("vpcmpgtb %ymm8,%ymm14,%ymm10");
			asm volatile ("vpaddb %ymm8,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm12,%ymm0,%ymm0");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpxor %ymm10,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm6,%ymm8,%ymm2");

			/* second 32-byte lane */
			asm volatile ("vpcmpgtb %ymm3,%ymm14,%ymm11");
			asm volatile ("vpaddb %ymm3,%ymm3,%ymm9");
			asm volatile ("vpxor %ymm7,%ymm5,%ymm13");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm5,%ymm9,%ymm9");

			asm volatile ("vpcmpgtb %ymm9,%ymm14,%ymm11");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm13,%ymm1,%ymm1");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm7,%ymm9,%ymm3");
		}

		/* single remaining disk */
		if (d == 0) {
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[0][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[0][i + 32]));

			asm volatile ("vpcmpgtb %ymm2,%ymm14,%ymm10");
			asm volatile ("vpaddb %ymm2,%ymm2,%ymm8");
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpxor %ymm10,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm4,%ymm8,%ymm2");

			asm volatile ("vpcmpgtb %ymm3,%ymm14,%ymm11");
			asm volatile ("vpaddb %ymm3,%ymm3,%ymm9");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm5,%ymm9,%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 Cauchy AVX2 implementation using the AES G23 Q recurrence.
 *
 * Process two data disks at a time and process the two 32-byte lanes
 * sequentially.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen2_avx2ext_g23(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];

	int g23_boundary = raid_g23_boundary(l - 1);

	raid_avx_begin();

	asm volatile ("vpxor %ymm14,%ymm14,%ymm14");
	asm volatile ("vpbroadcastb %0,%%ymm15" : : "m" (gfconst16.poly[0]));

	for (i = 0; i < size; i += 64) {
		int boundary = g23_boundary;

		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (v[l][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm1,%ymm3");

		d = l - 1;

		for (;;) {
			/*
			 * All transitions above 'boundary' are x2.
			 * Process two disks at a time without crossing the
			 * next G23 boundary.
			 */
			for (; d >= boundary + 2; d -= 2) {
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
				asm volatile ("vmovdqa %0,%%ymm6" : : "m" (v[d - 1][i]));
				asm volatile ("vmovdqa %0,%%ymm7" : : "m" (v[d - 1][i + 32]));

				/*
				 * First 32-byte lane.
				 *
				 * Fuse the two consecutive x2 Horner steps:
				 *
				 *     Q = 2 * Q + D[d]
				 *     Q = 2 * Q + D[d - 1]
				 *
				 * while updating P with both disks.
				 */
				asm volatile ("vpcmpgtb %ymm2,%ymm14,%ymm10");
				asm volatile ("vpaddb %ymm2,%ymm2,%ymm8");
				asm volatile ("vpxor %ymm6,%ymm4,%ymm12");
				asm volatile ("vpand %ymm15,%ymm10,%ymm10");
				asm volatile ("vpxor %ymm10,%ymm8,%ymm8");
				asm volatile ("vpxor %ymm4,%ymm8,%ymm8");
				asm volatile ("vpcmpgtb %ymm8,%ymm14,%ymm10");
				asm volatile ("vpaddb %ymm8,%ymm8,%ymm8");
				asm volatile ("vpxor %ymm12,%ymm0,%ymm0");
				asm volatile ("vpand %ymm15,%ymm10,%ymm10");
				asm volatile ("vpxor %ymm10,%ymm8,%ymm8");
				asm volatile ("vpxor %ymm6,%ymm8,%ymm2");

				/* Second 32-byte lane. */
				asm volatile ("vpcmpgtb %ymm3,%ymm14,%ymm11");
				asm volatile ("vpaddb %ymm3,%ymm3,%ymm9");
				asm volatile ("vpxor %ymm7,%ymm5,%ymm13");
				asm volatile ("vpand %ymm15,%ymm11,%ymm11");
				asm volatile ("vpxor %ymm11,%ymm9,%ymm9");
				asm volatile ("vpxor %ymm5,%ymm9,%ymm9");
				asm volatile ("vpcmpgtb %ymm9,%ymm14,%ymm11");
				asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
				asm volatile ("vpxor %ymm13,%ymm1,%ymm1");
				asm volatile ("vpand %ymm15,%ymm11,%ymm11");
				asm volatile ("vpxor %ymm11,%ymm9,%ymm9");
				asm volatile ("vpxor %ymm7,%ymm9,%ymm3");
			}

			/*
			 * The first x2 segment can have odd length.
			 * Consume its last x2 transition separately.
			 */
			if (d > boundary) {
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));

				asm volatile ("vpcmpgtb %ymm2,%ymm14,%ymm10");
				asm volatile ("vpcmpgtb %ymm3,%ymm14,%ymm11");
				asm volatile ("vpaddb %ymm2,%ymm2,%ymm2");
				asm volatile ("vpaddb %ymm3,%ymm3,%ymm3");
				asm volatile ("vpand %ymm15,%ymm10,%ymm10");
				asm volatile ("vpand %ymm15,%ymm11,%ymm11");
				asm volatile ("vpxor %ymm10,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm11,%ymm3,%ymm3");

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

				--d;
			}

			/*
			 * boundary == -1 identifies the final x2-only segment.
			 * At this point all remaining disks have been consumed.
			 */
			if (boundary < 0)
				break;

			/*
			 * d == boundary.
			 *
			 * Perform the fixed G23 x3 transition and then add
			 * the current disk to P and Q.
			 */
			asm volatile ("vmovdqa %ymm2,%ymm12");
			asm volatile ("vmovdqa %ymm3,%ymm13");

			asm volatile ("vpcmpgtb %ymm2,%ymm14,%ymm10");
			asm volatile ("vpcmpgtb %ymm3,%ymm14,%ymm11");
			asm volatile ("vpaddb %ymm2,%ymm2,%ymm2");
			asm volatile ("vpaddb %ymm3,%ymm3,%ymm3");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm10,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm11,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm12,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm13,%ymm3,%ymm3");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm5" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm5,%ymm3,%ymm3");

			--d;

			/*
			 * G23 boundaries are exactly 51 transitions apart.
			 * After d = 50 there are no more x3 transitions.
			 */
			boundary -= 51;
			if (boundary < 50)
				boundary = -1;
		}

		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();
}

void raid_gen2_avx2ext_raid(int nd, size_t size, void **vv)
{
	raid_gen2_avx2ext_x2(nd, size, vv);
}

void raid_gen2_avx2ext_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_gen2_avx2ext_x2(nd, size, vv);
	else
		raid_gen2_avx2ext_g23(nd, size, vv);
}

/*
 * GENz (triple parity with powers of 2^-1) AVX2 implementation
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
void raid_genz_avx2ext_raid(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	uint8_t *r;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];
	r = v[nd + 2];

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm3" : : "m" (gfconst16.half[0]));
	asm volatile ("vpbroadcastb %0,%%ymm11" : : "m" (gfconst16.low7[0]));
	asm volatile ("vpxor %ymm15,%ymm15,%ymm15");

	for (i = 0; i < size; i += 64) {
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm8" : : "m" (v[l][i + 32]));
		asm volatile ("vmovdqa %ymm0,%ymm1");
		asm volatile ("vmovdqa %ymm8,%ymm9");
		asm volatile ("vmovdqa %ymm0,%ymm2");
		asm volatile ("vmovdqa %ymm8,%ymm10");
		for (d = l - 1; d >= 0; --d) {
			asm volatile ("vpsllw $7,%ymm2,%ymm6");
			asm volatile ("vpsllw $7,%ymm10,%ymm14");
			asm volatile ("vpsrlw $1,%ymm2,%ymm2");
			asm volatile ("vpsrlw $1,%ymm10,%ymm10");
			asm volatile ("vpcmpgtb %ymm1,%ymm15,%ymm4");
			asm volatile ("vpcmpgtb %ymm9,%ymm15,%ymm12");
			asm volatile ("vpcmpgtb %ymm6,%ymm15,%ymm5");
			asm volatile ("vpcmpgtb %ymm14,%ymm15,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm11,%ymm2,%ymm2");
			asm volatile ("vpand %ymm11,%ymm10,%ymm10");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm12,%ymm12");
			asm volatile ("vpand %ymm3,%ymm5,%ymm5");
			asm volatile ("vpand %ymm3,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm5,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm13,%ymm10,%ymm10");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm12,%ymm10,%ymm10");
		}
		asm volatile ("vmovntdq %%ymm0,%0" : "=m" (p[i]));
		asm volatile ("vmovntdq %%ymm8,%0" : "=m" (p[i + 32]));
		asm volatile ("vmovntdq %%ymm1,%0" : "=m" (q[i]));
		asm volatile ("vmovntdq %%ymm9,%0" : "=m" (q[i + 32]));
		asm volatile ("vmovntdq %%ymm2,%0" : "=m" (r[i]));
		asm volatile ("vmovntdq %%ymm10,%0" : "=m" (r[i + 32]));
	}

	raid_avx_end();
}
#endif

#ifdef CONFIG_X86_64
/*
 * GEN3 (triple parity with Cauchy matrix) AVX2 implementation
 * using only Q *= 2.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen3_avx2ext_x2(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	uint8_t *r;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];
	r = v[nd + 2];

	/* special case with only one data disk */
	if (l == 0) {
		for (i = 0; i < 3; ++i)
			memcpy(v[1 + i], v[0], size);
		return;
	}

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm3" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm11" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 64) {
		/* last disk initializes P/Q without a preceding Q transition */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[l][i + 32]));

		asm volatile ("vmovdqa %ymm4,%ymm0");
		asm volatile ("vmovdqa %ymm4,%ymm1");
		asm volatile ("vmovdqa %ymm12,%ymm8");
		asm volatile ("vmovdqa %ymm12,%ymm9");

		asm volatile ("vpsrlw $4,%ymm4,%ymm5");
		asm volatile ("vpsrlw $4,%ymm12,%ymm13");
		asm volatile ("vpand %ymm11,%ymm4,%ymm4");
		asm volatile ("vpand %ymm11,%ymm12,%ymm12");
		asm volatile ("vpand %ymm11,%ymm5,%ymm5");
		asm volatile ("vpand %ymm11,%ymm13,%ymm13");

		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][1][1][0]));

		asm volatile ("vpshufb %ymm4,%ymm14,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
		asm volatile ("vpxor %ymm7,%ymm2,%ymm2");

		asm volatile ("vpshufb %ymm12,%ymm14,%ymm10");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
		asm volatile ("vpxor %ymm7,%ymm10,%ymm10");

		/* process two intermediate disks per iteration */
		for (d = l - 1; d > 1; d -= 2) {
			/* disk d */
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));

			/* Q *= 2 */
			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm3,%ymm5,%ymm5");
			asm volatile ("vpand %ymm3,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm11,%ymm4,%ymm4");
			asm volatile ("vpand %ymm11,%ymm12,%ymm12");
			asm volatile ("vpand %ymm11,%ymm5,%ymm5");
			asm volatile ("vpand %ymm11,%ymm13,%ymm13");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d][1][1][0]));

			asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

			asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm10,%ymm10");

			/* disk d - 1 */
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d - 1][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d - 1][i + 32]));

			/* Q *= 2 */
			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm3,%ymm5,%ymm5");
			asm volatile ("vpand %ymm3,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm11,%ymm4,%ymm4");
			asm volatile ("vpand %ymm11,%ymm12,%ymm12");
			asm volatile ("vpand %ymm11,%ymm5,%ymm5");
			asm volatile ("vpand %ymm11,%ymm13,%ymm13");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d - 1][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d - 1][1][1][0]));

			asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

			asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm10,%ymm10");
		}

		/* single remaining intermediate disk */
		if (d == 1) {
			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[1][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[1][i + 32]));

			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm3,%ymm5,%ymm5");
			asm volatile ("vpand %ymm3,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm11,%ymm4,%ymm4");
			asm volatile ("vpand %ymm11,%ymm12,%ymm12");
			asm volatile ("vpand %ymm11,%ymm5,%ymm5");
			asm volatile ("vpand %ymm11,%ymm13,%ymm13");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[1][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[1][1][1][0]));

			asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

			asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm10,%ymm10");
		}

		/* first disk with all coefficients at 1 */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[0][i + 32]));

		asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
		asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
		asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
		asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
		asm volatile ("vpand %ymm3,%ymm5,%ymm5");
		asm volatile ("vpand %ymm3,%ymm13,%ymm13");
		asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

		asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
		asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
		asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
		asm volatile ("vpxor %ymm12,%ymm10,%ymm10");

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

	raid_avx_end();
}

/*
 * GEN3 (triple parity with Cauchy matrix) AVX2 implementation
 * using the AES G23 Q recurrence.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen3_avx2ext_g23(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	uint8_t *r;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];
	r = v[nd + 2];

	/* special case with only one data disk */
	if (l == 0) {
		for (i = 0; i < 3; ++i)
			memcpy(v[1 + i], v[0], size);
		return;
	}

	int g23_boundary = raid_g23_boundary(l - 1);

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm3" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm11" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 64) {
		int boundary = g23_boundary;

		/* last disk initializes P/Q without a preceding Q transition */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[l][i + 32]));

		asm volatile ("vmovdqa %ymm4,%ymm0");
		asm volatile ("vmovdqa %ymm4,%ymm1");
		asm volatile ("vmovdqa %ymm12,%ymm8");
		asm volatile ("vmovdqa %ymm12,%ymm9");

		asm volatile ("vpsrlw $4,%ymm4,%ymm5");
		asm volatile ("vpsrlw $4,%ymm12,%ymm13");
		asm volatile ("vpand %ymm11,%ymm4,%ymm4");
		asm volatile ("vpand %ymm11,%ymm12,%ymm12");
		asm volatile ("vpand %ymm11,%ymm5,%ymm5");
		asm volatile ("vpand %ymm11,%ymm13,%ymm13");

		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][1][1][0]));

		asm volatile ("vpshufb %ymm4,%ymm14,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
		asm volatile ("vpxor %ymm7,%ymm2,%ymm2");

		asm volatile ("vpshufb %ymm12,%ymm14,%ymm10");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
		asm volatile ("vpxor %ymm7,%ymm10,%ymm10");

		d = l - 1;

		for (;;) {
			/*
			 * Process pairs entirely above the next G23 boundary.
			 * Disk 0 remains outside this loop.
			 */
			for (; d >= 2 && d >= boundary + 2; d -= 2) {
				/* disk d */
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));

				asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
				asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
				asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
				asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
				asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
				asm volatile ("vpand %ymm3,%ymm5,%ymm5");
				asm volatile ("vpand %ymm3,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
				asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

				asm volatile ("vpsrlw $4,%ymm4,%ymm5");
				asm volatile ("vpsrlw $4,%ymm12,%ymm13");
				asm volatile ("vpand %ymm11,%ymm4,%ymm4");
				asm volatile ("vpand %ymm11,%ymm12,%ymm12");
				asm volatile ("vpand %ymm11,%ymm5,%ymm5");
				asm volatile ("vpand %ymm11,%ymm13,%ymm13");

				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d][1][1][0]));

				asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

				asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm10,%ymm10");

				/* disk d - 1 */
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d - 1][i]));
				asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d - 1][i + 32]));

				asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
				asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
				asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
				asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
				asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
				asm volatile ("vpand %ymm3,%ymm5,%ymm5");
				asm volatile ("vpand %ymm3,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
				asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

				asm volatile ("vpsrlw $4,%ymm4,%ymm5");
				asm volatile ("vpsrlw $4,%ymm12,%ymm13");
				asm volatile ("vpand %ymm11,%ymm4,%ymm4");
				asm volatile ("vpand %ymm11,%ymm12,%ymm12");
				asm volatile ("vpand %ymm11,%ymm5,%ymm5");
				asm volatile ("vpand %ymm11,%ymm13,%ymm13");

				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d - 1][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d - 1][1][1][0]));

				asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

				asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm10,%ymm10");
			}

			/*
			 * The x2 segment may contain one remaining intermediate disk.
			 */
			if (d > boundary && d > 0) {
				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));

				asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
				asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
				asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
				asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
				asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
				asm volatile ("vpand %ymm3,%ymm5,%ymm5");
				asm volatile ("vpand %ymm3,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
				asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

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

				asm volatile ("vpsrlw $4,%ymm4,%ymm5");
				asm volatile ("vpsrlw $4,%ymm12,%ymm13");
				asm volatile ("vpand %ymm11,%ymm4,%ymm4");
				asm volatile ("vpand %ymm11,%ymm12,%ymm12");
				asm volatile ("vpand %ymm11,%ymm5,%ymm5");
				asm volatile ("vpand %ymm11,%ymm13,%ymm13");

				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d][1][1][0]));

				asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

				asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
				asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
				asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
				asm volatile ("vpxor %ymm6,%ymm10,%ymm10");

				--d;
			}

			if (boundary < 0)
				break;

			/*
			 * d == boundary: Q *= 3.
			 */
			asm volatile ("vmovdqa %ymm1,%ymm6");
			asm volatile ("vmovdqa %ymm9,%ymm7");

			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm3,%ymm5,%ymm5");
			asm volatile ("vpand %ymm3,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm6,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm7,%ymm9,%ymm9");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));

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

			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm11,%ymm4,%ymm4");
			asm volatile ("vpand %ymm11,%ymm12,%ymm12");
			asm volatile ("vpand %ymm11,%ymm5,%ymm5");
			asm volatile ("vpand %ymm11,%ymm13,%ymm13");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[d][1][1][0]));

			asm volatile ("vpshufb %ymm4,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm5,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm2,%ymm2");

			asm volatile ("vpshufb %ymm12,%ymm14,%ymm6");
			asm volatile ("vpshufb %ymm13,%ymm15,%ymm7");
			asm volatile ("vpxor %ymm7,%ymm6,%ymm6");
			asm volatile ("vpxor %ymm6,%ymm10,%ymm10");

			--d;

			boundary -= 51;
			if (boundary < 50)
				boundary = -1;
		}

		/*
		 * D0 always follows an x2 transition.
		 */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[0][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[0][i + 32]));

		asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
		asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
		asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
		asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
		asm volatile ("vpand %ymm3,%ymm5,%ymm5");
		asm volatile ("vpand %ymm3,%ymm13,%ymm13");
		asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

		asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
		asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
		asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
		asm volatile ("vpxor %ymm12,%ymm10,%ymm10");

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

	raid_avx_end();
}

void raid_gen3_avx2ext_raid(int nd, size_t size, void **vv)
{
	raid_gen3_avx2ext_x2(nd, size, vv);
}

void raid_gen3_avx2ext_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_gen3_avx2ext_x2(nd, size, vv);
	else
		raid_gen3_avx2ext_g23(nd, size, vv);
}
#endif

#ifdef CONFIG_X86_64
/*
 * GEN4 (quad parity with Cauchy matrix) AVX2 implementation
 * using only Q *= 2.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen4_avx2ext_x2(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	uint8_t *r;
	uint8_t *s;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];
	r = v[nd + 2];
	s = v[nd + 3];

	/* special case with only one data disk */
	if (l == 0) {
		for (i = 0; i < 4; ++i)
			memcpy(v[1 + i], v[0], size);
		return;
	}

	raid_avx_begin();

	/* keep the constants resident for the whole function */
	asm volatile ("vpbroadcastb %0,%%ymm6" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 64) {
		/* last disk initializes P/Q without a preceding Q transition */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[l][i + 32]));

		asm volatile ("vmovdqa %ymm4,%ymm0");
		asm volatile ("vmovdqa %ymm4,%ymm1");
		asm volatile ("vmovdqa %ymm12,%ymm8");
		asm volatile ("vmovdqa %ymm12,%ymm9");

		asm volatile ("vpsrlw $4,%ymm4,%ymm5");
		asm volatile ("vpand %ymm7,%ymm4,%ymm4");
		asm volatile ("vpand %ymm7,%ymm5,%ymm5");
		asm volatile ("vpsrlw $4,%ymm12,%ymm13");
		asm volatile ("vpand %ymm7,%ymm12,%ymm12");
		asm volatile ("vpand %ymm7,%ymm13,%ymm13");

		/* R */
		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][1][1][0]));
		asm volatile ("vpshufb %ymm4,%ymm14,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm3");
		asm volatile ("vpxor %ymm3,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm12,%ymm14,%ymm10");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm11");
		asm volatile ("vpxor %ymm11,%ymm10,%ymm10");

		/* S */
		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][2][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][2][1][0]));
		asm volatile ("vpshufb %ymm4,%ymm14,%ymm3");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm3,%ymm3");
		asm volatile ("vpshufb %ymm12,%ymm14,%ymm11");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm13");
		asm volatile ("vpxor %ymm13,%ymm11,%ymm11");

		/* intermediate disks */
		for (d = l - 1; d > 0; --d) {
			/* Q *= 2, first lane */
			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpand %ymm6,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm5,%ymm5");

			/* Q *= 2, second lane */
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm6,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm7,%ymm12,%ymm12");
			asm volatile ("vpand %ymm7,%ymm13,%ymm13");

			/* R */
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");

			/* S */
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
		}

		/* first disk with all coefficients at 1 */
		asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
		asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpand %ymm6,%ymm5,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm1,%ymm1");

		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[0][i]));
		asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
		asm volatile ("vpxor %ymm4,%ymm3,%ymm3");

		asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
		asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
		asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
		asm volatile ("vpand %ymm6,%ymm13,%ymm13");
		asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[0][i + 32]));
		asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
		asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
		asm volatile ("vpxor %ymm12,%ymm10,%ymm10");
		asm volatile ("vpxor %ymm12,%ymm11,%ymm11");

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

	raid_avx_end();
}

/*
 * GEN4 (quad parity with Cauchy matrix) AVX2 implementation
 * using the AES G23 Q recurrence.
 *
 * Note that it uses 16 registers, meaning that x64 is required.
 */
static __always_inline void raid_gen4_avx2ext_g23(int nd, size_t size, void **vv)
{
	uint8_t **v = (uint8_t **)vv;
	uint8_t *p;
	uint8_t *q;
	uint8_t *r;
	uint8_t *s;
	int d, l;
	size_t i;

	l = nd - 1;
	p = v[nd];
	q = v[nd + 1];
	r = v[nd + 2];
	s = v[nd + 3];

	if (l == 0) {
		for (i = 0; i < 4; ++i)
			memcpy(v[1 + i], v[0], size);
		return;
	}

	int g23_boundary = raid_g23_boundary(l - 1);

	raid_avx_begin();

	asm volatile ("vpbroadcastb %0,%%ymm6" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 64) {
		int boundary = g23_boundary;

		/* last disk */
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[l][i]));
		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[l][i + 32]));

		asm volatile ("vmovdqa %ymm4,%ymm0");
		asm volatile ("vmovdqa %ymm4,%ymm1");
		asm volatile ("vmovdqa %ymm12,%ymm8");
		asm volatile ("vmovdqa %ymm12,%ymm9");

		asm volatile ("vpsrlw $4,%ymm4,%ymm5");
		asm volatile ("vpand %ymm7,%ymm4,%ymm4");
		asm volatile ("vpand %ymm7,%ymm5,%ymm5");
		asm volatile ("vpsrlw $4,%ymm12,%ymm13");
		asm volatile ("vpand %ymm7,%ymm12,%ymm12");
		asm volatile ("vpand %ymm7,%ymm13,%ymm13");

		/* R */
		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][1][1][0]));
		asm volatile ("vpshufb %ymm4,%ymm14,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm3");
		asm volatile ("vpxor %ymm3,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm12,%ymm14,%ymm10");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm11");
		asm volatile ("vpxor %ymm11,%ymm10,%ymm10");

		/* S */
		asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[l][2][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfcauchypshufb[l][2][1][0]));
		asm volatile ("vpshufb %ymm4,%ymm14,%ymm3");
		asm volatile ("vpshufb %ymm5,%ymm15,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm3,%ymm3");
		asm volatile ("vpshufb %ymm12,%ymm14,%ymm11");
		asm volatile ("vpshufb %ymm13,%ymm15,%ymm13");
		asm volatile ("vpxor %ymm13,%ymm11,%ymm11");

		d = l - 1;

		while (boundary >= 0) {
			/* all transitions above boundary are x2 */
			for (; d > boundary; --d) {
				asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
				asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
				asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
				asm volatile ("vpand %ymm6,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm5,%ymm1,%ymm1");

				asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
				asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
				asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
				asm volatile ("vpsrlw $4,%ymm4,%ymm5");
				asm volatile ("vpand %ymm7,%ymm4,%ymm4");
				asm volatile ("vpand %ymm7,%ymm5,%ymm5");

				asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
				asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
				asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
				asm volatile ("vpand %ymm6,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

				asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));
				asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
				asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
				asm volatile ("vpsrlw $4,%ymm12,%ymm13");
				asm volatile ("vpand %ymm7,%ymm12,%ymm12");
				asm volatile ("vpand %ymm7,%ymm13,%ymm13");

				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
				asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
				asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm14,%ymm10,%ymm10");
				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
				asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
				asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
				asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm14,%ymm10,%ymm10");

				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
				asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
				asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
				asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
				asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
				asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
				asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
				asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
				asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
				asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
			}

			/*
			 * d == boundary: Q *= 3.
			 */
			asm volatile ("vmovdqa %ymm1,%ymm4");
			asm volatile ("vmovdqa %ymm9,%ymm12");

			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm6,%ymm5,%ymm5");
			asm volatile ("vpand %ymm6,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");
			asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
			asm volatile ("vpxor %ymm12,%ymm9,%ymm9");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));

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

			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm5,%ymm5");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm7,%ymm12,%ymm12");
			asm volatile ("vpand %ymm7,%ymm13,%ymm13");

			/* R */
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");

			/* S */
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");

			--d;
			boundary -= 51;
			if (boundary < 50)
				boundary = -1;
		}

		/*
		 * No G23 boundaries remain. Process intermediate disks
		 * down to D1 using only Q *= 2.
		 */
		for (; d > 0; --d) {
			asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
			asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpand %ymm6,%ymm5,%ymm5");
			asm volatile ("vpxor %ymm5,%ymm1,%ymm1");

			asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[d][i]));
			asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
			asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
			asm volatile ("vpsrlw $4,%ymm4,%ymm5");
			asm volatile ("vpand %ymm7,%ymm4,%ymm4");
			asm volatile ("vpand %ymm7,%ymm5,%ymm5");

			asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
			asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
			asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
			asm volatile ("vpand %ymm6,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

			asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[d][i + 32]));
			asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
			asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
			asm volatile ("vpsrlw $4,%ymm12,%ymm13");
			asm volatile ("vpand %ymm7,%ymm12,%ymm12");
			asm volatile ("vpand %ymm7,%ymm13,%ymm13");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm2,%ymm2");
			asm volatile ("vpxor %ymm14,%ymm10,%ymm10");

			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
			asm volatile ("vpshufb %ymm4,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm12,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
			asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
			asm volatile ("vpshufb %ymm5,%ymm14,%ymm15");
			asm volatile ("vpshufb %ymm13,%ymm14,%ymm14");
			asm volatile ("vpxor %ymm15,%ymm3,%ymm3");
			asm volatile ("vpxor %ymm14,%ymm11,%ymm11");
		}

		/*
		 * D0 always follows an x2 transition.
		 */
		asm volatile ("vpxor %ymm5,%ymm5,%ymm5");
		asm volatile ("vpcmpgtb %ymm1,%ymm5,%ymm5");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpand %ymm6,%ymm5,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm1,%ymm1");

		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (v[0][i]));
		asm volatile ("vpxor %ymm4,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm4,%ymm1,%ymm1");
		asm volatile ("vpxor %ymm4,%ymm2,%ymm2");
		asm volatile ("vpxor %ymm4,%ymm3,%ymm3");

		asm volatile ("vpxor %ymm13,%ymm13,%ymm13");
		asm volatile ("vpcmpgtb %ymm9,%ymm13,%ymm13");
		asm volatile ("vpaddb %ymm9,%ymm9,%ymm9");
		asm volatile ("vpand %ymm6,%ymm13,%ymm13");
		asm volatile ("vpxor %ymm13,%ymm9,%ymm9");

		asm volatile ("vmovdqa %0,%%ymm12" : : "m" (v[0][i + 32]));
		asm volatile ("vpxor %ymm12,%ymm8,%ymm8");
		asm volatile ("vpxor %ymm12,%ymm9,%ymm9");
		asm volatile ("vpxor %ymm12,%ymm10,%ymm10");
		asm volatile ("vpxor %ymm12,%ymm11,%ymm11");

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

	raid_avx_end();
}

void raid_gen4_avx2ext_raid(int nd, size_t size, void **vv)
{
	raid_gen4_avx2ext_x2(nd, size, vv);
}

void raid_gen4_avx2ext_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_gen4_avx2ext_x2(nd, size, vv);
	else
		raid_gen4_avx2ext_g23(nd, size, vv);
}
#endif

#ifdef CONFIG_X86_64
/*
 * GENX AVX2EXT implementation using only Q *= 2.
 */
static __always_inline void raid_genX_avx2ext_x2(int nd, size_t size,
	void **vv, int np)
{
	uint8_t **v = (uint8_t **)vv;
	size_t i;
	int d, l;

	l = nd - 1;

	/* special case with only one data disk */
	if (l == 0) {
		for (d = 0; d < np; ++d)
			memcpy(v[1 + d], v[0], size);
		return;
	}

	raid_avx_begin();

	asm volatile ("vpxor %ymm6,%ymm6,%ymm6");
	asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm15" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 32) {
		/* last disk initializes P/Q without a preceding Q transition */
		asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[l][i]));

		asm volatile ("vmovdqa %ymm10,%ymm0");
		asm volatile ("vmovdqa %ymm10,%ymm1");

		asm volatile ("vpsrlw $4,%ymm10,%ymm11");
		asm volatile ("vpand %ymm15,%ymm10,%ymm10");
		asm volatile ("vpand %ymm15,%ymm11,%ymm11");

		if (np >= 3) {
			asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][1][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm2,%ymm2");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
		}

		if (np >= 4) {
			asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfcauchypshufb[l][2][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][2][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm3,%ymm3");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
		}

		if (np >= 5) {
			asm volatile ("vbroadcasti128 %0,%%ymm4" : : "m" (raid_gfcauchypshufb[l][3][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][3][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm4,%ymm4");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
		}

		if (np >= 6) {
			asm volatile ("vbroadcasti128 %0,%%ymm5" : : "m" (raid_gfcauchypshufb[l][4][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][4][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm5,%ymm5");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
		}

		/* intermediate disks */
		for (d = l - 1; d > 0; --d) {
			asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[d][i]));

			/* Q *= 2 */
			asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpand %ymm14,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

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

			asm volatile ("vpsrlw $4,%ymm10,%ymm11");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");

			if (np >= 3) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
			}

			if (np >= 4) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm3,%ymm3");
				asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
			}

			if (np >= 5) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][3][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][3][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm4,%ymm4");
				asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
			}

			if (np >= 6) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][4][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][4][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
			}
		}

		/* first disk with all coefficients at 1 */
		asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[0][i]));

		/* final Q *= 2 */
		asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpand %ymm14,%ymm11,%ymm11");
		asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

		asm volatile ("vpxor %ymm10,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm10,%ymm1,%ymm1");
		if (np >= 3)
			asm volatile ("vpxor %ymm10,%ymm2,%ymm2");
		if (np >= 4)
			asm volatile ("vpxor %ymm10,%ymm3,%ymm3");
		if (np >= 5)
			asm volatile ("vpxor %ymm10,%ymm4,%ymm4");
		if (np >= 6)
			asm volatile ("vpxor %ymm10,%ymm5,%ymm5");

		asm volatile ("vmovntdq %%ymm0,%0" : "=m" (v[nd][i]));
		asm volatile ("vmovntdq %%ymm1,%0" : "=m" (v[nd + 1][i]));
		if (np >= 3)
			asm volatile ("vmovntdq %%ymm2,%0" : "=m" (v[nd + 2][i]));
		if (np >= 4)
			asm volatile ("vmovntdq %%ymm3,%0" : "=m" (v[nd + 3][i]));
		if (np >= 5)
			asm volatile ("vmovntdq %%ymm4,%0" : "=m" (v[nd + 4][i]));
		if (np >= 6)
			asm volatile ("vmovntdq %%ymm5,%0" : "=m" (v[nd + 5][i]));
	}

	raid_avx_end();
}

/*
 * GENX AVX2EXT implementation using the AES G23 Q recurrence.
 */
static __always_inline void raid_genX_avx2ext_g23(int nd, size_t size,
	void **vv, int np)
{
	uint8_t **v = (uint8_t **)vv;
	size_t i;
	int d, l;

	l = nd - 1;

	if (l == 0) {
		for (d = 0; d < np; ++d)
			memcpy(v[1 + d], v[0], size);
		return;
	}

	int g23_boundary = raid_g23_boundary(l - 1);

	raid_avx_begin();

	asm volatile ("vpxor %ymm6,%ymm6,%ymm6");
	asm volatile ("vpbroadcastb %0,%%ymm14" : : "m" (gfconst16.poly[0]));
	asm volatile ("vpbroadcastb %0,%%ymm15" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 32) {
		int boundary = g23_boundary;

		/* last disk */
		asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[l][i]));

		asm volatile ("vmovdqa %ymm10,%ymm0");
		asm volatile ("vmovdqa %ymm10,%ymm1");

		asm volatile ("vpsrlw $4,%ymm10,%ymm11");
		asm volatile ("vpand %ymm15,%ymm10,%ymm10");
		asm volatile ("vpand %ymm15,%ymm11,%ymm11");

		if (np >= 3) {
			asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfcauchypshufb[l][1][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][1][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm2,%ymm2");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
		}

		if (np >= 4) {
			asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfcauchypshufb[l][2][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][2][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm3,%ymm3");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
		}

		if (np >= 5) {
			asm volatile ("vbroadcasti128 %0,%%ymm4" : : "m" (raid_gfcauchypshufb[l][3][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][3][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm4,%ymm4");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
		}

		if (np >= 6) {
			asm volatile ("vbroadcasti128 %0,%%ymm5" : : "m" (raid_gfcauchypshufb[l][4][0][0]));
			asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[l][4][1][0]));
			asm volatile ("vpshufb %ymm10,%ymm5,%ymm5");
			asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
			asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
		}

		d = l - 1;

		while (boundary >= 0) {
			/* x2-only part of this segment */
			for (; d > boundary; --d) {
				asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[d][i]));

				asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
				asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
				asm volatile ("vpand %ymm14,%ymm11,%ymm11");
				asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

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

				asm volatile ("vpsrlw $4,%ymm10,%ymm11");
				asm volatile ("vpand %ymm15,%ymm10,%ymm10");
				asm volatile ("vpand %ymm15,%ymm11,%ymm11");

				if (np >= 3) {
					asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
					asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
					asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
					asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
					asm volatile ("vpxor %ymm12,%ymm2,%ymm2");
					asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
				}

				if (np >= 4) {
					asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
					asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
					asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
					asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
					asm volatile ("vpxor %ymm12,%ymm3,%ymm3");
					asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
				}

				if (np >= 5) {
					asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][3][0][0]));
					asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][3][1][0]));
					asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
					asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
					asm volatile ("vpxor %ymm12,%ymm4,%ymm4");
					asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
				}

				if (np >= 6) {
					asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][4][0][0]));
					asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][4][1][0]));
					asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
					asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
					asm volatile ("vpxor %ymm12,%ymm5,%ymm5");
					asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
				}
			}

			/*
			 * d == boundary: Q *= 3.
			 *
			 * ymm12 is free here and can temporarily preserve
			 * the old Q value.
			 */
			asm volatile ("vmovdqa %ymm1,%ymm12");

			/* Q *= 2 */
			asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpand %ymm14,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

			/* Q *= 3 */
			asm volatile ("vpxor %ymm12,%ymm1,%ymm1");

			asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[d][i]));

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

			asm volatile ("vpsrlw $4,%ymm10,%ymm11");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");

			if (np >= 3) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
			}

			if (np >= 4) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm3,%ymm3");
				asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
			}

			if (np >= 5) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][3][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][3][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm4,%ymm4");
				asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
			}

			if (np >= 6) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][4][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][4][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
			}

			--d;

			boundary -= 51;
			if (boundary < 50)
				boundary = -1;
		}

		/* final x2-only segment */
		for (; d > 0; --d) {
			asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[d][i]));

			asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
			asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
			asm volatile ("vpand %ymm14,%ymm11,%ymm11");
			asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

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

			asm volatile ("vpsrlw $4,%ymm10,%ymm11");
			asm volatile ("vpand %ymm15,%ymm10,%ymm10");
			asm volatile ("vpand %ymm15,%ymm11,%ymm11");

			if (np >= 3) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][1][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][1][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm2,%ymm2");
				asm volatile ("vpxor %ymm13,%ymm2,%ymm2");
			}

			if (np >= 4) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][2][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][2][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm3,%ymm3");
				asm volatile ("vpxor %ymm13,%ymm3,%ymm3");
			}

			if (np >= 5) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][3][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][3][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm4,%ymm4");
				asm volatile ("vpxor %ymm13,%ymm4,%ymm4");
			}

			if (np >= 6) {
				asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfcauchypshufb[d][4][0][0]));
				asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfcauchypshufb[d][4][1][0]));
				asm volatile ("vpshufb %ymm10,%ymm12,%ymm12");
				asm volatile ("vpshufb %ymm11,%ymm13,%ymm13");
				asm volatile ("vpxor %ymm12,%ymm5,%ymm5");
				asm volatile ("vpxor %ymm13,%ymm5,%ymm5");
			}
		}

		/*
		 * D0 always follows Q *= 2.
		 */
		asm volatile ("vmovdqa %0,%%ymm10" : : "m" (v[0][i]));

		asm volatile ("vpcmpgtb %ymm1,%ymm6,%ymm11");
		asm volatile ("vpaddb %ymm1,%ymm1,%ymm1");
		asm volatile ("vpand %ymm14,%ymm11,%ymm11");
		asm volatile ("vpxor %ymm11,%ymm1,%ymm1");

		asm volatile ("vpxor %ymm10,%ymm0,%ymm0");
		asm volatile ("vpxor %ymm10,%ymm1,%ymm1");
		if (np >= 3)
			asm volatile ("vpxor %ymm10,%ymm2,%ymm2");
		if (np >= 4)
			asm volatile ("vpxor %ymm10,%ymm3,%ymm3");
		if (np >= 5)
			asm volatile ("vpxor %ymm10,%ymm4,%ymm4");
		if (np >= 6)
			asm volatile ("vpxor %ymm10,%ymm5,%ymm5");

		asm volatile ("vmovntdq %%ymm0,%0" : "=m" (v[nd][i]));
		asm volatile ("vmovntdq %%ymm1,%0" : "=m" (v[nd + 1][i]));
		if (np >= 3)
			asm volatile ("vmovntdq %%ymm2,%0" : "=m" (v[nd + 2][i]));
		if (np >= 4)
			asm volatile ("vmovntdq %%ymm3,%0" : "=m" (v[nd + 3][i]));
		if (np >= 5)
			asm volatile ("vmovntdq %%ymm4,%0" : "=m" (v[nd + 4][i]));
		if (np >= 6)
			asm volatile ("vmovntdq %%ymm5,%0" : "=m" (v[nd + 5][i]));
	}

	raid_avx_end();
}
#endif

#ifdef CONFIG_X86_64
void raid_gen5_avx2ext_raid(int nd, size_t size, void **vv)
{
	raid_genX_avx2ext_x2(nd, size, vv, 5);
}

void raid_gen5_avx2ext_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_genX_avx2ext_x2(nd, size, vv, 5);
	else
		raid_genX_avx2ext_g23(nd, size, vv, 5);
}
#endif

#ifdef CONFIG_X86_64
void raid_gen6_avx2ext_raid(int nd, size_t size, void **vv)
{
	raid_genX_avx2ext_x2(nd, size, vv, 6);
}

void raid_gen6_avx2ext_aes(int nd, size_t size, void **vv)
{
	if (raid_g23_x2_only(nd))
		raid_genX_avx2ext_x2(nd, size, vv, 6);
	else
		raid_genX_avx2ext_g23(nd, size, vv, 6);
}
#endif

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

	(void)nr; /* unused, it's always 1 */

	/* if it's RAID5 uses the faster function */
	if (ip[0] == 0) {
		raid_rec1of1(id, nd, size, vv);
		return;
	}

	/* setup the coefficients matrix */
	G = A(ip[0], id[0]);

	/* invert it to solve the system of linear equations */
	V = inv(G);

	/* compute delta parity */
	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,%%ymm7" : : "m" (gfconst16.low4[0]));
	asm volatile ("vbroadcasti128 %0,%%ymm4" : : "m" (raid_gfmulpshufb[V][0][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm5" : : "m" (raid_gfmulpshufb[V][1][0]));

	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 ("vpsrlw  $4,%ymm0,%ymm2");
		asm volatile ("vpsrlw  $4,%ymm1,%ymm3");

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

		asm volatile ("vpshufb %ymm0,%ymm4,%ymm0");
		asm volatile ("vpshufb %ymm1,%ymm4,%ymm1");
		asm volatile ("vpshufb %ymm2,%ymm5,%ymm2");
		asm volatile ("vpshufb %ymm3,%ymm5,%ymm3");

		asm volatile ("vpxor   %ymm2,%ymm0,%ymm0");
		asm volatile ("vpxor   %ymm3,%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 two disks AVX2 implementation
 */
void raid_rec2_avx2(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];
	uint8_t *pa[N];
	uint8_t G[N * N];
	uint8_t V[N * N];
	size_t i;
	int j, k;

	(void)nr; /* unused, it's always 2 */

	/* setup the coefficients matrix */
	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	/* invert it to solve the system of linear equations */
	raid_invert(G, V, N);

	/* compute delta parity */
	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();

#ifdef CONFIG_X86_64
	asm volatile ("vpbroadcastb %0,%%ymm6" : : "m" (gfconst16.low4[0]));

	/* the inverse matrix V[] is constant for the whole recovery. */
	asm volatile ("vbroadcasti128 %0,%%ymm8" : : "m" (raid_gfmulpshufb[V[0]][0][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm9" : : "m" (raid_gfmulpshufb[V[0]][1][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm10" : : "m" (raid_gfmulpshufb[V[1]][0][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm11" : : "m" (raid_gfmulpshufb[V[1]][1][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm12" : : "m" (raid_gfmulpshufb[V[2]][0][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm13" : : "m" (raid_gfmulpshufb[V[2]][1][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm14" : : "m" (raid_gfmulpshufb[V[3]][0][0]));
	asm volatile ("vbroadcasti128 %0,%%ymm15" : : "m" (raid_gfmulpshufb[V[3]][1][0]));

	for (i = 0; i < size; i += 32) {
		/* d0 = p[0] ^ pa[0] */
		asm volatile ("vmovdqa %0,%%ymm0" : : "m" (p[0][i]));
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (pa[0][i]));
		asm volatile ("vpxor %ymm4,%ymm0,%ymm0");

		/* d1 = p[1] ^ pa[1] */
		asm volatile ("vmovdqa %0,%%ymm1" : : "m" (p[1][i]));
		asm volatile ("vmovdqa %0,%%ymm4" : : "m" (pa[1][i]));
		asm volatile ("vpxor %ymm4,%ymm1,%ymm1");

		/*
		 * Split both deltas into low/high nibbles once.
		 *
		 * ymm0 = d0 low
		 * ymm2 = d0 high
		 * ymm1 = d1 low
		 * ymm3 = d1 high
		 */
		asm volatile ("vpsrlw $4,%ymm0,%ymm2");
		asm volatile ("vpsrlw $4,%ymm1,%ymm3");
		asm volatile ("vpand %ymm6,%ymm0,%ymm0");
		asm volatile ("vpand %ymm6,%ymm2,%ymm2");
		asm volatile ("vpand %ymm6,%ymm1,%ymm1");
		asm volatile ("vpand %ymm6,%ymm3,%ymm3");

		/*
		 * pa[0] = V[0] * d0 ^ V[1] * d1
		 */
		asm volatile ("vpshufb %ymm0,%ymm8,%ymm4");
		asm volatile ("vpshufb %ymm2,%ymm9,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");

		asm volatile ("vpshufb %ymm1,%ymm10,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");
		asm volatile ("vpshufb %ymm3,%ymm11,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");

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

		/*
		 * pa[1] = V[2] * d0 ^ V[3] * d1
		 */
		asm volatile ("vpshufb %ymm0,%ymm12,%ymm4");
		asm volatile ("vpshufb %ymm2,%ymm13,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");

		asm volatile ("vpshufb %ymm1,%ymm14,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");
		asm volatile ("vpshufb %ymm3,%ymm15,%ymm5");
		asm volatile ("vpxor %ymm5,%ymm4,%ymm4");

		asm volatile ("vmovdqa %%ymm4,%0" : "=m" (pa[1][i]));
	}
#else /* CONFIG_X86_32 */
	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.low4[0]));

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

		asm volatile ("vpxor %ymm6,%ymm6,%ymm6");

		asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfmulpshufb[V[0]][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfmulpshufb[V[0]][1][0]));
		asm volatile ("vpsrlw  $4,%ymm0,%ymm5");
		asm volatile ("vpand   %ymm7,%ymm0,%ymm4");
		asm volatile ("vpand   %ymm7,%ymm5,%ymm5");
		asm volatile ("vpshufb %ymm4,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm3,%ymm3");
		asm volatile ("vpxor   %ymm2,%ymm6,%ymm6");
		asm volatile ("vpxor   %ymm3,%ymm6,%ymm6");

		asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfmulpshufb[V[1]][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfmulpshufb[V[1]][1][0]));
		asm volatile ("vpsrlw  $4,%ymm1,%ymm5");
		asm volatile ("vpand   %ymm7,%ymm1,%ymm4");
		asm volatile ("vpand   %ymm7,%ymm5,%ymm5");
		asm volatile ("vpshufb %ymm4,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm3,%ymm3");
		asm volatile ("vpxor   %ymm2,%ymm6,%ymm6");
		asm volatile ("vpxor   %ymm3,%ymm6,%ymm6");

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

		asm volatile ("vpxor %ymm6,%ymm6,%ymm6");

		asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfmulpshufb[V[2]][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfmulpshufb[V[2]][1][0]));
		asm volatile ("vpsrlw  $4,%ymm0,%ymm5");
		asm volatile ("vpand   %ymm7,%ymm0,%ymm4");
		asm volatile ("vpand   %ymm7,%ymm5,%ymm5");
		asm volatile ("vpshufb %ymm4,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm3,%ymm3");
		asm volatile ("vpxor   %ymm2,%ymm6,%ymm6");
		asm volatile ("vpxor   %ymm3,%ymm6,%ymm6");

		asm volatile ("vbroadcasti128 %0,%%ymm2" : : "m" (raid_gfmulpshufb[V[3]][0][0]));
		asm volatile ("vbroadcasti128 %0,%%ymm3" : : "m" (raid_gfmulpshufb[V[3]][1][0]));
		asm volatile ("vpsrlw  $4,%ymm1,%ymm5");
		asm volatile ("vpand   %ymm7,%ymm1,%ymm4");
		asm volatile ("vpand   %ymm7,%ymm5,%ymm5");
		asm volatile ("vpshufb %ymm4,%ymm2,%ymm2");
		asm volatile ("vpshufb %ymm5,%ymm3,%ymm3");
		asm volatile ("vpxor   %ymm2,%ymm6,%ymm6");
		asm volatile ("vpxor   %ymm3,%ymm6,%ymm6");

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

	raid_avx_end();
}

/*
 * RAID recovering AVX2 implementation
 */
void raid_recX_avx2(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];
	size_t i;
	int j, k;

	/* setup the coefficients matrix */
	for (j = 0; j < N; ++j)
		for (k = 0; k < N; ++k)
			G[j * N + k] = A(ip[j], id[k]);

	/* invert it to solve the system of linear equations */
	raid_invert(G, V, N);

	/* compute delta parity */
	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();

#ifdef CONFIG_X86_64
	const uint8_t *T[RAID_PARITY_MAX * RAID_PARITY_MAX];

	/* precompute shuffle table pointers */
	for (j = 0; j < N * N; ++j)
		T[j] = &raid_gfmulpshufb[V[j]][0][0];

	asm volatile ("vpbroadcastb %0,%%ymm15" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 32) {
		/* delta */
		asm volatile (
			"movq 0(%2), %%rax\n"
			"movq 0(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm1\n"
			"vpand %%ymm15, %%ymm12, %%ymm0\n"
			"vpand %%ymm15, %%ymm1, %%ymm1\n"
			"cmpq $1, %0\n"
			"jbe 1f\n"

			"movq 8(%2), %%rax\n"
			"movq 8(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm3\n"
			"vpand %%ymm15, %%ymm12, %%ymm2\n"
			"vpand %%ymm15, %%ymm3, %%ymm3\n"
			"cmpq $2, %0\n"
			"jbe 1f\n"

			"movq 16(%2), %%rax\n"
			"movq 16(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm5\n"
			"vpand %%ymm15, %%ymm12, %%ymm4\n"
			"vpand %%ymm15, %%ymm5, %%ymm5\n"
			"cmpq $3, %0\n"
			"jbe 1f\n"

			"movq 24(%2), %%rax\n"
			"movq 24(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm7\n"
			"vpand %%ymm15, %%ymm12, %%ymm6\n"
			"vpand %%ymm15, %%ymm7, %%ymm7\n"
			"cmpq $4, %0\n"
			"jbe 1f\n"

			"movq 32(%2), %%rax\n"
			"movq 32(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm9\n"
			"vpand %%ymm15, %%ymm12, %%ymm8\n"
			"vpand %%ymm15, %%ymm9, %%ymm9\n"
			"cmpq $5, %0\n"
			"jbe 1f\n"

			"movq 40(%2), %%rax\n"
			"movq 40(%3), %%rbx\n"
			"vmovdqa (%%rax, %1), %%ymm12\n"
			"vmovdqa (%%rbx, %1), %%ymm13\n"
			"vpxor %%ymm13, %%ymm12, %%ymm12\n"
			"vpsrlw $4, %%ymm12, %%ymm11\n"
			"vpand %%ymm15, %%ymm12, %%ymm10\n"
			"vpand %%ymm15, %%ymm11, %%ymm11\n"

			"1:\n"
			:
			: "r" ((uint64_t)N), "r" (i), "r" (p), "r" (pa)
			: "rax", "rbx", "cc", "memory"
		);

		/* reconstruct */
		for (j = 0; j < N; ++j) {
			asm volatile (
				"movq 0(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm14\n"
				"vpshufb %%ymm0, %%ymm13, %%ymm13\n"
				"vpshufb %%ymm1, %%ymm14, %%ymm14\n"
				"cmpq $1, %0\n"
				"jbe 1f\n"

				"movq 8(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm12\n"
				"vpshufb %%ymm2, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm13, %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm12\n"
				"vpshufb %%ymm3, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm14, %%ymm14\n"
				"cmpq $2, %0\n"
				"jbe 1f\n"

				"movq 16(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm12\n"
				"vpshufb %%ymm4, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm13, %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm12\n"
				"vpshufb %%ymm5, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm14, %%ymm14\n"
				"cmpq $3, %0\n"
				"jbe 1f\n"

				"movq 24(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm12\n"
				"vpshufb %%ymm6, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm13, %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm12\n"
				"vpshufb %%ymm7, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm14, %%ymm14\n"
				"cmpq $4, %0\n"
				"jbe 1f\n"

				"movq 32(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm12\n"
				"vpshufb %%ymm8, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm13, %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm12\n"
				"vpshufb %%ymm9, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm14, %%ymm14\n"
				"cmpq $5, %0\n"
				"jbe 1f\n"

				"movq 40(%2), %%rcx\n"
				"vbroadcasti128 0(%%rcx), %%ymm12\n"
				"vpshufb %%ymm10, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm13, %%ymm13\n"
				"vbroadcasti128 16(%%rcx), %%ymm12\n"
				"vpshufb %%ymm11, %%ymm12, %%ymm12\n"
				"vpxor %%ymm12, %%ymm14, %%ymm14\n"

				"1:\n"
				"vpxor %%ymm14, %%ymm13, %%ymm13\n"
				"movq %3, %%rax\n"
				"vmovdqa %%ymm13, (%%rax, %1)\n"
				:
				: "r" ((uint64_t)N), "r" (i), "r" (&T[j * N]), "r" (pa[j])
				: "rax", "rcx", "cc", "memory"
			);
		}
	}
#else /* CONFIG_X86_32 */
	uint8_t buffer_low[RAID_PARITY_MAX * 32 + 32];
	uint8_t buffer_high[RAID_PARITY_MAX * 32 + 32];
	uint8_t *pd_low = __align_ptr(buffer_low, 32);
	uint8_t *pd_high = __align_ptr(buffer_high, 32);

	asm volatile ("vpbroadcastb %0,%%ymm7" : : "m" (gfconst16.low4[0]));

	for (i = 0; i < size; i += 32) {
		/* delta */
		for (j = 0; j < N; ++j) {
			asm volatile (
				"vmovdqa %2, %%ymm0\n"
				"vmovdqa %3, %%ymm1\n"
				"vpxor   %%ymm1, %%ymm0, %%ymm0\n"
				"vpsrlw  $4, %%ymm0, %%ymm1\n"
				"vpand   %%ymm7, %%ymm0, %%ymm0\n"
				"vpand   %%ymm7, %%ymm1, %%ymm1\n"
				"vmovdqa %%ymm0, %0\n"
				"vmovdqa %%ymm1, %1\n"
				: "=m" (pd_low[j * 32]), "=m" (pd_high[j * 32])
				: "m" (p[j][i]), "m" (pa[j][i])
			);
		}

		/* reconstruct */
		for (j = 0; j < N; ++j) {
			asm volatile (
				"vpxor %%ymm0, %%ymm0, %%ymm0\n"
				"vpxor %%ymm1, %%ymm1, %%ymm1\n"
			);

			for (k = 0; k < N; ++k) {
				uint8_t m = V[j * N + k];

				asm volatile (
					"vbroadcasti128 %0, %%ymm2\n"
					"vbroadcasti128 %1, %%ymm3\n"
					"vmovdqa        %2, %%ymm4\n"
					"vmovdqa        %3, %%ymm5\n"
					"vpshufb        %%ymm4, %%ymm2, %%ymm2\n"
					"vpshufb        %%ymm5, %%ymm3, %%ymm3\n"
					"vpxor          %%ymm2, %%ymm0, %%ymm0\n"
					"vpxor          %%ymm3, %%ymm1, %%ymm1\n"
					:
					: "m" (raid_gfmulpshufb[m][0][0]), "m" (raid_gfmulpshufb[m][1][0]),
					"m" (pd_low[k * 32]), "m" (pd_high[k * 32])
				);
			}

			asm volatile (
				"vpxor   %%ymm1, %%ymm0, %%ymm0\n"
				"vmovdqa %%ymm0, %0\n"
				: "=m" (pa[j][i])
			);
		}
	}
#endif

	raid_avx_end();
}

void raid_register_avx2(void)
{
	if (raid_cpu_has_avx2()) {
		raid_gen_register(RAID_ALGO_CAUCHY_PAR1, "avx2", raid_gen1_avx2, RAID_POLY_ANY);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "avx2", raid_gen2_avx2_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "avx2", raid_gen2_avx2_aes, RAID_POLY_AES);
#ifdef CONFIG_X86_64
		raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "avx2e", raid_gen2_avx2ext_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR2, "avx2e", raid_gen2_avx2ext_aes, RAID_POLY_AES);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR3, "avx2e", raid_gen3_avx2ext_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR3, "avx2e", raid_gen3_avx2ext_aes, RAID_POLY_AES);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR4, "avx2e", raid_gen4_avx2ext_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR4, "avx2e", raid_gen4_avx2ext_aes, RAID_POLY_AES);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR5, "avx2e", raid_gen5_avx2ext_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR5, "avx2e", raid_gen5_avx2ext_aes, RAID_POLY_AES);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR6, "avx2e", raid_gen6_avx2ext_raid, RAID_POLY_RAID);
		raid_gen_register(RAID_ALGO_CAUCHY_PAR6, "avx2e", raid_gen6_avx2ext_aes, RAID_POLY_AES);
		raid_gen_register(RAID_ALGO_VANDERMONDE_PAR3, "avx2e", raid_genz_avx2ext_raid, RAID_POLY_RAID);
#endif

		raid_rec_register(RAID_ALGO_CAUCHY_PAR1, "avx2", raid_rec1_avx2, RAID_POLY_ANY);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR2, "avx2", raid_rec2_avx2, RAID_POLY_ANY);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR3, "avx2", raid_recX_avx2, RAID_POLY_ANY);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR4, "avx2", raid_recX_avx2, RAID_POLY_ANY);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR5, "avx2", raid_recX_avx2, RAID_POLY_ANY);
		raid_rec_register(RAID_ALGO_CAUCHY_PAR6, "avx2", raid_recX_avx2, RAID_POLY_ANY);
	}
}
#endif