chore: added M8192 tests & benchmarks

axiomhq · Dec 6, 2024 · 9c29541 · 9c29541
1 parent 43833b5
commit 9c29541
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 3 deletions.
diff --git a/benches/insert.rs b/benches/insert.rs
@@ -134,6 +134,17 @@ fn bench_ulysses(c: &mut Criterion) {
             }
         });
     });
+    group.bench_with_input("HyperTwoBits<8192>", &lines, |b, lines| {
+        let mut counter = HyperTwoBits::<h2b::M8192>::default();
+        b.iter(|| {
+            // for line in lines.chunks_exact(4) {
+            //     counter.insert4(&line[0], &line[1], &line[2], &line[3]);
+            // }
+            for line in lines {
+                counter.insert(line);
+            }
+        });
+    });
     group.bench_with_input("HyperThreeBits<64>", &lines, |b, lines| {
         let mut counter = HyperThreeBits::<h3b::M64>::default();
         b.iter(|| {
@@ -190,6 +201,14 @@ fn bench_ulysses(c: &mut Criterion) {
             }
         });
     });
+    group.bench_with_input("HyperThreeBits<8192>", &lines, |b, lines| {
+        let mut counter = HyperThreeBits::<h3b::M8192>::default();
+        b.iter(|| {
+            for line in lines {
+                counter.insert(line);
+            }
+        });
+    });
 
     group.finish();
 }

diff --git a/src/h2b.rs b/src/h2b.rs
@@ -4,7 +4,7 @@ mod tests;
 
 use std::hash::BuildHasher;
 
-pub use sketch::{Sketch, M1024, M128, M2048, M256, M4096, M512, M64};
+pub use sketch::{Sketch, M1024, M128, M2048, M256, M4096, M512, M64, M8192};
 
 use crate::AHasherDefaultBuilder;
 

diff --git a/src/h2b/tests.rs b/src/h2b/tests.rs
@@ -32,6 +32,10 @@ fn htb2048_size() {
 fn htb4096_size() {
     assert_eq!(std::mem::size_of::<HyperTwoBits<super::M4096>>(), 1040);
 }
+#[test]
+fn htb8192_size() {
+    assert_eq!(std::mem::size_of::<HyperTwoBits<super::M8192>>(), 2064);
+}
 
 #[allow(
     clippy::cast_precision_loss,

diff --git a/src/h3b.rs b/src/h3b.rs
@@ -4,7 +4,7 @@ mod tests;
 
 use std::hash::BuildHasher;
 
-pub use sketch::{Sketch, M1024, M128, M2048, M256, M4096, M512, M64};
+pub use sketch::{Sketch, M1024, M128, M2048, M256, M4096, M512, M64, M8192};
 
 use crate::AHasherDefaultBuilder;
 

diff --git a/src/h3b/sketch.rs b/src/h3b/sketch.rs
@@ -500,7 +500,7 @@ impl Sketch for M4096 {
     }
 }
 
-/// M = 4096 Sketch Implementation
+/// M = M8192 Sketch Implementation
 pub type M8192 = M128Reg<64>;
 
 impl Sketch for M8192 {
@@ -606,4 +606,8 @@ mod tests {
     fn test_m4096() {
         test::<M4096>();
     }
+    #[test]
+    fn test_m8192() {
+        test::<M8192>();
+    }
 }
diff --git a/src/h3b/tests.rs b/src/h3b/tests.rs
@@ -32,6 +32,10 @@ fn htb2048_size() {
 fn htb4096_size() {
     assert_eq!(std::mem::size_of::<HyperThreeBits<super::M4096>>(), 1552);
 }
+#[test]
+fn htb8192_size() {
+    assert_eq!(std::mem::size_of::<HyperThreeBits<super::M8192>>(), 3088);
+}
 
 #[allow(
     clippy::cast_precision_loss,