From 895d9c0abd66af3a2c2f5de5ed4a2305a33948f5 Mon Sep 17 00:00:00 2001
From: Kris Kwiatkowski <contact@amongbytes.com>
Date: Sun, 9 May 2021 18:11:18 +0100
Subject: [PATCH] bench ntt

---
 test/bench/kyber.cc | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/test/bench/kyber.cc b/test/bench/kyber.cc
index 53ae8c23..6657b4fb 100644
--- a/test/bench/kyber.cc
+++ b/test/bench/kyber.cc
@@ -92,7 +92,7 @@ static void BenchKyberDecaps(benchmark::State &st) {
     cpucycle(st, total);
 }
 
-static void BenchBaseMulAVX(benchmark::State &st) {
+static void BenchKyberBaseMulAVX(benchmark::State &st) {
     int64_t t, total = 0;
     __m256i r[32],a[32],b[32],data[32];
 
@@ -105,9 +105,22 @@ static void BenchBaseMulAVX(benchmark::State &st) {
     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(BenchKyberRejSampling);
 BENCHMARK(BenchKyberKeygen);
 BENCHMARK(BenchKyberEncaps);
 BENCHMARK(BenchKyberDecaps);
-BENCHMARK(BenchBaseMulAVX);
+BENCHMARK(BenchKyberBaseMulAVX);
+BENCHMARK(BenchKyberNttAVX);