From 80a97c12db6cdf93db8189acfbbd604919c50f59 Mon Sep 17 00:00:00 2001
From: "Chris T." <chris@succinct.xyz>
Date: Tue, 26 Nov 2024 13:22:30 -0800
Subject: [PATCH] fix: preprocessed trace bug (#1821)

---
 crates/core/machine/src/memory/program.rs |  2 +-
 crates/prover/src/lib.rs                  | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/crates/core/machine/src/memory/program.rs b/crates/core/machine/src/memory/program.rs
index 699e052c0d..4dc8b196ad 100644
--- a/crates/core/machine/src/memory/program.rs
+++ b/crates/core/machine/src/memory/program.rs
@@ -82,7 +82,7 @@ impl<F: PrimeField> MachineAir<F> for MemoryProgramChip {
         let mut values = zeroed_f_vec(padded_nb_rows * NUM_MEMORY_PROGRAM_PREPROCESSED_COLS);
         let chunk_size = std::cmp::max((nb_rows + 1) / num_cpus::get(), 1);
 
-        let memory = program.memory_image.iter().collect::<Vec<_>>();
+        let memory = program.memory_image.iter().sorted().collect::<Vec<_>>();
         values
             .chunks_mut(chunk_size * NUM_MEMORY_PROGRAM_PREPROCESSED_COLS)
             .enumerate()
diff --git a/crates/prover/src/lib.rs b/crates/prover/src/lib.rs
index f72f7c9df7..a6aa17187e 100644
--- a/crates/prover/src/lib.rs
+++ b/crates/prover/src/lib.rs
@@ -1534,4 +1534,14 @@ pub mod tests {
         setup_logger();
         test_e2e_with_deferred_proofs_prover::<DefaultProverComponents>(SP1ProverOpts::default())
     }
+
+    #[test]
+    fn test_deterministic_setup() {
+        setup_logger();
+        let prover = SP1Prover::<DefaultProverComponents>::new();
+        let program = test_artifacts::FIBONACCI_ELF;
+        let (pk, vk) = prover.setup(&program);
+        let pk2 = prover.setup(&program).0;
+        assert_eq!(pk.pk.commit, pk2.pk.commit);
+    }
 }