sync after gather (#638)

worldcoin · Nov 1, 2024 · 25bac46 · 25bac46
1 parent 0921a1a
commit 25bac46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/deploy/prod/common-values-iris-mpc.yaml b/deploy/prod/common-values-iris-mpc.yaml
@@ -1,4 +1,4 @@
-image: "ghcr.io/worldcoin/iris-mpc:v0.8.39"
+image: "ghcr.io/worldcoin/iris-mpc:v0.8.40"
 
 environment: prod
 replicaCount: 1

diff --git a/iris-mpc-gpu/src/server/actor.rs b/iris-mpc-gpu/src/server/actor.rs
@@ -1255,12 +1255,16 @@ impl ServerActor {
             .device(0)
             .htod_copy(valid_entries.iter().map(|&x| x as u8).collect::<Vec<_>>())?;
 
+        self.device_manager.device(0).synchronize()?;
+
         tracing::info!(party_id = self.party_id, "all_gather start");
 
         self.comms[0]
             .all_gather(&buffer_self, &mut buffer)
             .map_err(|e| eyre!(format!("{:?}", e)))?;
 
+        self.device_manager.device(0).synchronize()?;
+
         tracing::info!(party_id = self.party_id, "dtoh_sync_copy start");
 
         let results = self.device_manager.device(0).dtoh_sync_copy(&buffer)?;