From fc0b72618b9a35767c12d3bbfeb3638f5cab745e Mon Sep 17 00:00:00 2001
From: philsippl <philsippl@gmail.com>
Date: Fri, 27 Dec 2024 12:18:57 +0100
Subject: [PATCH] both limbs

---
 iris-mpc-gpu/src/dot/share_db.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/iris-mpc-gpu/src/dot/share_db.rs b/iris-mpc-gpu/src/dot/share_db.rs
index 6dee9a2c7..b0b5889c2 100644
--- a/iris-mpc-gpu/src/dot/share_db.rs
+++ b/iris-mpc-gpu/src/dot/share_db.rs
@@ -21,7 +21,7 @@ use cudarc::{
     },
     driver::{
         result::{self, malloc_async},
-        sys::CUdeviceptr,
+        sys::{CUdeviceptr, CU_MEMHOSTALLOC_PORTABLE},
         CudaFunction, CudaSlice, CudaStream, CudaView, DevicePtr, DeviceSlice, LaunchAsync,
     },
     nccl,
@@ -279,7 +279,13 @@ impl ShareDB {
                 let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
                     db.code_gr.limb_0[device_index] as *mut _,
                     max_size * self.code_length,
-                    0,
+                    CU_MEMHOSTALLOC_PORTABLE,
+                );
+
+                let _ = cudarc::driver::sys::lib().cuMemHostRegister_v2(
+                    db.code_gr.limb_1[device_index] as *mut _,
+                    max_size * self.code_length,
+                    CU_MEMHOSTALLOC_PORTABLE,
                 );
             }
         }