Integrate performance regression framework for proof verification in zkProgram zkprogram examples

Author: Shigoto-dev19

src/examples/zkprogram/gadgets.ts

@@ -1,4 +1,4 @@
-import { Field, Provable, Gadgets, ZkProgram } from 'o1js';
+import { Field, Gadgets, Provable, ZkProgram } from 'o1js';
 import { Performance } from '../../lib/testing/perf-regression.js';
 
 let cs = await Provable.constraintSystem(() => {
@@ -69,14 +69,26 @@ console.log('\nproving..');
 perfBitwise.start('prove', 'rot');
 let { proof: rotProof } = await BitwiseProver.rot();
 perfBitwise.end();
-if (!(await BitwiseProver.verify(rotProof))) throw Error('rot: Invalid proof');
+
+perfBitwise.start('verify', 'rot');
+const isValidRot = await BitwiseProver.verify(rotProof);
+perfBitwise.end();
+if (!isValidRot) throw Error('rot: Invalid proof');
 
 perfBitwise.start('prove', 'xor');
 let { proof: xorProof } = await BitwiseProver.xor();
 perfBitwise.end();
-if (!(await BitwiseProver.verify(xorProof))) throw Error('xor: Invalid proof');
+
+perfBitwise.start('verify', 'xor');
+const isValidXor = await BitwiseProver.verify(xorProof);
+perfBitwise.end();
+if (!isValidXor) throw Error('xor: Invalid proof');
 
 perfBitwise.start('prove', 'and');
 let { proof: andProof } = await BitwiseProver.and();
 perfBitwise.end();
-if (!(await BitwiseProver.verify(andProof))) throw Error('and: Invalid proof');
+
+perfBitwise.start('verify', 'and');
+const isValidAnd = await BitwiseProver.verify(andProof);
+perfBitwise.end();
+if (!isValidAnd) throw Error('and: Invalid proof');

src/examples/zkprogram/hash-chain.ts

@@ -50,6 +50,7 @@ let hashChainRecursive = Experimental.Recursive(hashChain);
 
 const cs = await hashChain.analyzeMethods();
 const perfHashChain = Performance.create(hashChain.name, cs);
+
 perfHashChain.start('compile');
 await hashChain.compile();
 perfHashChain.end();
@@ -61,7 +62,10 @@ perfHashChain.start('prove', 'chain');
 let { proof } = await hashChain.chain({ x, n });
 perfHashChain.end();
 
-assert(await hashChain.verify(proof), 'Proof invalid');
+perfHashChain.start('verify', 'chain');
+const isValid = await hashChain.verify(proof);
+perfHashChain.end();
+assert(isValid, 'Proof invalid');
 
 // check that the output is correct
 let z = Array.from({ length: n }, () => 0).reduce((y) => Poseidon.hash([y]), x);

src/examples/zkprogram/mutual-recursion.ts

@@ -1,4 +1,4 @@
-import { ZkProgram, Field, DynamicProof, Proof, VerificationKey, Undefined, verify } from 'o1js';
+import { DynamicProof, Field, Proof, Undefined, VerificationKey, ZkProgram, verify } from 'o1js';
 import { Performance } from '../../lib/testing/perf-regression.js';
 
 /**
@@ -72,14 +72,18 @@ perfAdd.start('prove', 'performAddition');
 const { proof: baseCase } = await add.performAddition(Field(5), dummyProof, multiplyVk);
 perfAdd.end();
 
+perfAdd.start('verify', 'performAddition');
 const validBaseCase = await verify(baseCase, addVk);
+perfAdd.end();
 console.log('ok?', validBaseCase);
 
 perfMultiply.start('prove', 'performMultiplication');
 const { proof: multiply1 } = await multiply.performMultiplication(Field(3), baseCase);
 perfMultiply.end();
 
+perfMultiply.start('verify', 'performMultiplication');
 const validMultiplication = await verify(multiply1, multiplyVk);
+perfMultiply.end();
 console.log('ok?', validMultiplication);
 
 console.log('Proving second (recursive) addition');

src/examples/zkprogram/program-small-big.ts

@@ -72,8 +72,8 @@ perfBig.start('prove', 'combinedHash');
 const { proof: proofBig } = await BigProgram.combinedHash(proofSmall.proof);
 perfBig.end();
 
-console.time('verify big');
+perfBig.start('verify', 'combinedHash');
 await verify(proofBig, verificationKeyBig);
-console.timeEnd('verify big');
+perfBig.end();
 
 console.log('Final Digest: ', proofBig.publicOutput.toHex());

src/examples/zkprogram/program-with-chunking.ts

@@ -1,4 +1,4 @@
-import { Field, Cache, Gadgets, ZkProgram } from 'o1js';
+import { Cache, Field, Gadgets, ZkProgram } from 'o1js';
 import { Performance } from '../../lib/testing/perf-regression.js';
 
 let MyProgram = ZkProgram({
@@ -36,6 +36,8 @@ perf.start('prove', 'baseCase');
 let { proof } = await MyProgram.baseCase(Field(0));
 perf.end();
 
-console.log('verify...');
+perf.start('verify', 'baseCase');
 let ok = await MyProgram.verify(proof);
+perf.end();
+
 console.log('ok?', ok);

src/examples/zkprogram/runtime-table/run.ts

@@ -1,6 +1,6 @@
 import { Field, verify } from 'o1js';
-import { PayrollRuntimeTableZkProgram } from './payroll.js';
 import { Performance } from '../../../lib/testing/perf-regression.js';
+import { PayrollRuntimeTableZkProgram } from './payroll.js';
 
 const cs = await PayrollRuntimeTableZkProgram.analyzeMethods();
 const perfPayroll = Performance.create(PayrollRuntimeTableZkProgram.name, cs);
@@ -26,4 +26,6 @@ let { proof } = await PayrollRuntimeTableZkProgram.verifyPayroll(
 );
 perfPayroll.end();
 
-verify(proof, verificationKey);
+perfPayroll.start('verify', 'verifyPayroll');
+await verify(proof, verificationKey);
+perfPayroll.end();

src/examples/zkprogram/side-loading/run.ts

@@ -1,12 +1,12 @@
 import { Field, MerkleTree, verify } from 'o1js';
+import { Performance } from '../../../lib/testing/perf-regression.js';
 import {
-  sideloadedProgram,
-  mainProgram,
-  SideloadedProgramProof,
-  MerkleTreeWitness,
   MainProgramState,
+  MerkleTreeWitness,
+  SideloadedProgramProof,
+  mainProgram,
+  sideloadedProgram,
 } from './dynamic-keys-merkletree.js';
-import { Performance } from '../../../lib/testing/perf-regression.js';
 
 const csSide = await sideloadedProgram.analyzeMethods();
 const csMain = await mainProgram.analyzeMethods();
@@ -56,7 +56,9 @@ const { proof: proof2 } = await mainProgram.validateUsingTree(
 );
 perfMain.end();
 
+perfMain.start('verify', 'validateUsingTree');
 const validProof2 = await verify(proof2, mainVk);
+perfMain.end();
 console.log('ok?', validProof2);
 
 console.log('\nProving different method of child program');