aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/restricted/aws/s2n/pq-crypto/kyber_r3/kyber512r3_ntt_avx2.h
blob: 361613235801b7ed301f0a83416c8f1decd7ea83 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#pragma once

#include <stdint.h>

#if defined(S2N_KYBER512R3_AVX2_BMI2)
#include <immintrin.h>

#define ntt_avx2_asm S2N_KYBER_512_R3_NAMESPACE(ntt_avx2_asm)
void ntt_avx2_asm(__m256i *r, const __m256i *qdata);

#define invntt_avx2_asm S2N_KYBER_512_R3_NAMESPACE(invntt_avx2_asm)
void invntt_avx2_asm(__m256i *r, const __m256i *qdata);

#define nttunpack_avx2_asm S2N_KYBER_512_R3_NAMESPACE(nttunpack_avx2_asm)
void nttunpack_avx2_asm(__m256i *r, const __m256i *qdata);

#define basemul_avx2_asm S2N_KYBER_512_R3_NAMESPACE(basemul_avx2_asm)
void basemul_avx2_asm(__m256i *r,
                 const __m256i *a,
                 const __m256i *b,
                 const __m256i *qdata);

#define ntttobytes_avx2_asm S2N_KYBER_512_R3_NAMESPACE(ntttobytes_avx2_asm)
void ntttobytes_avx2_asm(uint8_t *r, const __m256i *a, const __m256i *qdata);

#define nttfrombytes_avx2_asm S2N_KYBER_512_R3_NAMESPACE(nttfrombytes_avx2_asm)
void nttfrombytes_avx2_asm(__m256i *r, const uint8_t *a, const __m256i *qdata);
#endif