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

python311Packages.openai-triton: use requested number of cores for build

This commit is contained in:
ruro 2024-04-17 20:17:08 +03:00
parent 1042fd8b14
commit 68f745d578
No known key found for this signature in database
GPG Key ID: 58187C6CC59F5BB0
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)