Fix consecutive proofs bug

Author: jeremyfelder

crates/prover/src/constraint_framework/component.rs

@@ -597,8 +597,6 @@ impl<E: FrameworkEval + Sync> ComponentProver<IcicleBackend> for FrameworkCompon
 
         let mut col = DeviceVec::cuda_malloc(accum.col.len() as _).unwrap();
 
-        let d_exec_trace  = EXEC_TRACE.get().unwrap(); 
-
         let mut d_denom = DeviceVec::cuda_malloc(denom_inv.len() as _).unwrap();
         d_denom.copy_from_host(HostSlice::from_slice(unsafe {
             transmute(&denom_inv[..])})).unwrap();
@@ -608,18 +606,20 @@ impl<E: FrameworkEval + Sync> ComponentProver<IcicleBackend> for FrameworkCompon
             transmute(&accum.random_coeff_powers[..])
         })).unwrap();
 
-        icicle_m31::fri::compute_polynomial(
-            total_constraints as _,
-            eval_log_size,
-            domain_log_size,
-            trace_rows_dimension as _,
-            trace_cols_dimension as _,
-            &d_denom[..],
-            &d_rand_coeff_powers[..],
-            &d_exec_trace[..],
-            &mut col[..],
-        )
-        .unwrap();
+        unsafe {
+            icicle_m31::fri::compute_polynomial(
+                total_constraints as _,
+                eval_log_size,
+                domain_log_size,
+                trace_rows_dimension as _,
+                trace_cols_dimension as _,
+                &d_denom[..],
+                &d_rand_coeff_powers[..],
+                &(*EXEC_TRACE)[..],
+                &mut col[..],
+            )
+            .unwrap();
+        }
 
         let mut icicle_col = vec![QuarticExtensionField::zero(); accum.col.len()];
 

crates/prover/src/constraint_framework/mod.rs

@@ -17,9 +17,7 @@ use icicle_cuda_runtime::memory::DeviceVec;
 #[cfg(feature = "icicle")]
 use icicle_m31::field::ScalarField;
 #[cfg(feature = "icicle")]
-use std::sync::OnceLock;
-#[cfg(feature = "icicle")]
-pub static EXEC_TRACE: OnceLock<DeviceVec<ScalarField>> = OnceLock::new();
+pub static mut EXEC_TRACE: *mut DeviceVec<ScalarField> = std::ptr::null_mut();
 
 use std::array;
 use std::fmt::Debug;

crates/prover/src/examples/wide_fibonacci/mod.rs

@@ -349,7 +349,9 @@ mod tests {
             h.copy_from_host(HostSlice::from_slice(unsafe { transmute(b_transposed.as_slice()) }))
                 .unwrap();
 
-            EXEC_TRACE.get_or_init(|| h);
+            unsafe {
+                EXEC_TRACE = &mut h;   
+            }
 
             icicle_m31::fri::precompute_fri_twiddles(log_n_instances).unwrap();