From c1727a7b498ac2685c5abef574e3c548b9939da2 Mon Sep 17 00:00:00 2001
From: Weikeng Chen <w.k@berkeley.edu>
Date: Wed, 28 Jun 2023 01:31:17 -0700
Subject: [PATCH] Add the microbenchmark for Anemoi (#314)

* add the microbenchmark for Anemoi

* add the code

* full ver

* undelete
---
 api/Cargo.toml        |   5 ++
 api/benches/anemoi.rs | 107 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+)
 create mode 100644 api/benches/anemoi.rs

diff --git a/api/Cargo.toml b/api/Cargo.toml
index a5f664fb..1719e3ff 100644
--- a/api/Cargo.toml
+++ b/api/Cargo.toml
@@ -23,6 +23,11 @@ name = 'merkle_tree'
 path = 'benches/merkle_tree.rs'
 harness = false
 
+[[bench]]
+name = 'anemoi'
+path = 'benches/anemoi.rs'
+harness = false
+
 [package]
 name = 'noah'
 version = '0.5.0'
diff --git a/api/benches/anemoi.rs b/api/benches/anemoi.rs
new file mode 100644
index 00000000..3b233ff1
--- /dev/null
+++ b/api/benches/anemoi.rs
@@ -0,0 +1,107 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use merlin::Transcript;
+use noah_algebra::bn254::{BN254PairingEngine, BN254Scalar};
+use noah_algebra::prelude::*;
+use noah_crypto::anemoi_jive::{AnemoiJive, AnemoiJive254, ANEMOI_JIVE_BN254_SALTS};
+use noah_plonk::plonk::constraint_system::{ConstraintSystem, TurboCS};
+use noah_plonk::plonk::indexer::{indexer, indexer_with_lagrange, PlonkVK};
+use noah_plonk::plonk::prover::prover;
+use noah_plonk::plonk::verifier::verifier;
+use noah_plonk::poly_commit::kzg_poly_com::KZGCommitmentScheme;
+use noah_plonk::poly_commit::pcs::PolyComScheme;
+
+fn anemoi(c: &mut Criterion) {
+    let mut cs = TurboCS::new();
+    cs.load_anemoi_jive_parameters::<AnemoiJive254>();
+
+    let va = cs.new_variable(BN254Scalar::from(1u64));
+    let vb = cs.new_variable(BN254Scalar::from(2u64));
+    let vc = cs.new_variable(BN254Scalar::from(3u64));
+
+    let trace = AnemoiJive254::eval_jive_with_trace(
+        &[BN254Scalar::from(1u64), BN254Scalar::from(2u64)],
+        &[BN254Scalar::from(3u64), ANEMOI_JIVE_BN254_SALTS[0]],
+    );
+
+    for _ in 0..500 {
+        let _ = cs.jive_crh::<AnemoiJive254>(&trace, &[va, vb, vc], ANEMOI_JIVE_BN254_SALTS[0]);
+    }
+    println!("number of constraints: {}", cs.size);
+    cs.pad();
+    let witness = cs.get_and_clear_witness();
+
+    let mut prng = test_rng();
+    let pcs = KZGCommitmentScheme::<BN254PairingEngine>::new(8195, &mut prng);
+
+    let prover_params = indexer(&cs, &pcs).unwrap();
+    let verifier_params = prover_params.get_verifier_params_ref();
+
+    let mut single_group = c.benchmark_group("prover");
+    single_group.sample_size(10);
+    single_group.bench_function("batch of 500".to_string(), |b| {
+        b.iter(|| {
+            let mut transcript = Transcript::new(b"TestTurboPlonk");
+            prover(
+                &mut prng,
+                &mut transcript,
+                &pcs,
+                &cs,
+                &prover_params,
+                &witness,
+            )
+            .unwrap()
+        });
+    });
+    single_group.finish();
+
+    let mut transcript = Transcript::new(b"TestTurboPlonk");
+    let proof = prover(
+        &mut prng,
+        &mut transcript,
+        &pcs,
+        &cs,
+        &prover_params,
+        &witness,
+    )
+    .unwrap();
+
+    let mut single_group = c.benchmark_group("verifier");
+    single_group.sample_size(10);
+    single_group.bench_function("batch of 500".to_string(), |b| {
+        b.iter(|| {
+            let mut transcript = Transcript::new(b"TestTurboPlonk");
+            verifier(
+                &mut transcript,
+                &pcs.shrink_to_verifier_only(),
+                &cs.shrink_to_verifier_only(),
+                verifier_params,
+                &[],
+                &proof,
+            )
+            .unwrap()
+        });
+    });
+    single_group.finish();
+
+    let mut single_group = c.benchmark_group("indexer");
+    single_group.sample_size(10);
+    single_group.bench_function("batch of 500".to_string(), |b| {
+        b.iter(|| {
+            let _ = indexer_with_lagrange(&cs, &pcs, Some(&pcs), None).unwrap();
+        });
+    });
+    single_group.finish();
+
+    let vk = PlonkVK::<KZGCommitmentScheme<BN254PairingEngine>>::from(verifier_params.clone());
+    let mut single_group = c.benchmark_group("re-indexer");
+    single_group.sample_size(10);
+    single_group.bench_function("batch of 500".to_string(), |b| {
+        b.iter(|| {
+            let _ = indexer_with_lagrange(&cs, &pcs, Some(&pcs), Some(vk.clone())).unwrap();
+        });
+    });
+    single_group.finish();
+}
+
+criterion_group!(benches, anemoi);
+criterion_main!(benches);