fix offset

worldcoin · Jan 28, 2025 · 2d0ed57 · 2d0ed57
1 parent 8aa7c52
commit 2d0ed57
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/temp-branch-build-and-push.yaml b/.github/workflows/temp-branch-build-and-push.yaml
@@ -3,7 +3,7 @@ name: Branch - Build and push docker image
 on:
   push:
     branches:
-      - "test-benchmark"
+      - "fix/test-benchmark-chunked-page-lock"
 
 concurrency:
   group: '${{ github.workflow }} @ ${{ github.event.pull_request.head.label || github.head_ref || github.ref }}'

diff --git a/deploy/stage/common-values-iris-mpc.yaml b/deploy/stage/common-values-iris-mpc.yaml
@@ -1,4 +1,4 @@
-image: "ghcr.io/worldcoin/iris-mpc:a76986f433af1327232a946766a9e1467048cf4a"
+image: "ghcr.io/worldcoin/iris-mpc:1654b9cbda7d59684ce026c04efeaa89f12f329d"
 
 environment: stage
 replicaCount: 1

diff --git a/iris-mpc-gpu/src/helpers/mod.rs b/iris-mpc-gpu/src/helpers/mod.rs
@@ -183,17 +183,18 @@ pub fn register_host_memory(
         chunk_offset + chunk_length
     );
     let size = chunk_length / device_manager.device_count();
+    let offset = chunk_offset / device_manager.device_count();
     for (device_index, device) in device_manager.devices().iter().enumerate() {
         device.bind_to_thread().unwrap();
         unsafe {
             let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
-                (db.limb_0[device_index] + (chunk_offset * code_length) as u64) as *mut _,
+                (db.limb_0[device_index] + (offset * code_length) as u64) as *mut _,
                 size * code_length,
                 CU_MEMHOSTALLOC_PORTABLE,
             );
 
             let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
-                (db.limb_1[device_index] + (chunk_offset * code_length) as u64) as *mut _,
+                (db.limb_1[device_index] + (offset * code_length) as u64) as *mut _,
                 size * code_length,
                 CU_MEMHOSTALLOC_PORTABLE,
             );