compiler nits

succinctlabs · Sep 18, 2024 · 2e8b0a8 · 2e8b0a8
1 parent ca95892
commit 2e8b0a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/crates/sdk/src/lib.rs b/crates/sdk/src/lib.rs
@@ -81,7 +81,7 @@ impl ProverClient {
                 #[cfg(not(feature = "cuda"))]
                 prover: Box::new(CpuProver::new()),
                 #[cfg(feature = "cuda")]
-                prover: Box::new(CudaProver::new()),
+                prover: Box::new(CudaProver::new(SP1Prover::new())),
             },
             "network" => {
                 cfg_if! {

diff --git a/crates/sdk/src/provers/cuda.rs b/crates/sdk/src/provers/cuda.rs
@@ -17,10 +17,9 @@ pub struct CudaProver {
 
 impl CudaProver {
     /// Creates a new [CudaProver].
-    pub fn new() -> Self {
-        let prover = SP1Prover::new();
+    pub fn new(prover: SP1Prover) -> Self {
         let cuda_prover = SP1CudaProver::new();
-        Self { prover, cuda_prover }
+        Self { prover, cuda_prover: cuda_prover.expect("Failed to initialize CUDA prover") }
     }
 }
 
@@ -102,6 +101,6 @@ impl Prover<DefaultProverComponents> for CudaProver {
 
 impl Default for CudaProver {
     fn default() -> Self {
-        Self::new()
+        Self::new(SP1Prover::new())
     }
 }