up

worldcoin · Sep 1, 2024 · e09014a · e09014a
1 parent df65aee
commit e09014a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/iris-mpc-gpu/src/helpers/device_manager.rs b/iris-mpc-gpu/src/helpers/device_manager.rs
@@ -29,7 +29,7 @@ impl DeviceManager {
     pub fn init() -> Self {
         let mut devices = vec![];
         for i in 0..CudaDevice::count().unwrap() {
-            devices.push(CudaDevice::new(i as usize).unwrap());
+            devices.push(CudaDevice::new_with_stream(i as usize).unwrap());
         }
 
         tracing::info!("Found {} devices", devices.len());