Update benchmarks. Add kzg bench

benchmark/kzg.js

@@ -0,0 +1,103 @@
+import { compare, utils as butils } from 'micro-bmark';
+import { loadKZG } from 'kzg-wasm';
+import * as kzg from '../esm/kzg.js';
+import { jsonGZ, __dirname } from '../test/util.js';
+import { trustedSetup as s_fast } from '@paulmillr/trusted-setups/fast.js';
+import { trustedSetup } from '@paulmillr/trusted-setups';
+
+// Test cases
+const VIEM = Object.fromEntries(
+  [
+    'blob-to-kzg-commitment',
+    'blobs',
+    'compute-blob-kzg-proof',
+    'compute-kzg-proof',
+    'invalid-blobs',
+    'verify-blob-kzg-proof-batch',
+    'verify-blob-kzg-proof',
+    'verify-kzg-proof',
+  ].map((i) => [i, jsonGZ(`../test/vectors/viem/test/kzg/${i}.json.gz`)])
+);
+
+const strip0x = (items) => items.map((i) => i.substring(2)).join('');
+
+export async function main() {
+  const opts = {
+    n1: 4096,
+    n2: 65,
+    g1: strip0x(trustedSetup.g1_lagrange),
+    g2: strip0x(trustedSetup.g2_monomial),
+  };
+  const wasmKZG = await loadKZG(opts);
+  const nobleKZG = new kzg.KZG(s_fast);
+
+  await compare('init', 1, {
+    wasm: () => loadKZG(opts),
+    noble: () => new kzg.KZG(s_fast),
+  });
+
+  const i0 = VIEM['blob-to-kzg-commitment'][1].input;
+  deepStrictEqual(
+    nobleKZG.blobToKzgCommitment(i0.blob),
+    wasmKZG.blobToKZGCommitment(i0.blob).toLowerCase()
+  );
+  await compare('blobToKzgCommitment', 5, {
+    wasm: () => wasmKZG.blobToKZGCommitment(i0.blob),
+    noble: () => nobleKZG.blobToKzgCommitment(i0.blob),
+  });
+
+  const i1 = VIEM['compute-kzg-proof'][0].input;
+  await compare('computeKzgProof', 100, {
+    // wasm: () => wasmKZG.computeKzgProof(unhex(i1.blob), unhex(i1.z)),
+    noble: () => nobleKZG.computeProof(i1.blob, i1.z),
+  });
+
+  const i2 = VIEM['compute-blob-kzg-proof'][3].input;
+  deepStrictEqual(
+    nobleKZG.computeBlobProof(i2.blob, i2.commitment),
+    wasmKZG.computeBlobKZGProof(i2.blob, i2.commitment).toLowerCase()
+  );
+  await compare('computeBlobKzgProof', 5, {
+    wasm: () => wasmKZG.computeBlobKZGProof(i2.blob, i2.commitment),
+    noble: () => nobleKZG.computeBlobProof(i2.blob, i2.commitment),
+  });
+
+  const i3 = VIEM['verify-kzg-proof'][0].input;
+  deepStrictEqual(
+    nobleKZG.verifyProof(i3.commitment, i3.z, i3.y, i3.proof),
+    wasmKZG.verifyKZGProof(i3.commitment, i3.z, i3.y, i3.proof)
+  );
+  await compare('verifyKzgProof', 200, {
+    wasm: () => wasmKZG.verifyKZGProof(i3.commitment, i3.z, i3.y, i3.proof),
+    noble: () => nobleKZG.verifyProof(i3.commitment, i3.z, i3.y, i3.proof),
+  });
+
+  const i4 = VIEM['verify-blob-kzg-proof'][0].input;
+  deepStrictEqual(
+    nobleKZG.verifyBlobProof(i4.blob, i4.commitment, i4.proof),
+    wasmKZG.verifyBlobKZGProof(i4.blob, i4.commitment, i4.proof)
+  );
+  await compare('verifyBlobKzgProof', 100, {
+    wasm: () => wasmKZG.verifyBlobKZGProof(i4.blob, i4.commitment, i4.proof),
+    noble: () => nobleKZG.verifyBlobProof(i4.blob, i4.commitment, i4.proof),
+  });
+
+  const i5 = VIEM['verify-blob-kzg-proof-batch'][1].input;
+  deepStrictEqual(
+    nobleKZG.verifyBlobProofBatch(i5.blobs, i5.commitments, i5.proofs),
+    wasmKZG.verifyBlobKZGProofBatch(i5.blobs, i5.commitments, i5.proofs)
+  );
+  await compare('verifyBlobKzgProofBatch', 10, {
+    wasm: () => wasmKZG.verifyBlobKZGProofBatch(i5.blobs, i5.commitments, i5.proofs),
+    noble: () => nobleKZG.verifyBlobProofBatch(i5.blobs, i5.commitments, i5.proofs),
+  });
+
+  butils.logMem();
+}
+
+// ESM is broken.
+import url from 'node:url';
+import { deepStrictEqual } from 'node:assert';
+if (import.meta.url === url.pathToFileURL(process.argv[1]).href) {
+  main();
+}

benchmark/package.json

@@ -19,6 +19,7 @@
     "@hazae41/cubane": "0.1.14",
     "@lodestar/types": "^1.5.0",
     "ethers": "6.11.0",
+    "kzg-wasm": "^0.5.0",
     "verkle-cryptography-wasm": "0.4.8",
     "viem": "2.7.8"
   }

benchmark/verkle.js

@@ -21,14 +21,15 @@ export async function main() {
   const SAMPLES = {
     commitToScalars: 10_000,
     createProof: 50,
-    getTreeKey: 10_000,
-    hashCommitment: 1_000_000,
-    serializeCommitment: 2_000_000,
-    updateCommitment: 10_000,
+    getTreeKey: 100_000,
+    hashCommitment: 2_000_000,
+    serializeCommitment: 3_000_000,
+    updateCommitment: 500_000,
     verifyExecutionWitnessPreState: 10_000,
     verifyProof: 100,
   };
   for (const k in VERKLE_MONOREPO) {
+    //if (k !== 'getTreeKey') continue;
     const v = VERKLE_MONOREPO[k][0];
     const args = v.arguments;
     await compare(`${k}`, SAMPLES[k], {