mirror of
https://github.com/amadvance/snapraid.git
synced 2026-10-26 02:49:02 -04:00
Optimize gen2 and gen3 kernels for AVX-512 GFNI with coefficient preloading
This commit is contained in:
parent
76ed8d643a
commit
3759c9356b
1 changed files with 940 additions and 8 deletions
948
raid/gfni.c
948
raid/gfni.c
|
|
@ -118,6 +118,472 @@ static __always_inline void raid_genX_avx2gfni_raid(int nd, size_t size, void **
|
|||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GEN2 (RAID6 with powers of 2) AVX512 GFNI implementation
|
||||
*/
|
||||
static __always_inline void raid_gen2_avx512gfni_raid_gen(int nd, size_t size, void **vv)
|
||||
{
|
||||
uint8_t **v = (uint8_t **)vv;
|
||||
size_t i;
|
||||
int d;
|
||||
|
||||
if (nd == 1) {
|
||||
if (v[1] != v[0])
|
||||
memcpy(v[1], v[0], size);
|
||||
if (v[2] != v[0])
|
||||
memcpy(v[2], v[0], size);
|
||||
return;
|
||||
}
|
||||
|
||||
raid_avx_begin();
|
||||
|
||||
/* preload as many Q coefficient matrices as possible */
|
||||
asm volatile ("vpbroadcastq %0,%%zmm4" : : "m" (raid_gfcauchyaffine_raid[1][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm5" : : "m" (raid_gfcauchyaffine_raid[2][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm6" : : "m" (raid_gfcauchyaffine_raid[3][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm7" : : "m" (raid_gfcauchyaffine_raid[4][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm8" : : "m" (raid_gfcauchyaffine_raid[5][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm9" : : "m" (raid_gfcauchyaffine_raid[6][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm10" : : "m" (raid_gfcauchyaffine_raid[7][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm11" : : "m" (raid_gfcauchyaffine_raid[8][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm12" : : "m" (raid_gfcauchyaffine_raid[9][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm13" : : "m" (raid_gfcauchyaffine_raid[10][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm14" : : "m" (raid_gfcauchyaffine_raid[11][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm15" : : "m" (raid_gfcauchyaffine_raid[12][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm16" : : "m" (raid_gfcauchyaffine_raid[13][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm17" : : "m" (raid_gfcauchyaffine_raid[14][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm18" : : "m" (raid_gfcauchyaffine_raid[15][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm19" : : "m" (raid_gfcauchyaffine_raid[16][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm20" : : "m" (raid_gfcauchyaffine_raid[17][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm21" : : "m" (raid_gfcauchyaffine_raid[18][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm22" : : "m" (raid_gfcauchyaffine_raid[19][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm23" : : "m" (raid_gfcauchyaffine_raid[20][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm24" : : "m" (raid_gfcauchyaffine_raid[21][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm25" : : "m" (raid_gfcauchyaffine_raid[22][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm26" : : "m" (raid_gfcauchyaffine_raid[23][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm27" : : "m" (raid_gfcauchyaffine_raid[24][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm28" : : "m" (raid_gfcauchyaffine_raid[25][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm29" : : "m" (raid_gfcauchyaffine_raid[26][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm30" : : "m" (raid_gfcauchyaffine_raid[27][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm31" : : "m" (raid_gfcauchyaffine_raid[28][0][0]));
|
||||
|
||||
for (i = 0; i < size; i += 64) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
|
||||
asm volatile ("vmovdqa64 %zmm0,%zmm1");
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[1][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm4,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 2)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[2][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm5,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 3)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[3][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm6,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 4)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[4][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm7,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 5)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[5][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm8,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 6)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[6][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm9,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 7)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[7][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm10,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 8)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[8][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm11,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 9)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[9][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm12,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 10)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[10][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm13,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 11)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[11][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm14,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 12)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[12][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm15,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 13)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[13][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm16,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 14)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[14][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm17,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 15)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[15][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm18,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 16)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[16][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm19,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 17)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[17][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm20,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 18)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[18][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm21,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 19)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[19][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm22,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 20)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[20][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm23,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 21)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[21][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm24,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 22)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[22][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm25,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 23)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[23][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm26,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 24)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[24][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm27,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 25)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[25][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm28,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 26)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[26][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm29,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 27)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[27][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm30,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 28)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[28][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm31,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 29)
|
||||
goto store;
|
||||
|
||||
/*
|
||||
* No more registers are available for resident coefficient
|
||||
* tables. Process D29 and later with the normal loop.
|
||||
*/
|
||||
for (d = 29; d < nd; ++d) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[d][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vpbroadcastq %0,%%zmm3" : : "m" (raid_gfcauchyaffine_raid[d][0][0]));
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm3,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
}
|
||||
|
||||
store:
|
||||
asm volatile ("vmovntdq %%zmm0,%0" : "=m" (v[nd][i]));
|
||||
asm volatile ("vmovntdq %%zmm1,%0" : "=m" (v[nd + 1][i]));
|
||||
}
|
||||
|
||||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GEN3 (triple parity with Cauchy matrix) AVX512 GFNI implementation
|
||||
*/
|
||||
static __always_inline void raid_gen3_avx512gfni_raid_gen(int nd, size_t size, void **vv)
|
||||
{
|
||||
uint8_t **v = (uint8_t **)vv;
|
||||
size_t i;
|
||||
int d;
|
||||
|
||||
if (nd == 1) {
|
||||
if (v[1] != v[0])
|
||||
memcpy(v[1], v[0], size);
|
||||
if (v[2] != v[0])
|
||||
memcpy(v[2], v[0], size);
|
||||
if (v[3] != v[0])
|
||||
memcpy(v[3], v[0], size);
|
||||
return;
|
||||
}
|
||||
|
||||
raid_avx_begin();
|
||||
|
||||
/* preload as many Q and R coefficient matrices as possible */
|
||||
asm volatile ("vpbroadcastq %0,%%zmm6" : : "m" (raid_gfcauchyaffine_raid[1][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm7" : : "m" (raid_gfcauchyaffine_raid[1][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm8" : : "m" (raid_gfcauchyaffine_raid[2][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm9" : : "m" (raid_gfcauchyaffine_raid[2][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm10" : : "m" (raid_gfcauchyaffine_raid[3][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm11" : : "m" (raid_gfcauchyaffine_raid[3][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm12" : : "m" (raid_gfcauchyaffine_raid[4][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm13" : : "m" (raid_gfcauchyaffine_raid[4][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm14" : : "m" (raid_gfcauchyaffine_raid[5][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm15" : : "m" (raid_gfcauchyaffine_raid[5][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm16" : : "m" (raid_gfcauchyaffine_raid[6][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm17" : : "m" (raid_gfcauchyaffine_raid[6][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm18" : : "m" (raid_gfcauchyaffine_raid[7][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm19" : : "m" (raid_gfcauchyaffine_raid[7][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm20" : : "m" (raid_gfcauchyaffine_raid[8][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm21" : : "m" (raid_gfcauchyaffine_raid[8][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm22" : : "m" (raid_gfcauchyaffine_raid[9][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm23" : : "m" (raid_gfcauchyaffine_raid[9][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm24" : : "m" (raid_gfcauchyaffine_raid[10][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm25" : : "m" (raid_gfcauchyaffine_raid[10][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm26" : : "m" (raid_gfcauchyaffine_raid[11][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm27" : : "m" (raid_gfcauchyaffine_raid[11][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm28" : : "m" (raid_gfcauchyaffine_raid[12][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm29" : : "m" (raid_gfcauchyaffine_raid[12][1][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm30" : : "m" (raid_gfcauchyaffine_raid[13][0][0]));
|
||||
asm volatile ("vpbroadcastq %0,%%zmm31" : : "m" (raid_gfcauchyaffine_raid[13][1][0]));
|
||||
|
||||
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 %0,%%zmm3" : : "m" (v[1][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm6,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm7,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 2)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[2][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm8,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm9,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 3)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[3][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm10,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm11,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 4)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[4][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm12,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm13,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 5)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[5][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm14,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm15,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 6)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[6][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm16,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm17,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 7)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[7][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm18,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm19,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 8)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[8][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm20,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm21,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 9)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[9][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm22,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm23,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 10)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[10][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm24,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm25,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 11)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[11][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm26,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm27,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 12)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[12][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm28,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm29,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 13)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[13][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm30,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm31,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 14)
|
||||
goto store;
|
||||
|
||||
/*
|
||||
* No more registers are available for resident coefficient
|
||||
* tables. Process D14 and later with the normal loop.
|
||||
*/
|
||||
for (d = 14; d < nd; ++d) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[d][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
|
||||
asm volatile ("vpbroadcastq %0,%%zmm4" : : "m" (raid_gfcauchyaffine_raid[d][0][0]));
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm4,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
|
||||
asm volatile ("vpbroadcastq %0,%%zmm5" : : "m" (raid_gfcauchyaffine_raid[d][1][0]));
|
||||
asm volatile ("vgf2p8affineqb $0,%zmm5,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
}
|
||||
|
||||
store:
|
||||
asm volatile ("vmovntdq %%zmm0,%0" : "=m" (v[nd][i]));
|
||||
asm volatile ("vmovntdq %%zmm1,%0" : "=m" (v[nd + 1][i]));
|
||||
asm volatile ("vmovntdq %%zmm2,%0" : "=m" (v[nd + 2][i]));
|
||||
}
|
||||
|
||||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GENX AVX512 GFNI implementation for the RAID polynomial using direct
|
||||
* Cauchy coefficients.
|
||||
|
|
@ -321,6 +787,472 @@ static __always_inline void raid_genX_avx2gfni_aes(int nd, size_t size, void **v
|
|||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GEN2 (RAID6 with powers of 3) AVX512 GFNI implementation
|
||||
*/
|
||||
static __always_inline void raid_gen2_avx512gfni_aes_gen(int nd, size_t size, void **vv)
|
||||
{
|
||||
uint8_t **v = (uint8_t **)vv;
|
||||
size_t i;
|
||||
int d;
|
||||
|
||||
if (nd == 1) {
|
||||
if (v[1] != v[0])
|
||||
memcpy(v[1], v[0], size);
|
||||
if (v[2] != v[0])
|
||||
memcpy(v[2], v[0], size);
|
||||
return;
|
||||
}
|
||||
|
||||
raid_avx_begin();
|
||||
|
||||
/* preload as many Q coefficient multipliers as possible */
|
||||
asm volatile ("vpbroadcastb %0,%%zmm4" : : "m" (raid_gfcauchy_aes[1][1]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm5" : : "m" (raid_gfcauchy_aes[1][2]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm6" : : "m" (raid_gfcauchy_aes[1][3]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (raid_gfcauchy_aes[1][4]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (raid_gfcauchy_aes[1][5]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm9" : : "m" (raid_gfcauchy_aes[1][6]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm10" : : "m" (raid_gfcauchy_aes[1][7]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm11" : : "m" (raid_gfcauchy_aes[1][8]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm12" : : "m" (raid_gfcauchy_aes[1][9]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm13" : : "m" (raid_gfcauchy_aes[1][10]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm14" : : "m" (raid_gfcauchy_aes[1][11]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm15" : : "m" (raid_gfcauchy_aes[1][12]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm16" : : "m" (raid_gfcauchy_aes[1][13]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm17" : : "m" (raid_gfcauchy_aes[1][14]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm18" : : "m" (raid_gfcauchy_aes[1][15]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm19" : : "m" (raid_gfcauchy_aes[1][16]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm20" : : "m" (raid_gfcauchy_aes[1][17]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm21" : : "m" (raid_gfcauchy_aes[1][18]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm22" : : "m" (raid_gfcauchy_aes[1][19]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm23" : : "m" (raid_gfcauchy_aes[1][20]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm24" : : "m" (raid_gfcauchy_aes[1][21]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm25" : : "m" (raid_gfcauchy_aes[1][22]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm26" : : "m" (raid_gfcauchy_aes[1][23]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm27" : : "m" (raid_gfcauchy_aes[1][24]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm28" : : "m" (raid_gfcauchy_aes[1][25]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm29" : : "m" (raid_gfcauchy_aes[1][26]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm30" : : "m" (raid_gfcauchy_aes[1][27]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm31" : : "m" (raid_gfcauchy_aes[1][28]));
|
||||
|
||||
for (i = 0; i < size; i += 64) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm0" : : "m" (v[0][i]));
|
||||
asm volatile ("vmovdqa64 %zmm0,%zmm1");
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[1][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm4,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 2)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[2][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm5,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 3)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[3][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm6,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 4)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[4][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm7,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 5)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[5][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm8,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 6)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[6][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm9,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 7)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[7][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm10,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 8)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[8][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm11,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 9)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[9][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm12,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 10)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[10][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm13,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 11)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[11][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm14,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 12)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[12][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm15,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 13)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[13][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm16,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 14)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[14][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm17,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 15)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[15][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm18,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 16)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[16][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm19,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 17)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[17][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm20,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 18)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[18][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm21,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 19)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[19][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm22,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 20)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[20][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm23,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 21)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[21][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm24,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 22)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[22][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm25,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 23)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[23][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm26,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 24)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[24][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm27,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 25)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[25][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm28,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 26)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[26][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm29,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 27)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[27][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm30,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 28)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[28][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm31,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
if (nd == 29)
|
||||
goto store;
|
||||
|
||||
/*
|
||||
* No more registers are available for resident coefficient
|
||||
* tables. Process D29 and later with the normal loop.
|
||||
*/
|
||||
for (d = 29; d < nd; ++d) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm2" : : "m" (v[d][i]));
|
||||
asm volatile ("vpxorq %zmm2,%zmm0,%zmm0");
|
||||
asm volatile ("vpbroadcastb %0,%%zmm3" : : "m" (raid_gfcauchy_aes[1][d]));
|
||||
asm volatile ("vgf2p8mulb %zmm3,%zmm2,%zmm3");
|
||||
asm volatile ("vpxorq %zmm3,%zmm1,%zmm1");
|
||||
}
|
||||
|
||||
store:
|
||||
asm volatile ("vmovntdq %%zmm0,%0" : "=m" (v[nd][i]));
|
||||
asm volatile ("vmovntdq %%zmm1,%0" : "=m" (v[nd + 1][i]));
|
||||
}
|
||||
|
||||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GEN3 (triple parity with Cauchy matrix) AVX512 GFNI implementation
|
||||
*/
|
||||
static __always_inline void raid_gen3_avx512gfni_aes_gen(int nd, size_t size, void **vv)
|
||||
{
|
||||
uint8_t **v = (uint8_t **)vv;
|
||||
size_t i;
|
||||
int d;
|
||||
|
||||
if (nd == 1) {
|
||||
if (v[1] != v[0])
|
||||
memcpy(v[1], v[0], size);
|
||||
if (v[2] != v[0])
|
||||
memcpy(v[2], v[0], size);
|
||||
if (v[3] != v[0])
|
||||
memcpy(v[3], v[0], size);
|
||||
return;
|
||||
}
|
||||
|
||||
raid_avx_begin();
|
||||
|
||||
/* preload as many Q and R coefficient multipliers as possible */
|
||||
asm volatile ("vpbroadcastb %0,%%zmm6" : : "m" (raid_gfcauchy_aes[1][1]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm7" : : "m" (raid_gfcauchy_aes[2][1]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm8" : : "m" (raid_gfcauchy_aes[1][2]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm9" : : "m" (raid_gfcauchy_aes[2][2]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm10" : : "m" (raid_gfcauchy_aes[1][3]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm11" : : "m" (raid_gfcauchy_aes[2][3]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm12" : : "m" (raid_gfcauchy_aes[1][4]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm13" : : "m" (raid_gfcauchy_aes[2][4]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm14" : : "m" (raid_gfcauchy_aes[1][5]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm15" : : "m" (raid_gfcauchy_aes[2][5]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm16" : : "m" (raid_gfcauchy_aes[1][6]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm17" : : "m" (raid_gfcauchy_aes[2][6]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm18" : : "m" (raid_gfcauchy_aes[1][7]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm19" : : "m" (raid_gfcauchy_aes[2][7]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm20" : : "m" (raid_gfcauchy_aes[1][8]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm21" : : "m" (raid_gfcauchy_aes[2][8]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm22" : : "m" (raid_gfcauchy_aes[1][9]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm23" : : "m" (raid_gfcauchy_aes[2][9]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm24" : : "m" (raid_gfcauchy_aes[1][10]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm25" : : "m" (raid_gfcauchy_aes[2][10]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm26" : : "m" (raid_gfcauchy_aes[1][11]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm27" : : "m" (raid_gfcauchy_aes[2][11]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm28" : : "m" (raid_gfcauchy_aes[1][12]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm29" : : "m" (raid_gfcauchy_aes[2][12]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm30" : : "m" (raid_gfcauchy_aes[1][13]));
|
||||
asm volatile ("vpbroadcastb %0,%%zmm31" : : "m" (raid_gfcauchy_aes[2][13]));
|
||||
|
||||
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 %0,%%zmm3" : : "m" (v[1][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm6,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm7,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 2)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[2][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm8,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm9,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 3)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[3][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm10,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm11,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 4)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[4][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm12,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm13,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 5)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[5][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm14,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm15,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 6)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[6][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm16,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm17,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 7)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[7][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm18,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm19,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 8)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[8][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm20,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm21,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 9)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[9][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm22,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm23,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 10)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[10][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm24,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm25,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 11)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[11][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm26,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm27,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 12)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[12][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm28,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm29,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 13)
|
||||
goto store;
|
||||
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[13][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
asm volatile ("vgf2p8mulb %zmm30,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
asm volatile ("vgf2p8mulb %zmm31,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
if (nd == 14)
|
||||
goto store;
|
||||
|
||||
/*
|
||||
* No more registers are available for resident coefficient
|
||||
* tables. Process D14 and later with the normal loop.
|
||||
*/
|
||||
for (d = 14; d < nd; ++d) {
|
||||
asm volatile ("vmovdqa64 %0,%%zmm3" : : "m" (v[d][i]));
|
||||
asm volatile ("vpxorq %zmm3,%zmm0,%zmm0");
|
||||
|
||||
asm volatile ("vpbroadcastb %0,%%zmm4" : : "m" (raid_gfcauchy_aes[1][d]));
|
||||
asm volatile ("vgf2p8mulb %zmm4,%zmm3,%zmm4");
|
||||
asm volatile ("vpxorq %zmm4,%zmm1,%zmm1");
|
||||
|
||||
asm volatile ("vpbroadcastb %0,%%zmm5" : : "m" (raid_gfcauchy_aes[2][d]));
|
||||
asm volatile ("vgf2p8mulb %zmm5,%zmm3,%zmm5");
|
||||
asm volatile ("vpxorq %zmm5,%zmm2,%zmm2");
|
||||
}
|
||||
|
||||
store:
|
||||
asm volatile ("vmovntdq %%zmm0,%0" : "=m" (v[nd][i]));
|
||||
asm volatile ("vmovntdq %%zmm1,%0" : "=m" (v[nd + 1][i]));
|
||||
asm volatile ("vmovntdq %%zmm2,%0" : "=m" (v[nd + 2][i]));
|
||||
}
|
||||
|
||||
raid_avx_end();
|
||||
}
|
||||
|
||||
/*
|
||||
* GENX AVX512 GFNI implementation for the AES polynomial using direct
|
||||
* Cauchy coefficients.
|
||||
|
|
@ -1150,11 +2082,11 @@ void raid_gen2_avx2gfni_raid(int nd, size_t size, void **vv)
|
|||
}
|
||||
|
||||
/*
|
||||
* GEN2 (RAID6 with powers of 2) GFNI implementation
|
||||
* GEN2 (RAID6 with powers of 2) AVX512 GFNI implementation
|
||||
*/
|
||||
void raid_gen2_avx512gfni_raid(int nd, size_t size, void **vv)
|
||||
{
|
||||
raid_genX_avx512gfni_raid(nd, size, vv, 2);
|
||||
raid_gen2_avx512gfni_raid_gen(nd, size, vv);
|
||||
}
|
||||
|
||||
/*
|
||||
|
|
@ -1166,11 +2098,11 @@ void raid_gen3_avx2gfni_raid(int nd, size_t size, void **vv)
|
|||
}
|
||||
|
||||
/*
|
||||
* GEN3 (triple parity with Cauchy matrix) GFNI implementation
|
||||
* GEN3 (triple parity with Cauchy matrix) AVX512 GFNI implementation
|
||||
*/
|
||||
void raid_gen3_avx512gfni_raid(int nd, size_t size, void **vv)
|
||||
{
|
||||
raid_genX_avx512gfni_raid(nd, size, vv, 3);
|
||||
raid_gen3_avx512gfni_raid_gen(nd, size, vv);
|
||||
}
|
||||
|
||||
/*
|
||||
|
|
@ -1230,11 +2162,11 @@ void raid_gen2_avx2gfni_aes(int nd, size_t size, void **vv)
|
|||
}
|
||||
|
||||
/*
|
||||
* GEN2 (RAID6 with powers of 3) GFNI implementation
|
||||
* GEN2 (RAID6 with powers of 3) AVX512 GFNI implementation
|
||||
*/
|
||||
void raid_gen2_avx512gfni_aes(int nd, size_t size, void **vv)
|
||||
{
|
||||
raid_genX_avx512gfni_aes(nd, size, vv, 2);
|
||||
raid_gen2_avx512gfni_aes_gen(nd, size, vv);
|
||||
}
|
||||
|
||||
/*
|
||||
|
|
@ -1246,11 +2178,11 @@ void raid_gen3_avx2gfni_aes(int nd, size_t size, void **vv)
|
|||
}
|
||||
|
||||
/*
|
||||
* GEN3 (triple parity with Cauchy matrix) GFNI implementation
|
||||
* GEN3 (triple parity with Cauchy matrix) AVX512 GFNI implementation
|
||||
*/
|
||||
void raid_gen3_avx512gfni_aes(int nd, size_t size, void **vv)
|
||||
{
|
||||
raid_genX_avx512gfni_aes(nd, size, vv, 3);
|
||||
raid_gen3_avx512gfni_aes_gen(nd, size, vv);
|
||||
}
|
||||
|
||||
/*
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue