bump cudarc (#964)

* bump cudarc * remove borrows * lock
worldcoin · Jan 24, 2025 · ee0d900 · ee0d900
1 parent d11df4b
commit ee0d900
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 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,7 @@ repository.workspace = true
 
 [dependencies]
 bincode = "1.3.3"
-cudarc = { version = "0.12", features = ["cuda-12020", "nccl"] }
+cudarc = { version = "0.13.3", 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,9 +85,15 @@ async fn main() -> eyre::Result<()> {
         for i in 0..n_devices {
             devs[i].bind_to_thread().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();
+            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();
         }
 
         for dev in devs.iter() {