Give rdtset more configuration flexibility

Signed-off-by: Pavel Abramov <uncle.decart@gmail.com>

Author: uncleDecart

conf/config.yaml

@@ -8,6 +8,14 @@ run:
   stressor: true
   tests_path: "tests"
 
+  resctrl:
+    - mask: "0x7ff0"
+      cpus: "3"
+    - mask: "0x7fff"
+      cpus: "0,2,4,6,8,10,12,14"
+    - mask: "0x000f"
+      cpus: "1,5,7,9,11,13,15"
+
 cpus:
   t_core: "3"
   enable_turbo: false

main.py

@@ -221,7 +221,7 @@ def _run_caterpillar(
             "caterpillar:latest",
             "/bin/bash",
             "-c",
-            f'stdbuf -oL -eL /usr/sbin/rdtset -t "l3={l3_cache_mask};cpu={t_core}" '
+            f"stdbuf -oL -eL /usr/sbin/rdtset {self._compose_rdtset()} "
             f"-c {t_core} -k /opt/benchmarking/caterpillar/caterpillar -c {t_core} -s {self.config.caterpillar.n_cycles}",
         ]
 
@@ -509,6 +509,13 @@ def _start_stressor(self) -> None:
         except subprocess.CalledProcessError:
             print("Warning: Could not check stressor status")
 
+    def _compose_rdtset(self) -> str:
+        ans = ""
+        for group in self.config.run.resctrl:
+            ans += f' -t "l3={group.mask};cpu={group.cpus}"'
+
+        return ans
+
     @staticmethod
     def _run_command(cmd: List[str]) -> CompletedProcess | None:
         """Run a command and return its exit code."""

notebook/02_Megabenchmark.ipynb

@@ -17,6 +17,7 @@ dependencies = [
     "pyextremes>=2.3.3",
     "pymc>=5.26.1",
     "pytest>=8.4.2",
+    "scikit-learn>=1.7.2",
     "scipy>=1.16.3",
     "seaborn>=0.13.2",
     "statsmodels>=0.14.5",