test: add e2e deferral test (#2481)

Author: stephenh-axiom-xyz

Cargo.lock

@@ -32,6 +32,9 @@ openvm-cuda-backend = { workspace = true, optional = true }
 [dev-dependencies]
 openvm-rv32im-circuit = { workspace = true, features = ["test-utils"] }
 openvm-rv32im-transpiler = { workspace = true }
+openvm-deferral-circuit = { workspace = true }
+openvm-deferral-transpiler = { workspace = true }
+openvm-instructions = { workspace = true }
 openvm-toolchain-tests = { workspace = true }
 openvm-transpiler = { workspace = true }
 test-case = { workspace = true }

crates/continuations-v2/Cargo.toml

@@ -1,14 +1,18 @@
 [workspace]
 [package]
-name = "openvm-continautions-test-programs"
+name = "openvm-continuations-test-programs"
 version = "0.0.0"
 edition = "2021"
 
 [dependencies]
-openvm = { git = "https://github.com/openvm-org/openvm.git", branch = "develop-v2", default-features = false }
+openvm = { path = "../../toolchain/openvm", default-features = false }
+openvm-platform = { path = "../../toolchain/platform", default-features = false }
+openvm-custom-insn = { path = "../../toolchain/custom_insn", default-features = false }
+openvm-deferral-guest = { path = "../../../extensions/deferral/guest", default-features = false }
 
 [features]
 default = []
+std = ["openvm/std"]
 
 [profile.release]
 panic = "abort"

crates/continuations-v2/programs/Cargo.lock

@@ -1,5 +1,5 @@
-#![no_main]
-#![no_std]
+#![cfg_attr(target_os = "zkvm", no_main)]
+#![cfg_attr(target_os = "zkvm", no_std)]
 
 use openvm::io::{read, reveal_u32};
 

crates/continuations-v2/programs/Cargo.toml

@@ -0,0 +1,36 @@
+#![cfg_attr(target_os = "zkvm", no_main)]
+#![cfg_attr(target_os = "zkvm", no_std)]
+
+use openvm_deferral_guest::{deferred_compute, get_deferred_output, Commit};
+
+openvm::entry!(main);
+
+const INPUT_COMMIT_0: Commit = [0x11; 32];
+const INPUT_COMMIT_1: Commit = [0x22; 32];
+const INPUT_COMMIT_2: Commit = [0x33; 32];
+const EXPECTED_OUTPUT_0_IDX1: [u8; 8] = [2, 4, 7, 11, 16, 22, 29, 37];
+const EXPECTED_OUTPUT_1_IDX1: [u8; 8] = [9, 16, 22, 27, 31, 34, 36, 37];
+const EXPECTED_OUTPUT_2_IDX1: [u8; 8] = [10, 19, 28, 37, 46, 55, 64, 73];
+const EXPECTED_OUTPUT_0_IDX2: [u8; 8] = [3, 5, 8, 12, 17, 23, 30, 38];
+
+pub fn main() {
+    let output_key_0 = deferred_compute::<1>(&INPUT_COMMIT_0);
+    let mut output_0 = [0u8; EXPECTED_OUTPUT_0_IDX1.len()];
+    get_deferred_output::<1>(&mut output_0, &output_key_0);
+    assert_eq!(output_0, EXPECTED_OUTPUT_0_IDX1);
+
+    let output_key_1 = deferred_compute::<1>(&INPUT_COMMIT_1);
+    let mut output_1 = [0u8; EXPECTED_OUTPUT_1_IDX1.len()];
+    get_deferred_output::<1>(&mut output_1, &output_key_1);
+    assert_eq!(output_1, EXPECTED_OUTPUT_1_IDX1);
+
+    let output_key_2 = deferred_compute::<1>(&INPUT_COMMIT_2);
+    let mut output_2 = [0u8; EXPECTED_OUTPUT_2_IDX1.len()];
+    get_deferred_output::<1>(&mut output_2, &output_key_2);
+    assert_eq!(output_2, EXPECTED_OUTPUT_2_IDX1);
+
+    let output_key_3 = deferred_compute::<2>(&INPUT_COMMIT_0);
+    let mut output_3 = [0u8; EXPECTED_OUTPUT_0_IDX2.len()];
+    get_deferred_output::<2>(&mut output_3, &output_key_3);
+    assert_eq!(output_3, EXPECTED_OUTPUT_0_IDX2);
+}

crates/continuations-v2/programs/examples/fibonacci.rs

@@ -30,9 +30,10 @@ pub fn generate_proving_ctx(
     AirProvingContext<CpuBackend<BabyBearPoseidon2Config>>,
     Vec<[F; POSEIDON2_WIDTH]>,
 ) {
-    assert_eq!(
-        absent_trace_pvs.is_some(),
-        matches!(proofs_type, ProofsType::Deferral) && proofs.len() == 1
+    assert!(
+        absent_trace_pvs.is_none()
+            || (matches!(proofs_type, ProofsType::Deferral) && proofs.len() == 1),
+        "absent_trace_pvs is only valid for single-proof deferral aggregation"
     );
     let mut proof_idxs = vec![];
     let (num_rows, def_flag) = match proofs_type {