chore: update peerdas-kzg library

Cargo.toml

@@ -99,7 +99,7 @@ delay_map = "0.3"
 derivative = "2"
 dirs = "3"
 either = "1.9"
-peerdas-kzg = { git = "https://github.com/crate-crypto/peerdas-kzg", rev = "bb8295011cf27dc663699539c7f9a17fe273e896", package = "eip7594" }
+peerdas-kzg = { git = "https://github.com/crate-crypto/peerdas-kzg", rev = "55ae9e9011d792a2998d242c2a71d822ba909fa9", package = "eip7594" }
 discv5 = { version = "0.4.1", features = ["libp2p"] }
 env_logger = "0.9"
 error-chain = "0.12"

crypto/kzg/Cargo.toml

@@ -19,3 +19,12 @@ ethereum_hashing = { workspace = true }
 c-kzg = { workspace = true }
 peerdas-kzg = { workspace = true }
 mockall = { workspace = true }
+
+[dev-dependencies]
+criterion = { workspace = true }
+serde_json = { workspace = true }
+eth2_network_config = { workspace = true }
+
+[[bench]]
+name = "benchmark"
+harness = false

crypto/kzg/benches/benchmark.rs

@@ -0,0 +1,27 @@
+use c_kzg::KzgSettings;
+use criterion::{criterion_group, criterion_main, Criterion};
+use eth2_network_config::TRUSTED_SETUP_BYTES;
+use kzg::TrustedSetup;
+use peerdas_kzg::{PeerDASContext, TrustedSetup as PeerDASTrustedSetup};
+
+pub fn bench_init_context(c: &mut Criterion) {
+    c.bench_function(&format!("Initialize context peerdas-kzg"), |b| {
+        b.iter(|| {
+            const NUM_THREADS: usize = 1;
+            let trusted_setup = PeerDASTrustedSetup::default();
+            PeerDASContext::with_threads(&trusted_setup, NUM_THREADS)
+        })
+    });
+    c.bench_function(&format!("Initialize context c-kzg (4844)"), |b| {
+        b.iter(|| {
+            let trusted_setup: TrustedSetup = serde_json::from_reader(TRUSTED_SETUP_BYTES)
+                .map_err(|e| format!("Unable to read trusted setup file: {}", e))
+                .expect("should have trusted setup");
+            KzgSettings::load_trusted_setup(&trusted_setup.g1_points(), &trusted_setup.g2_points())
+                .unwrap()
+        })
+    });
+}
+
+criterion_group!(benches, bench_init_context);
+criterion_main!(benches);

crypto/kzg/src/lib.rs

@@ -17,7 +17,7 @@ use mockall::automock;
 
 pub use peerdas_kzg::{
     constants::{BYTES_PER_CELL, CELLS_PER_EXT_BLOB},
-    Cell, CellID, CellRef, TrustedSetup as PeerDASTrustedSetup,
+    Cell, CellIndex as CellID, CellRef, TrustedSetup as PeerDASTrustedSetup,
 };
 use peerdas_kzg::{prover::ProverError, verifier::VerifierError, PeerDASContext};
 pub type CellsAndKzgProofs = ([Cell; CELLS_PER_EXT_BLOB], [KzgProof; CELLS_PER_EXT_BLOB]);
@@ -181,7 +181,6 @@ impl Kzg {
 
         let (cells, proofs) = self
             .context
-            .prover_ctx()
             .compute_cells_and_kzg_proofs(blob_bytes)
             .map_err(Error::ProverKZG)?;
 
@@ -212,7 +211,7 @@ impl Kzg {
             .iter()
             .map(|commitment| commitment.as_ref())
             .collect();
-        let verification_result = self.context.verifier_ctx().verify_cell_kzg_proof_batch(
+        let verification_result = self.context.verify_cell_kzg_proof_batch(
             commitments.to_vec(),
             rows,
             columns,
@@ -236,7 +235,6 @@ impl Kzg {
     ) -> Result<CellsAndKzgProofs, Error> {
         let (cells, proofs) = self
             .context
-            .prover_ctx()
             .recover_cells_and_proofs(cell_ids.to_vec(), cells.to_vec())
             .map_err(Error::ProverKZG)?;