From a97ea4ef459f6d14059fd4216f36a8461091c612 Mon Sep 17 00:00:00 2001 From: Alex Gherghisan Date: Wed, 24 Apr 2024 09:14:15 +0000 Subject: [PATCH] feat: configure prover as separate process --- .../aztec-node/src/aztec-node/config.ts | 14 +++- .../aztec-node/src/aztec-node/server.ts | 7 +- yarn-project/aztec/package.json | 1 + yarn-project/aztec/src/cli/cli.ts | 3 + yarn-project/aztec/src/cli/cmds/start_node.ts | 9 +++ .../aztec/src/cli/cmds/start_prover.ts | 50 +++++++++++++ yarn-project/aztec/src/cli/util.ts | 5 ++ yarn-project/aztec/tsconfig.json | 3 + .../src/interfaces/prover-client.ts | 3 + .../kernel/kernel_circuit_public_inputs.ts | 8 ++ .../public_kernel_circuit_public_inputs.ts | 8 ++ ...kernel_tail_circuit_private_inputs.test.ts | 7 ++ ...blic_kernel_tail_circuit_private_inputs.ts | 8 ++ .../composed/integration_l1_publisher.test.ts | 3 +- yarn-project/prover-client/package.json | 5 +- .../prover-client/src/dummy-prover.ts | 23 ++++++ .../prover-client/src/prover-pool/index.ts | 1 + .../src/prover-pool/prover-agent.ts | 3 +- .../src/prover-pool/prover-pool.ts | 33 ++++++++- .../src/prover-pool/proving-error.ts | 9 +++ .../prover-client/src/prover-pool/rpc.ts | 73 +++++++++++++++++++ .../src/prover/test_circuit_prover.ts | 5 +- .../prover-client/src/tx-prover/tx-prover.ts | 39 ++++------ yarn-project/yarn.lock | 1 + 24 files changed, 287 insertions(+), 34 deletions(-) create mode 100644 yarn-project/aztec/src/cli/cmds/start_prover.ts create mode 100644 yarn-project/prover-client/src/prover-pool/proving-error.ts create mode 100644 yarn-project/prover-client/src/prover-pool/rpc.ts diff --git a/yarn-project/aztec-node/src/aztec-node/config.ts b/yarn-project/aztec-node/src/aztec-node/config.ts index 125ec88b2878..4b644ca3b783 100644 --- a/yarn-project/aztec-node/src/aztec-node/config.ts +++ b/yarn-project/aztec-node/src/aztec-node/config.ts @@ -17,6 +17,8 @@ export type AztecNodeConfig = ArchiverConfig & /** A URL for an archiver service that the node will use. */ archiverUrl?: string; + + proverAgents: number; }; /** @@ -24,15 +26,21 @@ export type AztecNodeConfig = ArchiverConfig & * @returns A valid aztec node config. */ export function getConfigEnvVars(): AztecNodeConfig { - const { SEQ_DISABLED, PROVER_DISABLED } = process.env; + const { SEQ_DISABLED, PROVER_DISABLED, ARCHIVER_URL, PROVER_AGENTS = '1' } = process.env; + let proverAgents = parseInt(PROVER_AGENTS, 10); + if (Number.isNaN(proverAgents)) { + proverAgents = 1; + } + const allEnvVars: AztecNodeConfig = { ...getSequencerVars(), ...getArchiverVars(), ...getP2PConfigEnvVars(), ...getWorldStateVars(), disableSequencer: !!SEQ_DISABLED, - disableProver: !!PROVER_DISABLED, - archiverUrl: process.env.ARCHIVER_URL, + archiverUrl: ARCHIVER_URL, + disableProver: PROVER_DISABLED === '1', + proverAgents, }; return allEnvVars; diff --git a/yarn-project/aztec-node/src/aztec-node/server.ts b/yarn-project/aztec-node/src/aztec-node/server.ts index 5a57d935a312..7ec441d29a48 100644 --- a/yarn-project/aztec-node/src/aztec-node/server.ts +++ b/yarn-project/aztec-node/src/aztec-node/server.ts @@ -52,6 +52,7 @@ import { initStoreForRollup, openTmpStore } from '@aztec/kv-store/utils'; import { SHA256Trunc, StandardTree } from '@aztec/merkle-tree'; import { AztecKVTxPool, type P2P, createP2PClient } from '@aztec/p2p'; import { DummyProver, TxProver } from '@aztec/prover-client'; +import { ProverPool } from '@aztec/prover-client/prover-pool'; import { type GlobalVariableBuilder, SequencerClient, getGlobalVariableBuilder } from '@aztec/sequencer-client'; import { PublicProcessorFactory, WASMSimulator } from '@aztec/simulator'; import { @@ -151,7 +152,7 @@ export class AztecNodeService implements AztecNode { const simulationProvider = await getSimulationProvider(config, log); const prover = config.disableProver ? await DummyProver.new() - : await TxProver.new(config, worldStateSynchronizer, simulationProvider); + : await TxProver.new(worldStateSynchronizer, ProverPool.testPool(simulationProvider, config.proverAgents)); // now create the sequencer const sequencer = config.disableSequencer @@ -194,6 +195,10 @@ export class AztecNodeService implements AztecNode { return this.sequencer; } + public getProver(): ProverClient { + return this.prover; + } + /** * Method to return the currently deployed L1 contract addresses. * @returns - The currently deployed L1 contract addresses. diff --git a/yarn-project/aztec/package.json b/yarn-project/aztec/package.json index d9f79c69f3b7..9f91f14b3ea6 100644 --- a/yarn-project/aztec/package.json +++ b/yarn-project/aztec/package.json @@ -42,6 +42,7 @@ "@aztec/noir-contracts.js": "workspace:^", "@aztec/p2p": "workspace:^", "@aztec/protocol-contracts": "workspace:^", + "@aztec/prover-client": "workspace:^", "@aztec/pxe": "workspace:^", "abitype": "^0.8.11", "commander": "^11.1.0", diff --git a/yarn-project/aztec/src/cli/cli.ts b/yarn-project/aztec/src/cli/cli.ts index 525b3e5d741d..ec6a48eb709e 100644 --- a/yarn-project/aztec/src/cli/cli.ts +++ b/yarn-project/aztec/src/cli/cli.ts @@ -56,6 +56,9 @@ export function getProgram(userLog: LogFn, debugLogger: DebugLogger): Command { } else if (options.p2pBootstrap) { const { startP2PBootstrap } = await import('./cmds/start_p2p_bootstrap.js'); await startP2PBootstrap(options, signalHandlers, userLog, debugLogger); + } else if (options.prover) { + const { startProver } = await import('./cmds/start_prover.js'); + services = await startProver(options, signalHandlers, userLog); } if (services.length) { const rpcServer = createNamespacedJsonRpcServer(services, debugLogger); diff --git a/yarn-project/aztec/src/cli/cmds/start_node.ts b/yarn-project/aztec/src/cli/cmds/start_node.ts index 63a5ab7aa9b9..35fb127f6075 100644 --- a/yarn-project/aztec/src/cli/cmds/start_node.ts +++ b/yarn-project/aztec/src/cli/cmds/start_node.ts @@ -6,6 +6,7 @@ import { import { NULL_KEY } from '@aztec/ethereum'; import { type ServerList } from '@aztec/foundation/json-rpc/server'; import { type LogFn } from '@aztec/foundation/log'; +import { createProvingJobSourceServer } from '@aztec/prover-client/prover-pool'; import { type PXEServiceConfig, createPXERpcServer, getPXEServiceConfig } from '@aztec/pxe'; import { mnemonicToAccount, privateKeyToAccount } from 'viem/accounts'; @@ -64,7 +65,10 @@ export const startNode = async ( } if (!options.prover) { + userLog(`Prover is disabled, using mocked proofs`); nodeConfig.disableProver = true; + } else { + nodeConfig = mergeEnvVarsAndCliOptions(nodeConfig, parseModuleOptions(options.prover)); } if (!nodeConfig.disableSequencer && nodeConfig.disableProver) { @@ -78,6 +82,11 @@ export const startNode = async ( // Add node to services list services.push({ node: nodeServer }); + if (!nodeConfig.disableProver) { + const provingJobSource = createProvingJobSourceServer(node.getProver().getProvingJobSource()); + services.push({ provingJobSource }); + } + // Add node stop function to signal handlers signalHandlers.push(node.stop); diff --git a/yarn-project/aztec/src/cli/cmds/start_prover.ts b/yarn-project/aztec/src/cli/cmds/start_prover.ts new file mode 100644 index 000000000000..103ca97c8df0 --- /dev/null +++ b/yarn-project/aztec/src/cli/cmds/start_prover.ts @@ -0,0 +1,50 @@ +import { type ProvingJobSource } from '@aztec/circuit-types'; +import { ProverPool, createProvingJobSourceClient } from '@aztec/prover-client/prover-pool'; + +import { type ServiceStarter, parseModuleOptions } from '../util.js'; + +type ProverOptions = Partial<{ + proverUrl: string; + agents: string; + acvmBinaryPath?: string; + bbBinaryPath?: string; + simulate?: string; +}>; + +export const startProver: ServiceStarter = async (options, signalHandlers, logger) => { + const proverOptions: ProverOptions = parseModuleOptions(options.prover); + let source: ProvingJobSource; + + if (typeof proverOptions.proverUrl === 'string') { + logger(`Connecting to prover at ${proverOptions.proverUrl}`); + source = createProvingJobSourceClient(proverOptions.proverUrl, 'provingJobSource'); + } else { + throw new Error('Starting prover without an orchestrator is not supported'); + } + + const agentCount = proverOptions.agents ? parseInt(proverOptions.agents, 10) : 1; + if (agentCount === 0 || !Number.isSafeInteger(agentCount)) { + throw new Error('Cannot start prover without agents'); + } + + let pool: ProverPool; + if (proverOptions.simulate) { + pool = ProverPool.testPool(undefined, agentCount); + } else if (proverOptions.acvmBinaryPath && proverOptions.bbBinaryPath) { + pool = ProverPool.nativePool( + { + acvmBinaryPath: proverOptions.acvmBinaryPath, + bbBinaryPath: proverOptions.bbBinaryPath, + }, + agentCount, + ); + } else { + throw new Error('Cannot start prover without simulation or native prover options'); + } + + logger(`Starting ${agentCount} prover agents`); + await pool.start(source); + signalHandlers.push(() => pool.stop()); + + return Promise.resolve([]); +}; diff --git a/yarn-project/aztec/src/cli/util.ts b/yarn-project/aztec/src/cli/util.ts index 1a14868638be..db16f546c7b5 100644 --- a/yarn-project/aztec/src/cli/util.ts +++ b/yarn-project/aztec/src/cli/util.ts @@ -3,10 +3,15 @@ import { type AztecNodeConfig } from '@aztec/aztec-node'; import { type AccountManager, type Fr } from '@aztec/aztec.js'; import { type L1ContractAddresses, l1ContractsNames } from '@aztec/ethereum'; import { EthAddress } from '@aztec/foundation/eth-address'; +import { type ServerList } from '@aztec/foundation/json-rpc/server'; import { type LogFn, createConsoleLogger } from '@aztec/foundation/log'; import { type P2PConfig } from '@aztec/p2p'; import { type PXEService, type PXEServiceConfig } from '@aztec/pxe'; +export interface ServiceStarter { + (options: T, signalHandlers: (() => Promise)[], logger: LogFn): Promise; +} + /** * Checks if the object has l1Contracts property * @param obj - The object to check diff --git a/yarn-project/aztec/tsconfig.json b/yarn-project/aztec/tsconfig.json index 457e0776787e..bc45846794b2 100644 --- a/yarn-project/aztec/tsconfig.json +++ b/yarn-project/aztec/tsconfig.json @@ -51,6 +51,9 @@ { "path": "../protocol-contracts" }, + { + "path": "../prover-client" + }, { "path": "../pxe" } diff --git a/yarn-project/circuit-types/src/interfaces/prover-client.ts b/yarn-project/circuit-types/src/interfaces/prover-client.ts index 5c2ef55b8d57..8e55d3a2dbbb 100644 --- a/yarn-project/circuit-types/src/interfaces/prover-client.ts +++ b/yarn-project/circuit-types/src/interfaces/prover-client.ts @@ -1,4 +1,5 @@ import { type BlockProver } from './block-prover.js'; +import { type ProvingJobSource } from './proving-job.js'; /** * The interface to the prover client. @@ -8,4 +9,6 @@ export interface ProverClient extends BlockProver { start(): Promise; stop(): Promise; + + getProvingJobSource(): ProvingJobSource; } diff --git a/yarn-project/circuits.js/src/structs/kernel/kernel_circuit_public_inputs.ts b/yarn-project/circuits.js/src/structs/kernel/kernel_circuit_public_inputs.ts index 84637a61f919..8c3f4ad3eb02 100644 --- a/yarn-project/circuits.js/src/structs/kernel/kernel_circuit_public_inputs.ts +++ b/yarn-project/circuits.js/src/structs/kernel/kernel_circuit_public_inputs.ts @@ -78,4 +78,12 @@ export class KernelCircuitPublicInputs { RevertCode.OK, ); } + + toString() { + return this.toBuffer().toString('hex'); + } + + static fromString(str: string) { + return KernelCircuitPublicInputs.fromBuffer(Buffer.from(str, 'hex')); + } } diff --git a/yarn-project/circuits.js/src/structs/kernel/public_kernel_circuit_public_inputs.ts b/yarn-project/circuits.js/src/structs/kernel/public_kernel_circuit_public_inputs.ts index 207a1db74d52..87e2e9545e97 100644 --- a/yarn-project/circuits.js/src/structs/kernel/public_kernel_circuit_public_inputs.ts +++ b/yarn-project/circuits.js/src/structs/kernel/public_kernel_circuit_public_inputs.ts @@ -51,6 +51,14 @@ export class PublicKernelCircuitPublicInputs { ); } + toString() { + return this.toBuffer().toString('hex'); + } + + static fromString(str: string) { + return PublicKernelCircuitPublicInputs.fromBuffer(Buffer.from(str, 'hex')); + } + get needsSetup() { return !this.endNonRevertibleData.publicCallStack[1].isEmpty(); } diff --git a/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.test.ts b/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.test.ts index e53f3f2df214..3e2767cbda35 100644 --- a/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.test.ts +++ b/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.test.ts @@ -14,4 +14,11 @@ describe('PublicKernelTailCircuitPrivateInputs', () => { expect(original).toEqual(serialized); expect(original).not.toBe(serialized); }); + + it('serializes to string and back', () => { + const original = makePublicKernelTailCircuitPrivateInputs(123); + const str = original.toString(); + const deserialized = PublicKernelTailCircuitPrivateInputs.fromString(str); + expect(original).toEqual(deserialized); + }); }); diff --git a/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.ts b/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.ts index 5399146b7e5c..a1cf6b9d8ff6 100644 --- a/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.ts +++ b/yarn-project/circuits.js/src/structs/kernel/public_kernel_tail_circuit_private_inputs.ts @@ -41,6 +41,14 @@ export class PublicKernelTailCircuitPrivateInputs { ); } + toString() { + return this.toBuffer().toString('hex'); + } + + static fromString(str: string) { + return PublicKernelTailCircuitPrivateInputs.fromBuffer(Buffer.from(str, 'hex')); + } + static fromBuffer(buffer: Buffer | BufferReader) { const reader = BufferReader.asReader(buffer); return new PublicKernelTailCircuitPrivateInputs( diff --git a/yarn-project/end-to-end/src/composed/integration_l1_publisher.test.ts b/yarn-project/end-to-end/src/composed/integration_l1_publisher.test.ts index b72df9815b38..bea0b56d8aa9 100644 --- a/yarn-project/end-to-end/src/composed/integration_l1_publisher.test.ts +++ b/yarn-project/end-to-end/src/composed/integration_l1_publisher.test.ts @@ -36,6 +36,7 @@ import { openTmpStore } from '@aztec/kv-store/utils'; import { AvailabilityOracleAbi, InboxAbi, OutboxAbi, RollupAbi } from '@aztec/l1-artifacts'; import { SHA256Trunc, StandardTree } from '@aztec/merkle-tree'; import { TxProver } from '@aztec/prover-client'; +import { ProverPool } from '@aztec/prover-client/prover-pool'; import { type L1Publisher, getL1Publisher } from '@aztec/sequencer-client'; import { WASMSimulator } from '@aztec/simulator'; import { MerkleTrees, ServerWorldStateSynchronizer, type WorldStateConfig } from '@aztec/world-state'; @@ -143,7 +144,7 @@ describe('L1Publisher integration', () => { }; const worldStateSynchronizer = new ServerWorldStateSynchronizer(tmpStore, builderDb, blockSource, worldStateConfig); await worldStateSynchronizer.start(); - builder = await TxProver.new({}, worldStateSynchronizer, new WASMSimulator()); + builder = await TxProver.new(worldStateSynchronizer, ProverPool.testPool(new WASMSimulator(), 4)); l2Proof = Buffer.alloc(0); publisher = getL1Publisher({ diff --git a/yarn-project/prover-client/package.json b/yarn-project/prover-client/package.json index 132403cbb7ff..64580446844a 100644 --- a/yarn-project/prover-client/package.json +++ b/yarn-project/prover-client/package.json @@ -2,7 +2,10 @@ "name": "@aztec/prover-client", "version": "0.1.0", "type": "module", - "exports": "./dest/index.js", + "exports": { + ".": "./dest/index.js", + "./prover-pool": "./dest/prover-pool/index.js" + }, "bin": { "bb-cli": "./dest/bb/index.js" }, diff --git a/yarn-project/prover-client/src/dummy-prover.ts b/yarn-project/prover-client/src/dummy-prover.ts index de61267c7384..ed9b1fdf06ab 100644 --- a/yarn-project/prover-client/src/dummy-prover.ts +++ b/yarn-project/prover-client/src/dummy-prover.ts @@ -4,6 +4,9 @@ import { PROVING_STATUS, type ProcessedTx, type ProverClient, + type ProvingJob, + type ProvingJobSource, + type ProvingRequest, type ProvingSuccess, type ProvingTicket, } from '@aztec/circuit-types'; @@ -11,6 +14,8 @@ import { type GlobalVariables, makeEmptyProof } from '@aztec/circuits.js'; import { type Fr } from '@aztec/foundation/fields'; export class DummyProver implements ProverClient { + jobs = new DummyProvingJobSource(); + public start(): Promise { return Promise.resolve(); } @@ -54,4 +59,22 @@ export class DummyProver implements ProverClient { setBlockCompleted(): Promise { return Promise.resolve(); } + + getProvingJobSource(): ProvingJobSource { + return this.jobs; + } +} + +class DummyProvingJobSource implements ProvingJobSource { + getProvingJob(): Promise | null> { + return Promise.resolve(null); + } + + rejectProvingJob(): Promise { + return Promise.resolve(); + } + + resolveProvingJob(): Promise { + return Promise.resolve(); + } } diff --git a/yarn-project/prover-client/src/prover-pool/index.ts b/yarn-project/prover-client/src/prover-pool/index.ts index eaae01068bcf..353448bcbcef 100644 --- a/yarn-project/prover-client/src/prover-pool/index.ts +++ b/yarn-project/prover-client/src/prover-pool/index.ts @@ -1,3 +1,4 @@ export * from './prover-agent.js'; export * from './memory-proving-queue.js'; export * from './prover-pool.js'; +export * from './rpc.js'; diff --git a/yarn-project/prover-client/src/prover-pool/prover-agent.ts b/yarn-project/prover-client/src/prover-pool/prover-agent.ts index 030bcc35f908..96df3851bfad 100644 --- a/yarn-project/prover-client/src/prover-pool/prover-agent.ts +++ b/yarn-project/prover-client/src/prover-pool/prover-agent.ts @@ -10,6 +10,7 @@ import { RunningPromise } from '@aztec/foundation/running-promise'; import { elapsed } from '@aztec/foundation/timer'; import { type CircuitProver } from '../prover/interface.js'; +import { ProvingError } from './proving-error.js'; export class ProverAgent { private runningPromise?: RunningPromise; @@ -45,7 +46,7 @@ export class ProverAgent { this.log.error( `Error processing proving job id=${job.id} type=${ProvingRequestType[job.request.type]}: ${err}`, ); - await queue.rejectProvingJob(job.id, err as Error); + await queue.rejectProvingJob(job.id, new ProvingError(String(err))); } }, this.intervalMs); diff --git a/yarn-project/prover-client/src/prover-pool/prover-pool.ts b/yarn-project/prover-client/src/prover-pool/prover-pool.ts index 407b0a19fe65..0750b41520e1 100644 --- a/yarn-project/prover-client/src/prover-pool/prover-pool.ts +++ b/yarn-project/prover-client/src/prover-pool/prover-pool.ts @@ -1,6 +1,11 @@ import { type ProvingJobSource } from '@aztec/circuit-types'; import { type SimulationProvider } from '@aztec/simulator'; +import { mkdtemp } from 'fs/promises'; +import { tmpdir } from 'os'; +import { join } from 'path'; + +import { BBNativeRollupProver, type BBProverConfig } from '../prover/bb_prover.js'; import { TestCircuitProver } from '../prover/test_circuit_prover.js'; import { ProverAgent } from './prover-agent.js'; @@ -43,10 +48,34 @@ export class ProverPool { this.running = false; } - static testPool(simulationProvider: SimulationProvider, size = 1, agentPollIntervalMS = 10): ProverPool { + static testPool(simulationProvider?: SimulationProvider, size = 1, agentPollIntervalMS = 10): ProverPool { return new ProverPool( size, - i => new ProverAgent(new TestCircuitProver(simulationProvider), agentPollIntervalMS, `${i}`), + i => new ProverAgent(new TestCircuitProver(simulationProvider), agentPollIntervalMS, `test-prover-${i}`), ); } + + static nativePool( + { acvmBinaryPath, bbBinaryPath }: Pick, + size: number, + agentPollIntervalMS = 10, + ): ProverPool { + // TODO generate keys ahead of time so that each agent doesn't have to do it + return new ProverPool(size, async i => { + const [acvmWorkingDirectory, bbWorkingDirectory] = await Promise.all([ + mkdtemp(join(tmpdir(), 'acvm-')), + mkdtemp(join(tmpdir(), 'bb-')), + ]); + return new ProverAgent( + await BBNativeRollupProver.new({ + acvmBinaryPath, + acvmWorkingDirectory, + bbBinaryPath, + bbWorkingDirectory, + }), + agentPollIntervalMS, + `bb-prover-${i}`, + ); + }); + } } diff --git a/yarn-project/prover-client/src/prover-pool/proving-error.ts b/yarn-project/prover-client/src/prover-pool/proving-error.ts new file mode 100644 index 000000000000..982ec0cdc0e7 --- /dev/null +++ b/yarn-project/prover-client/src/prover-pool/proving-error.ts @@ -0,0 +1,9 @@ +export class ProvingError extends Error { + override toString() { + return this.message; + } + + static fromString(message: string) { + return new ProvingError(message); + } +} diff --git a/yarn-project/prover-client/src/prover-pool/rpc.ts b/yarn-project/prover-client/src/prover-pool/rpc.ts new file mode 100644 index 000000000000..a7debf7bff21 --- /dev/null +++ b/yarn-project/prover-client/src/prover-pool/rpc.ts @@ -0,0 +1,73 @@ +import { type ProvingJobSource } from '@aztec/circuit-types'; +import { + BaseOrMergeRollupPublicInputs, + BaseParityInputs, + BaseRollupInputs, + KernelCircuitPublicInputs, + MergeRollupInputs, + ParityPublicInputs, + Proof, + PublicKernelCircuitPrivateInputs, + PublicKernelCircuitPublicInputs, + PublicKernelTailCircuitPrivateInputs, + RootParityInputs, + RootRollupInputs, + RootRollupPublicInputs, +} from '@aztec/circuits.js'; +import { createJsonRpcClient, makeFetch } from '@aztec/foundation/json-rpc/client'; +import { JsonRpcServer } from '@aztec/foundation/json-rpc/server'; + +import { ProvingError } from './proving-error.js'; + +export function createProvingJobSourceServer(queue: ProvingJobSource): JsonRpcServer { + return new JsonRpcServer( + queue, + { + BaseParityInputs, + BaseOrMergeRollupPublicInputs, + BaseRollupInputs, + MergeRollupInputs, + ParityPublicInputs, + Proof, + RootParityInputs, + RootRollupInputs, + RootRollupPublicInputs, + PublicKernelCircuitPrivateInputs, + PublicKernelCircuitPublicInputs, + PublicKernelTailCircuitPrivateInputs, + KernelCircuitPublicInputs, + ProvingError, + }, + {}, + ); +} + +export function createProvingJobSourceClient( + url: string, + namespace?: string, + fetch = makeFetch([1, 2, 3], false), +): ProvingJobSource { + return createJsonRpcClient( + url, + { + BaseParityInputs, + BaseOrMergeRollupPublicInputs, + BaseRollupInputs, + MergeRollupInputs, + ParityPublicInputs, + Proof, + RootParityInputs, + RootRollupInputs, + RootRollupPublicInputs, + PublicKernelCircuitPrivateInputs, + PublicKernelCircuitPublicInputs, + PublicKernelTailCircuitPrivateInputs, + KernelCircuitPublicInputs, + ProvingError, + }, + {}, + false, + namespace, + fetch, + ) as ProvingJobSource; +} diff --git a/yarn-project/prover-client/src/prover/test_circuit_prover.ts b/yarn-project/prover-client/src/prover/test_circuit_prover.ts index 63a1f74a1c2e..37e1d6b68208 100644 --- a/yarn-project/prover-client/src/prover/test_circuit_prover.ts +++ b/yarn-project/prover-client/src/prover/test_circuit_prover.ts @@ -49,7 +49,7 @@ export class TestCircuitProver implements CircuitProver { private wasmSimulator = new WASMSimulator(); constructor( - private simulationProvider: SimulationProvider, + private simulationProvider?: SimulationProvider, private logger = createDebugLogger('aztec:test-prover'), ) {} @@ -93,7 +93,8 @@ export class TestCircuitProver implements CircuitProver { public async getBaseRollupProof(input: BaseRollupInputs): Promise<[BaseOrMergeRollupPublicInputs, Proof]> { const witnessMap = convertSimulatedBaseRollupInputsToWitnessMap(input); - const witness = await this.simulationProvider.simulateCircuit(witnessMap, SimulatedBaseRollupArtifact); + const simulationProvider = this.simulationProvider ?? this.wasmSimulator; + const witness = await simulationProvider.simulateCircuit(witnessMap, SimulatedBaseRollupArtifact); const result = convertSimulatedBaseRollupOutputsFromWitnessMap(witness); diff --git a/yarn-project/prover-client/src/tx-prover/tx-prover.ts b/yarn-project/prover-client/src/tx-prover/tx-prover.ts index 90daea5a7fa4..bc45cc582bf9 100644 --- a/yarn-project/prover-client/src/tx-prover/tx-prover.ts +++ b/yarn-project/prover-client/src/tx-prover/tx-prover.ts @@ -1,37 +1,30 @@ import { type ProcessedTx } from '@aztec/circuit-types'; -import { type BlockResult, type ProverClient, type ProvingTicket } from '@aztec/circuit-types/interfaces'; +import { + type BlockResult, + type ProverClient, + type ProvingJobSource, + type ProvingTicket, +} from '@aztec/circuit-types/interfaces'; import { type Fr, type GlobalVariables } from '@aztec/circuits.js'; -import { type SimulationProvider } from '@aztec/simulator'; import { type WorldStateSynchronizer } from '@aztec/world-state'; -import { type ProverConfig } from '../config.js'; import { type VerificationKeys, getVerificationKeys } from '../mocks/verification_keys.js'; import { ProvingOrchestrator } from '../orchestrator/orchestrator.js'; import { MemoryProvingQueue } from '../prover-pool/memory-proving-queue.js'; -import { ProverAgent } from '../prover-pool/prover-agent.js'; -import { ProverPool } from '../prover-pool/prover-pool.js'; -import { TestCircuitProver } from '../prover/test_circuit_prover.js'; +import { type ProverPool } from '../prover-pool/prover-pool.js'; /** * A prover accepting individual transaction requests */ export class TxProver implements ProverClient { private orchestrator: ProvingOrchestrator; - private proverPool: ProverPool; private queue = new MemoryProvingQueue(); constructor( private worldStateSynchronizer: WorldStateSynchronizer, - simulationProvider: SimulationProvider, protected vks: VerificationKeys, - agentCount = 4, - agentPollIntervalMS = 10, + private proverPool?: ProverPool, ) { - this.proverPool = new ProverPool( - agentCount, - i => new ProverAgent(new TestCircuitProver(simulationProvider), agentPollIntervalMS, `${i}`), - ); - this.orchestrator = new ProvingOrchestrator(worldStateSynchronizer.getLatest(), this.queue); } @@ -39,14 +32,14 @@ export class TxProver implements ProverClient { * Starts the prover instance */ public async start() { - await this.proverPool.start(this.queue); + await this.proverPool?.start(this.queue); } /** * Stops the prover instance */ public async stop() { - await this.proverPool.stop(); + await this.proverPool?.stop(); } /** @@ -55,12 +48,8 @@ export class TxProver implements ProverClient { * @param worldStateSynchronizer - An instance of the world state * @returns An instance of the prover, constructed and started. */ - public static async new( - config: ProverConfig, - worldStateSynchronizer: WorldStateSynchronizer, - simulationProvider: SimulationProvider, - ) { - const prover = new TxProver(worldStateSynchronizer, simulationProvider, getVerificationKeys()); + public static async new(worldStateSynchronizer: WorldStateSynchronizer, proverPool?: ProverPool) { + const prover = new TxProver(worldStateSynchronizer, getVerificationKeys(), proverPool); await prover.start(); return prover; } @@ -112,4 +101,8 @@ export class TxProver implements ProverClient { public setBlockCompleted(): Promise { return this.orchestrator.setBlockCompleted(); } + + getProvingJobSource(): ProvingJobSource { + return this.queue; + } } diff --git a/yarn-project/yarn.lock b/yarn-project/yarn.lock index f7a6a3e7d1c6..9b79cfcc6a7b 100644 --- a/yarn-project/yarn.lock +++ b/yarn-project/yarn.lock @@ -207,6 +207,7 @@ __metadata: "@aztec/noir-contracts.js": "workspace:^" "@aztec/p2p": "workspace:^" "@aztec/protocol-contracts": "workspace:^" + "@aztec/prover-client": "workspace:^" "@aztec/pxe": "workspace:^" "@jest/globals": ^29.5.0 "@types/jest": ^29.5.0