Update program-small-big.ts to use the performance regression framework

Author: Shigoto-dev19

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

@@ -1,4 +1,5 @@
 import { Bytes, Field, Hash, Poseidon, UInt8, ZkProgram, verify } from 'o1js';
+import { Performance } from '../../lib/testing/perf-regression.js';
 
 const SmallProgram = ZkProgram({
   name: 'small-program',
@@ -52,21 +53,24 @@ console.log('small program rows: ', csSmall.poseidonHash.rows);
 const csBig = await BigProgram.analyzeMethods();
 console.log('big program rows: ', csBig.combinedHash.rows, '\n');
 
-console.time('compile small');
+const perfSmall = Performance.create(SmallProgram.name, csSmall);
+const perfBig = Performance.create(BigProgram.name, csBig);
+
+perfSmall.start('compile');
 await SmallProgram.compile();
-console.timeEnd('compile small');
+perfSmall.end();
 
-console.time('compile big');
+perfBig.start('compile');
 const { verificationKey: verificationKeyBig } = await BigProgram.compile();
-console.timeEnd('compile big');
+perfBig.end();
 
-console.time('prove small');
+perfSmall.start('prove', 'poseidonHash');
 const proofSmall = await SmallProgram.poseidonHash(Field.random());
-console.timeEnd('prove small');
+perfSmall.end();
 
-console.time('prove big');
+perfBig.start('prove', 'combinedHash');
 const { proof: proofBig } = await BigProgram.combinedHash(proofSmall.proof);
-console.timeEnd('prove big');
+perfBig.end();
 
 console.time('verify big');
 await verify(proofBig, verificationKeyBig);

tests/perf-regression/perf-regression.sh

@@ -20,6 +20,7 @@ STOP_AFTER=2 ./run src/examples/zkprogram/hash-chain.ts --bundle "$MODE"
 STOP_AFTER=4 ./run src/examples/zkprogram/gadgets.ts --bundle "$MODE"
 STOP_AFTER=6 ./run src/examples/zkprogram/side-loading/run.ts --bundle "$MODE"
 STOP_AFTER=4 ./run src/examples/zkprogram/runtime-table/run.ts --bundle "$MODE"
+STOP_AFTER=4 ./run src/examples/zkprogram/program-small-big.ts --bundle "$MODE"
 
 # Run CS + zkApps performance regression tests
 ./run tests/perf-regression/perf-regression.ts --bundle "$MODE"