From 7c3fe4ea760f4773f0eb1a98ed3ba493a36015e2 Mon Sep 17 00:00:00 2001
From: Douglas Rumbaugh <dbr4@psu.edu>
Date: Wed, 17 Sep 2025 17:52:22 -0400
Subject: Last updates for dissertation

---
 benchmarks/tail-latency/knn_query_parm_sweep.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'benchmarks/tail-latency/knn_query_parm_sweep.cpp')

diff --git a/benchmarks/tail-latency/knn_query_parm_sweep.cpp b/benchmarks/tail-latency/knn_query_parm_sweep.cpp
index 1766c74..dcbe7cf 100644
--- a/benchmarks/tail-latency/knn_query_parm_sweep.cpp
+++ b/benchmarks/tail-latency/knn_query_parm_sweep.cpp
@@ -107,26 +107,28 @@ int main(int argc, char **argv) {
   auto data = read_vector_file<Rec, W2V_SIZE>(d_fname, n);
   auto queries = read_knn_queries<QP>(q_fname, 100, 1);
 
-  size_t buffer_size = 1000;
-  std::vector<size_t> policies = {0, 1};
+  std::vector<size_t> buffer_sizes = {100, 500, 1000, 10000};
+  std::vector<size_t> policies = {0, 1, 2};
 
   std::vector<size_t> thread_counts = {8};
   std::vector<double> modifiers = {0};
-  std::vector<size_t> scale_factors = {2, 4, 6, 8, 16, 32, 128}; 
+  std::vector<size_t> scale_factors = {2, 8}; 
 
   size_t insert_threads = 1;
   size_t query_threads = 1;
 
   reccnt = n;
 
+  for (auto bs : buffer_sizes) {
   for (auto pol : policies) {
     for (auto internal_thread_cnt : thread_counts) {
       for (auto mod : modifiers) {
         for (auto sf : scale_factors) {
-          auto policy = get_policy<Shard, Q>(sf, buffer_size, pol, n, mod);
+          auto policy = get_policy<Shard, Q>(sf, bs, pol, n, mod);
           auto config = Conf(std::move(policy));
           config.recon_enable_maint_on_flush = true;
           config.recon_maint_disabled = false;
+          config.buffer_size = bs;
           config.buffer_flush_trigger = config.buffer_size;
           config.maximum_threads = internal_thread_cnt;
 
@@ -212,6 +214,7 @@ int main(int argc, char **argv) {
       }
     }
   }
+  }
 
   fflush(stderr);
 }
-- 
cgit v1.2.3