mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-23 16:08:59 +00:00
1525 lines
54 KiB
C
1525 lines
54 KiB
C
#include "SABER_params.h"
|
|
#include "poly.h"
|
|
|
|
|
|
#define L (SABER_N / 64)
|
|
|
|
static inline __m256i mul_add(__m256i a, __m256i b, __m256i c) {
|
|
return _mm256_add_epi16(_mm256_mullo_epi16(a, b), c);
|
|
}
|
|
|
|
static void schoolbook_avx_acc(__m256i *c, const __m256i *a, const __m256i *b) {
|
|
__m256i a0, a1, a2, a3, a4, a5, a6, a7, b0, b1, b2, b3, b4, b5, b6, b7;
|
|
__m256i temp;
|
|
|
|
a0 = a[0];
|
|
a1 = a[1];
|
|
a2 = a[2];
|
|
a3 = a[3];
|
|
a4 = a[4];
|
|
a5 = a[5];
|
|
a6 = a[6];
|
|
a7 = a[7];
|
|
|
|
b0 = b[0];
|
|
b1 = b[1];
|
|
b2 = b[2];
|
|
b3 = b[3];
|
|
b4 = b[4];
|
|
b5 = b[5];
|
|
b6 = b[6];
|
|
b7 = b[7];
|
|
|
|
c[0] = mul_add(a0, b0, c[0]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b1);
|
|
temp = mul_add(a1, b0, temp);
|
|
c[1] = _mm256_add_epi16(temp, c[1]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b2);
|
|
temp = mul_add(a1, b1, temp);
|
|
temp = mul_add(a2, b0, temp);
|
|
c[2] = _mm256_add_epi16(temp, c[2]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b3);
|
|
temp = mul_add(a1, b2, temp);
|
|
temp = mul_add(a2, b1, temp);
|
|
temp = mul_add(a3, b0, temp);
|
|
c[3] = _mm256_add_epi16(temp, c[3]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b4);
|
|
temp = mul_add(a1, b3, temp);
|
|
temp = mul_add(a3, b1, temp);
|
|
temp = mul_add(a4, b0, temp);
|
|
temp = mul_add(a2, b2, temp);
|
|
c[4] = _mm256_add_epi16(temp, c[4]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b5);
|
|
temp = mul_add(a1, b4, temp);
|
|
temp = mul_add(a2, b3, temp);
|
|
temp = mul_add(a3, b2, temp);
|
|
temp = mul_add( a4, b1, temp);
|
|
temp = mul_add(a5, b0, temp);
|
|
c[5] = _mm256_add_epi16(temp, c[5]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b6);
|
|
temp = mul_add(a1, b5, temp);
|
|
temp = mul_add(a5, b1, temp);
|
|
temp = mul_add(a6, b0, temp);
|
|
temp = mul_add(a2, b4, temp);
|
|
temp = mul_add(a3, b3, temp);
|
|
temp = mul_add(a4, b2, temp);
|
|
c[6] = _mm256_add_epi16(temp, c[6]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b7);
|
|
temp = mul_add(a1, b6, temp);
|
|
temp = mul_add(a6, b1, temp);
|
|
temp = mul_add(a7, b0, temp);
|
|
temp = mul_add(a2, b5, temp);
|
|
temp = mul_add(a3, b4, temp);
|
|
temp = mul_add(a4, b3, temp);
|
|
temp = mul_add(a5, b2, temp);
|
|
c[7] = _mm256_add_epi16(temp, c[7]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[8]);
|
|
temp = mul_add(a1, b7, temp);
|
|
temp = mul_add(a7, b1, temp);
|
|
temp = mul_add(a[8], b0, temp);
|
|
temp = mul_add(a2, b6, temp);
|
|
temp = mul_add(a3, b5, temp);
|
|
temp = mul_add(a4, b4, temp);
|
|
temp = mul_add(a5, b3, temp);
|
|
temp = mul_add(a6, b2, temp);
|
|
c[8] = _mm256_add_epi16(temp, c[8]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[9]);
|
|
temp = mul_add(a1, b[8], temp);
|
|
temp = mul_add(a[8], b1, temp);
|
|
temp = mul_add(a[9], b0, temp);
|
|
temp = mul_add(a2, b7, temp);
|
|
temp = mul_add(a3, b6, temp);
|
|
temp = mul_add(a4, b5, temp);
|
|
temp = mul_add(a5, b4, temp);
|
|
temp = mul_add(a6, b3, temp);
|
|
temp = mul_add(a7, b2, temp);
|
|
c[9] = _mm256_add_epi16(temp, c[9]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[10]);
|
|
temp = mul_add(a1, b[9], temp);
|
|
temp = mul_add(a[9], b1, temp);
|
|
temp = mul_add(a[10], b0, temp);
|
|
temp = mul_add(a2, b[8], temp);
|
|
temp = mul_add(a3, b7, temp);
|
|
temp = mul_add(a4, b6, temp);
|
|
temp = mul_add(a5, b5, temp);
|
|
temp = mul_add(a6, b4, temp);
|
|
temp = mul_add(a7, b3, temp);
|
|
temp = mul_add(a[8], b2, temp);
|
|
c[10] = _mm256_add_epi16(temp, c[10]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[11]);
|
|
temp = mul_add(a1, b[10], temp);
|
|
temp = mul_add(a[10], b1, temp);
|
|
temp = mul_add(a[11], b0, temp);
|
|
temp = mul_add(a2, b[9], temp);
|
|
temp = mul_add(a3, b[8], temp);
|
|
temp = mul_add(a4, b7, temp);
|
|
temp = mul_add(a5, b6, temp);
|
|
temp = mul_add(a6, b5, temp);
|
|
temp = mul_add(a7, b4, temp);
|
|
temp = mul_add(a[8], b3, temp);
|
|
temp = mul_add(a[9], b2, temp);
|
|
c[11] = _mm256_add_epi16(temp, c[11]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[12]);
|
|
temp = mul_add(a1, b[11], temp);
|
|
temp = mul_add(a[11], b1, temp);
|
|
temp = mul_add(a[12], b0, temp);
|
|
temp = mul_add(a2, b[10], temp);
|
|
temp = mul_add(a3, b[9], temp);
|
|
temp = mul_add(a4, b[8], temp);
|
|
temp = mul_add(a5, b7, temp);
|
|
temp = mul_add(a6, b6, temp);
|
|
temp = mul_add(a7, b5, temp);
|
|
temp = mul_add(a[8], b4, temp);
|
|
temp = mul_add(a[9], b3, temp);
|
|
temp = mul_add(a[10], b2, temp);
|
|
c[12] = _mm256_add_epi16(temp, c[12]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[13]);
|
|
temp = mul_add(a1, b[12], temp);
|
|
temp = mul_add(a[12], b1, temp);
|
|
temp = mul_add(a[13], b0, temp);
|
|
temp = mul_add(a2, b[11], temp);
|
|
temp = mul_add(a3, b[10], temp);
|
|
temp = mul_add(a4, b[9], temp);
|
|
temp = mul_add(a5, b[8], temp);
|
|
temp = mul_add(a6, b7, temp);
|
|
temp = mul_add(a7, b6, temp);
|
|
temp = mul_add(a[8], b5, temp);
|
|
temp = mul_add(a[9], b4, temp);
|
|
temp = mul_add(a[10], b3, temp);
|
|
temp = mul_add(a[11], b2, temp);
|
|
c[13] = _mm256_add_epi16(temp, c[13]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[14]);
|
|
temp = mul_add(a1, b[13], temp);
|
|
temp = mul_add(a[13], b1, temp);
|
|
temp = mul_add(a[14], b0, temp);
|
|
temp = mul_add(a2, b[12], temp);
|
|
temp = mul_add(a3, b[11], temp);
|
|
temp = mul_add(a4, b[10], temp);
|
|
temp = mul_add(a5, b[9], temp);
|
|
temp = mul_add(a6, b[8], temp);
|
|
temp = mul_add(a7, b7, temp);
|
|
temp = mul_add(a[8], b6, temp);
|
|
temp = mul_add(a[9], b5, temp);
|
|
temp = mul_add(a[10], b4, temp);
|
|
temp = mul_add(a[11], b3, temp);
|
|
temp = mul_add(a[12], b2, temp);
|
|
c[14] = _mm256_add_epi16(temp, c[14]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[15]);
|
|
temp = mul_add(a1, b[14], temp);
|
|
temp = mul_add(a[14], b1, temp);
|
|
temp = mul_add(a[15], b0, temp);
|
|
temp = mul_add(a2, b[13], temp);
|
|
temp = mul_add(a3, b[12], temp);
|
|
temp = mul_add(a4, b[11], temp);
|
|
temp = mul_add(a5, b[10], temp);
|
|
temp = mul_add(a6, b[9], temp);
|
|
temp = mul_add(a7, b[8], temp);
|
|
temp = mul_add(a[8], b7, temp);
|
|
temp = mul_add(a[9], b6, temp);
|
|
temp = mul_add(a[10], b5, temp);
|
|
temp = mul_add(a[11], b4, temp);
|
|
temp = mul_add(a[12], b3, temp);
|
|
temp = mul_add(a[13], b2, temp);
|
|
c[15] = _mm256_add_epi16(temp, c[15]);
|
|
|
|
a0 = a[14];
|
|
a1 = a[15];
|
|
a2 = a[13];
|
|
a3 = a[12];
|
|
a4 = a[11];
|
|
a5 = a[10];
|
|
a6 = a[9];
|
|
a7 = a[8];
|
|
|
|
b0 = b[14];
|
|
b1 = b[15];
|
|
b2 = b[13];
|
|
b3 = b[12];
|
|
b4 = b[11];
|
|
b5 = b[10];
|
|
b6 = b[9];
|
|
b7 = b[8];
|
|
|
|
temp = _mm256_mullo_epi16(a[1], b1);
|
|
temp = mul_add(a[2], b0, temp);
|
|
temp = mul_add(a[3], b2, temp);
|
|
temp = mul_add(a[4], b3, temp);
|
|
temp = mul_add(a[5], b4, temp);
|
|
temp = mul_add(a[6], b5, temp);
|
|
temp = mul_add(a[7], b6, temp);
|
|
temp = mul_add(a7, b7, temp);
|
|
temp = mul_add(a6, b[7], temp);
|
|
temp = mul_add(a5, b[6], temp);
|
|
temp = mul_add(a4, b[5], temp);
|
|
temp = mul_add(a3, b[4], temp);
|
|
temp = mul_add(a2, b[3], temp);
|
|
temp = mul_add(a0, b[2], temp);
|
|
temp = mul_add(a1, b[1], temp);
|
|
c[16] = _mm256_add_epi16(temp, c[16]);
|
|
|
|
temp = _mm256_mullo_epi16(a[2], b1);
|
|
temp = mul_add(a[3], b0, temp);
|
|
temp = mul_add(a[4], b2, temp);
|
|
temp = mul_add(a[5], b3, temp);
|
|
temp = mul_add(a[6], b4, temp);
|
|
temp = mul_add(a[7], b5, temp);
|
|
temp = mul_add(a7, b6, temp);
|
|
temp = mul_add(a6, b7, temp);
|
|
temp = mul_add(a5, b[7], temp);
|
|
temp = mul_add(a4, b[6], temp);
|
|
temp = mul_add(a3, b[5], temp);
|
|
temp = mul_add(a2, b[4], temp);
|
|
temp = mul_add(a0, b[3], temp);
|
|
temp = mul_add(a1, b[2], temp);
|
|
c[17] = _mm256_add_epi16(temp, c[17]);
|
|
|
|
temp = _mm256_mullo_epi16(a[3], b1);
|
|
temp = mul_add(a[4], b0, temp);
|
|
temp = mul_add(a[5], b2, temp);
|
|
temp = mul_add(a[6], b3, temp);
|
|
temp = mul_add(a[7], b4, temp);
|
|
temp = mul_add(a7, b5, temp);
|
|
temp = mul_add(a6, b6, temp);
|
|
temp = mul_add(a5, b7, temp);
|
|
temp = mul_add(a4, b[7], temp);
|
|
temp = mul_add(a3, b[6], temp);
|
|
temp = mul_add(a2, b[5], temp);
|
|
temp = mul_add(a0, b[4], temp);
|
|
temp = mul_add(a1, b[3], temp);
|
|
c[18] = _mm256_add_epi16(temp, c[18]);
|
|
|
|
temp = _mm256_mullo_epi16(a[4], b1);
|
|
temp = mul_add(a[5], b0, temp);
|
|
temp = mul_add(a[6], b2, temp);
|
|
temp = mul_add(a[7], b3, temp);
|
|
temp = mul_add(a7, b4, temp);
|
|
temp = mul_add(a6, b5, temp);
|
|
temp = mul_add(a5, b6, temp);
|
|
temp = mul_add(a4, b7, temp);
|
|
temp = mul_add(a3, b[7], temp);
|
|
temp = mul_add(a2, b[6], temp);
|
|
temp = mul_add(a0, b[5], temp);
|
|
temp = mul_add(a1, b[4], temp);
|
|
c[19] = _mm256_add_epi16(temp, c[19]);
|
|
|
|
temp = _mm256_mullo_epi16(a[5], b1);
|
|
temp = mul_add(a[6], b0, temp);
|
|
temp = mul_add(a[7], b2, temp);
|
|
temp = mul_add(a7, b3, temp);
|
|
temp = mul_add(a6, b4, temp);
|
|
temp = mul_add(a5, b5, temp);
|
|
temp = mul_add(a4, b6, temp);
|
|
temp = mul_add(a3, b7, temp);
|
|
temp = mul_add(a2, b[7], temp);
|
|
temp = mul_add(a0, b[6], temp);
|
|
temp = mul_add(a1, b[5], temp);
|
|
c[20] = _mm256_add_epi16(temp, c[20]);
|
|
|
|
temp = _mm256_mullo_epi16(a[6], b1);
|
|
temp = mul_add(a[7], b0, temp);
|
|
temp = mul_add(a7, b2, temp);
|
|
temp = mul_add(a6, b3, temp);
|
|
temp = mul_add(a5, b4, temp);
|
|
temp = mul_add(a4, b5, temp);
|
|
temp = mul_add(a3, b6, temp);
|
|
temp = mul_add(a2, b7, temp);
|
|
temp = mul_add(a0, b[7], temp);
|
|
temp = mul_add(a1, b[6], temp);
|
|
c[21] = _mm256_add_epi16(temp, c[21]);
|
|
|
|
temp = _mm256_mullo_epi16(a[7], b1);
|
|
temp = mul_add(a7, b0, temp);
|
|
temp = mul_add(a6, b2, temp);
|
|
temp = mul_add(a5, b3, temp);
|
|
temp = mul_add(a4, b4, temp);
|
|
temp = mul_add(a3, b5, temp);
|
|
temp = mul_add(a2, b6, temp);
|
|
temp = mul_add(a0, b7, temp);
|
|
temp = mul_add(a1, b[7], temp);
|
|
c[22] = _mm256_add_epi16(temp, c[22]);
|
|
|
|
temp = _mm256_mullo_epi16(a7, b1);
|
|
temp = mul_add(a6, b0, temp);
|
|
temp = mul_add(a5, b2, temp);
|
|
temp = mul_add(a4, b3, temp);
|
|
temp = mul_add(a3, b4, temp);
|
|
temp = mul_add(a2, b5, temp);
|
|
temp = mul_add(a0, b6, temp);
|
|
temp = mul_add(a1, b7, temp);
|
|
c[23] = _mm256_add_epi16(temp, c[23]);
|
|
|
|
temp = _mm256_mullo_epi16(a6, b1);
|
|
temp = mul_add(a5, b0, temp);
|
|
temp = mul_add(a4, b2, temp);
|
|
temp = mul_add(a3, b3, temp);
|
|
temp = mul_add(a2, b4, temp);
|
|
temp = mul_add(a0, b5, temp);
|
|
temp = mul_add(a1, b6, temp);
|
|
c[24] = _mm256_add_epi16(temp, c[24]);
|
|
|
|
temp = _mm256_mullo_epi16(a5, b1);
|
|
temp = mul_add(a4, b0, temp);
|
|
temp = mul_add(a3, b2, temp);
|
|
temp = mul_add(a2, b3, temp);
|
|
temp = mul_add(a0, b4, temp);
|
|
temp = mul_add(a1, b5, temp);
|
|
c[25] = _mm256_add_epi16(temp, c[25]);
|
|
|
|
temp = _mm256_mullo_epi16(a4, b1);
|
|
temp = mul_add(a3, b0, temp);
|
|
temp = mul_add(a2, b2, temp);
|
|
temp = mul_add(a0, b3, temp);
|
|
temp = mul_add(a1, b4, temp);
|
|
c[26] = _mm256_add_epi16(temp, c[26]);
|
|
|
|
temp = _mm256_mullo_epi16(a3, b1);
|
|
temp = mul_add(a2, b0, temp);
|
|
temp = mul_add(a0, b2, temp);
|
|
temp = mul_add(a1, b3, temp);
|
|
c[27] = _mm256_add_epi16(temp, c[27]);
|
|
|
|
temp = _mm256_mullo_epi16(a2, b1);
|
|
temp = mul_add(a0, b0, temp);
|
|
temp = mul_add(a1, b2, temp);
|
|
c[28] = _mm256_add_epi16(temp, c[28]);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b1);
|
|
temp = mul_add(a1, b0, temp);
|
|
c[29] = _mm256_add_epi16(temp, c[29]);
|
|
|
|
c[30] = mul_add(a1, b1, c[30]);
|
|
|
|
c[31] = _mm256_set_epi64x(0, 0, 0, 0);
|
|
}
|
|
|
|
|
|
static void schoolbook_avx(__m256i *c, const __m256i *a, const __m256i *b) {
|
|
__m256i a0, a1, a2, a3, a4, a5, a6, a7, b0, b1, b2, b3, b4, b5, b6, b7;
|
|
__m256i temp;
|
|
|
|
a0 = a[0];
|
|
a1 = a[1];
|
|
a2 = a[2];
|
|
a3 = a[3];
|
|
a4 = a[4];
|
|
a5 = a[5];
|
|
a6 = a[6];
|
|
a7 = a[7];
|
|
|
|
b0 = b[0];
|
|
b1 = b[1];
|
|
b2 = b[2];
|
|
b3 = b[3];
|
|
b4 = b[4];
|
|
b5 = b[5];
|
|
b6 = b[6];
|
|
b7 = b[7];
|
|
|
|
c[0] = _mm256_mullo_epi16(a0, b0);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b1);
|
|
c[1] = mul_add(a1, b0, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b2);
|
|
temp = mul_add(a1, b1, temp);
|
|
c[2] = mul_add(a2, b0, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b3);
|
|
temp = mul_add(a1, b2, temp);
|
|
temp = mul_add(a2, b1, temp);
|
|
c[3] = mul_add(a3, b0, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b4);
|
|
temp = mul_add(a1, b3, temp);
|
|
temp = mul_add(a3, b1, temp);
|
|
temp = mul_add(a4, b0, temp);
|
|
c[4] = mul_add(a2, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b5);
|
|
temp = mul_add(a1, b4, temp);
|
|
temp = mul_add(a2, b3, temp);
|
|
temp = mul_add(a3, b2, temp);
|
|
temp = mul_add( a4, b1, temp);
|
|
c[5] = mul_add(a5, b0, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b6);
|
|
temp = mul_add(a1, b5, temp);
|
|
temp = mul_add(a5, b1, temp);
|
|
temp = mul_add(a6, b0, temp);
|
|
temp = mul_add(a2, b4, temp);
|
|
temp = mul_add(a3, b3, temp);
|
|
c[6] = mul_add(a4, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b7);
|
|
temp = mul_add(a1, b6, temp);
|
|
temp = mul_add(a6, b1, temp);
|
|
temp = mul_add(a7, b0, temp);
|
|
temp = mul_add(a2, b5, temp);
|
|
temp = mul_add(a3, b4, temp);
|
|
temp = mul_add(a4, b3, temp);
|
|
c[7] = mul_add(a5, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[8]);
|
|
temp = mul_add(a1, b7, temp);
|
|
temp = mul_add(a7, b1, temp);
|
|
temp = mul_add(a[8], b0, temp);
|
|
temp = mul_add(a2, b6, temp);
|
|
temp = mul_add(a3, b5, temp);
|
|
temp = mul_add(a4, b4, temp);
|
|
temp = mul_add(a5, b3, temp);
|
|
c[8] = mul_add(a6, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[9]);
|
|
temp = mul_add(a1, b[8], temp);
|
|
temp = mul_add(a[8], b1, temp);
|
|
temp = mul_add(a[9], b0, temp);
|
|
temp = mul_add(a2, b7, temp);
|
|
temp = mul_add(a3, b6, temp);
|
|
temp = mul_add(a4, b5, temp);
|
|
temp = mul_add(a5, b4, temp);
|
|
temp = mul_add(a6, b3, temp);
|
|
c[9] = mul_add(a7, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[10]);
|
|
temp = mul_add(a1, b[9], temp);
|
|
temp = mul_add(a[9], b1, temp);
|
|
temp = mul_add(a[10], b0, temp);
|
|
temp = mul_add(a2, b[8], temp);
|
|
temp = mul_add(a3, b7, temp);
|
|
temp = mul_add(a4, b6, temp);
|
|
temp = mul_add(a5, b5, temp);
|
|
temp = mul_add(a6, b4, temp);
|
|
temp = mul_add(a7, b3, temp);
|
|
c[10] = mul_add(a[8], b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[11]);
|
|
temp = mul_add(a1, b[10], temp);
|
|
temp = mul_add(a[10], b1, temp);
|
|
temp = mul_add(a[11], b0, temp);
|
|
temp = mul_add(a2, b[9], temp);
|
|
temp = mul_add(a3, b[8], temp);
|
|
temp = mul_add(a4, b7, temp);
|
|
temp = mul_add(a5, b6, temp);
|
|
temp = mul_add(a6, b5, temp);
|
|
temp = mul_add(a7, b4, temp);
|
|
temp = mul_add(a[8], b3, temp);
|
|
c[11] = mul_add(a[9], b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[12]);
|
|
temp = mul_add(a1, b[11], temp);
|
|
temp = mul_add(a[11], b1, temp);
|
|
temp = mul_add(a[12], b0, temp);
|
|
temp = mul_add(a2, b[10], temp);
|
|
temp = mul_add(a3, b[9], temp);
|
|
temp = mul_add(a4, b[8], temp);
|
|
temp = mul_add(a5, b7, temp);
|
|
temp = mul_add(a6, b6, temp);
|
|
temp = mul_add(a7, b5, temp);
|
|
temp = mul_add(a[8], b4, temp);
|
|
temp = mul_add(a[9], b3, temp);
|
|
c[12] = mul_add(a[10], b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[13]);
|
|
temp = mul_add(a1, b[12], temp);
|
|
temp = mul_add(a[12], b1, temp);
|
|
temp = mul_add(a[13], b0, temp);
|
|
temp = mul_add(a2, b[11], temp);
|
|
temp = mul_add(a3, b[10], temp);
|
|
temp = mul_add(a4, b[9], temp);
|
|
temp = mul_add(a5, b[8], temp);
|
|
temp = mul_add(a6, b7, temp);
|
|
temp = mul_add(a7, b6, temp);
|
|
temp = mul_add(a[8], b5, temp);
|
|
temp = mul_add(a[9], b4, temp);
|
|
temp = mul_add(a[10], b3, temp);
|
|
c[13] = mul_add(a[11], b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[14]);
|
|
temp = mul_add(a1, b[13], temp);
|
|
temp = mul_add(a[13], b1, temp);
|
|
temp = mul_add(a[14], b0, temp);
|
|
temp = mul_add(a2, b[12], temp);
|
|
temp = mul_add(a3, b[11], temp);
|
|
temp = mul_add(a4, b[10], temp);
|
|
temp = mul_add(a5, b[9], temp);
|
|
temp = mul_add(a6, b[8], temp);
|
|
temp = mul_add(a7, b7, temp);
|
|
temp = mul_add(a[8], b6, temp);
|
|
temp = mul_add(a[9], b5, temp);
|
|
temp = mul_add(a[10], b4, temp);
|
|
temp = mul_add(a[11], b3, temp);
|
|
c[14] = mul_add(a[12], b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b[15]);
|
|
temp = mul_add(a1, b[14], temp);
|
|
temp = mul_add(a[14], b1, temp);
|
|
temp = mul_add(a[15], b0, temp);
|
|
temp = mul_add(a2, b[13], temp);
|
|
temp = mul_add(a3, b[12], temp);
|
|
temp = mul_add(a4, b[11], temp);
|
|
temp = mul_add(a5, b[10], temp);
|
|
temp = mul_add(a6, b[9], temp);
|
|
temp = mul_add(a7, b[8], temp);
|
|
temp = mul_add(a[8], b7, temp);
|
|
temp = mul_add(a[9], b6, temp);
|
|
temp = mul_add(a[10], b5, temp);
|
|
temp = mul_add(a[11], b4, temp);
|
|
temp = mul_add(a[12], b3, temp);
|
|
c[15] = mul_add(a[13], b2, temp);
|
|
|
|
// unrolled second triangle
|
|
a0 = a[14];
|
|
a1 = a[15];
|
|
a2 = a[13];
|
|
a3 = a[12];
|
|
a4 = a[11];
|
|
a5 = a[10];
|
|
a6 = a[9];
|
|
a7 = a[8];
|
|
|
|
b0 = b[14];
|
|
b1 = b[15];
|
|
b2 = b[13];
|
|
b3 = b[12];
|
|
b4 = b[11];
|
|
b5 = b[10];
|
|
b6 = b[9];
|
|
b7 = b[8];
|
|
|
|
temp = _mm256_mullo_epi16(a[1], b1);
|
|
temp = mul_add(a[2], b0, temp);
|
|
temp = mul_add(a[3], b2, temp);
|
|
temp = mul_add(a[4], b3, temp);
|
|
temp = mul_add(a[5], b4, temp);
|
|
temp = mul_add(a[6], b5, temp);
|
|
temp = mul_add(a[7], b6, temp);
|
|
temp = mul_add(a7, b7, temp);
|
|
temp = mul_add(a6, b[7], temp);
|
|
temp = mul_add(a5, b[6], temp);
|
|
temp = mul_add(a4, b[5], temp);
|
|
temp = mul_add(a3, b[4], temp);
|
|
temp = mul_add(a2, b[3], temp);
|
|
temp = mul_add(a0, b[2], temp);
|
|
c[16] = mul_add(a1, b[1], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[2], b1);
|
|
temp = mul_add(a[3], b0, temp);
|
|
temp = mul_add(a[4], b2, temp);
|
|
temp = mul_add(a[5], b3, temp);
|
|
temp = mul_add(a[6], b4, temp);
|
|
temp = mul_add(a[7], b5, temp);
|
|
temp = mul_add(a7, b6, temp);
|
|
temp = mul_add(a6, b7, temp);
|
|
temp = mul_add(a5, b[7], temp);
|
|
temp = mul_add(a4, b[6], temp);
|
|
temp = mul_add(a3, b[5], temp);
|
|
temp = mul_add(a2, b[4], temp);
|
|
temp = mul_add(a0, b[3], temp);
|
|
c[17] = mul_add(a1, b[2], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[3], b1);
|
|
temp = mul_add(a[4], b0, temp);
|
|
temp = mul_add(a[5], b2, temp);
|
|
temp = mul_add(a[6], b3, temp);
|
|
temp = mul_add(a[7], b4, temp);
|
|
temp = mul_add(a7, b5, temp);
|
|
temp = mul_add(a6, b6, temp);
|
|
temp = mul_add(a5, b7, temp);
|
|
temp = mul_add(a4, b[7], temp);
|
|
temp = mul_add(a3, b[6], temp);
|
|
temp = mul_add(a2, b[5], temp);
|
|
temp = mul_add(a0, b[4], temp);
|
|
c[18] = mul_add(a1, b[3], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[4], b1);
|
|
temp = mul_add(a[5], b0, temp);
|
|
temp = mul_add(a[6], b2, temp);
|
|
temp = mul_add(a[7], b3, temp);
|
|
temp = mul_add(a7, b4, temp);
|
|
temp = mul_add(a6, b5, temp);
|
|
temp = mul_add(a5, b6, temp);
|
|
temp = mul_add(a4, b7, temp);
|
|
temp = mul_add(a3, b[7], temp);
|
|
temp = mul_add(a2, b[6], temp);
|
|
temp = mul_add(a0, b[5], temp);
|
|
c[19] = mul_add(a1, b[4], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[5], b1);
|
|
temp = mul_add(a[6], b0, temp);
|
|
temp = mul_add(a[7], b2, temp);
|
|
temp = mul_add(a7, b3, temp);
|
|
temp = mul_add(a6, b4, temp);
|
|
temp = mul_add(a5, b5, temp);
|
|
temp = mul_add(a4, b6, temp);
|
|
temp = mul_add(a3, b7, temp);
|
|
temp = mul_add(a2, b[7], temp);
|
|
temp = mul_add(a0, b[6], temp);
|
|
c[20] = mul_add(a1, b[5], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[6], b1);
|
|
temp = mul_add(a[7], b0, temp);
|
|
temp = mul_add(a7, b2, temp);
|
|
temp = mul_add(a6, b3, temp);
|
|
temp = mul_add(a5, b4, temp);
|
|
temp = mul_add(a4, b5, temp);
|
|
temp = mul_add(a3, b6, temp);
|
|
temp = mul_add(a2, b7, temp);
|
|
temp = mul_add(a0, b[7], temp);
|
|
c[21] = mul_add(a1, b[6], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a[7], b1);
|
|
temp = mul_add(a7, b0, temp);
|
|
temp = mul_add(a6, b2, temp);
|
|
temp = mul_add(a5, b3, temp);
|
|
temp = mul_add(a4, b4, temp);
|
|
temp = mul_add(a3, b5, temp);
|
|
temp = mul_add(a2, b6, temp);
|
|
temp = mul_add(a0, b7, temp);
|
|
c[22] = mul_add(a1, b[7], temp);
|
|
|
|
temp = _mm256_mullo_epi16(a7, b1);
|
|
temp = mul_add(a6, b0, temp);
|
|
temp = mul_add(a5, b2, temp);
|
|
temp = mul_add(a4, b3, temp);
|
|
temp = mul_add(a3, b4, temp);
|
|
temp = mul_add(a2, b5, temp);
|
|
temp = mul_add(a0, b6, temp);
|
|
c[23] = mul_add(a1, b7, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a6, b1);
|
|
temp = mul_add(a5, b0, temp);
|
|
temp = mul_add(a4, b2, temp);
|
|
temp = mul_add(a3, b3, temp);
|
|
temp = mul_add(a2, b4, temp);
|
|
temp = mul_add(a0, b5, temp);
|
|
c[24] = mul_add(a1, b6, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a5, b1);
|
|
temp = mul_add(a4, b0, temp);
|
|
temp = mul_add(a3, b2, temp);
|
|
temp = mul_add(a2, b3, temp);
|
|
temp = mul_add(a0, b4, temp);
|
|
c[25] = mul_add(a1, b5, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a4, b1);
|
|
temp = mul_add(a3, b0, temp);
|
|
temp = mul_add(a2, b2, temp);
|
|
temp = mul_add(a0, b3, temp);
|
|
c[26] = mul_add(a1, b4, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a3, b1);
|
|
temp = mul_add(a2, b0, temp);
|
|
temp = mul_add(a0, b2, temp);
|
|
c[27] = mul_add(a1, b3, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a2, b1);
|
|
temp = mul_add(a0, b0, temp);
|
|
c[28] = mul_add(a1, b2, temp);
|
|
|
|
temp = _mm256_mullo_epi16(a0, b1);
|
|
c[29] = mul_add(a1, b0, temp);
|
|
|
|
c[30] = _mm256_mullo_epi16(a1, b1);
|
|
|
|
c[31] = _mm256_set_epi64x(0, 0, 0, 0);
|
|
}
|
|
|
|
static void transpose(__m256i *M) {
|
|
__m256i r0, r1, r2, r3, r4, r5, r6, r7, r8, r9, r10, r11;
|
|
__m256i temp, temp0, temp1, temp2;
|
|
|
|
r0 = _mm256_unpacklo_epi16(M[0], M[1]);
|
|
r1 = _mm256_unpacklo_epi16(M[2], M[3]);
|
|
r2 = _mm256_unpacklo_epi16(M[4], M[5]);
|
|
r3 = _mm256_unpacklo_epi16(M[6], M[7]);
|
|
r4 = _mm256_unpacklo_epi16(M[8], M[9]);
|
|
r5 = _mm256_unpacklo_epi16(M[10], M[11]);
|
|
r6 = _mm256_unpacklo_epi16(M[12], M[13]);
|
|
r7 = _mm256_unpacklo_epi16(M[14], M[15]);
|
|
|
|
temp = _mm256_unpacklo_epi32(r0, r1);
|
|
temp0 = _mm256_unpacklo_epi32(r2, r3);
|
|
temp1 = _mm256_unpacklo_epi32(r4, r5);
|
|
temp2 = _mm256_unpacklo_epi32(r6, r7);
|
|
|
|
r8 = _mm256_unpackhi_epi32(r0, r1);
|
|
r9 = _mm256_unpackhi_epi32(r2, r3);
|
|
r10 = _mm256_unpackhi_epi32(r4, r5);
|
|
r11 = _mm256_unpackhi_epi32(r6, r7);
|
|
|
|
r0 = _mm256_unpacklo_epi64(temp, temp0);
|
|
r2 = _mm256_unpackhi_epi64(temp, temp0);
|
|
r1 = _mm256_unpacklo_epi64(temp1, temp2);
|
|
r3 = _mm256_unpackhi_epi64(temp1, temp2);
|
|
|
|
temp = _mm256_unpackhi_epi16(M[0], M[1]);
|
|
temp0 = _mm256_unpackhi_epi16(M[2], M[3]);
|
|
temp1 = _mm256_unpackhi_epi16(M[4], M[5]);
|
|
temp2 = _mm256_unpackhi_epi16(M[6], M[7]);
|
|
|
|
r4 = _mm256_unpackhi_epi16(M[8], M[9]);
|
|
M[0] = _mm256_permute2f128_si256(r0, r1, 0x20);
|
|
M[8] = _mm256_permute2f128_si256(r0, r1, 0x31);
|
|
M[1] = _mm256_permute2f128_si256(r2, r3, 0x20);
|
|
M[9] = _mm256_permute2f128_si256(r2, r3, 0x31);
|
|
r5 = _mm256_unpackhi_epi16(M[10], M[11]);
|
|
r6 = _mm256_unpackhi_epi16(M[12], M[13]);
|
|
r7 = _mm256_unpackhi_epi16(M[14], M[15]);
|
|
|
|
r0 = _mm256_unpacklo_epi64(r8, r9);
|
|
r1 = _mm256_unpacklo_epi64(r10, r11);
|
|
r2 = _mm256_unpackhi_epi64(r8, r9);
|
|
r3 = _mm256_unpackhi_epi64(r10, r11);
|
|
|
|
M[3] = _mm256_permute2f128_si256(r2, r3, 0x20);
|
|
M[11] = _mm256_permute2f128_si256(r2, r3, 0x31);
|
|
M[2] = _mm256_permute2f128_si256(r0, r1, 0x20);
|
|
M[10] = _mm256_permute2f128_si256(r0, r1, 0x31);
|
|
|
|
r0 = _mm256_unpacklo_epi32(temp, temp0);
|
|
r1 = _mm256_unpacklo_epi32(temp1, temp2);
|
|
r2 = _mm256_unpacklo_epi32(r4, r5);
|
|
r3 = _mm256_unpacklo_epi32(r6, r7);
|
|
|
|
r8 = _mm256_unpacklo_epi64(r0, r1);
|
|
r10 = _mm256_unpackhi_epi64(r0, r1);
|
|
r9 = _mm256_unpacklo_epi64(r2, r3);
|
|
r11 = _mm256_unpackhi_epi64(r2, r3);
|
|
|
|
M[4] = _mm256_permute2f128_si256(r8, r9, 0x20);
|
|
M[12] = _mm256_permute2f128_si256(r8, r9, 0x31);
|
|
M[5] = _mm256_permute2f128_si256(r10, r11, 0x20);
|
|
M[13] = _mm256_permute2f128_si256(r10, r11, 0x31);
|
|
|
|
r0 = _mm256_unpackhi_epi32(temp, temp0);
|
|
r1 = _mm256_unpackhi_epi32(temp1, temp2);
|
|
r2 = _mm256_unpackhi_epi32(r4, r5);
|
|
r3 = _mm256_unpackhi_epi32(r6, r7);
|
|
|
|
r4 = _mm256_unpacklo_epi64(r0, r1);
|
|
r6 = _mm256_unpackhi_epi64(r0, r1);
|
|
r5 = _mm256_unpacklo_epi64(r2, r3);
|
|
r7 = _mm256_unpackhi_epi64(r2, r3);
|
|
|
|
M[6] = _mm256_permute2f128_si256(r4, r5, 0x20);
|
|
M[14] = _mm256_permute2f128_si256(r4, r5, 0x31);
|
|
M[7] = _mm256_permute2f128_si256(r6, r7, 0x20);
|
|
M[15] = _mm256_permute2f128_si256(r6, r7, 0x31);
|
|
}
|
|
|
|
static void batch_64coefficient_multiplications(toom4_points_product *c_eval, const __m256i *a, const toom4_points *b_eval, int accumulate) {
|
|
toom4_points a_eval;// Holds evaluation (a & b) for 7 Karatsuba at a time
|
|
__m256i r0_avx, r1_avx, r2_avx, r3_avx;
|
|
__m256i *va = (__m256i *)a_eval.coeffs;
|
|
__m256i *vb = (__m256i *)b_eval->coeffs;
|
|
__m256i *vc = (__m256i *)c_eval->coeffs;
|
|
|
|
//------------------AVX evaluation for 1st poly-----------------------
|
|
r0_avx = a[0 * L + 0];
|
|
r1_avx = a[0 * L + 1];
|
|
r2_avx = a[0 * L + 2];
|
|
r3_avx = a[0 * L + 3];
|
|
|
|
va[0] = r0_avx;
|
|
va[1] = r1_avx;
|
|
va[2] = r2_avx;
|
|
va[3] = r3_avx;
|
|
va[4] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8] = _mm256_add_epi16(va[6], va[7]);
|
|
//------------------AVX evaluation for 1st poly ends------------------
|
|
|
|
//------------------AVX evaluation for 2nd poly-----------------------
|
|
r0_avx = a[1 * L + 0];
|
|
r1_avx = a[1 * L + 1];
|
|
r2_avx = a[1 * L + 2];
|
|
r3_avx = a[1 * L + 3];
|
|
|
|
va[0 + 9] = r0_avx;
|
|
va[1 + 9] = r1_avx;
|
|
va[2 + 9] = r2_avx;
|
|
va[3 + 9] = r3_avx;
|
|
va[4 + 9] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 9] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 9] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 9] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 9] = _mm256_add_epi16(va[6 + 9], va[7 + 9]);
|
|
//------------------AVX evaluation for 2nd poly ends------------------
|
|
|
|
//------------------AVX evaluation for 3rd poly-----------------------
|
|
r0_avx = a[2 * L + 0];
|
|
r1_avx = a[2 * L + 1];
|
|
r2_avx = a[2 * L + 2];
|
|
r3_avx = a[2 * L + 3];
|
|
|
|
va[0 + 18] = r0_avx;
|
|
va[1 + 18] = r1_avx;
|
|
va[2 + 18] = r2_avx;
|
|
va[3 + 18] = r3_avx;
|
|
va[4 + 18] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 18] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 18] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 18] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 18] = _mm256_add_epi16(va[6 + 18], va[7 + 18]);
|
|
//------------------AVX evaluation for 3rd poly ends------------------
|
|
|
|
//------------------AVX evaluation for 4th poly-----------------------
|
|
r0_avx = a[3 * L + 0];
|
|
r1_avx = a[3 * L + 1];
|
|
r2_avx = a[3 * L + 2];
|
|
r3_avx = a[3 * L + 3];
|
|
|
|
va[0 + 27] = r0_avx;
|
|
va[1 + 27] = r1_avx;
|
|
va[2 + 27] = r2_avx;
|
|
va[3 + 27] = r3_avx;
|
|
va[4 + 27] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 27] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 27] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 27] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 27] = _mm256_add_epi16(va[6 + 27], va[7 + 27]);
|
|
//------------------AVX evaluation for 4th poly ends------------------
|
|
|
|
//------------------AVX evaluation for 5th poly-----------------------
|
|
r0_avx = a[4 * L + 0];
|
|
r1_avx = a[4 * L + 1];
|
|
r2_avx = a[4 * L + 2];
|
|
r3_avx = a[4 * L + 3];
|
|
|
|
va[0 + 36] = r0_avx;
|
|
va[1 + 36] = r1_avx;
|
|
va[2 + 36] = r2_avx;
|
|
va[3 + 36] = r3_avx;
|
|
va[4 + 36] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 36] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 36] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 36] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 36] = _mm256_add_epi16(va[6 + 36], va[7 + 36]);
|
|
//------------------AVX evaluation for 5th poly ends------------------
|
|
|
|
//------------------AVX evaluation for 6th poly-----------------------
|
|
r0_avx = a[5 * L + 0];
|
|
r1_avx = a[5 * L + 1];
|
|
r2_avx = a[5 * L + 2];
|
|
r3_avx = a[5 * L + 3];
|
|
|
|
va[0 + 45] = r0_avx;
|
|
va[1 + 45] = r1_avx;
|
|
va[2 + 45] = r2_avx;
|
|
va[3 + 45] = r3_avx;
|
|
va[4 + 45] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 45] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 45] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 45] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 45] = _mm256_add_epi16(va[6 + 45], va[7 + 45]);
|
|
//------------------AVX evaluation for 6th poly ends------------------
|
|
|
|
//------------------AVX evaluation for 7th poly-----------------------
|
|
r0_avx = a[6 * L + 0];
|
|
r1_avx = a[6 * L + 1];
|
|
r2_avx = a[6 * L + 2];
|
|
r3_avx = a[6 * L + 3];
|
|
|
|
va[0 + 54] = r0_avx;
|
|
va[1 + 54] = r1_avx;
|
|
va[2 + 54] = r2_avx;
|
|
va[3 + 54] = r3_avx;
|
|
va[4 + 54] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
va[5 + 54] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
va[6 + 54] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
va[7 + 54] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
va[8 + 54] = _mm256_add_epi16(va[6 + 54], va[7 + 54]);
|
|
//------------------AVX evaluation for 7th poly ends------------------
|
|
|
|
//-----------------Forward transposes--------------------------------------
|
|
transpose(va);
|
|
transpose(va + 16);
|
|
transpose(va + 32);
|
|
transpose(va + 48);
|
|
//-----------------Forward transposes ends---------------------------------
|
|
|
|
if (accumulate == 0) {
|
|
schoolbook_avx(vc, va, vb);
|
|
schoolbook_avx(vc + 32, va + 16, vb + 16);
|
|
schoolbook_avx(vc + 64, va + 32, vb + 32);
|
|
schoolbook_avx(vc + 96, va + 48, vb + 48);
|
|
} else {
|
|
schoolbook_avx_acc(vc, va, vb);
|
|
schoolbook_avx_acc(vc + 32, va + 16, vb + 16);
|
|
schoolbook_avx_acc(vc + 64, va + 32, vb + 32);
|
|
schoolbook_avx_acc(vc + 96, va + 48, vb + 48);
|
|
}
|
|
}
|
|
|
|
static void karatsuba_eval(__m256i *b_eval, const __m256i *b) {
|
|
__m256i r0_avx, r1_avx, r2_avx, r3_avx;
|
|
|
|
//-------1st poly----------------------------------------------------
|
|
r0_avx = b[0 * L + 0];
|
|
r1_avx = b[0 * L + 1];
|
|
r2_avx = b[0 * L + 2];
|
|
r3_avx = b[0 * L + 3];
|
|
|
|
b_eval[0] = r0_avx;
|
|
b_eval[1] = r1_avx;
|
|
b_eval[2] = r2_avx;
|
|
b_eval[3] = r3_avx;
|
|
b_eval[4] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8] = _mm256_add_epi16(b_eval[6], b_eval[7]);
|
|
|
|
//-------2nd poly----------------------------------------------------
|
|
r0_avx = b[1 * L + 0];
|
|
r1_avx = b[1 * L + 1];
|
|
r2_avx = b[1 * L + 2];
|
|
r3_avx = b[1 * L + 3];
|
|
|
|
b_eval[0 + 9] = r0_avx;
|
|
b_eval[1 + 9] = r1_avx;
|
|
b_eval[2 + 9] = r2_avx;
|
|
b_eval[3 + 9] = r3_avx;
|
|
b_eval[4 + 9] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 9] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 9] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 9] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 9] = _mm256_add_epi16(b_eval[6 + 9], b_eval[7 + 9]);
|
|
|
|
//-------3rd poly----------------------------------------------------
|
|
r0_avx = b[2 * L + 0];
|
|
r1_avx = b[2 * L + 1];
|
|
r2_avx = b[2 * L + 2];
|
|
r3_avx = b[2 * L + 3];
|
|
|
|
b_eval[0 + 18] = r0_avx;
|
|
b_eval[1 + 18] = r1_avx;
|
|
b_eval[2 + 18] = r2_avx;
|
|
b_eval[3 + 18] = r3_avx;
|
|
b_eval[4 + 18] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 18] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 18] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 18] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 18] = _mm256_add_epi16(b_eval[6 + 18], b_eval[7 + 18]);
|
|
|
|
//-------4th poly----------------------------------------------------
|
|
r0_avx = b[3 * L + 0];
|
|
r1_avx = b[3 * L + 1];
|
|
r2_avx = b[3 * L + 2];
|
|
r3_avx = b[3 * L + 3];
|
|
|
|
b_eval[0 + 27] = r0_avx;
|
|
b_eval[1 + 27] = r1_avx;
|
|
b_eval[2 + 27] = r2_avx;
|
|
b_eval[3 + 27] = r3_avx;
|
|
b_eval[4 + 27] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 27] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 27] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 27] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 27] = _mm256_add_epi16(b_eval[6 + 27], b_eval[7 + 27]);
|
|
|
|
//-------5th poly----------------------------------------------------
|
|
r0_avx = b[4 * L + 0];
|
|
r1_avx = b[4 * L + 1];
|
|
r2_avx = b[4 * L + 2];
|
|
r3_avx = b[4 * L + 3];
|
|
|
|
b_eval[0 + 36] = r0_avx;
|
|
b_eval[1 + 36] = r1_avx;
|
|
b_eval[2 + 36] = r2_avx;
|
|
b_eval[3 + 36] = r3_avx;
|
|
b_eval[4 + 36] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 36] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 36] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 36] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 36] = _mm256_add_epi16(b_eval[6 + 36], b_eval[7 + 36]);
|
|
|
|
//-------6th poly----------------------------------------------------
|
|
r0_avx = b[5 * L + 0];
|
|
r1_avx = b[5 * L + 1];
|
|
r2_avx = b[5 * L + 2];
|
|
r3_avx = b[5 * L + 3];
|
|
|
|
b_eval[0 + 45] = r0_avx;
|
|
b_eval[1 + 45] = r1_avx;
|
|
b_eval[2 + 45] = r2_avx;
|
|
b_eval[3 + 45] = r3_avx;
|
|
b_eval[4 + 45] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 45] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 45] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 45] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 45] = _mm256_add_epi16(b_eval[6 + 45], b_eval[7 + 45]);
|
|
|
|
//-------7th poly----------------------------------------------------
|
|
r0_avx = b[6 * L + 0];
|
|
r1_avx = b[6 * L + 1];
|
|
r2_avx = b[6 * L + 2];
|
|
r3_avx = b[6 * L + 3];
|
|
|
|
b_eval[0 + 54] = r0_avx;
|
|
b_eval[1 + 54] = r1_avx;
|
|
b_eval[2 + 54] = r2_avx;
|
|
b_eval[3 + 54] = r3_avx;
|
|
b_eval[4 + 54] = _mm256_add_epi16(r0_avx, r1_avx);
|
|
b_eval[5 + 54] = _mm256_add_epi16(r2_avx, r3_avx);
|
|
b_eval[6 + 54] = _mm256_add_epi16(r0_avx, r2_avx);
|
|
b_eval[7 + 54] = _mm256_add_epi16(r1_avx, r3_avx);
|
|
b_eval[8 + 54] = _mm256_add_epi16(b_eval[6 + 54], b_eval[7 + 54]);
|
|
|
|
//--------------Evaluating B poly ends-------------------------------
|
|
transpose(b_eval);
|
|
transpose(b_eval + 16);
|
|
transpose(b_eval + 32);
|
|
transpose(b_eval + 48);
|
|
}
|
|
|
|
static void karatsuba_interp(__m256i *result_final0, __m256i *result_final1, __m256i *result_final2, __m256i *result_final3, __m256i *result_final4, __m256i *result_final5, __m256i *result_final6, const __m256i *c_eval) {
|
|
__m256i res_avx0, res_avx1, res_avx2, res_avx3, res_avx4, res_avx5, res_avx6, res_avx7; // to hold each 64X64 poly mul results
|
|
__m256i temp, c6_avx, c7_avx, c8_avx, c20_avx, c21_avx, c22_avx, c23_avx, c24_avx;
|
|
|
|
//------------------------AVX interpolation for 1st poly external-------------------
|
|
res_avx0 = c_eval[0];
|
|
res_avx2 = c_eval[1];
|
|
res_avx4 = c_eval[2];
|
|
res_avx6 = c_eval[3];
|
|
c6_avx = c_eval[6];
|
|
c7_avx = c_eval[7];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[8], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[16];
|
|
res_avx3 = c_eval[17];
|
|
res_avx5 = c_eval[18];
|
|
res_avx7 = c_eval[19];
|
|
c22_avx = c_eval[22];
|
|
c23_avx = c_eval[23];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[21], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[24], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[20], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[5], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[4], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final0[0] = res_avx0;
|
|
result_final0[1] = res_avx1;
|
|
result_final0[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final0[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final0[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final0[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final0[6] = res_avx6;
|
|
result_final0[7] = res_avx7;
|
|
//------------------------AVX interpolation for 1st poly ends--------------
|
|
|
|
|
|
//------------------------AVX interpolation for 2nd poly external-------------------
|
|
res_avx0 = c_eval[9]; //c_eval0
|
|
res_avx2 = c_eval[10]; //c_eval1
|
|
res_avx4 = c_eval[11]; //c_eval2
|
|
res_avx6 = c_eval[12]; //c_eval3
|
|
c6_avx = c_eval[15]; //c_eval6
|
|
c7_avx = c_eval[32]; //c_eval7
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[33], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[25]; //c_eval0
|
|
res_avx3 = c_eval[26]; //c_eval1
|
|
res_avx5 = c_eval[27]; //c_eval2
|
|
res_avx7 = c_eval[28]; //c_eval3
|
|
c22_avx = c_eval[31];
|
|
c23_avx = c_eval[48];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[30], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[49], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[29], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[14], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[13], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final1[0] = res_avx0;
|
|
result_final1[1] = res_avx1;
|
|
result_final1[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final1[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final1[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final1[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final1[6] = res_avx6;
|
|
result_final1[7] = res_avx7;
|
|
//------------------------AVX interpolation for 2nd poly ends--------------
|
|
|
|
//------------------------AVX interpolation for 3rd poly external-------------------
|
|
res_avx0 = c_eval[34]; //c_eval0
|
|
res_avx2 = c_eval[35]; //c_eval1
|
|
res_avx4 = c_eval[36];
|
|
res_avx6 = c_eval[37];
|
|
c6_avx = c_eval[40];
|
|
c7_avx = c_eval[41];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[42], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[50]; //c_eval0
|
|
res_avx3 = c_eval[51]; //c_eval1
|
|
res_avx5 = c_eval[52];
|
|
res_avx7 = c_eval[53];
|
|
c22_avx = c_eval[56];
|
|
c23_avx = c_eval[57];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[55], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[58], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[54], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[39], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[38], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final2[0] = res_avx0;
|
|
result_final2[1] = res_avx1;
|
|
result_final2[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final2[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final2[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final2[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final2[6] = res_avx6;
|
|
result_final2[7] = res_avx7;
|
|
//------------------------AVX interpolation for 3rd poly ends--------------
|
|
|
|
//------------------------AVX interpolation for 4th poly external-------------------
|
|
res_avx0 = c_eval[43];
|
|
res_avx2 = c_eval[44];
|
|
res_avx4 = c_eval[45];
|
|
res_avx6 = c_eval[46];
|
|
c6_avx = c_eval[65];
|
|
c7_avx = c_eval[66];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[67], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[59];
|
|
res_avx3 = c_eval[60];
|
|
res_avx5 = c_eval[61];
|
|
res_avx7 = c_eval[62];
|
|
c22_avx = c_eval[81];
|
|
c23_avx = c_eval[82];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[80], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[83], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[63], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[64], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[47], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final3[0] = res_avx0;
|
|
result_final3[1] = res_avx1;
|
|
result_final3[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final3[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final3[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final3[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final3[6] = res_avx6;
|
|
result_final3[7] = res_avx7;
|
|
//------------------------AVX interpolation for 4th poly ends--------------
|
|
|
|
//------------------------AVX interpolation for 5th poly external-------------------
|
|
res_avx0 = c_eval[68];
|
|
res_avx2 = c_eval[69];
|
|
res_avx4 = c_eval[70];
|
|
res_avx6 = c_eval[71];
|
|
c6_avx = c_eval[74];
|
|
c7_avx = c_eval[75];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[76], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[84];
|
|
res_avx3 = c_eval[85];
|
|
res_avx5 = c_eval[86];
|
|
res_avx7 = c_eval[87];
|
|
c22_avx = c_eval[90];
|
|
c23_avx = c_eval[91];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[89], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[92], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[88], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[73], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[72], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final4[0] = res_avx0;
|
|
result_final4[1] = res_avx1;
|
|
result_final4[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final4[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final4[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final4[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final4[6] = res_avx6;
|
|
result_final4[7] = res_avx7;
|
|
//------------------------AVX interpolation for 5th poly ends--------------
|
|
|
|
//------------------------AVX interpolation for 6th poly external-------------------
|
|
res_avx0 = c_eval[77];
|
|
res_avx2 = c_eval[78];
|
|
res_avx4 = c_eval[79];
|
|
res_avx6 = c_eval[96];
|
|
c6_avx = c_eval[99];
|
|
c7_avx = c_eval[100];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[101], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[93];
|
|
res_avx3 = c_eval[94];
|
|
res_avx5 = c_eval[95];
|
|
res_avx7 = c_eval[112];
|
|
c22_avx = c_eval[115];
|
|
c23_avx = c_eval[116];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[114], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[117], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[113], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[98], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[97], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final5[0] = res_avx0;
|
|
result_final5[1] = res_avx1;
|
|
result_final5[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final5[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final5[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final5[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final5[6] = res_avx6;
|
|
result_final5[7] = res_avx7;
|
|
//------------------------AVX interpolation for 6th poly ends--------------
|
|
|
|
//------------------------AVX interpolation for 7th poly external-------------------
|
|
res_avx0 = c_eval[102];
|
|
res_avx2 = c_eval[103];
|
|
res_avx4 = c_eval[104];
|
|
res_avx6 = c_eval[105];
|
|
c6_avx = c_eval[108];
|
|
c7_avx = c_eval[109];
|
|
|
|
c8_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[110], c6_avx), c7_avx);
|
|
|
|
res_avx1 = c_eval[118];
|
|
res_avx3 = c_eval[119];
|
|
res_avx5 = c_eval[120];
|
|
res_avx7 = c_eval[121];
|
|
c22_avx = c_eval[124];
|
|
c23_avx = c_eval[125];
|
|
|
|
c21_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[123], res_avx5), res_avx7);
|
|
c24_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[126], c22_avx), c23_avx);
|
|
c20_avx = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[122], res_avx1), res_avx3);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[107], res_avx4), res_avx6);
|
|
res_avx5 = _mm256_add_epi16(res_avx5, temp);
|
|
temp = _mm256_sub_epi16(_mm256_sub_epi16(c_eval[106], res_avx0), res_avx2);
|
|
res_avx1 = _mm256_add_epi16(res_avx1, temp);
|
|
c22_avx = _mm256_add_epi16(c22_avx, c8_avx);
|
|
res_avx6 = _mm256_add_epi16(res_avx6, c21_avx);
|
|
res_avx2 = _mm256_add_epi16(res_avx2, c20_avx);
|
|
c7_avx = _mm256_add_epi16(c7_avx, c24_avx);
|
|
c6_avx = _mm256_sub_epi16(_mm256_sub_epi16(c6_avx, res_avx0), res_avx4);
|
|
c22_avx = _mm256_sub_epi16(_mm256_sub_epi16(c22_avx, res_avx1), res_avx5);
|
|
c7_avx = _mm256_sub_epi16(_mm256_sub_epi16(c7_avx, res_avx2), res_avx6);
|
|
c23_avx = _mm256_sub_epi16(_mm256_sub_epi16(c23_avx, res_avx3), res_avx7);
|
|
|
|
result_final6[0] = res_avx0;
|
|
result_final6[1] = res_avx1;
|
|
result_final6[2] = _mm256_add_epi16(res_avx2, c6_avx);
|
|
result_final6[3] = _mm256_add_epi16(res_avx3, c22_avx);
|
|
result_final6[4] = _mm256_add_epi16(res_avx4, c7_avx);
|
|
result_final6[5] = _mm256_add_epi16(res_avx5, c23_avx);
|
|
result_final6[6] = res_avx6;
|
|
result_final6[7] = res_avx7;
|
|
//------------------------AVX interpolation for 7th poly ends--------------
|
|
}
|
|
|
|
void PQCLEAN_FIRESABER_AVX2_toom4_mul_A_by_B_eval(toom4_points_product *c_eval, const poly *a, const toom4_points *b_eval, int accumulate) {
|
|
size_t i;
|
|
__m256i r0_avx, r1_avx, r2_avx, r3_avx, r4_avx, r5_avx, r6_avx;
|
|
__m256i aw_avx[7 * L];
|
|
__m256i *va = (__m256i *)a->coeffs;
|
|
|
|
for (i = 0; i < L; i++) {
|
|
r0_avx = va[0 * L + i];
|
|
r1_avx = va[1 * L + i];
|
|
r2_avx = va[2 * L + i];
|
|
r3_avx = va[3 * L + i];
|
|
r4_avx = _mm256_add_epi16(r0_avx, r2_avx);
|
|
r5_avx = _mm256_add_epi16(r1_avx, r3_avx);
|
|
aw_avx[2 * L + i] = _mm256_add_epi16(r4_avx, r5_avx);
|
|
aw_avx[3 * L + i] = _mm256_sub_epi16(r4_avx, r5_avx);
|
|
r4_avx = _mm256_slli_epi16(r0_avx, 2);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r2_avx);
|
|
r4_avx = _mm256_slli_epi16(r4_avx, 1);
|
|
r5_avx = _mm256_slli_epi16(r1_avx, 2);
|
|
r5_avx = _mm256_add_epi16(r5_avx, r3_avx);
|
|
aw_avx[4 * L + i] = _mm256_add_epi16(r4_avx, r5_avx);
|
|
aw_avx[5 * L + i] = _mm256_sub_epi16(r4_avx, r5_avx);
|
|
r4_avx = _mm256_slli_epi16(r3_avx, 3);
|
|
r6_avx = _mm256_slli_epi16(r2_avx, 2);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r6_avx);
|
|
r6_avx = _mm256_slli_epi16(r1_avx, 1);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r6_avx);
|
|
aw_avx[1 * L + i] = _mm256_add_epi16(r4_avx, r0_avx);
|
|
aw_avx[6 * L + i] = r0_avx;
|
|
aw_avx[0 * L + i] = r3_avx;
|
|
}
|
|
|
|
batch_64coefficient_multiplications(c_eval, aw_avx, b_eval, accumulate);
|
|
}
|
|
|
|
void PQCLEAN_FIRESABER_AVX2_toom4_eval(toom4_points *b_eval, const poly *b) {
|
|
size_t i;
|
|
__m256i bw_avx[7 * L];
|
|
__m256i r0_avx, r1_avx, r2_avx, r3_avx, r4_avx, r5_avx, r6_avx;
|
|
__m256i *vb = (__m256i *)b->coeffs;
|
|
__m256i *vb_eval = (__m256i *)b_eval->coeffs;
|
|
|
|
for (i = 0; i < L; i++) {
|
|
r0_avx = vb[0 * L + i];
|
|
r1_avx = vb[1 * L + i];
|
|
r2_avx = vb[2 * L + i];
|
|
r3_avx = vb[3 * L + i];
|
|
r4_avx = _mm256_add_epi16(r0_avx, r2_avx);
|
|
r5_avx = _mm256_add_epi16(r1_avx, r3_avx);
|
|
bw_avx[2 * L + i] = _mm256_add_epi16(r4_avx, r5_avx);
|
|
bw_avx[3 * L + i] = _mm256_sub_epi16(r4_avx, r5_avx);
|
|
r4_avx = _mm256_slli_epi16(r0_avx, 2);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r2_avx);
|
|
r4_avx = _mm256_slli_epi16(r4_avx, 1);
|
|
r5_avx = _mm256_slli_epi16(r1_avx, 2);
|
|
r5_avx = _mm256_add_epi16(r5_avx, r3_avx);
|
|
bw_avx[4 * L + i] = _mm256_add_epi16(r4_avx, r5_avx);
|
|
bw_avx[5 * L + i] = _mm256_sub_epi16(r4_avx, r5_avx);
|
|
r4_avx = _mm256_slli_epi16(r3_avx, 3);
|
|
r6_avx = _mm256_slli_epi16(r2_avx, 2);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r6_avx);
|
|
r6_avx = _mm256_slli_epi16(r1_avx, 1);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r6_avx);
|
|
bw_avx[1 * L + i] = _mm256_add_epi16(r4_avx, r0_avx);
|
|
bw_avx[6 * L + i] = r0_avx;
|
|
bw_avx[0 * L + i] = r3_avx;
|
|
}
|
|
|
|
karatsuba_eval(vb_eval, bw_avx);
|
|
}
|
|
|
|
|
|
void PQCLEAN_FIRESABER_AVX2_toom4_interp(poly *res, const toom4_points_product *c_eval) {
|
|
size_t i;
|
|
__m256i r0_avx, r1_avx, r2_avx, r3_avx, r4_avx, r5_avx, r6_avx, temp_avx;
|
|
__m256i w1_avx[2 * L], w2_avx[2 * L], w3_avx[2 * L], w4_avx[2 * L], w5_avx[2 * L], w6_avx[2 * L], w7_avx[2 * L];
|
|
__m256i res_full[32];
|
|
__m256i *vc = (__m256i *)c_eval->coeffs;
|
|
__m256i *vres = (__m256i *)res->coeffs;
|
|
|
|
transpose(vc);
|
|
transpose(vc + 16);
|
|
transpose(vc + 32);
|
|
transpose(vc + 48);
|
|
transpose(vc + 64);
|
|
transpose(vc + 80);
|
|
transpose(vc + 96);
|
|
transpose(vc + 112);
|
|
|
|
karatsuba_interp(w1_avx, w2_avx, w3_avx, w4_avx, w5_avx, w6_avx, w7_avx, vc);
|
|
|
|
for (i = 0; i < 2 * L; i++) {
|
|
r0_avx = w1_avx[i];
|
|
r1_avx = w2_avx[i];
|
|
r2_avx = w3_avx[i];
|
|
r3_avx = w4_avx[i];
|
|
r4_avx = w5_avx[i];
|
|
r5_avx = w6_avx[i];
|
|
r6_avx = w7_avx[i];
|
|
|
|
r1_avx = _mm256_add_epi16(r1_avx, r4_avx);
|
|
r5_avx = _mm256_sub_epi16(r5_avx, r4_avx);
|
|
r3_avx = _mm256_sub_epi16(r3_avx, r2_avx);
|
|
r3_avx = _mm256_srli_epi16(r3_avx, 1);
|
|
r4_avx = _mm256_sub_epi16(r4_avx, r0_avx);
|
|
temp_avx = _mm256_slli_epi16(r6_avx, 6);
|
|
|
|
r4_avx = _mm256_sub_epi16(r4_avx, temp_avx);
|
|
r4_avx = _mm256_slli_epi16(r4_avx, 1);
|
|
r4_avx = _mm256_add_epi16(r4_avx, r5_avx);
|
|
r2_avx = _mm256_add_epi16(r2_avx, r3_avx);
|
|
temp_avx = _mm256_slli_epi16(r2_avx, 6);
|
|
|
|
r1_avx = _mm256_sub_epi16(r1_avx, temp_avx);
|
|
r1_avx = _mm256_sub_epi16(r1_avx, r2_avx);
|
|
r2_avx = _mm256_sub_epi16(r2_avx, r6_avx);
|
|
r2_avx = _mm256_sub_epi16(r2_avx, r0_avx);
|
|
temp_avx = _mm256_mullo_epi16(r2_avx, _mm256_set1_epi16(45));
|
|
|
|
r1_avx = _mm256_add_epi16(r1_avx, temp_avx);
|
|
temp_avx = _mm256_slli_epi16(r2_avx, 3);
|
|
|
|
r4_avx = _mm256_sub_epi16(r4_avx, temp_avx);
|
|
r4_avx = _mm256_mullo_epi16(r4_avx, _mm256_set1_epi16(-21845)); // -21845 = 1/3 (mod 2^16)
|
|
r4_avx = _mm256_srli_epi16(r4_avx, 3);
|
|
r5_avx = _mm256_add_epi16(r5_avx, r1_avx);
|
|
temp_avx = _mm256_slli_epi16(r3_avx, 4);
|
|
|
|
r1_avx = _mm256_add_epi16(r1_avx, temp_avx);
|
|
r1_avx = _mm256_mullo_epi16(r1_avx, _mm256_set1_epi16(-29127)); // -29127 = 1/9 (mod 2^16)
|
|
r1_avx = _mm256_srli_epi16(r1_avx, 1);
|
|
r3_avx = _mm256_add_epi16(r1_avx, r3_avx);
|
|
r3_avx = _mm256_sub_epi16(_mm256_set1_epi16(0), r3_avx);
|
|
temp_avx = _mm256_mullo_epi16(r1_avx, _mm256_set1_epi16(30));
|
|
temp_avx = _mm256_sub_epi16(temp_avx, r5_avx);
|
|
temp_avx = _mm256_mullo_epi16(temp_avx, _mm256_set1_epi16(-4369)); // -4369 = 1/15 (mod 2^16)
|
|
|
|
r5_avx = _mm256_srli_epi16(temp_avx, 2);
|
|
r2_avx = _mm256_sub_epi16(r2_avx, r4_avx);
|
|
r1_avx = _mm256_sub_epi16(r1_avx, r5_avx);
|
|
|
|
if (i < L) {
|
|
res_full[0 * L + i] = r6_avx;
|
|
res_full[1 * L + i] = r5_avx;
|
|
res_full[2 * L + i] = r4_avx;
|
|
res_full[3 * L + i] = r3_avx;
|
|
res_full[4 * L + i] = r2_avx;
|
|
res_full[5 * L + i] = r1_avx;
|
|
res_full[6 * L + i] = r0_avx;
|
|
} else {
|
|
res_full[0 * L + i] = _mm256_add_epi16(res_full[0 * L + i], r6_avx);
|
|
res_full[1 * L + i] = _mm256_add_epi16(res_full[1 * L + i], r5_avx);
|
|
res_full[2 * L + i] = _mm256_add_epi16(res_full[2 * L + i], r4_avx);
|
|
res_full[3 * L + i] = _mm256_add_epi16(res_full[3 * L + i], r3_avx);
|
|
res_full[4 * L + i] = _mm256_add_epi16(res_full[4 * L + i], r2_avx);
|
|
res_full[5 * L + i] = _mm256_add_epi16(res_full[5 * L + i], r1_avx);
|
|
res_full[6 * L + i] = r0_avx;
|
|
}
|
|
}
|
|
|
|
// Reduction by X^256 + 1
|
|
for (i = 0; i < 16; i++) {
|
|
vres[i] = _mm256_sub_epi16(res_full[i], res_full[i + 16]);
|
|
}
|
|
}
|