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

python311Packages.openai-triton: use requested number of cores for build
This commit is contained in:
Aleksana 2024-04-20 19:19:23 +08:00 committed by GitHub
commit 33156229e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)