tasks: override cpu count in tests

faasm · Aug 30, 2023 · 469e8cb · 469e8cb
1 parent c725c04
commit 469e8cb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tasks/tests.py b/tasks/tests.py
@@ -4,6 +4,8 @@
 from subprocess import run
 from tasks.util.env import FAABRIC_STATIC_BUILD_DIR, PROJ_ROOT
 
+IS_CI = "HOST_TYPE" in environ and environ["HOST_TYPE"] == "ci"
+
 TEST_ENV = {
     "LOG_LEVEL": "info",
     "PLANNER_HOST": "localhost",
@@ -23,6 +25,10 @@
     "UBSAN_OPTIONS": "print_stacktrace=1:halt_on_error=1",
 }
 
+# If executing in CI, give us some extra CPU cores to run the tests
+if IS_CI:
+    TEST_ENV["OVERRIDE_CPU_COUNT"] = 5
+
 
 @task(default=True)
 def tests(