fix test

worldcoin · Sep 10, 2024 · e519e22 · e519e22
1 parent 8f582cd
commit e519e22
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/iris-mpc-gpu/src/dot/share_db.rs b/iris-mpc-gpu/src/dot/share_db.rs
@@ -945,7 +945,8 @@ mod tests {
         for i in 0..DB_SIZE * QUERY_SIZE / n_devices {
             assert_eq!(
                 (gpu_result[0][i] + gpu_result[1][i] + gpu_result[2][i]),
-                (db.db[i / (DB_SIZE / n_devices)].mask & db.db[i % (DB_SIZE / n_devices)].mask)
+                (db.db[i / (DB_SIZE / n_devices)].mask
+                    & db.db[(i % (DB_SIZE / n_devices)) * n_devices].mask)
                     .count_ones() as u16
             );
         }