Add zipfian argument to the calibration scripts

mitdbg · May 1, 2024 · f0b610f · f0b610f
1 parent ebabb2a
commit f0b610f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/tools/calibration/transactions/chbenchmark/COND b/tools/calibration/transactions/chbenchmark/COND
@@ -13,6 +13,8 @@ COND_INSTANCES = {
   instance: instance.replace(".", "_").replace("db.", "") for instance in INSTANCES
 }
 
+ZIPFIAN_ALPHA = 5.0
+
 combine(
   name="all",
   deps=[
@@ -36,6 +38,7 @@ for instance in INSTANCES:
           "txn-warehouses": 1740,
           "txn-config-file": "aurora.config",
           "schema-name": "chbenchmark",
+          "txn-zipfian-alpha": ZIPFIAN_ALPHA,
           "instance": instance,
         },
       )

diff --git a/tools/calibration/transactions/chbenchmark/run_instance.sh b/tools/calibration/transactions/chbenchmark/run_instance.sh
@@ -38,6 +38,10 @@ function extract_named_arguments() {
     if [[ $phys_arg =~ --instance=.+ ]]; then
       instance=${phys_arg:11}
     fi
+
+    if [[ $phys_arg =~ --txn-zipfian-alpha=.+ ]]; then
+      txn_zipfian_alpha=${phys_arg:20}
+    fi
   done
 }
 
@@ -74,7 +78,8 @@ RECORD_DETAILED_STATS=1 python3 -m pytpcc.tpcc aurora \
   --duration $run_for_s \
   --clients $t_clients \
   --scalefactor 1 \
-  --lat-sample-prob 0.25
+  --lat-sample-prob 0.25 \
+  --txn-zipfian-alpha $txn_zipfian_alpha
 popd
 
 >&2 echo "Waiting 10 seconds before retrieving metrics..."