feat, test: benchmark prefix_rank

SGSSGene · Jul 22, 2024 · 8039adc · 8039adc
1 parent 0ec503a
commit 8039adc
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/test_fmindex-collection/rankvector/checkRankVector.cpp b/src/test_fmindex-collection/rankvector/checkRankVector.cpp
@@ -371,6 +371,7 @@ struct Bench : ankerl::nanobench::Bench {
 
 struct Benchs {
     Bench bench_rank{"rank()"};
+    Bench bench_prefix_rank{"prefix_rank()"};
     Bench bench_symbol{"symbol()"};
     Bench bench_ctor{"c'tor"};
 };
@@ -383,7 +384,7 @@ TEMPLATE_TEST_CASE("benchmark vectors c'tor,symbol() and rank() operations", "[R
         return;
     }
 
-    auto& [bench_rank, bench_symbol, bench_ctor] = benchs_256;
+    auto& [bench_rank, bench_prefix_rank, bench_symbol, bench_ctor] = benchs_256;
 
 
     auto vector_name = getName<Vector>();
@@ -415,7 +416,12 @@ TEMPLATE_TEST_CASE("benchmark vectors c'tor,symbol() and rank() operations", "[R
         });
 
         bench_rank.minEpochIterations(2'000'000).run(vector_name, [&]() {
-            auto v = vec.rank(rng.bounded(text.size()), rng.bounded(4)+1);
+            auto v = vec.rank(rng.bounded(text.size()), rng.bounded(256));
+            ankerl::nanobench::doNotOptimizeAway(v);
+        });
+
+        bench_prefix_rank.minEpochIterations(2'000'000).run(vector_name, [&]() {
+            auto v = vec.prefix_rank(rng.bounded(text.size()), rng.bounded(256));
             ankerl::nanobench::doNotOptimizeAway(v);
         });
 
@@ -436,7 +442,7 @@ TEMPLATE_TEST_CASE("benchmark vectors c'tor,symbol() and rank() operations, dna4
         return;
     }
 
-    auto& [bench_rank, bench_symbol, bench_ctor] = benchs_5;
+    auto& [bench_rank, bench_prefix_rank, bench_symbol, bench_ctor] = benchs_5;
 
 
     auto vector_name = getName<Vector>();
@@ -472,6 +478,10 @@ TEMPLATE_TEST_CASE("benchmark vectors c'tor,symbol() and rank() operations, dna4
             ankerl::nanobench::doNotOptimizeAway(v);
         });
 
+        bench_prefix_rank.minEpochIterations(2'000'000).run(vector_name, [&]() {
+            auto v = vec.prefix_rank(rng.bounded(text.size()), rng.bounded(4)+1);
+            ankerl::nanobench::doNotOptimizeAway(v);
+        });
         {
             auto ofs     = std::stringstream{};
             auto archive = cereal::BinaryOutputArchive{ofs};