arnaucube/sonobe-playground
1
0
Fork 0
mirror of https://github.com/arnaucube/sonobe-playground.git synced 2026-01-14 01:51:29 +01:00
Code Issues Projects Releases Wiki Activity

Multifolding transformation

Browse Source
This commit is contained in:
Piotr Mikołajczyk
2024-10-02 13:03:07 +02:00
parent 82b20639be
commit 23b0dc36ec
5 changed files with 82 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
src/folding.rs
View File

@@ -1,17 +1,18 @@
use ark_bn254::{constraints::GVar, Bn254, Fr, G1Projective as G1};
use ark_crypto_primitives::sponge::poseidon::PoseidonConfig;
use ark_grumpkin::{constraints::GVar as GVar2, Projective as G2};
use itertools::Itertools;
use sonobe::{
commitment::{kzg::KZG, pedersen::Pedersen},
folding::{hypernova::HyperNova, nova::Nova},
frontend::circom::CircomFCircuit,
transcript::poseidon::poseidon_canonical_config,
FoldingScheme,
FoldingScheme, MultiFolding,
};
pub type NovaFolding =
Nova<G1, GVar, G2, GVar2, CircomFCircuit<Fr>, KZG<'static, Bn254>, Pedersen<G2>, false>;
pub type HyperNovaFolding = HyperNova<
pub type HyperNovaFolding<const N: usize, const M: usize> = HyperNova<
G1,
GVar,
G2,
@@ -19,8 +20,8 @@ pub type HyperNovaFolding = HyperNova<
CircomFCircuit<Fr>,
KZG<'static, Bn254>,
Pedersen<G2>,
1,
1,
N,
M,
false,
>;
@@ -36,8 +37,11 @@ pub trait FoldingSchemeExt: FoldingScheme<G1, G2, CircomFCircuit<Fr>> {
) -> Self::PreprocessorParam;
fn transform_inputs(
&self,
full_input: Vec<Vec<Fr>>,
) -> impl Iterator<Item = StepInput<Self::MultiCommittedInstanceWithWitness>>;
initial_state: Vec<Fr>,
rng: &mut impl rand::RngCore,
) -> Vec<StepInput<Self::MultiCommittedInstanceWithWitness>>;
}
impl FoldingSchemeExt for NovaFolding {
@@ -49,16 +53,22 @@ impl FoldingSchemeExt for NovaFolding {
}
fn transform_inputs(
&self,
full_input: Vec<Vec<Fr>>,
) -> impl Iterator<Item = StepInput<Self::MultiCommittedInstanceWithWitness>> {
full_input.into_iter().map(|input| StepInput {
external_inputs: input,
other_instances: None,
})
_initial_state: Vec<Fr>,
_rng: &mut impl rand::RngCore,
) -> Vec<StepInput<Self::MultiCommittedInstanceWithWitness>> {
full_input
.into_iter()
.map(|input| StepInput {
external_inputs: input,
other_instances: None,
})
.collect()
}
}
impl FoldingSchemeExt for HyperNovaFolding {
impl<const N: usize, const M: usize> FoldingSchemeExt for HyperNovaFolding<N, M> {
fn prepreprocess(
poseidon_config: PoseidonConfig<Fr>,
circuit: CircomFCircuit<Fr>,
@@ -67,12 +77,52 @@ impl FoldingSchemeExt for HyperNovaFolding {
}
fn transform_inputs(
&self,
full_input: Vec<Vec<Fr>>,
) -> impl Iterator<Item = StepInput<Self::MultiCommittedInstanceWithWitness>> {
full_input.into_iter().map(|input| StepInput {
external_inputs: input,
other_instances: Some((vec![], vec![])),
})
initial_state: Vec<Fr>,
rng: &mut impl rand::RngCore,
) -> Vec<StepInput<Self::MultiCommittedInstanceWithWitness>> {
full_input
.into_iter()
.chunks(M + N - 1)
.into_iter()
.map(|chunk| {
let chunk = chunk.collect::<Vec<_>>();
let (running, rest) = chunk.split_at(M - 1);
let (incoming, [single]) = rest.split_at(N - 1) else {
panic!("Invalid input chunk size");
};
let lcccs = running
.iter()
.map(|instance| {
self.new_running_instance(
&mut *rng,
initial_state.clone(),
instance.clone(),
)
.expect("Failed to create running instance")
})
.collect();
let cccs = incoming
.iter()
.map(|instance| {
self.new_incoming_instance(
&mut *rng,
initial_state.clone(),
instance.clone(),
)
.expect("Failed to create incoming instance")
})
.collect();
StepInput {
external_inputs: single.clone(),
other_instances: Some((lcccs, cccs)),
}
})
.collect()
}
}

20
src/main.rs
View File

@@ -3,7 +3,7 @@ use std::time::Instant;
use scenario_config::ScenarioConfig;
use crate::folding::{
FoldingSchemeExt, HyperNovaFolding, NovaFolding, prepare_folding, verify_folding,
prepare_folding, verify_folding, FoldingSchemeExt, HyperNovaFolding, NovaFolding,
};
mod circuit;
@@ -22,14 +22,17 @@ fn measure<T, Action: FnOnce() -> T>(action_name: &str, action: Action) -> T {
fn scenario<FS: FoldingSchemeExt>(config: ScenarioConfig, rng: &mut impl rand::RngCore) {
// ============== FOLDING PREPARATION ==========================================================
let start_state = config.start_ivc_state.clone();
let (mut folding, folding_vp) = measure("Prepare folding", || {
prepare_folding::<FS>(&config.circuit, config.start_ivc_state.clone(), rng)
prepare_folding::<FS>(&config.circuit, start_state.clone(), rng)
});
// ============== FOLDING ======================================================================
let input = config.input().to_vec();
for (i, step_input) in FS::transform_inputs(input).enumerate() {
let input = measure("Transform input", || {
folding.transform_inputs(config.input().to_vec(), start_state.clone(), &mut *rng)
});
for (i, step_input) in input.into_iter().enumerate() {
measure(&format!("Prove_step {i}"), || {
folding
.prove_step(
@@ -57,9 +60,12 @@ fn main() {
let mut rng = rand::rngs::OsRng;
let config = ScenarioConfig::new();
println!("========== Nova folding scheme ==========");
println!("========== Nova folding scheme ====================");
scenario::<NovaFolding>(config.clone(), &mut rng);
println!("========== HyperNova folding scheme ==========");
scenario::<HyperNovaFolding>(config, &mut rng);
println!("========== HyperNova<1,1> folding scheme ==========");
scenario::<HyperNovaFolding<1, 1>>(config.clone(), &mut rng);
println!("========== HyperNova<2,2> folding scheme ==========");
scenario::<HyperNovaFolding<2, 2>>(config, &mut rng);
}

2
src/scenario_config.rs
View File

@@ -15,7 +15,7 @@ pub struct ScenarioConfig {
impl ScenarioConfig {
pub fn new() -> Self {
Self {
num_steps: 5,
num_steps: 6,
start_ivc_state: vec![Fr::zero(); 2],
circuit: measure("Prepare circuit", create_circuit),
input: measure("Prepare input", prepare_input),