use staged

o1-labs · Mar 5, 2024 · f27f6d8 · f27f6d8
1 parent 46b1300
commit f27f6d8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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 :