From 308a1308bd3b788da82e2355f9b932c6c476e18b Mon Sep 17 00:00:00 2001
From: Francesco Nigro <nigro.fra@gmail.com>
Date: Fri, 17 May 2024 12:48:56 +0200
Subject: [PATCH] Adding --cpuset-cpus tfb argument to set it on the server
 container

---
 toolset/run-tests.py              | 5 +++++
 toolset/utils/benchmark_config.py | 1 +
 toolset/utils/docker_helper.py    | 8 ++++++++
 3 files changed, 14 insertions(+)

diff --git a/toolset/run-tests.py b/toolset/run-tests.py
index 413ba37379c..c226edece26 100644
--- a/toolset/run-tests.py
+++ b/toolset/run-tests.py
@@ -67,6 +67,11 @@ def main(argv=None):
         ''')
 
     # Suite options
+    # CPU set  options
+    parser.add_argument(
+        '--cpuset-cpus',
+        default=None,
+        help='The cpu set to run framework container on')
     parser.add_argument(
         '--audit',
         action='store_true',
diff --git a/toolset/utils/benchmark_config.py b/toolset/utils/benchmark_config.py
index 0a580586420..48755e9f460 100755
--- a/toolset/utils/benchmark_config.py
+++ b/toolset/utils/benchmark_config.py
@@ -52,6 +52,7 @@ def __init__(self, args):
         self.database_docker_host = None
         self.client_docker_host = None
         self.network = None
+        self.cpuset_cpus = args.cpuset_cpus
         self.test_container_memory = args.test_container_memory
         self.extra_docker_runtime_args = args.extra_docker_runtime_args
 
diff --git a/toolset/utils/docker_helper.py b/toolset/utils/docker_helper.py
index c58bb71d6a2..1f3ea692ee1 100644
--- a/toolset/utils/docker_helper.py
+++ b/toolset/utils/docker_helper.py
@@ -189,6 +189,13 @@ def watch_container(docker_container, docker_file):
                 'soft': 99
             }]
 
+            cpuset_cpus = ''
+
+            if self.benchmarker.config.cpuset_cpus is not None:
+                    cpuset_cpus = self.benchmarker.config.cpuset_cpus
+
+            log("Running docker container with cpu set: %s" %cpuset_cpus)
+
             docker_cmd = ''
             if hasattr(test, 'docker_cmd'):
                 docker_cmd = test.docker_cmd
@@ -235,6 +242,7 @@ def watch_container(docker_container, docker_file):
                 sysctls=sysctl,
                 remove=True,
                 log_config={'type': None},
+                cpuset_cpus=cpuset_cpus,
                 **extra_docker_args
                 )