From f27f6d8c7065ffc840d4f3e80e17e44de02da653 Mon Sep 17 00:00:00 2001 From: Gregor Date: Tue, 5 Mar 2024 17:27:01 +0100 Subject: [PATCH] use staged --- src/base/snark0.ml | 4 ++-- src/base/snark_intf.ml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/base/snark0.ml b/src/base/snark0.ml index 15f8f6ee0..473b7117a 100644 --- a/src/base/snark0.ml +++ b/src/base/snark0.ml @@ -1395,7 +1395,7 @@ module Run = struct (* start an as_prover / exists block and return a function to finish it and witness a given list of fields *) let as_prover_manual (size_to_witness : int) : - field array option -> Field.t array = + (field array option -> Field.t array) Staged.t = let s = !state in let old_as_prover = Run_state.as_prover s in (* enter the as_prover block *) @@ -1426,7 +1426,7 @@ module Run = struct | true, None -> failwith "Expected values to witness" in - finish_computation + Staged.stage finish_computation let run_unchecked x = finalize_is_running (fun () -> diff --git a/src/base/snark_intf.ml b/src/base/snark_intf.ml index bd2706888..a6416ca31 100644 --- a/src/base/snark_intf.ml +++ b/src/base/snark_intf.ml @@ -1403,7 +1403,7 @@ module type Run_basic = sig manual_callbacks (* Callback, low-level version of [as_prover] and [exists]. *) - val as_prover_manual : int -> field array option -> Field.t array + val as_prover_manual : int -> (field array option -> Field.t array) Staged.t (** Generate the public input vector for a given statement. *) val generate_public_input :