Merge pull request #2553 from o1-labs/leon/small-big

feat: small and big zk program for cache regression

Author: bleepbloopsify

scripts/tests/check-cache-regressions.sh

@@ -9,7 +9,7 @@ source ./scripts/lib/ux.sh
 ./run ./src/tests/cache/simple-regression.ts --bundle --mode check --tarball ./tests/test-artifacts/cache/simple-regression.tar.gz
 
 # This pin is generated in ./dump-cache-regressions.sh
-ARTIFACT_PIN=2025-10-14T14:08:25-04:00
+ARTIFACT_PIN=2025-10-14T14:16:00-04:00
 
 WORKDIR=tests/test-artifacts/cache/
 mkdir -p $WORKDIR
@@ -22,8 +22,9 @@ WORKDIR=tests/test-artifacts/cache/$ARTIFACT_PIN
 # Regression checks
 ./run ./src/tests/cache/simple-regression.ts --bundle --mode check --tarball $WORKDIR/simple-regression.tar.gz
 ./run ./src/tests/cache/complex-regression.ts --bundle --mode check --tarball $WORKDIR/complex-regression.tar.gz
-./run ./src/tests/cache/rsa-regression.ts --bundle --keep --mode check --tarball $WORKDIR/rsa-regression.tar.gz
-./run ./src/tests/cache/sideloading-regression.ts --bundle --keep --mode check --tarball $WORKDIR/sideloading-regression.tar.gz
-./run ./src/tests/cache/runtime-table-regression.ts --bundle --keep --mode check --tarball $WORKDIR/runtime-table-regression.tar.gz
+./run ./src/tests/cache/rsa-regression.ts --bundle --mode check --tarball $WORKDIR/rsa-regression.tar.gz
+./run ./src/tests/cache/sideloading-regression.ts --bundle --mode check --tarball $WORKDIR/sideloading-regression.tar.gz
+./run ./src/tests/cache/runtime-table-regression.ts --bundle --mode check --tarball $WORKDIR/runtime-table-regression.tar.gz
+./run ./src/tests/cache/small-big-regression.ts --bundle --mode check --tarball $WORKDIR/small-big-regression.tar.gz
 
 echo "Artifacts checked successfully!"

scripts/tests/dump-cache-regressions.sh

@@ -9,10 +9,11 @@ WORKDIR=tests/test-artifacts/cache/$ARTIFACT_PIN/
 mkdir -p $WORKDIR
 
 ./run ./src/tests/cache/simple-regression.ts --bundle --mode dump --tarball $WORKDIR/simple-regression.tar.gz
-./run ./src/tests/cache/complex-regression.ts --bundle --keep --mode dump --tarball $WORKDIR/complex-regression.tar.gz
-./run ./src/tests/cache/rsa-regression.ts --bundle --keep --mode dump --tarball $WORKDIR/rsa-regression.tar.gz
-./run ./src/tests/cache/sideloading-regression.ts --bundle --keep --mode dump --tarball $WORKDIR/sideloading-regression.tar.gz
-./run ./src/tests/cache/runtime-table-regression.ts --bundle --keep --mode dump --tarball $WORKDIR/runtime-table-regression.tar.gz
+./run ./src/tests/cache/complex-regression.ts --bundle --mode dump --tarball $WORKDIR/complex-regression.tar.gz
+./run ./src/tests/cache/rsa-regression.ts --bundle --mode dump --tarball $WORKDIR/rsa-regression.tar.gz
+./run ./src/tests/cache/sideloading-regression.ts --bundle --mode dump --tarball $WORKDIR/sideloading-regression.tar.gz
+./run ./src/tests/cache/runtime-table-regression.ts --bundle --mode dump --tarball $WORKDIR/runtime-table-regression.tar.gz
+./run ./src/tests/cache/small-big-regression.ts --bundle --mode dump --tarball $WORKDIR/small-big-regression.tar.gz
 
 gcloud storage cp --recursive $WORKDIR gs://o1js-ci/tests/cache/fixtures/
 

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

@@ -0,0 +1,47 @@
+import { Bytes, Field, Hash, Poseidon, UInt8, ZkProgram } from 'o1js';
+
+export const SmallProgram = ZkProgram({
+  name: 'small-program',
+  publicOutput: Field,
+
+  methods: {
+    poseidonHash: {
+      privateInputs: [Field],
+      async method(preimage: Field) {
+        const digest = Poseidon.hash([preimage]);
+
+        return { publicOutput: digest };
+      },
+    },
+  },
+});
+
+class PoseidonProof extends SmallProgram.Proof {}
+class Bytes32 extends Bytes(32) {}
+
+export const BigProgram = ZkProgram({
+  name: 'big-program',
+  publicOutput: Bytes32,
+
+  methods: {
+    combinedHash: {
+      privateInputs: [PoseidonProof],
+      async method(poseidonProof: PoseidonProof) {
+        poseidonProof.verify();
+
+        const poseidonOutput = poseidonProof.publicOutput;
+        const outputBits = poseidonOutput.toBits();
+
+        const byteChunks = Array.from({ length: 32 }, (_, i) => outputBits.slice(i * 8, i * 8 + 8));
+        const outputBytes = byteChunks.map((byteChunks) => UInt8.fromBits(byteChunks));
+
+        const hash1 = Hash.SHA2_256.hash(outputBytes);
+        const hash2 = Hash.BLAKE2B.hash(hash1);
+        const hash3 = Hash.SHA3_512.hash(Bytes.from([...hash1.bytes, ...hash2.bytes]));
+        const hash4 = Hash.Keccak256.hash(hash3);
+
+        return { publicOutput: hash4 };
+      },
+    },
+  },
+});

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

@@ -0,0 +1,34 @@
+import { Field, verify } from 'o1js';
+import { Performance } from '../../../lib/testing/perf-regression.js';
+import { BigProgram, SmallProgram } from './program-small-big.js';
+
+const csSmall = await SmallProgram.analyzeMethods();
+console.log('small program rows: ', csSmall.poseidonHash.rows);
+
+const csBig = await BigProgram.analyzeMethods();
+console.log('big program rows: ', csBig.combinedHash.rows, '\n');
+
+const perfSmall = Performance.create(SmallProgram.name, csSmall);
+const perfBig = Performance.create(BigProgram.name, csBig);
+
+perfSmall.start('compile');
+await SmallProgram.compile();
+perfSmall.end();
+
+perfBig.start('compile');
+const { verificationKey: verificationKeyBig } = await BigProgram.compile();
+perfBig.end();
+
+perfSmall.start('prove', 'poseidonHash');
+const proofSmall = await SmallProgram.poseidonHash(Field.random());
+perfSmall.end();
+
+perfBig.start('prove', 'combinedHash');
+const { proof: proofBig } = await BigProgram.combinedHash(proofSmall.proof);
+perfBig.end();
+
+perfBig.start('verify', 'combinedHash');
+await verify(proofBig, verificationKeyBig);
+perfBig.end();
+
+console.log('Final Digest: ', proofBig.publicOutput.toHex());

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

@@ -0,0 +1,31 @@
+import minimist from 'minimist';
+import assert from 'node:assert';
+import { Field } from 'o1js';
+import { BigProgram, SmallProgram } from '../../examples/zkprogram/small-big/program-small-big.js';
+import { CacheHarness } from './harness.js';
+
+const { mode, tarball } = minimist(process.argv.slice(2));
+
+const harness = await CacheHarness({ mode, tarball });
+
+const { verificationKey: smallVk } = await SmallProgram.compile({ cache: harness.cache });
+harness.check(smallVk, 'smallVk');
+
+const { verificationKey: bigVk } = await BigProgram.compile({ cache: harness.cache });
+harness.check(bigVk, 'bigVk');
+
+{
+  const { proof } = await SmallProgram.poseidonHash(Field.random());
+  {
+    const ok = await harness.verify(proof, 'smallVk');
+    assert.equal(ok, true, 'small proof should verify');
+  }
+
+  const { proof: derived } = await BigProgram.combinedHash(proof);
+  {
+    const ok = await harness.verify(derived, 'bigVk');
+    assert.equal(ok, true, 'derived proof should verify');
+  }
+}
+
+await harness.finish();

src/tests/cache/small-big-regression.ts

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