Async circuits, pt. 1 - async circuit runners

src/examples/benchmarks/foreign-field.ts

@@ -21,11 +21,11 @@ main();
 console.timeEnd('running constant version');
 
 console.time('running witness generation & checks');
-Provable.runAndCheck(main);
+await Provable.runAndCheck(main);
 console.timeEnd('running witness generation & checks');
 
 console.time('creating constraint system');
-let cs = Provable.constraintSystem(main);
+let cs = await Provable.constraintSystem(main);
 console.timeEnd('creating constraint system');
 
 console.log(cs.summary());

src/examples/benchmarks/hash-witness.ts

@@ -19,5 +19,5 @@ function main(nMuls: number) {
 }
 
 tic('run and check');
-Provable.runAndCheck(() => main(nPermutations));
+await Provable.runAndCheck(() => main(nPermutations));
 toc();

src/examples/benchmarks/keccak-witness.ts

@@ -3,7 +3,7 @@ import { Hash, Bytes, Provable } from 'o1js';
 let Bytes32 = Bytes(32);
 
 console.time('keccak witness');
-Provable.runAndCheck(() => {
+await Provable.runAndCheck(() => {
   let bytes = Provable.witness(Bytes32.provable, () => Bytes32.random());
   Hash.Keccak256.hash(bytes);
 });

src/examples/benchmarks/mul-web.ts

@@ -20,8 +20,8 @@ function main(nMuls: number) {
   }
 }
 
-function getRows(nMuls: number) {
-  let { rows } = Provable.constraintSystem(() => main(nMuls));
+async function getRows(nMuls: number) {
+  let { rows } = await Provable.constraintSystem(() => main(nMuls));
   return rows;
 }
 
@@ -52,7 +52,7 @@ function picklesCircuit(nMuls: number) {
 
 // the script
 
-console.log('circuit size (without pickles overhead)', getRows(nMuls));
+console.log('circuit size (without pickles overhead)', await getRows(nMuls));
 
 if (withPickles) {
   let circuit = picklesCircuit(nMuls);

src/examples/benchmarks/mul-witness.ts

@@ -19,5 +19,5 @@ function main(nMuls: number) {
 }
 
 tic('run and check');
-Provable.runAndCheck(() => main(nMuls));
+await Provable.runAndCheck(() => main(nMuls));
 toc();

src/examples/benchmarks/mul.ts

@@ -19,8 +19,8 @@ function main(nMuls: number) {
   }
 }
 
-function getRows(nMuls: number) {
-  let { rows } = Provable.constraintSystem(() => main(nMuls));
+async function getRows(nMuls: number) {
+  let { rows } = await Provable.constraintSystem(() => main(nMuls));
   return rows;
 }
 
@@ -48,7 +48,7 @@ function picklesCircuit(nMuls: number) {
   });
 }
 
-console.log('circuit size (without pickles overhead)', getRows(nMuls));
+console.log('circuit size (without pickles overhead)', await getRows(nMuls));
 
 if (withPickles) {
   let circuit = picklesCircuit(nMuls);

src/examples/constraint-system.ts

@@ -2,13 +2,15 @@ import { Field, Poseidon, Provable } from 'o1js';
 
 let hash = Poseidon.hash([Field(1), Field(-1)]);
 
-let { rows, digest, publicInputSize, print } = Provable.constraintSystem(() => {
-  let x = Provable.witness(Field, () => Field(1));
-  let y = Provable.witness(Field, () => Field(-1));
-  x.add(y).assertEquals(Field(0));
-  let z = Poseidon.hash([x, y]);
-  z.assertEquals(hash);
-});
+let { rows, digest, publicInputSize, print } = await Provable.constraintSystem(
+  () => {
+    let x = Provable.witness(Field, () => Field(1));
+    let y = Provable.witness(Field, () => Field(-1));
+    x.add(y).assertEquals(Field(0));
+    let z = Poseidon.hash([x, y]);
+    z.assertEquals(hash);
+  }
+);
 
 print();
 console.log({ rows, digest, publicInputSize });

src/examples/crypto/ecdsa/run.ts

@@ -13,12 +13,12 @@ let signature = Ecdsa.sign(message.toBytes(), privateKey.toBigInt());
 // investigate the constraint system generated by ECDSA verify
 
 console.time('ecdsa verify only (build constraint system)');
-let csEcdsa = ecdsa.analyzeMethods();
+let csEcdsa = await ecdsa.analyzeMethods();
 console.timeEnd('ecdsa verify only (build constraint system)');
 console.log(csEcdsa.verifySignedHash.summary());
 
 console.time('keccak + ecdsa verify (build constraint system)');
-let cs = keccakAndEcdsa.analyzeMethods();
+let cs = await keccakAndEcdsa.analyzeMethods();
 console.timeEnd('keccak + ecdsa verify (build constraint system)');
 console.log(cs.verifyEcdsa.summary());
 

src/examples/crypto/foreign-field.ts

@@ -96,7 +96,7 @@ class MyContract extends SmartContract {
     this.x.set(x.assertAlmostReduced());
   }
 }
-MyContract.analyzeMethods(); // works
+await MyContract.analyzeMethods(); // works
 
 // btw - we support any finite field up to 259 bits. for example, the seqp256k1 base field:
 let Fseqp256k1 = createForeignField((1n << 256n) - (1n << 32n) - 0b1111010001n);

src/examples/crypto/sha256/run.ts

@@ -6,7 +6,7 @@ console.timeEnd('compile');
 
 let preimage = Bytes12.fromString('hello world!');
 
-console.log('sha256 rows:', SHA256Program.analyzeMethods().sha256.rows);
+console.log('sha256 rows:', (await SHA256Program.analyzeMethods()).sha256.rows);
 
 console.time('prove');
 let proof = await SHA256Program.sha256(preimage);

src/examples/encryption.ts

@@ -30,7 +30,7 @@ console.log(`Recovered message: "${decryptedMessage}"`);
 
 // the same but in a checked computation
 
-Provable.runAndCheck(() => {
+await Provable.runAndCheck(() => {
   // encrypt
   let cipherText = Encryption.encrypt(messageFields, publicKey);
 
@@ -76,7 +76,7 @@ console.log(`Recovered message: "${decryptedMessage}"`);
 
 // the same but in a checked computation
 
-Provable.runAndCheck(() => {
+await Provable.runAndCheck(() => {
   // encrypt
   let cipherText = Encryption.encrypt(messageFields, publicKey);
 

src/examples/internals/advanced-provable-types.ts

@@ -61,7 +61,7 @@ expect(accountUpdateRecovered.lazyAuthorization).not.toEqual(
  * -) witness() and asProver() blocks are executed
  * -) constraints are checked; failing assertions throw an error
  */
-Provable.runAndCheck(() => {
+await Provable.runAndCheck(() => {
   /**
    * Provable.witness() is used to introduce all values to the circuit which are not hard-coded constants.
    *
@@ -99,7 +99,7 @@ Provable.runAndCheck(() => {
  * -) fields don't have actual values attached to them; they're purely abstract variables
  * -) constraints are not checked
  */
-let result = Provable.constraintSystem(() => {
+let result = await Provable.constraintSystem(() => {
   /**
    * In compile mode, witness() returns
    * - abstract variables without values for fields
@@ -140,7 +140,7 @@ console.log(
  *
  * This is why we have this custom way of witnessing account updates, with the `skipCheck` option.
  */
-result = Provable.constraintSystem(() => {
+result = await Provable.constraintSystem(() => {
   let { accountUpdate: accountUpdateWitness } = AccountUpdate.witness(
     Empty,
     () => ({ accountUpdate, result: undefined }),
@@ -156,7 +156,7 @@ console.log(
  * To relate an account update to the hash which is the public input, we need to perform the hash in-circuit.
  * This is takes several 100 constraints, and is basically the minimal size of a zkApp method.
  */
-result = Provable.constraintSystem(() => {
+result = await Provable.constraintSystem(() => {
   let { accountUpdate: accountUpdateWitness } = AccountUpdate.witness(
     Empty,
     () => ({ accountUpdate, result: undefined }),

src/examples/matrix-mul.ts

@@ -55,9 +55,9 @@ function circuit(): Field[][] {
   return matrixMul(x, y);
 }
 
-let { rows } = Provable.constraintSystem(circuit);
+let { rows } = await Provable.constraintSystem(circuit);
 let result: Field[][];
-Provable.runAndCheck(() => {
+await Provable.runAndCheck(() => {
   let result_ = circuit();
   Provable.asProver(() => {
     result = result_.map((x) => x.map((y) => y.toConstant()));

src/examples/simple-zkapp.ts

@@ -92,7 +92,7 @@ if (doProofs) {
   await SimpleZkapp.compile();
   console.timeEnd('compile');
 } else {
-  SimpleZkapp.analyzeMethods();
+  await SimpleZkapp.analyzeMethods();
 }
 
 console.log('deploy');

src/examples/zkapps/dex/happy-path-with-proofs.ts

@@ -21,9 +21,9 @@ let tx, balances, oldBalances;
 
 let { Dex, DexTokenHolder, getTokenBalances } = createDex();
 
-TokenContract.analyzeMethods();
-DexTokenHolder.analyzeMethods();
-Dex.analyzeMethods();
+await TokenContract.analyzeMethods();
+await DexTokenHolder.analyzeMethods();
+await Dex.analyzeMethods();
 
 if (proofsEnabled) {
   tic('compile (token)');

src/examples/zkapps/dex/run-live.ts

@@ -47,9 +47,9 @@ if (!useCustomLocalNetwork) {
   await ensureFundedAccount(senderKey.toBase58());
 }
 
-TokenContract.analyzeMethods();
-DexTokenHolder.analyzeMethods();
-Dex.analyzeMethods();
+await TokenContract.analyzeMethods();
+await DexTokenHolder.analyzeMethods();
+await Dex.analyzeMethods();
 
 tic('compile (token)');
 await TokenContract.compile();

src/examples/zkapps/dex/run.ts

@@ -24,7 +24,7 @@ console.log('TOKEN X ID\t', TokenId.toBase58(tokenIds.X));
 console.log('TOKEN Y ID\t', TokenId.toBase58(tokenIds.Y));
 console.log('-------------------------------------------------');
 
-TokenContract.analyzeMethods();
+await TokenContract.analyzeMethods();
 if (proofsEnabled) {
   console.log('compile (token)...');
   await TokenContract.compile();
@@ -57,8 +57,8 @@ async function main({ withVesting }: { withVesting: boolean }) {
   let { Dex, DexTokenHolder, getTokenBalances } = createDex(options);
 
   // analyze methods for quick error feedback
-  DexTokenHolder.analyzeMethods();
-  Dex.analyzeMethods();
+  await DexTokenHolder.analyzeMethods();
+  await Dex.analyzeMethods();
 
   if (proofsEnabled) {
     // compile & deploy all zkApps

src/examples/zkapps/dex/upgradability.ts

@@ -38,8 +38,8 @@ async function atomicActionsTest({ withVesting }: { withVesting: boolean }) {
   let { Dex, DexTokenHolder, getTokenBalances } = createDex(options);
 
   // analyze methods for quick error feedback
-  DexTokenHolder.analyzeMethods();
-  Dex.analyzeMethods();
+  await DexTokenHolder.analyzeMethods();
+  await Dex.analyzeMethods();
 
   if (proofsEnabled) {
     // compile & deploy all zkApps
@@ -249,8 +249,8 @@ async function upgradeabilityTests({ withVesting }: { withVesting: boolean }) {
   } = createDex(options);
 
   // analyze methods for quick error feedback
-  DexTokenHolder.analyzeMethods();
-  Dex.analyzeMethods();
+  await DexTokenHolder.analyzeMethods();
+  await Dex.analyzeMethods();
 
   // compile & deploy all zkApps
   console.log('compile (token contract)...');

src/examples/zkapps/reducer/actions-as-merkle-list.ts

@@ -119,7 +119,7 @@ let zkapp = new ActionsContract(zkappAddress);
 await ActionsContract.compile();
 console.log(
   `rows for ${MAX_UPDATES_WITH_ACTIONS} updates with actions`,
-  ActionsContract.analyzeMethods().assertContainsAddress.rows
+  (await ActionsContract.analyzeMethods()).assertContainsAddress.rows
 );
 let deployTx = await Mina.transaction(sender, () => zkapp.deploy());
 await deployTx.sign([senderKey, zkappKey]).send();

src/examples/zkapps/reducer/map.ts

@@ -86,7 +86,7 @@ let k = 1 << 4;
 
 let Local = Mina.LocalBlockchain();
 Mina.setActiveInstance(Local);
-let cs = StorageContract.analyzeMethods();
+let cs = await StorageContract.analyzeMethods();
 
 console.log(`method size for a "mapping" contract with ${k} entries`);
 console.log('get rows:', cs['get'].rows);

src/examples/zkapps/reducer/reducer-composite.ts

@@ -7,7 +7,6 @@ import {
   SmartContract,
   Mina,
   AccountUpdate,
-  isReady,
   Bool,
   Struct,
   Reducer,
@@ -16,8 +15,6 @@ import {
 import assert from 'node:assert/strict';
 import { getProfiler } from '../../utils/profiler.js';
 
-await isReady;
-
 class MaybeIncrement extends Struct({
   isIncrement: Bool,
   otherData: Field,

src/examples/zkapps/reducer/reducer.ts

@@ -83,7 +83,7 @@ if (doProofs) {
   // TODO: if we don't do this, then `analyzeMethods()` will be called during `runUnchecked()` in the tx callback below,
   // which currently fails due to `finalize_is_running` in snarky not resetting internal state, and instead setting is_running unconditionally to false,
   // so we can't nest different snarky circuit runners
-  CounterZkapp.analyzeMethods();
+  await CounterZkapp.analyzeMethods();
 }
 
 console.log('deploy');

src/examples/zkapps/sudoku/index.ts

@@ -13,7 +13,7 @@ const zkAppAddress = zkAppPrivateKey.toPublicKey();
 // create an instance of the smart contract
 const zkApp = new SudokuZkApp(zkAppAddress);
 
-let methods = SudokuZkApp.analyzeMethods();
+let methods = await SudokuZkApp.analyzeMethods();
 console.log(
   'first 5 gates of submitSolution method:',
   ...methods.submitSolution.gates.slice(0, 5)

src/examples/zkprogram/gadgets.ts

@@ -1,6 +1,6 @@
 import { Field, Provable, Gadgets, ZkProgram } from 'o1js';
 
-let cs = Provable.constraintSystem(() => {
+let cs = await Provable.constraintSystem(() => {
   let f = Provable.witness(Field, () => Field(12));
 
   let res1 = Gadgets.rotate64(f, 2, 'left');

src/lib/account-update.unit-test.ts

@@ -53,17 +53,15 @@ function createAccountUpdate() {
   let accountUpdate = createAccountUpdate();
 
   // TODO remove restriction "This function can't be run outside of a checked computation."
-  Provable.runAndCheck(() => {
-    let hash = accountUpdate.hash();
+  let hash = accountUpdate.hash();
 
-    // if we clone the accountUpdate, hash should be the same
-    let accountUpdate2 = AccountUpdate.clone(accountUpdate);
-    expect(accountUpdate2.hash()).toEqual(hash);
+  // if we clone the accountUpdate, hash should be the same
+  let accountUpdate2 = AccountUpdate.clone(accountUpdate);
+  expect(accountUpdate2.hash()).toEqual(hash);
 
-    // if we change something on the cloned accountUpdate, the hash should become different
-    AccountUpdate.setValue(accountUpdate2.update.appState[0], Field(1));
-    expect(accountUpdate2.hash()).not.toEqual(hash);
-  });
+  // if we change something on the cloned accountUpdate, the hash should become different
+  AccountUpdate.setValue(accountUpdate2.update.appState[0], Field(1));
+  expect(accountUpdate2.hash()).not.toEqual(hash);
 }
 
 // converts account update to a public input that's consistent with the ocaml implementation

src/lib/circuit-value.test.ts

@@ -15,7 +15,7 @@ describe('circuit', () => {
   });
 
   it('Provable.if in snark', () => {
-    Provable.runAndCheck(() => {
+    Provable.runAndCheckSync(() => {
       let x = Provable.witness(Int64, () => Int64.from(-1));
       let y = Provable.witness(Int64, () => Int64.from(-2));
       let b = Provable.witness(Bool, () => Bool(true));
@@ -81,7 +81,7 @@ describe('circuit', () => {
   it('Provable.assertEqual', () => {
     const FieldAndBool = Struct({ x: Field, b: Bool });
 
-    Provable.runAndCheck(() => {
+    Provable.runAndCheckSync(() => {
       let x = Provable.witness(Field, () => Field(1));
       let b = Provable.witness(Bool, () => Bool(true));
 
@@ -123,7 +123,7 @@ describe('circuit', () => {
       });
     }
 
-    Provable.runAndCheck(() => {
+    Provable.runAndCheckSync(() => {
       let x = Provable.witness(Field, () => Field(1));
       let b = Provable.witness(Bool, () => Bool(true));
       let pk = Provable.witness(PublicKey, () => pk1);