re-enable streaming no heartbeat (#390)

* check deletion index * Revert "Revert "Stream DB entries to memory (#356)" (#384)" This reverts commit 844045d. * some bounds checks * disable heartbeat * bump * bump more --------- Co-authored-by: Wojciech Sromek <157375010+wojciechsromek@users.noreply.github.com>
worldcoin · Sep 17, 2024 · de6b4ab · de6b4ab
1 parent fcd26d0
commit de6b4ab
Show file tree

Hide file tree

Showing 8 changed files with 390 additions and 387 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:v0.6.4"
+image: "ghcr.io/worldcoin/iris-mpc:v0.6.5"
 
 environment: stage
 replicaCount: 1

diff --git a/iris-mpc-common/src/galois_engine.rs b/iris-mpc-common/src/galois_engine.rs
@@ -419,8 +419,6 @@ pub mod degree4 {
 
                 let res = 0.5f64 - (dot_codes as i16) as f64 / (2f64 * dot_masks as f64);
 
-                println!("{} {}", dot_codes, dot_masks);
-
                 // Without rotations
                 if rot_idx == 15 {
                     assert_float_eq!(

diff --git a/iris-mpc-gpu/benches/matmul.rs b/iris-mpc-gpu/benches/matmul.rs
@@ -38,7 +38,8 @@ fn bench_memcpy(c: &mut Criterion) {
     let preprocessed_query = preprocess_query(&query);
     let streams = device_manager.fork_streams();
     let blass = device_manager.create_cublas(&streams);
-    let (db_slices, db_sizes) = engine.load_db(&db, DB_SIZE, DB_SIZE, false);
+    let mut db_slices = engine.alloc_db(DB_SIZE);
+    let db_sizes = engine.load_full_db(&mut db_slices, &db);
 
     group.throughput(Throughput::Elements((DB_SIZE * QUERY_SIZE / 31) as u64));
     group.sample_size(10);