From b08de0c82cfe55348648327325b53836a51b13f5 Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Fri, 1 Dec 2023 11:32:59 -0500 Subject: [PATCH] Add additional Aurora configs for resource scaling model data collection --- tools/calibration/load/aurora/COND | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tools/calibration/load/aurora/COND b/tools/calibration/load/aurora/COND index 3999960e..5393bc40 100644 --- a/tools/calibration/load/aurora/COND +++ b/tools/calibration/load/aurora/COND @@ -74,3 +74,24 @@ for instance in INSTANCE_TYPES: for clients, query_idx in product(NUM_CLIENTS, range(NUM_QUERIES)) ], ) + +for instance in ["r6g_4xlarge", "r6g_8xlarge"]: + run_experiment_group( + name=f"{instance}-imdb_extended_100g", + run="python3 -m brad.calibration.measure_load", + experiments=[ + ExperimentInstance( + name=f"{instance}-imdb_extended_100g-1-q{query_idx}", + options={ + "num-clients": 1, + "specific-query-idx": query_idx, + "run-for-s": RUN_FOR_S, + "avg-gap-s": AVG_GAP_S, + "wait-before-start": WAIT_BEFORE_START, + "query-file": "../../../../workloads/IMDB_100GB/scaling_20/queries.sql", + "engine": "aurora", + }, + ) + for query_idx in range(NUM_QUERIES) + ], + )