nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-10-03 21:40:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.jax-cuda12-plugin: cleanup postInstall

Browse Source
This commit is contained in:
Gaetan Lepage
2026-09-05 16:50:01 +02:00
parent c3c18a3616
commit d7fd799186
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkgs/development/python-modules/jax-cuda12-plugin/default.nix
View File

@@ -70,9 +70,10 @@ buildPythonPackage {
# * https://github.com/NixOS/nixpkgs/pull/375186
# for more info.
postInstall = ''
mkdir -p $out/${python.sitePackages}/jax_cuda12_plugin/cuda/bin
ln -s ${lib.getExe' cudaPackages.cuda_nvcc "ptxas"} $out/${python.sitePackages}/jax_cuda12_plugin/cuda/bin
ln -s ${lib.getExe' cudaPackages.cuda_nvcc "nvlink"} $out/${python.sitePackages}/jax_cuda12_plugin/cuda/bin
export BINPATH="$out/${python.sitePackages}/jax_cuda12_plugin/cuda/bin"
mkdir -p $BINPATH
ln -s ${lib.getExe' cudaPackages.cuda_nvcc "ptxas"} $BINPATH/ptxas
ln -s ${lib.getExe' cudaPackages.cuda_nvcc "nvlink"} $BINPATH/nvlink
'';
# jax-cuda12-plugin contains shared libraries that open other shared libraries via dlopen