fix(cuda-prover): dont discard stderr

succinctlabs · Sep 18, 2024 · 23c9acf · 23c9acf
1 parent 7457d8b
commit 23c9acf
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/crates/cuda/src/lib.rs b/crates/cuda/src/lib.rs
@@ -120,6 +120,22 @@ impl SP1CudaProver {
             .spawn()
             .expect("failed to start Docker container");
 
+        let stderr = child.stderr.take().unwrap();
+        std::thread::spawn(move || {
+            let mut reader = BufReader::new(stderr);
+            let mut buffer = [0; 1024];
+            loop {
+                match reader.read(&mut buffer) {
+                    Ok(0) => break,
+                    Ok(n) => {
+                        std::io::stderr().write_all(&buffer[..n]).unwrap();
+                        std::io::stderr().flush().unwrap();
+                    }
+                    Err(_) => break,
+                }
+            }
+        });
+
         let stdout = child.stdout.take().unwrap();
         std::thread::spawn(move || {
             let mut reader = BufReader::new(stdout);