await streams

worldcoin · Jan 28, 2025 · 5c64f64 · 5c64f64
1 parent 42bb753
commit 5c64f64
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/iris-mpc-gpu/src/server/actor.rs b/iris-mpc-gpu/src/server/actor.rs
@@ -1212,6 +1212,8 @@ impl ServerActor {
             let now = std::time::Instant::now();
             tracing::info!("Collected enough match distances, starting bucket calculation");
 
+            self.device_manager.await_streams(&batch_streams);
+
             let indices = match_distances_indices
                 .iter()
                 .enumerate()
@@ -1315,6 +1317,8 @@ impl ServerActor {
                 }
             }
 
+            self.device_manager.await_streams(&batch_streams);
+
             tracing::info!("Bucket calculation took {:?}", now.elapsed());
         }