Add cpuset_cpus stage parameter and update README

illinois-cs241 · Oct 6, 2023 · b7bf84f · b7bf84f
1 parent e3f01f3
commit b7bf84f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -37,6 +37,8 @@ stages = [{
   "image": "alpine:3.5",
   # memory cap (optional, defaults to 2GB)
   "memory": "2g",
+  # set of cpus to use, e.g. 0-3 for the first four or 0,2 for the first and third (optional, defaults to all)
+  "cpuset_cpus": "0-7",
   # whether to allow networking capabilities (optional, defaults to True)
   "networking": True,
   # whether to switch on privileged mode (optional, defaults to False)

diff --git a/chainlink/__init__.py b/chainlink/__init__.py
@@ -85,6 +85,7 @@ async def _run_stage(self, stage, mount, environ):
         # TODO find a way to set disk limit?
         options = {
             "cpu_period": 100000,  # microseconds
+            "cpuset_cpus": stage.get("cpuset_cpus", None),
             "detach": True,
             "entrypoint": stage.get("entrypoint", None),
             "environment": environ,