2 files changed, 10 insertions, 10 deletions

diff --git a/benchmarks/tail-latency/config_sweep.cpp b/benchmarks/tail-latency/config_sweep.cpp
index d973ee5..abc3f86 100644
--- a/benchmarks/tail-latency/config_sweep.cpp
+++ b/benchmarks/tail-latency/config_sweep.cpp
@@ -53,7 +53,7 @@ int main(int argc, char **argv) {
     for (size_t j=0; j<buffers.size(); j++) {
     for (size_t k=0; k<sfs.size(); k++) {
         auto policy = get_policy<Shard, Q>(sfs[k], buffers[j], policies[l]);
-        auto extension = new Ext(policy, buffers[j]/4, buffers[j]);
+        auto extension = new Ext(std::move(policy));
 
         /* warmup structure w/ 10% of records */
         size_t warmup = .1 * n;
@@ -63,7 +63,7 @@ int main(int argc, char **argv) {
             }
         }
 
-        extension->await_next_epoch();
+        extension->await_version();
 
         TIMER_INIT();
 
@@ -74,10 +74,10 @@ int main(int argc, char **argv) {
             }
             TIMER_STOP();
 
-            //fprintf(stdout, "%ld\t%ld\t%d\t%ld\n", sfs[k], buffers[j], policies[l], TIMER_RESULT());
+            fprintf(stdout, "I\t%ld\t%ld\t%d\t%ld\n", sfs[k], buffers[j], policies[l], TIMER_RESULT());
         }
 
-        extension->await_next_epoch();
+        extension->await_version();
         
         /* repeat the queries a bunch of times */
         for (size_t l=0; l<10; l++) {
@@ -88,7 +88,7 @@ int main(int argc, char **argv) {
             res.get();
             TIMER_STOP();
 
-            fprintf(stdout, "%ld\t%ld\t%d\t%ld\n", sfs[k], buffers[j], policies[l], TIMER_RESULT());
+            fprintf(stdout, "Q\t%ld\t%ld\t%d\t%ld\n", sfs[k], buffers[j], policies[l], TIMER_RESULT());
         }
         }
 
@@ -97,7 +97,7 @@ int main(int argc, char **argv) {
         auto res =extension->query(std::move(p));
 
         fprintf(stderr, "%ld\n", res.get());
-        extension->await_next_epoch();
+        extension->await_version();
         delete extension;
     }}}
 
diff --git a/benchmarks/tail-latency/fixed_shard_count.cpp b/benchmarks/tail-latency/fixed_shard_count.cpp
index fecf599..e980bcf 100644
--- a/benchmarks/tail-latency/fixed_shard_count.cpp
+++ b/benchmarks/tail-latency/fixed_shard_count.cpp
@@ -51,8 +51,8 @@ int main(int argc, char **argv) {
     size_t buffer_size = 8000;
 
     for (size_t i=0; i<shard_counts.size(); i++) {
-        auto policy = new de::FixedShardCountPolicy<Shard,Q>(buffer_size, shard_counts[i], n);
-        auto extension = new Ext(policy, buffer_size / 4, buffer_size);
+        auto policy = get_policy<Shard, Q>(shard_counts[i], buffer_size, 4, n);
+        auto extension = new Ext(std::move(policy));
 
         /* warmup structure w/ 10% of records */
         size_t warmup = .1 * n;
@@ -62,7 +62,7 @@ int main(int argc, char **argv) {
             }
         }
 
-        extension->await_next_epoch();
+        extension->await_version();
 
         TIMER_INIT();
 
@@ -76,7 +76,7 @@ int main(int argc, char **argv) {
 
         auto insert_tput = (size_t) ((double) (n - warmup) / (double) TIMER_RESULT() *1.0e9);
 
-        extension->await_next_epoch();
+        extension->await_version();
         
         /* repeat the queries a bunch of times */
         TIMER_START();