use uninitialized memory

reilabs · Nov 27, 2024 · 300b7e8 · 300b7e8
1 parent 849569d
commit 300b7e8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/prover/src/gpu/cuda/mod.rs b/prover/src/gpu/cuda/mod.rs
@@ -1,6 +1,6 @@
 //! This module contains GPU acceleration logic for Nvidia CUDA devices.
 
-use std::{cell::RefCell, marker::PhantomData};
+use std::{cell::RefCell, marker::PhantomData, mem::MaybeUninit};
 
 use air::{AuxRandElements, PartitionOptions};
 use miden_gpu::{cuda::{constraint::CudaConstraintCommitment, merkle::MerkleTree, trace_lde::CudaTraceLde}, HashFn};
@@ -33,9 +33,9 @@ where
     D: Digest + From<[Felt; DIGEST_SIZE]> + Into<[Felt; DIGEST_SIZE]>,
     R: RandomCoin<BaseField = Felt, Hasher = H> + Send,
 {
-    main: RefCell<&'g mut [Felt]>,
-    aux: RefCell<&'g mut [Felt]>,
-    ce: RefCell<&'g mut [Felt]>,
+    main: RefCell<&'g mut [MaybeUninit<Felt>]>,
+    aux: RefCell<&'g mut [MaybeUninit<Felt>]>,
+    ce: RefCell<&'g mut [MaybeUninit<Felt>]>,
 
     pub execution_prover: ExecutionProver<H, R>,
     pub hash_fn: HashFn,
@@ -48,7 +48,7 @@ where
     D: Digest + From<[Felt; DIGEST_SIZE]> + Into<[Felt; DIGEST_SIZE]>,
     R: RandomCoin<BaseField = Felt, Hasher = H> + Send,
 {
-    pub fn new(execution_prover: ExecutionProver<H, R>, hash_fn: HashFn, main: &'g mut [Felt], aux: &'g mut [Felt], ce: &'g mut [Felt]) -> Self {
+    pub fn new(execution_prover: ExecutionProver<H, R>, hash_fn: HashFn, main: &'g mut [MaybeUninit<Felt>], aux: &'g mut [MaybeUninit<Felt>], ce: &'g mut [MaybeUninit<Felt>]) -> Self {
         CudaExecutionProver {
             main: RefCell::new(main),
             aux: RefCell::new(aux),