Limiting the number of parallel builds for python CI

Signed-off-by: Fabian Sauter <[email protected]>

.github/workflows/python_tests.yml

@@ -20,7 +20,9 @@ jobs:
     - name: Install Python Requirements
       run: pip3 install --user -r python/test/requirements-dev.txt
     - name: Python Build
-      run: pip3 install --user .
+      env:
+        KOMPUTE_PYTHON_NUM_PARALLEL_THREADS: 2
+      run: pip3 install --user . -v
     - name: Python run Tests
       run: |
         export VK_ICD_FILENAMES=/swiftshader/vk_swiftshader_icd.json

setup.py

@@ -66,6 +66,9 @@ def build_extension(self, ext):
             env['CXXFLAGS'] += ' -fPIC'
             cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg]
             build_args += ['--', '-j']
+            # Optional environment variable to limit the number of parallel jobs for GitHub actions to reduce RAM usage
+            if 'KOMPUTE_PYTHON_NUM_PARALLEL_THREADS' in env:
+                build_args += env['KOMPUTE_PYTHON_NUM_PARALLEL_THREADS']
 
         if not os.path.exists(self.build_temp):
             os.makedirs(self.build_temp)