colin
/
nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
parent 8b31f30ed9 68f745d578
commit 33156229e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/development/python-modules/openai-triton/default.nix
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)