revert nccl changes

worldcoin · Jan 25, 2025 · fad8d51 · fad8d51
1 parent 4d15e1a
commit fad8d51
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/iris-mpc-gpu/Cargo.toml b/iris-mpc-gpu/Cargo.toml
@@ -9,7 +9,10 @@ repository.workspace = true
 
 [dependencies]
 bincode = "1.3.3"
-cudarc = { version = "0.13.3", features = ["cuda-12020", "nccl"] }
+cudarc = { git = "https://github.com/worldcoin/cudarc-fork.git", features = [
+    "cuda-12020",
+    "nccl",
+] }
 eyre.workspace = true
 tracing.workspace = true
 bytemuck.workspace = true

diff --git a/iris-mpc-gpu/src/bin/nccl.rs b/iris-mpc-gpu/src/bin/nccl.rs
@@ -85,15 +85,9 @@ async fn main() -> eyre::Result<()> {
         for i in 0..n_devices {
             devs[i].bind_to_thread().unwrap();
 
-            comms[i]
-                .broadcast(slices[i].as_ref(), &mut slices1[i], 0)
-                .unwrap();
-            comms[i]
-                .broadcast(slices[i].as_ref(), &mut slices2[i], 1)
-                .unwrap();
-            comms[i]
-                .broadcast(slices[i].as_ref(), &mut slices3[i], 2)
-                .unwrap();
+            comms[i].broadcast(&slices[i], &mut slices1[i], 0).unwrap();
+            comms[i].broadcast(&slices[i], &mut slices2[i], 1).unwrap();
+            comms[i].broadcast(&slices[i], &mut slices3[i], 2).unwrap();
         }
 
         for dev in devs.iter() {