Use len(os.sched_getaffinity(0)) instead of os.cpu_count() (MagicStac…

…k#591) Co-authored-by: Fantix King <fantix.king@gmail.com>
edgarrmondragon · Aug 19, 2024 · a073237 · a073237
1 parent 1fa0240
commit a073237
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -13,6 +13,7 @@ jobs:
   test:
     runs-on: ${{ matrix.os }}
     strategy:
+      fail-fast: false
       matrix:
         python-version:
         - "3.8"

diff --git a/setup.py b/setup.py
@@ -176,7 +176,11 @@ def build_libuv(self):
             cmd,
             cwd=LIBUV_BUILD_DIR, env=env, check=True)
 
-        j_flag = '-j{}'.format(os.cpu_count() or 1)
+        try:
+            njobs = len(os.sched_getaffinity(0))
+        except AttributeError:
+            njobs = os.cpu_count()
+        j_flag = '-j{}'.format(njobs or 1)
         c_flag = "CFLAGS={}".format(env['CFLAGS'])
         subprocess.run(
             ['make', j_flag, c_flag],