minor fixes for ctx

Author: AlexOnTheStorm

crates/proof-compression/src/proof_system/boojum.rs

@@ -86,10 +86,11 @@ where
     }
 
     fn init_context(domain_size: Self::ContextConfig) -> anyhow::Result<Self::Context> {
+        let start = std::time::Instant::now();
         let config =
             ProverContextConfig::default().with_smallest_supported_domain_size(domain_size);
         let context = Self::Context::create_with_config(config).context("gpu prover context");
-
+        println!("compression ctx initialization takes {:?}", start.elapsed());
         context
     }
     fn get_context_config_from_hint(hint: &Self::FinalizationHint) -> Self::ContextConfig {

crates/proof-compression/src/proof_system/fflonk.rs

@@ -78,12 +78,14 @@ impl SnarkWrapperProofSystem for FflonkSnarkWrapper {
     }
 
     fn init_context(compact_raw_crs: &Self::CRS) -> anyhow::Result<Self::Context> {
+        let start = std::time::Instant::now();
         let domain_size = 1 << ::fflonk::fflonk_cpu::L1_VERIFIER_DOMAIN_SIZE_LOG;
         let context = DeviceContextWithSingleDevice::init_from_preloaded_crs::<Self::Allocator>(
             domain_size,
             compact_raw_crs,
         )
         .unwrap();
+        println!("fflonk ctx initialization takes {:?}", start.elapsed());
         Ok(context)
     }
 
@@ -143,7 +145,7 @@ impl SnarkWrapperProofSystemExt for FflonkSnarkWrapper {
     }
 
     fn generate_precomputation_and_vk(
-        _ctx: &Self::Context,
+        ctx: Self::Context,
         setup_assembly: Self::SetupAssembly,
         _hardcoded_finalization_hint: Self::FinalizationHint,
     ) -> anyhow::Result<(Self::Precomputation, Self::VK)> {
@@ -158,6 +160,7 @@ impl SnarkWrapperProofSystemExt for FflonkSnarkWrapper {
                 )
             })?;
         let vk = device_setup.get_verification_key();
+        drop(ctx);
         Ok((
             FflonkSnarkVerifierCircuitDeviceSetupWrapper(device_setup),
             vk,

crates/proof-compression/src/proof_system/interface.rs

@@ -107,7 +107,7 @@ pub trait SnarkWrapperProofSystemExt: SnarkWrapperProofSystem {
     type SetupAssembly;
     fn synthesize_for_setup(circuit: Self::Circuit) -> Self::SetupAssembly;
     fn generate_precomputation_and_vk(
-        _: &Self::Context,
+        _: Self::Context,
         _: Self::SetupAssembly,
         _: Self::FinalizationHint,
     ) -> anyhow::Result<(Self::Precomputation, Self::VK)>;

crates/proof-compression/src/proof_system/plonk.rs

@@ -274,7 +274,7 @@ impl SnarkWrapperProofSystemExt for PlonkSnarkWrapper {
     }
 
     fn generate_precomputation_and_vk(
-        _: &Self::Context,
+        _: Self::Context,
         _: Self::SetupAssembly,
         _: Self::FinalizationHint,
     ) -> anyhow::Result<(Self::Precomputation, Self::VK)> {

crates/proof-compression/src/step/snark_wrapper.rs

@@ -134,7 +134,7 @@ pub trait SnarkWrapperStepExt: SnarkWrapperProofSystemExt + SnarkWrapperStep {
         let ctx = <Self as SnarkWrapperProofSystem>::init_context(&crs)?;
         let (precomputation, vk) =
             <Self as SnarkWrapperProofSystemExt>::generate_precomputation_and_vk(
-                &ctx,
+                ctx,
                 setup_assembly,
                 finalization_hint,
             )?;