From e4f1682e66ff16871692ff84768da8ff4e2497bd Mon Sep 17 00:00:00 2001
From: Alexander Kalistratov <alexander.kalistratov@intel.com>
Date: Tue, 18 Jul 2023 17:57:22 +0200
Subject: [PATCH] Add M16G preset to main workloads

---
 dpbench/configs/bench_info/black_scholes.toml     | 4 ++++
 dpbench/configs/bench_info/dbscan.toml            | 6 ++++++
 dpbench/configs/bench_info/gpairs.toml            | 7 +++++++
 dpbench/configs/bench_info/kmeans.toml            | 7 +++++++
 dpbench/configs/bench_info/knn.toml               | 9 +++++++++
 dpbench/configs/bench_info/l2_norm.toml           | 5 +++++
 dpbench/configs/bench_info/pairwise_distance.toml | 5 +++++
 dpbench/configs/bench_info/pca.toml               | 4 ++++
 dpbench/configs/bench_info/rambo.toml             | 4 ++++
 dpbench/console/run.py                            | 2 +-
 10 files changed, 52 insertions(+), 1 deletion(-)

diff --git a/dpbench/configs/bench_info/black_scholes.toml b/dpbench/configs/bench_info/black_scholes.toml
index d71f0a3d..ff917b9c 100644
--- a/dpbench/configs/bench_info/black_scholes.toml
+++ b/dpbench/configs/bench_info/black_scholes.toml
@@ -36,6 +36,10 @@ output_args = [
 nopt = 524288
 seed = 777777
 
+[benchmark.parameters.M16Gb]
+nopt = 67108864
+seed = 777777
+
 [benchmark.parameters.M]
 nopt = 134217728
 seed = 777777
diff --git a/dpbench/configs/bench_info/dbscan.toml b/dpbench/configs/bench_info/dbscan.toml
index c6ee4e8f..9d986101 100644
--- a/dpbench/configs/bench_info/dbscan.toml
+++ b/dpbench/configs/bench_info/dbscan.toml
@@ -27,6 +27,12 @@ n_features = 10
 centers = 10
 seed = 777777
 
+[benchmark.parameters.M16Gb]
+n_samples = 8192
+n_features = 10
+centers = 10
+seed = 777777
+
 [benchmark.parameters.M]
 n_samples = 8192
 n_features = 10
diff --git a/dpbench/configs/bench_info/gpairs.toml b/dpbench/configs/bench_info/gpairs.toml
index a19f9eea..4a2c5ad6 100644
--- a/dpbench/configs/bench_info/gpairs.toml
+++ b/dpbench/configs/bench_info/gpairs.toml
@@ -48,6 +48,13 @@ nbins = 20
 rmax = 50
 rmin = 0.1
 
+[benchmark.parameters.M16Gb]
+nopt = 4096
+seed = 1234
+nbins = 20
+rmax = 50
+rmin = 0.1
+
 [benchmark.parameters.M]
 nopt = 8192
 seed = 1234
diff --git a/dpbench/configs/bench_info/kmeans.toml b/dpbench/configs/bench_info/kmeans.toml
index a1dd500f..584091e6 100644
--- a/dpbench/configs/bench_info/kmeans.toml
+++ b/dpbench/configs/bench_info/kmeans.toml
@@ -37,6 +37,13 @@ seed = 7777777
 ndims = 2
 ncentroids = 10
 
+[benchmark.parameters.M16Gb]
+npoints = 1048576
+niters = 30
+seed = 7777777
+ndims = 2
+ncentroids = 10
+
 [benchmark.parameters.M]
 npoints = 1048576
 niters = 30
diff --git a/dpbench/configs/bench_info/knn.toml b/dpbench/configs/bench_info/knn.toml
index 927da1b4..c631e94b 100644
--- a/dpbench/configs/bench_info/knn.toml
+++ b/dpbench/configs/bench_info/knn.toml
@@ -45,6 +45,15 @@ seed_test = 777777
 seed_train = 0
 k = 5
 
+[benchmark.parameters.M16Gb]
+test_size = 8388608
+train_size = 1024
+data_dim = 16
+classes_num = 3
+seed_test = 777777
+seed_train = 0
+k = 5
+
 [benchmark.parameters.M]
 test_size = 8388608
 train_size = 1024
diff --git a/dpbench/configs/bench_info/l2_norm.toml b/dpbench/configs/bench_info/l2_norm.toml
index 37cb31a1..e5e29ea6 100644
--- a/dpbench/configs/bench_info/l2_norm.toml
+++ b/dpbench/configs/bench_info/l2_norm.toml
@@ -28,6 +28,11 @@ npoints = 32768
 dims = 3
 seed = 777777
 
+[benchmark.parameters.M16Gb]
+npoints = 134217728
+dims = 3
+seed = 777777
+
 [benchmark.parameters.M]
 npoints = 268435456
 dims = 3
diff --git a/dpbench/configs/bench_info/pairwise_distance.toml b/dpbench/configs/bench_info/pairwise_distance.toml
index f3819f7c..ac133a08 100644
--- a/dpbench/configs/bench_info/pairwise_distance.toml
+++ b/dpbench/configs/bench_info/pairwise_distance.toml
@@ -30,6 +30,11 @@ npoints = 1024
 dims = 3
 seed = 7777777
 
+[benchmark.parameters.M16Gb]
+npoints = 21846
+dims = 3
+seed = 7777777
+
 [benchmark.parameters.M]
 npoints = 32768
 dims = 3
diff --git a/dpbench/configs/bench_info/pca.toml b/dpbench/configs/bench_info/pca.toml
index 9f2e6072..001b5286 100644
--- a/dpbench/configs/bench_info/pca.toml
+++ b/dpbench/configs/bench_info/pca.toml
@@ -27,6 +27,10 @@ expected_failure_implementations = ["numba_dpex_n"]
 npoints = 1024
 dims = 128
 
+[benchmark.parameters.M16Gb]
+npoints = 1048576
+dims = 128
+
 [benchmark.parameters.M]
 npoints = 1048576
 dims = 128
diff --git a/dpbench/configs/bench_info/rambo.toml b/dpbench/configs/bench_info/rambo.toml
index e502506e..0f7301c0 100644
--- a/dpbench/configs/bench_info/rambo.toml
+++ b/dpbench/configs/bench_info/rambo.toml
@@ -32,6 +32,10 @@ output_args = [
 nevts = 32768
 nout = 4
 
+[benchmark.parameters.M16Gb]
+nevts = 16777216
+nout = 4
+
 [benchmark.parameters.M]
 nevts = 8388608
 nout = 4
diff --git a/dpbench/console/run.py b/dpbench/console/run.py
index e5179cbb..e22433dd 100644
--- a/dpbench/console/run.py
+++ b/dpbench/console/run.py
@@ -27,7 +27,7 @@ def add_run_arguments(parser: argparse.ArgumentParser):
     parser.add_argument(
         "-p",
         "--preset",
-        choices=["S", "M", "L"],
+        choices=["S", "M16Gb", "M", "L"],
         type=str,
         nargs="?",
         default="S",