소스 검색

Add bench for rejection sampling

tags/v0.0.1
Henry Case 3 년 전
부모
커밋
128b5406cc
1개의 변경된 파일15개의 추가작업 그리고 0개의 파일을 삭제
  1. +15
    -0
      test/bench/kyber.cc

+ 15
- 0
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);

불러오는 중...
취소
저장