Merge pull request #305120 from RuRo/fix-triton-max-jobs

python311Packages.openai-triton: use requested number of cores for build
2024-04-20 19:19:23 +08:00
parent 8b31f30ed9 68f745d578
commit 33156229e2
1 changed files with 3 additions and 0 deletions
@@ -115,6 +115,9 @@ buildPythonPackage rec {

  # Avoid GLIBCXX mismatch with other cuda-enabled python packages
  preConfigure = ''
+    # Ensure that the build process uses the requested number of cores
+    export MAX_JOBS="$NIX_BUILD_CORES"
+
    # Upstream's setup.py tries to write cache somewhere in ~/
    export HOME=$(mktemp -d)