@aztec/end-to-end 0.87.5 → 0.87.7

package/dest/bench/client_flows/data_extractor.js

@@ -4,86 +4,14 @@ import { createLogger, logger } from '@aztec/foundation/log';
 import { Timer } from '@aztec/foundation/timer';
 import { WASMSimulator } from '@aztec/simulator/client';
 import { Decoder } from 'msgpackr';
-import assert from 'node:assert';
 import { readFile, readdir, writeFile } from 'node:fs/promises';
 import { join } from 'node:path';
-const logLevel = [
-    'silent',
-    'fatal',
-    'error',
-    'warn',
-    'info',
-    'verbose',
-    'debug',
-    'trace'
-];
-const GATE_TYPES = [
-    'ecc_op',
-    'busread',
-    'lookup',
-    'pub_inputs',
-    'arithmetic',
-    'delta_range',
-    'elliptic',
-    'aux',
-    'poseidon2_external',
-    'poseidon2_internal',
-    'overflow'
-];
-export class ProxyLogger {
-    static instance;
-    logs = [];
-    constructor(){}
-    static create() {
-        ProxyLogger.instance = new ProxyLogger();
-    }
-    static getInstance() {
-        return ProxyLogger.instance;
-    }
-    createLogger(prefix) {
-        return new Proxy(createLogger(prefix), {
-            get: (target, prop)=>{
-                if (logLevel.includes(prop)) {
-                    return function(...data) {
-                        const loggingFn = prop;
-                        const args = [
-                            loggingFn,
-                            prefix,
-                            ...data
-                        ];
-                        ProxyLogger.getInstance().handleLog(...args);
-                        target[loggingFn].call(this, ...[
-                            data[0],
-                            data[1]
-                        ]);
-                    };
-                } else {
-                    return target[prop];
-                }
-            }
-        });
-    }
-    handleLog(type, prefix, message, data) {
-        this.logs.unshift({
-            type,
-            prefix,
-            message,
-            data,
-            timestamp: Date.now()
-        });
-    }
-    flushLogs() {
-        this.logs = [];
-    }
-    getLogs() {
-        return this.logs;
-    }
-}
+import { ProxyLogger, generateBenchmark } from './benchmark.js';
 async function createProver(config = {}, log) {
-    const simulationProvider = new WASMSimulator();
+    const simulator = new WASMSimulator();
     if (!config.bbBinaryPath || !config.bbWorkingDirectory) {
         return {
-            prover: new BBWASMBundlePrivateKernelProver(simulationProvider, 16, log),
+            prover: new BBWASMBundlePrivateKernelProver(simulator, 16, log),
             type: 'wasm'
         };
     } else {
@@ -92,30 +20,11 @@ async function createProver(config = {}, log) {
             prover: await BBNativePrivateKernelProver.new({
                 bbSkipCleanup: false,
                 ...bbConfig
-            }, simulationProvider, log),
+            }, simulator, log),
             type: 'native'
         };
     }
 }
-function getMinimumTrace(logs) {
-    const minimumMessage = 'Trace details:';
-    const minimumMessageIndex = logs.findIndex((log)=>log.message.includes(minimumMessage));
-    const candidateLogs = logs.slice(minimumMessageIndex - GATE_TYPES.length, minimumMessageIndex);
-    const traceLogs = candidateLogs.filter((log)=>GATE_TYPES.some((type)=>log.message.includes(type))).map((log)=>log.message.split(/\t|\n/)).flat().map((log)=>log.replace(/\(mem: .*\)/, '').trim()).filter(Boolean);
-    const traceSizes = traceLogs.map((log)=>{
-        const [gateType, gateSizeStr] = log.replace(/\n.*\)$/, '').replace(/bb - /, '').split(':').map((s)=>s.trim());
-        const gateSize = parseInt(gateSizeStr);
-        assert(GATE_TYPES.includes(gateType), `Gate type ${gateType} is not recognized`);
-        return {
-            [gateType]: gateSize
-        };
-    });
-    assert(traceSizes.length === GATE_TYPES.length, 'Decoded trace sizes do not match expected amount of gate types');
-    return traceSizes.reduce((acc, curr)=>({
-            ...acc,
-            ...curr
-        }), {});
-}
 async function main() {
     ProxyLogger.create();
     const proxyLogger = ProxyLogger.getInstance();
@@ -146,7 +55,7 @@ async function main() {
         const profileFile = await readFile(join(ivcFolder, flow, 'profile.json'));
         const profile = JSON.parse(profileFile.toString());
         const privateExecutionSteps = profile.steps.map((step, i)=>({
-                functionName: step.fnName,
+                functionName: step.functionName,
                 gateCount: step.gateCount,
                 bytecode: stepsFromFile[i].bytecode,
                 // TODO(AD) do we still want to take this from witness.json?
@@ -154,10 +63,9 @@ async function main() {
                 vk: stepsFromFile[i].vk,
                 timings: {
                     witgen: step.timings.witgen,
-                    gateCount: step.timings.witgen
+                    gateCount: step.timings.gateCount
                 }
             }));
-        let stats;
         let error;
         let currentLogs = [];
         let provingTime;
@@ -172,35 +80,10 @@ async function main() {
         // Extract logs from this run from the proxy and write them to disk unconditionally
         currentLogs = proxyLogger.getLogs();
         await writeFile(join(ivcFolder, flow, 'logs.json'), JSON.stringify(currentLogs, null, 2));
-        if (!error) {
-            stats = currentLogs[0].data;
+        if (!profile.stats.timings.proving) {
+            profile.stats.timings.proving = provingTime;
         }
-        const minimumTrace = getMinimumTrace(currentLogs);
-        const steps = profile.steps.reduce((acc, step, i)=>{
-            const previousAccGateCount = i === 0 ? 0 : acc[i - 1].accGateCount;
-            return [
-                ...acc,
-                {
-                    fnName: step.fnName,
-                    gateCount: step.gateCount,
-                    accGateCount: previousAccGateCount + step.gateCount,
-                    timings: {
-                        witgen: step.timings.witgen
-                    }
-                }
-            ];
-        }, []);
-        const totalGateCount = steps[steps.length - 1].accGateCount;
-        const benchmark = {
-            syncTime: profile.syncTime,
-            provingTime,
-            proverType,
-            minimumTrace: minimumTrace,
-            totalGateCount,
-            stats,
-            steps,
-            error
-        };
+        const benchmark = generateBenchmark(flow, currentLogs, profile.stats, privateExecutionSteps, proverType, error);
         await writeFile(join(ivcFolder, flow, 'benchmark.json'), JSON.stringify(benchmark, null, 2));
         proxyLogger.flushLogs();
     }

package/dest/bench/utils.d.ts

@@ -23,6 +23,13 @@ type MetricFilter = {
     name: string;
     unit?: string;
 };
+export type GithubActionBenchmarkResult = {
+    name: string;
+    value: number;
+    range?: string;
+    unit: string;
+    extra?: string;
+};
 /**
  * Assembles and sends multiple transactions simultaneously to the node in context.
  * Each tx is the result of calling makeCall.

package/dest/bench/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/bench/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,SAAS,EAAa,KAAK,MAAM,EAAE,KAAK,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAExF,OAAO,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AAClF,OAAO,EAAE,KAAK,UAAU,EAA2C,MAAM,mBAAmB,CAAC;AAC7F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAA4C,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAKxH,OAAO,EAAE,KAAK,eAAe,EAAE,KAAK,YAAY,EAAS,MAAM,sBAAsB,CAAC;AAEtF;;GAEG;AACH,wBAAsB,cAAc,CAClC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG;IAC5B,6BAA6B,CAAC,OAAO,EAAE,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,CAAC;IACtE,kFAAkF;IAClF,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;;;;;GAwBF;AAED,KAAK,YAAY,GAAG;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAuEF;;;;;;;;GAQG;AACH,wBAAsB,OAAO,CAC3B,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,eAAe,EACxB,QAAQ,EAAE,oBAAoB,EAC9B,kBAAkB,GAAE,OAAe,GAClC,OAAO,CAAC,MAAM,EAAE,CAAC,CAMnB;AAED,wBAAsB,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,CAAC,EAAE,QAAQ,iBAI3F;AAED;;;;;;GAMG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC,CAiBvG"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/bench/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,SAAS,EAAa,KAAK,MAAM,EAAE,KAAK,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAExF,OAAO,EAAE,oBAAoB,EAAE,MAAM,4CAA4C,CAAC;AAClF,OAAO,EAAE,KAAK,UAAU,EAA2C,MAAM,mBAAmB,CAAC;AAC7F,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,KAAK,EAA4C,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAKxH,OAAO,EAAE,KAAK,eAAe,EAAE,KAAK,YAAY,EAAS,MAAM,sBAAsB,CAAC;AAEtF;;GAEG;AACH,wBAAsB,cAAc,CAClC,IAAI,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG;IAC5B,6BAA6B,CAAC,OAAO,EAAE,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,CAAC;IACtE,kFAAkF;IAClF,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;;;;;GAwBF;AAED,KAAK,YAAY,GAAG;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,MAAM,CAAC;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAGF,MAAM,MAAM,2BAA2B,GAAG;IACxC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC;AA8DF;;;;;;;;GAQG;AACH,wBAAsB,OAAO,CAC3B,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,eAAe,EACxB,QAAQ,EAAE,oBAAoB,EAC9B,kBAAkB,GAAE,OAAe,GAClC,OAAO,CAAC,MAAM,EAAE,CAAC,CAMnB;AAED,wBAAsB,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE,UAAU,CAAC,EAAE,QAAQ,iBAI3F;AAED;;;;;;GAMG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC,CAiBvG"}

package/dest/e2e_p2p/shared.js

@@ -27,7 +27,9 @@ export const submitComplexTxsTo = async (logger, spamContract, numTxs, opts = {}
 export const createPXEServiceAndSubmitTransactions = async (logger, node, numTxs, fundedAccount)=>{
     const rpcConfig = getRpcConfig();
     rpcConfig.proverEnabled = false;
-    const pxeService = await createPXEService(node, rpcConfig, true);
+    const pxeService = await createPXEService(node, rpcConfig, {
+        useLogSuffix: true
+    });
     const account = await getSchnorrAccount(pxeService, fundedAccount.secret, fundedAccount.signingKey, fundedAccount.salt);
     await account.register();
     const wallet = await account.getWallet();
@@ -41,7 +43,9 @@ export const createPXEServiceAndSubmitTransactions = async (logger, node, numTxs
 export async function createPXEServiceAndPrepareTransactions(logger, node, numTxs, fundedAccount) {
     const rpcConfig = getRpcConfig();
     rpcConfig.proverEnabled = false;
-    const pxe = await createPXEService(node, rpcConfig, true);
+    const pxe = await createPXEService(node, rpcConfig, {
+        useLogSuffix: true
+    });
     const account = await getSchnorrAccount(pxe, fundedAccount.secret, fundedAccount.signingKey, fundedAccount.salt);
     await account.register();
     const wallet = await account.getWallet();

package/dest/fixtures/utils.js

@@ -20,9 +20,9 @@ import { SponsoredFPCContract } from '@aztec/noir-contracts.js/SponsoredFPC';
 import { getVKTreeRoot } from '@aztec/noir-protocol-circuits-types/vk-tree';
 import { protocolContractTreeRoot } from '@aztec/protocol-contracts';
 import { createProverNode } from '@aztec/prover-node';
-import { createPXEServiceWithSimulationProvider, getPXEServiceConfig } from '@aztec/pxe/server';
-import { WASMSimulator } from '@aztec/simulator/client';
-import { SimulationProviderRecorderWrapper } from '@aztec/simulator/testing';
+import { createPXEServiceWithSimulator, getPXEServiceConfig } from '@aztec/pxe/server';
+import { MemoryCircuitRecorder, SimulatorRecorderWrapper, WASMSimulator } from '@aztec/simulator/client';
+import { FileCircuitRecorder } from '@aztec/simulator/testing';
 import { getContractClassFromArtifact, getContractInstanceFromDeployParams } from '@aztec/stdlib/contract';
 import { getConfigEnvVars as getTelemetryConfig, initTelemetryClient } from '@aztec/telemetry-client';
 import { BenchmarkTelemetryClient } from '@aztec/telemetry-client/bench';
@@ -96,9 +96,12 @@ export const setupL1Contracts = async (l1RpcUrls, account, logger, args = {}, ch
     if (!configuredDataDirectory) {
         pxeServiceConfig.dataDirectory = path.join(tmpdir(), randomBytes(8).toString('hex'));
     }
-    const simulationProvider = new WASMSimulator();
-    const simulationProviderWithRecorder = new SimulationProviderRecorderWrapper(simulationProvider);
-    const pxe = await createPXEServiceWithSimulationProvider(aztecNode, simulationProviderWithRecorder, pxeServiceConfig, useLogSuffix);
+    const simulator = new WASMSimulator();
+    const recorder = process.env.CIRCUIT_RECORD_DIR ? new FileCircuitRecorder(process.env.CIRCUIT_RECORD_DIR) : new MemoryCircuitRecorder();
+    const simulatorWithRecorder = new SimulatorRecorderWrapper(simulator, recorder);
+    const pxe = await createPXEServiceWithSimulator(aztecNode, simulatorWithRecorder, pxeServiceConfig, {
+        useLogSuffix
+    });
     const teardown = async ()=>{
         if (!configuredDataDirectory) {
             try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aztec/end-to-end",
-  "version": "0.87.5",
+  "version": "0.87.7",
   "type": "module",
   "exports": "./dest/index.js",
   "inherits": [
@@ -25,39 +25,39 @@
     "formatting": "run -T prettier --check ./src && run -T eslint ./src"
   },
   "dependencies": {
-    "@aztec/accounts": "0.87.5",
-    "@aztec/archiver": "0.87.5",
-    "@aztec/aztec": "0.87.5",
-    "@aztec/aztec-node": "0.87.5",
-    "@aztec/aztec.js": "0.87.5",
-    "@aztec/bb-prover": "0.87.5",
-    "@aztec/blob-lib": "0.87.5",
-    "@aztec/blob-sink": "0.87.5",
-    "@aztec/bot": "0.87.5",
-    "@aztec/cli": "0.87.5",
-    "@aztec/constants": "0.87.5",
-    "@aztec/entrypoints": "0.87.5",
-    "@aztec/epoch-cache": "0.87.5",
-    "@aztec/ethereum": "0.87.5",
-    "@aztec/foundation": "0.87.5",
-    "@aztec/kv-store": "0.87.5",
-    "@aztec/l1-artifacts": "0.87.5",
-    "@aztec/merkle-tree": "0.87.5",
-    "@aztec/noir-contracts.js": "0.87.5",
-    "@aztec/noir-noirc_abi": "0.87.5",
-    "@aztec/noir-protocol-circuits-types": "0.87.5",
-    "@aztec/noir-test-contracts.js": "0.87.5",
-    "@aztec/p2p": "0.87.5",
-    "@aztec/protocol-contracts": "0.87.5",
-    "@aztec/prover-client": "0.87.5",
-    "@aztec/prover-node": "0.87.5",
-    "@aztec/pxe": "0.87.5",
-    "@aztec/sequencer-client": "0.87.5",
-    "@aztec/simulator": "0.87.5",
-    "@aztec/stdlib": "0.87.5",
-    "@aztec/telemetry-client": "0.87.5",
-    "@aztec/validator-client": "0.87.5",
-    "@aztec/world-state": "0.87.5",
+    "@aztec/accounts": "0.87.7",
+    "@aztec/archiver": "0.87.7",
+    "@aztec/aztec": "0.87.7",
+    "@aztec/aztec-node": "0.87.7",
+    "@aztec/aztec.js": "0.87.7",
+    "@aztec/bb-prover": "0.87.7",
+    "@aztec/blob-lib": "0.87.7",
+    "@aztec/blob-sink": "0.87.7",
+    "@aztec/bot": "0.87.7",
+    "@aztec/cli": "0.87.7",
+    "@aztec/constants": "0.87.7",
+    "@aztec/entrypoints": "0.87.7",
+    "@aztec/epoch-cache": "0.87.7",
+    "@aztec/ethereum": "0.87.7",
+    "@aztec/foundation": "0.87.7",
+    "@aztec/kv-store": "0.87.7",
+    "@aztec/l1-artifacts": "0.87.7",
+    "@aztec/merkle-tree": "0.87.7",
+    "@aztec/noir-contracts.js": "0.87.7",
+    "@aztec/noir-noirc_abi": "0.87.7",
+    "@aztec/noir-protocol-circuits-types": "0.87.7",
+    "@aztec/noir-test-contracts.js": "0.87.7",
+    "@aztec/p2p": "0.87.7",
+    "@aztec/protocol-contracts": "0.87.7",
+    "@aztec/prover-client": "0.87.7",
+    "@aztec/prover-node": "0.87.7",
+    "@aztec/pxe": "0.87.7",
+    "@aztec/sequencer-client": "0.87.7",
+    "@aztec/simulator": "0.87.7",
+    "@aztec/stdlib": "0.87.7",
+    "@aztec/telemetry-client": "0.87.7",
+    "@aztec/validator-client": "0.87.7",
+    "@aztec/world-state": "0.87.7",
     "@iarna/toml": "^2.2.5",
     "@jest/globals": "^29.5.0",
     "@noble/curves": "^1.0.0",

package/src/bench/client_flows/benchmark.ts

@@ -0,0 +1,336 @@
+import type {
+  ContractFunctionInteraction,
+  DeployMethod,
+  DeployOptions,
+  Logger,
+  ProfileMethodOptions,
+} from '@aztec/aztec.js';
+import { createLogger } from '@aztec/foundation/log';
+import { type PrivateExecutionStep, serializePrivateExecutionSteps } from '@aztec/stdlib/kernel';
+import type { ProvingStats, ProvingTimings, SimulationStats, SimulationTimings } from '@aztec/stdlib/tx';
+
+import assert from 'node:assert';
+import { mkdir, writeFile } from 'node:fs/promises';
+import { join } from 'node:path';
+
+import type { GithubActionBenchmarkResult } from '../utils.js';
+
+const logger = createLogger('bench:profile_capture');
+
+const logLevel = ['silent', 'fatal', 'error', 'warn', 'info', 'verbose', 'debug', 'trace'] as const;
+type LogLevel = (typeof logLevel)[number];
+
+export type Log = {
+  type: LogLevel;
+  timestamp: number;
+  prefix: string;
+  message: string;
+  data: any;
+};
+
+const GATE_TYPES = [
+  'ecc_op',
+  'busread',
+  'lookup',
+  'pub_inputs',
+  'arithmetic',
+  'delta_range',
+  'elliptic',
+  'aux',
+  'poseidon2_external',
+  'poseidon2_internal',
+  'overflow',
+] as const;
+
+type GateType = (typeof GATE_TYPES)[number];
+
+type StructuredTrace = {
+  [k in GateType]: number;
+};
+
+export class ProxyLogger {
+  private static instance: ProxyLogger;
+  private logs: Log[] = [];
+
+  private constructor() {}
+
+  static create() {
+    ProxyLogger.instance = new ProxyLogger();
+  }
+
+  static getInstance() {
+    return ProxyLogger.instance;
+  }
+
+  createLogger(prefix: string): Logger {
+    return new Proxy(createLogger(prefix), {
+      get: (target: Logger, prop: keyof Logger) => {
+        if (logLevel.includes(prop as (typeof logLevel)[number])) {
+          return function (this: Logger, ...data: Parameters<Logger[LogLevel]>) {
+            const loggingFn = prop as LogLevel;
+            const args = [loggingFn, prefix, ...data] as Parameters<ProxyLogger['handleLog']>;
+            ProxyLogger.getInstance().handleLog(...args);
+            target[loggingFn].call(this, ...[data[0], data[1]]);
+          };
+        } else {
+          return target[prop];
+        }
+      },
+    });
+  }
+
+  private handleLog(type: (typeof logLevel)[number], prefix: string, message: string, data: any) {
+    this.logs.unshift({ type, prefix, message, data, timestamp: Date.now() });
+  }
+
+  public flushLogs() {
+    this.logs = [];
+  }
+
+  public getLogs() {
+    return this.logs;
+  }
+}
+
+export type ProverType = 'wasm' | 'native';
+
+type CallRecording = {
+  // Number of times the function has been called
+  calls: number;
+  // Maximum time taken by the function (in ms)
+  max: number;
+  // Minimum time taken by the function (in ms)
+  min: number;
+  // Average time taken by the function (in ms)
+  avg: number;
+  // Total time spent in the function, computed as sum of all calls (in ms)
+  total: number;
+};
+
+type Step = Pick<PrivateExecutionStep, 'functionName' | 'gateCount'> & {
+  time: number;
+  accGateCount?: number;
+  oracles: Record<string, CallRecording>;
+};
+
+type ClientFlowBenchmark = {
+  name: string;
+  timings: Omit<ProvingTimings & SimulationTimings, 'perFunction'> & { witgen: number };
+  maxMemory: number;
+  rpc: Record<string, CallRecording>;
+  proverType: ProverType;
+  minimumTrace: StructuredTrace;
+  totalGateCount: number;
+  steps: Step[];
+  error: string | undefined;
+};
+
+function getMinimumTrace(logs: Log[]): StructuredTrace {
+  const minimumMessage = 'Minimum required block sizes for structured trace';
+  const minimumMessageIndex = logs.findIndex(log => log.message.includes(minimumMessage));
+  const candidateLogs = logs.slice(minimumMessageIndex - GATE_TYPES.length, minimumMessageIndex + 5);
+
+  const traceLogs = candidateLogs
+    .filter(log => GATE_TYPES.some(type => log.message.includes(type)))
+    .map(log => log.message.split(/\t|\n/))
+    .flat()
+    .map(log => log.replace(/\(mem: .*\)/, '').trim())
+    .filter(Boolean);
+
+  const traceSizes = traceLogs.map(log => {
+    const [gateType, gateSizeStr] = log
+      .replace(/\n.*\)$/, '')
+      .replace(/bb - /, '')
+      .split(':')
+      .map(s => s.trim());
+    const gateSize = parseInt(gateSizeStr);
+    assert(GATE_TYPES.includes(gateType as GateType), `Gate type ${gateType} is not recognized`);
+    return { [gateType]: gateSize };
+  });
+
+  assert(traceSizes.length === GATE_TYPES.length, 'Decoded trace sizes do not match expected amount of gate types');
+  return traceSizes.reduce((acc, curr) => ({ ...acc, ...curr }), {}) as StructuredTrace;
+}
+
+function getMaxMemory(logs: Log[]): number {
+  const candidateLogs = logs.slice(0, 100).filter(log => /\(mem: .*MiB\)/.test(log.message));
+  const usage = candidateLogs.map(log => {
+    const memStr = log ? log.message.slice(log.message.indexOf('(mem: ') + 6, log.message.indexOf('MiB') - 3) : '';
+    return memStr ? parseInt(memStr) : 0;
+  });
+  return Math.max(...usage);
+}
+
+export function generateBenchmark(
+  flow: string,
+  logs: Log[],
+  stats: ProvingStats | SimulationStats,
+  privateExecutionSteps: PrivateExecutionStep[],
+  proverType: ProverType,
+  error: string | undefined,
+): ClientFlowBenchmark {
+  let maxMemory = 0;
+  let minimumTrace: StructuredTrace;
+  try {
+    minimumTrace = getMinimumTrace(logs);
+    maxMemory = getMaxMemory(logs);
+  } catch {
+    logger.warn(`Failed obtain minimum trace and max memory for ${flow}. Did you run with REAL_PROOFS=1?`);
+  }
+
+  const steps = privateExecutionSteps.reduce<Step[]>((acc, step, i) => {
+    const previousAccGateCount = i === 0 ? 0 : acc[i - 1].accGateCount!;
+    return [
+      ...acc,
+      {
+        functionName: step.functionName,
+        gateCount: step.gateCount,
+        accGateCount: previousAccGateCount + step.gateCount!,
+        time: step.timings.witgen,
+        oracles: Object.entries(step.timings.oracles ?? {}).reduce(
+          (acc, [oracleName, oracleData]) => {
+            const total = oracleData.times.reduce((sum, time) => sum + time, 0);
+            const calls = oracleData.times.length;
+            acc[oracleName] = {
+              calls,
+              max: Math.max(...oracleData.times),
+              min: Math.min(...oracleData.times),
+              total,
+              avg: total / calls,
+            };
+            return acc;
+          },
+          {} as Record<string, CallRecording>,
+        ),
+      },
+    ];
+  }, []);
+  const timings = stats.timings;
+  const totalGateCount = steps[steps.length - 1].accGateCount;
+  return {
+    name: flow,
+    timings: {
+      total: timings.total,
+      sync: timings.sync!,
+      proving: (timings as ProvingTimings).proving,
+      unaccounted: timings.unaccounted,
+      witgen: timings.perFunction.reduce((acc, fn) => acc + fn.time, 0),
+    },
+    rpc: Object.entries(stats.nodeRPCCalls ?? {}).reduce(
+      (acc, [RPCName, RPCCalls]) => {
+        const total = RPCCalls.times.reduce((sum, time) => sum + time, 0);
+        const calls = RPCCalls.times.length;
+        acc[RPCName] = {
+          calls,
+          max: Math.max(...RPCCalls.times),
+          min: Math.min(...RPCCalls.times),
+          total,
+          avg: total / calls,
+        };
+        return acc;
+      },
+      {} as Record<string, CallRecording>,
+    ),
+    maxMemory,
+    proverType,
+    minimumTrace: minimumTrace!,
+    totalGateCount: totalGateCount!,
+    steps,
+    error,
+  };
+}
+
+export function convertProfileToGHBenchmark(benchmark: ClientFlowBenchmark): GithubActionBenchmarkResult[] {
+  const totalRPCCalls = Object.values(benchmark.rpc).reduce((acc, call) => acc + call.calls, 0);
+  const benches = [
+    {
+      name: `${benchmark.name}/witgen`,
+      value: benchmark.timings.witgen,
+      unit: 'ms',
+    },
+
+    {
+      name: `${benchmark.name}/total`,
+      value: benchmark.timings.total,
+      unit: 'ms',
+    },
+    {
+      name: `${benchmark.name}/sync`,
+      value: benchmark.timings.sync!,
+      unit: 'ms',
+    },
+    {
+      name: `${benchmark.name}/unaccounted`,
+      value: benchmark.timings.unaccounted,
+      unit: 'ms',
+    },
+
+    {
+      name: `${benchmark.name}/total_gate_count`,
+      value: benchmark.totalGateCount,
+      unit: 'gates',
+    },
+    {
+      name: `${benchmark.name}/rpc`,
+      value: totalRPCCalls,
+      unit: 'calls',
+    },
+  ];
+  if (benchmark.timings.proving) {
+    benches.push({
+      name: `${benchmark.name}/proving`,
+      value: benchmark.timings.proving,
+      unit: 'ms',
+    });
+  }
+  if (benchmark.maxMemory) {
+    benches.push({
+      name: `${benchmark.name}/max_memory`,
+      value: benchmark.maxMemory,
+      unit: 'MiB',
+    });
+  }
+  return benches;
+}
+
+export async function captureProfile(
+  label: string,
+  interaction: ContractFunctionInteraction | DeployMethod,
+  opts?: Omit<ProfileMethodOptions & DeployOptions, 'profileMode'>,
+  expectedSteps?: number,
+) {
+  // Make sure the proxy logger starts from a clean slate
+  ProxyLogger.getInstance().flushLogs();
+  const result = await interaction.profile({ ...opts, profileMode: 'full', skipProofGeneration: false });
+  const logs = ProxyLogger.getInstance().getLogs();
+  if (expectedSteps !== undefined && result.executionSteps.length !== expectedSteps) {
+    throw new Error(`Expected ${expectedSteps} execution steps, got ${result.executionSteps.length}`);
+  }
+  const benchmark = generateBenchmark(label, logs, result.stats, result.executionSteps, 'wasm', undefined);
+
+  const ivcFolder = process.env.CAPTURE_IVC_FOLDER;
+  if (ivcFolder) {
+    logger.info(`Capturing client ivc execution profile for ${label}`);
+
+    const resultsDirectory = join(ivcFolder, label);
+    logger.info(`Writing private execution steps to ${resultsDirectory}`);
+    await mkdir(resultsDirectory, { recursive: true });
+    // Write the client IVC files read by the prover.
+    const ivcInputsPath = join(resultsDirectory, 'ivc-inputs.msgpack');
+    await writeFile(ivcInputsPath, serializePrivateExecutionSteps(result.executionSteps));
+    await writeFile(join(resultsDirectory, 'logs.json'), JSON.stringify(logs, null, 2));
+    await writeFile(join(resultsDirectory, 'benchmark.json'), JSON.stringify(benchmark, null, 2));
+    logger.info(`Wrote private execution steps to ${resultsDirectory}`);
+  }
+
+  const benchOutput = process.env.BENCH_OUTPUT;
+  if (benchOutput) {
+    await mkdir(benchOutput, { recursive: true });
+    const ghBenchmark = convertProfileToGHBenchmark(benchmark);
+    const benchFile = join(benchOutput, `${label}.bench.json`);
+    await writeFile(benchFile, JSON.stringify(ghBenchmark));
+    logger.info(`Wrote benchmark to ${benchFile}`);
+  }
+
+  return result;
+}