summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/buildstream/_platform/platform.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/buildstream/_platform/platform.py b/src/buildstream/_platform/platform.py
index dba60ddca..0fbe145f9 100644
--- a/src/buildstream/_platform/platform.py
+++ b/src/buildstream/_platform/platform.py
@@ -22,6 +22,8 @@ import platform
import sys
import resource
+import psutil
+
from .._exceptions import PlatformError, ImplError
@@ -68,7 +70,7 @@ class Platform():
return cls._instance
def get_cpu_count(self, cap=None):
- cpu_count = len(os.sched_getaffinity(0))
+ cpu_count = len(psutil.Process().cpu_affinity())
if cap is None:
return cpu_count
else: