|
|
@ -1,13 +1,12 @@ |
|
|
|
use std::{env, fs::File, time::Instant};
|
|
|
|
use std::{fs::File, io, time::Instant};
|
|
|
|
|
|
|
|
use ark_bls12_381::{Bls12_381, Fr};
|
|
|
|
use ark_serialize::Write;
|
|
|
|
use ark_serialize::{CanonicalDeserialize, CanonicalSerialize, Write};
|
|
|
|
use ark_std::test_rng;
|
|
|
|
use hyperplonk::{
|
|
|
|
prelude::{CustomizedGates, HyperPlonkErrors, MockCircuit},
|
|
|
|
HyperPlonkSNARK,
|
|
|
|
};
|
|
|
|
use rayon::ThreadPoolBuilder;
|
|
|
|
use subroutines::{
|
|
|
|
pcs::{
|
|
|
|
prelude::{MultilinearKzgPCS, MultilinearUniversalParams},
|
|
|
@ -16,37 +15,58 @@ use subroutines::{ |
|
|
|
poly_iop::PolyIOP,
|
|
|
|
};
|
|
|
|
|
|
|
|
const SUPPORTED_SIZE: usize = 15;
|
|
|
|
const SUPPORTED_SIZE: usize = 20;
|
|
|
|
const MIN_NUM_VARS: usize = 8;
|
|
|
|
const MAX_NUM_VARS: usize = 15;
|
|
|
|
const MAX_NUM_VARS: usize = 20;
|
|
|
|
const MIN_CUSTOM_DEGREE: usize = 1;
|
|
|
|
const MAX_CUSTOM_DEGREE: usize = 32;
|
|
|
|
const MAX_NUM_THREADS: usize = 24;
|
|
|
|
const HIGH_DEGREE_TEST_NV: usize = 15;
|
|
|
|
|
|
|
|
fn main() -> Result<(), HyperPlonkErrors> {
|
|
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
let thread = args[1].parse().unwrap_or(MAX_NUM_THREADS);
|
|
|
|
let thread = rayon::current_num_threads();
|
|
|
|
println!("start benchmark with #{} threads", thread);
|
|
|
|
let mut rng = test_rng();
|
|
|
|
let pcs_srs = MultilinearKzgPCS::<Bls12_381>::gen_srs_for_testing(&mut rng, SUPPORTED_SIZE)?;
|
|
|
|
ThreadPoolBuilder::new()
|
|
|
|
.num_threads(thread)
|
|
|
|
.build()
|
|
|
|
.unwrap();
|
|
|
|
let pcs_srs = {
|
|
|
|
match read_srs() {
|
|
|
|
Ok(p) => p,
|
|
|
|
Err(_e) => {
|
|
|
|
let srs =
|
|
|
|
MultilinearKzgPCS::<Bls12_381>::gen_srs_for_testing(&mut rng, SUPPORTED_SIZE)?;
|
|
|
|
write_srs(&srs);
|
|
|
|
srs
|
|
|
|
},
|
|
|
|
}
|
|
|
|
};
|
|
|
|
bench_jellyfish_plonk(&pcs_srs, thread)?;
|
|
|
|
println!();
|
|
|
|
bench_vanilla_plonk(&pcs_srs, thread)?;
|
|
|
|
for degree in MIN_CUSTOM_DEGREE..MAX_CUSTOM_DEGREE {
|
|
|
|
println!();
|
|
|
|
for degree in MIN_CUSTOM_DEGREE..=MAX_CUSTOM_DEGREE {
|
|
|
|
bench_high_degree_plonk(&pcs_srs, degree, thread)?;
|
|
|
|
println!();
|
|
|
|
}
|
|
|
|
println!();
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn read_srs() -> Result<MultilinearUniversalParams<Bls12_381>, io::Error> {
|
|
|
|
let mut f = File::open("srs.params")?;
|
|
|
|
Ok(MultilinearUniversalParams::<Bls12_381>::deserialize_unchecked(&mut f).unwrap())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn write_srs(pcs_srs: &MultilinearUniversalParams<Bls12_381>) {
|
|
|
|
let mut f = File::create("srs.params").unwrap();
|
|
|
|
pcs_srs.serialize_uncompressed(&mut f).unwrap();
|
|
|
|
}
|
|
|
|
|
|
|
|
fn bench_vanilla_plonk(
|
|
|
|
pcs_srs: &MultilinearUniversalParams<Bls12_381>,
|
|
|
|
thread: usize,
|
|
|
|
) -> Result<(), HyperPlonkErrors> {
|
|
|
|
let filename = format!("vanilla threads {}.txt", thread);
|
|
|
|
let mut file = File::create(filename).unwrap();
|
|
|
|
for nv in MIN_NUM_VARS..MAX_NUM_VARS {
|
|
|
|
for nv in MIN_NUM_VARS..=MAX_NUM_VARS {
|
|
|
|
let vanilla_gate = CustomizedGates::vanilla_plonk_gate();
|
|
|
|
bench_mock_circuit_zkp_helper(&mut file, nv, &vanilla_gate, &pcs_srs)?;
|
|
|
|
}
|
|
|
@ -54,6 +74,20 @@ fn bench_vanilla_plonk( |
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn bench_jellyfish_plonk(
|
|
|
|
pcs_srs: &MultilinearUniversalParams<Bls12_381>,
|
|
|
|
thread: usize,
|
|
|
|
) -> Result<(), HyperPlonkErrors> {
|
|
|
|
let filename = format!("jellyfish threads {}.txt", thread);
|
|
|
|
let mut file = File::create(filename).unwrap();
|
|
|
|
for nv in MIN_NUM_VARS..=MAX_NUM_VARS {
|
|
|
|
let jf_gate = CustomizedGates::jellyfish_turbo_plonk_gate();
|
|
|
|
bench_mock_circuit_zkp_helper(&mut file, nv, &jf_gate, &pcs_srs)?;
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn bench_high_degree_plonk(
|
|
|
|
pcs_srs: &MultilinearUniversalParams<Bls12_381>,
|
|
|
|
degree: usize,
|
|
|
@ -61,10 +95,9 @@ fn bench_high_degree_plonk( |
|
|
|
) -> Result<(), HyperPlonkErrors> {
|
|
|
|
let filename = format!("high degree {} thread {}.txt", degree, thread);
|
|
|
|
let mut file = File::create(filename).unwrap();
|
|
|
|
for nv in MIN_NUM_VARS..MAX_NUM_VARS {
|
|
|
|
let vanilla_gate = CustomizedGates::mock_gate(2, degree);
|
|
|
|
bench_mock_circuit_zkp_helper(&mut file, nv, &vanilla_gate, &pcs_srs)?;
|
|
|
|
}
|
|
|
|
println!("custom gate of degree {}", degree);
|
|
|
|
let vanilla_gate = CustomizedGates::mock_gate(2, degree);
|
|
|
|
bench_mock_circuit_zkp_helper(&mut file, HIGH_DEGREE_TEST_NV, &vanilla_gate, &pcs_srs)?;
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
@ -84,17 +117,6 @@ fn bench_mock_circuit_zkp_helper( |
|
|
|
};
|
|
|
|
|
|
|
|
//==========================================================
|
|
|
|
let start = Instant::now();
|
|
|
|
for _ in 0..repetition {
|
|
|
|
let circuit = MockCircuit::<Fr>::new(1 << nv, gate);
|
|
|
|
assert!(circuit.is_satisfied());
|
|
|
|
}
|
|
|
|
println!(
|
|
|
|
"mock circuit gen for {} variables: {} ns",
|
|
|
|
nv,
|
|
|
|
start.elapsed().as_nanos() / repetition as u128
|
|
|
|
);
|
|
|
|
|
|
|
|
let circuit = MockCircuit::<Fr>::new(1 << nv, gate);
|
|
|
|
assert!(circuit.is_satisfied());
|
|
|
|
let index = circuit.index;
|
|
|
@ -123,17 +145,21 @@ fn bench_mock_circuit_zkp_helper( |
|
|
|
let _proof =
|
|
|
|
<PolyIOP<Fr> as HyperPlonkSNARK<Bls12_381, MultilinearKzgPCS<Bls12_381>>>::prove(
|
|
|
|
&pk,
|
|
|
|
&circuit.witnesses[0].coeff_ref(),
|
|
|
|
&circuit.public_inputs,
|
|
|
|
&circuit.witnesses,
|
|
|
|
)?;
|
|
|
|
}
|
|
|
|
let t = start.elapsed().as_micros() / repetition as u128;
|
|
|
|
|
|
|
|
println!(
|
|
|
|
"proving for {} variables: {} us",
|
|
|
|
nv,
|
|
|
|
start.elapsed().as_micros() / repetition as u128
|
|
|
|
);
|
|
|
|
file.write_all(format!("{} {}\n", nv, t).as_ref()).unwrap();
|
|
|
|
|
|
|
|
let proof = <PolyIOP<Fr> as HyperPlonkSNARK<Bls12_381, MultilinearKzgPCS<Bls12_381>>>::prove(
|
|
|
|
&pk,
|
|
|
|
&circuit.witnesses[0].coeff_ref(),
|
|
|
|
&circuit.public_inputs,
|
|
|
|
&circuit.witnesses,
|
|
|
|
)?;
|
|
|
|
//==========================================================
|
|
|
@ -143,7 +169,7 @@ fn bench_mock_circuit_zkp_helper( |
|
|
|
let verify =
|
|
|
|
<PolyIOP<Fr> as HyperPlonkSNARK<Bls12_381, MultilinearKzgPCS<Bls12_381>>>::verify(
|
|
|
|
&vk,
|
|
|
|
&circuit.witnesses[0].coeff_ref(),
|
|
|
|
&circuit.public_inputs,
|
|
|
|
&proof,
|
|
|
|
)?;
|
|
|
|
assert!(verify);
|
|
|
|