try cuMemAllocHost_v2

worldcoin · Jan 24, 2025 · 943c796 · 943c796
1 parent 4e80dbc
commit 943c796
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
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:c2e7d7a099f189ecedea11947c34d55965b41f7c"
+image: "ghcr.io/worldcoin/iris-mpc:e9ee28e5441ee6b7e9c24e68a1c32bc87867bd2e"
 
 environment: stage
 replicaCount: 1

diff --git a/deploy/stage/smpcv2-1-stage/values-iris-mpc.yaml b/deploy/stage/smpcv2-1-stage/values-iris-mpc.yaml
@@ -72,7 +72,7 @@ env:
     value: "true"
 
   - name: SMPC__DB_CHUNKS_BUCKET_NAME
-    value: "iris-mpc-db-exporter-store-node-1-stage-eu-north-1"
+    value: "temp-iris-mpc-db-exporter-stage-1--eun1-az3--x-s3"
 
   - name: SMPC__DB_CHUNKS_FOLDER_NAME
     value: "even_odd_binary_output_16k"

diff --git a/iris-mpc-gpu/src/helpers/mod.rs b/iris-mpc-gpu/src/helpers/mod.rs
@@ -180,16 +180,14 @@ pub fn register_host_memory(
     for (device_index, device) in device_manager.devices().iter().enumerate() {
         device.bind_to_thread().unwrap();
         unsafe {
-            let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
+            let _ = cudarc::driver::sys::lib().cuMemAllocHost_v2(
                 db.limb_0[device_index] as *mut _,
                 max_size * code_length,
-                CU_MEMHOSTALLOC_PORTABLE,
             );
 
-            let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
+            let _ = cudarc::driver::sys::lib().cuMemAllocHost_v2(
                 db.limb_1[device_index] as *mut _,
                 max_size * code_length,
-                CU_MEMHOSTALLOC_PORTABLE,
             );
         }
     }