diff --git a/test/bench/kyber.cc b/test/bench/kyber.cc index d9ab6ec5..c571915f 100644 --- a/test/bench/kyber.cc +++ b/test/bench/kyber.cc @@ -10,6 +10,7 @@ extern "C" { #include "kem/kyber/kyber512/avx2/indcpa.h" #include "kem/kyber/kyber512/avx2/kem.h" + #include "kem/kyber/kyber512/avx2/rejsample.h" } auto cpucycle = [](benchmark::State &st, int64_t cycles) { @@ -30,6 +31,19 @@ static void BenchKyberMatK2(benchmark::State &st) { cpucycle(st, total); } +static void BenchKyberRejSampling(benchmark::State &st) { + int64_t t, total = 0; + int16_t a[256]; + uint8_t seed[168*3]; + for (auto _ : st) { + t = benchmark::cycleclock::Now(); + PQCLEAN_KYBER512_AVX2_rej_uniform_avx(a, seed); + total += benchmark::cycleclock::Now() - t; + benchmark::DoNotOptimize(a); + } + cpucycle(st, total); +} + static void BenchKyberKeygen(benchmark::State &st) { int64_t t, total = 0; uint8_t sk[1632]; @@ -77,6 +91,7 @@ static void BenchKyberDecaps(benchmark::State &st) { cpucycle(st, total); } BENCHMARK(BenchKyberMatK2); +BENCHMARK(BenchKyberRejSampling); BENCHMARK(BenchKyberKeygen); BENCHMARK(BenchKyberEncaps); BENCHMARK(BenchKyberDecaps);