kris/pqc
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bench ntt

Browse Source
This commit is contained in:
Henry Case 2021-05-09 18:11:18 +01:00
parent 395896dc92
commit 895d9c0abd
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
test/bench/kyber.cc
View File

@ -92,7 +92,7 @@ static void BenchKyberDecaps(benchmark::State &st) {
cpucycle(st, total); cpucycle(st, total);
} }
static void BenchBaseMulAVX(benchmark::State &st) { static void BenchKyberBaseMulAVX(benchmark::State &st) {
int64_t t, total = 0; int64_t t, total = 0;
__m256i r[32],a[32],b[32],data[32]; __m256i r[32],a[32],b[32],data[32];
@ -105,9 +105,22 @@ static void BenchBaseMulAVX(benchmark::State &st) {
cpucycle(st, total); cpucycle(st, total);
} }
static void BenchKyberNttAVX(benchmark::State &st) {
int64_t t, total = 0;
__m256i r[32],data[32];
for (auto _ : st) {
t = benchmark::cycleclock::Now();
PQCLEAN_KYBER512_AVX2_ntt_avx(r, data);
total += benchmark::cycleclock::Now() - t;
benchmark::DoNotOptimize(r);
}
cpucycle(st, total);
}
BENCHMARK(BenchKyberMatK2); BENCHMARK(BenchKyberMatK2);
BENCHMARK(BenchKyberRejSampling); BENCHMARK(BenchKyberRejSampling);
BENCHMARK(BenchKyberKeygen); BENCHMARK(BenchKyberKeygen);
BENCHMARK(BenchKyberEncaps); BENCHMARK(BenchKyberEncaps);
BENCHMARK(BenchKyberDecaps); BENCHMARK(BenchKyberDecaps);
BENCHMARK(BenchBaseMulAVX); BENCHMARK(BenchKyberBaseMulAVX);
BENCHMARK(BenchKyberNttAVX);