minor fix

worldcoin · Jan 13, 2025 · 4d26e4e · 4d26e4e
1 parent c9c60a2
commit 4d26e4e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/iris-mpc-gpu/tests/threshold.rs b/iris-mpc-gpu/tests/threshold.rs
@@ -1,4 +1,4 @@
-#[cfg(feature = "gpu_dependent")]
+// #[cfg(feature = "gpu_dependent")]
 mod threshold_test {
     use cudarc::{
         driver::{CudaDevice, CudaStream},
@@ -110,7 +110,7 @@ mod threshold_test {
         let mod_ = 1u64 << (16 + B_BITS);
         let mut res = Vec::with_capacity(code_input.len());
         for (c, m) in code_input.into_iter().zip(mask_input) {
-            let r = ((m as u64) * A - ((c as u64) << B_BITS)) % mod_;
+            let r = ((m as u64) * A - ((c as u64) << B_BITS) - 1) % mod_;
             let msb = r >> (B_BITS + 16 - 1) & 1 == 1;
             res.push(msb)
         }
@@ -226,7 +226,6 @@ mod threshold_test {
     }
 
     #[tokio::test]
-    #[ignore]
     async fn test_threshold() -> eyre::Result<()> {
         install_tracing();
         env::set_var("NCCL_P2P_LEVEL", "LOC");