fix: tweak bls/bls_dkg benchmarks to work faster in -sanity-check mode

UdjinM6 · kwvg · commit 61bba13bc220 · 2025-07-16T11:56:40.000Z
diff --git a/src/bench/bls.cpp b/src/bench/bls.cpp
@@ -49,7 +49,7 @@ static void BLS_PubKeyAggregate_Normal(benchmark::Bench& bench)
     CBLSPublicKey pubKey2 = secKey2.GetPublicKey();
 
     // Benchmark.
-    bench.minEpochIterations(100).run([&] {
+    bench.minEpochIterations(bench.output() ? 100 : 1).run([&] {
         pubKey1.AggregateInsecure(pubKey2);
     });
 }
@@ -126,10 +126,10 @@ static void BLS_Verify_LargeBlock(size_t txCount, benchmark::Bench& bench, uint3
     std::vector<CBLSSignature> sigs;
     std::vector<uint256> msgHashes;
     std::vector<bool> invalid;
-    BuildTestVectors(txCount, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
+    BuildTestVectors(bench.output() ? txCount : 1, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
 
     // Benchmark.
-    bench.minEpochIterations(epoch_iters).run([&] {
+    bench.minEpochIterations(bench.output() ? epoch_iters : 1).run([&] {
         for (size_t i = 0; i < pubKeys.size(); i++) {
             bool ok = sigs[i].VerifyInsecure(pubKeys[i], msgHashes[i]);
             assert(ok);
@@ -154,10 +154,10 @@ static void BLS_Verify_LargeBlockSelfAggregated(size_t txCount, benchmark::Bench
     std::vector<CBLSSignature> sigs;
     std::vector<uint256> msgHashes;
     std::vector<bool> invalid;
-    BuildTestVectors(txCount, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
+    BuildTestVectors(bench.output() ? txCount : 1, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
 
     // Benchmark.
-    bench.minEpochIterations(epoch_iters).run([&] {
+    bench.minEpochIterations(bench.output() ? epoch_iters : 1).run([&] {
         CBLSSignature aggSig = CBLSSignature::AggregateInsecure(sigs);
         bool ok = aggSig.VerifyInsecureAggregated(pubKeys, msgHashes);
         assert(ok);
@@ -181,12 +181,12 @@ static void BLS_Verify_LargeAggregatedBlock(size_t txCount, benchmark::Bench& be
     std::vector<CBLSSignature> sigs;
     std::vector<uint256> msgHashes;
     std::vector<bool> invalid;
-    BuildTestVectors(txCount, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
+    BuildTestVectors(bench.output() ? txCount : 1, 0, pubKeys, secKeys, sigs, msgHashes, invalid);
 
     CBLSSignature aggSig = CBLSSignature::AggregateInsecure(sigs);
 
     // Benchmark.
-    bench.minEpochIterations(epoch_iters).run([&] {
+    bench.minEpochIterations(bench.output() ? epoch_iters : 1).run([&] {
         bool ok = aggSig.VerifyInsecureAggregated(pubKeys, msgHashes);
         assert(ok);
     });
@@ -224,7 +224,7 @@ static void BLS_Verify_LargeAggregatedBlock1000PreVerified(benchmark::Bench& ben
     }
 
     // Benchmark.
-    bench.minEpochIterations(10).run([&] {
+    bench.minEpochIterations(bench.output() ? 10 : 1).run([&] {
         std::vector<CBLSPublicKey> nonvalidatedPubKeys;
         std::vector<uint256> nonvalidatedHashes;
         nonvalidatedPubKeys.reserve(pubKeys.size());
@@ -255,13 +255,13 @@ static void BLS_Verify_Batched(benchmark::Bench& bench)
     std::vector<CBLSSignature> sigs;
     std::vector<uint256> msgHashes;
     std::vector<bool> invalid;
-    BuildTestVectors(1000, 10, pubKeys, secKeys, sigs, msgHashes, invalid);
+    BuildTestVectors(bench.output() ? 1000 : 1, 10, pubKeys, secKeys, sigs, msgHashes, invalid);
 
     // Benchmark.
     size_t i = 0;
     size_t j = 0;
     size_t batchSize = 16;
-    bench.minEpochIterations(1000).run([&] {
+    bench.minEpochIterations(bench.output() ? 1000 : 1).run([&] {
         j++;
         if ((j % batchSize) != 0) {
             return;
@@ -311,7 +311,7 @@ static void BLS_Verify_BatchedParallel(benchmark::Bench& bench)
     std::vector<CBLSSignature> sigs;
     std::vector<uint256> msgHashes;
     std::vector<bool> invalid;
-    BuildTestVectors(1000, 10, pubKeys, secKeys, sigs, msgHashes, invalid);
+    BuildTestVectors(bench.output() ? 1000 : 1, 10, pubKeys, secKeys, sigs, msgHashes, invalid);
 
     std::list<std::pair<size_t, std::future<bool>>> futures;
 
@@ -324,7 +324,7 @@ static void BLS_Verify_BatchedParallel(benchmark::Bench& bench)
     blsWorker.Start();
 
     // Benchmark.
-    bench.minEpochIterations(1000).run([&] {
+    bench.minEpochIterations(bench.output() ? 1000 : 1).run([&] {
         if (futures.size() < 100) {
             while (futures.size() < 10000) {
                 size_t i = 0;
diff --git a/src/bench/bls_dkg.cpp b/src/bench/bls_dkg.cpp
@@ -113,6 +113,10 @@ static void BLSDKG_GenerateContributions(benchmark::Bench& bench, uint32_t epoch
     blsWorker.Start();
     std::vector<CBLSId> ids;
     std::vector<Member> members;
+    if (!bench.output()) {
+        epoch_iters = 1;
+        quorumSize = 1;
+    }
     for (const int i : irange::range(quorumSize)) {
         uint256 id;
         WriteLE64(id.begin(), i + 1);
@@ -136,6 +140,10 @@ static void BLSDKG_GenerateContributions(benchmark::Bench& bench, uint32_t epoch
 
 static void BLSDKG_InitDKG(benchmark::Bench& bench, uint32_t epoch_iters, int quorumSize)
 {
+    if (!bench.output()) {
+        epoch_iters = 1;
+        quorumSize = 1;
+    }
     bench.minEpochIterations(epoch_iters).run([&] {
         DKG d(quorumSize);
     });
@@ -151,6 +159,12 @@ static void BLSDKG_InitDKG(benchmark::Bench& bench, uint32_t epoch_iters, int qu
 #define BENCH_BuildQuorumVerificationVectors(name, quorumSize, epoch_iters) \
     static void BLSDKG_BuildQuorumVerificationVectors_##name##_##quorumSize(benchmark::Bench& bench) \
     { \
+        if (!bench.output()) { \
+            std::unique_ptr<DKG> ptr = std::make_unique<DKG>(1); \
+            ptr->Bench_BuildQuorumVerificationVectors(bench, 1); \
+            ptr.reset(); \
+            return; \
+        } \
         std::unique_ptr<DKG> ptr = std::make_unique<DKG>(quorumSize); \
         ptr->Bench_BuildQuorumVerificationVectors(bench, epoch_iters); \
         ptr.reset(); \
@@ -160,9 +174,15 @@ static void BLSDKG_InitDKG(benchmark::Bench& bench, uint32_t epoch_iters, int qu
 #define BENCH_VerifyContributionShares(name, quorumSize, invalidCount, aggregated, epoch_iters) \
     static void BLSDKG_VerifyContributionShares_##name##_##quorumSize(benchmark::Bench& bench) \
     { \
-      std::unique_ptr<DKG> ptr = std::make_unique<DKG>(quorumSize); \
-      ptr->Bench_VerifyContributionShares(bench, invalidCount, aggregated, epoch_iters); \
-      ptr.reset(); \
+        if (!bench.output()) { \
+            std::unique_ptr<DKG> ptr = std::make_unique<DKG>(1); \
+            ptr->Bench_VerifyContributionShares(bench, invalidCount, aggregated, 1); \
+            ptr.reset(); \
+            return; \
+        } \
+        std::unique_ptr<DKG> ptr = std::make_unique<DKG>(quorumSize); \
+        ptr->Bench_VerifyContributionShares(bench, invalidCount, aggregated, epoch_iters); \
+        ptr.reset(); \
     } \
     BENCHMARK(BLSDKG_VerifyContributionShares_##name##_##quorumSize)
 
