ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-10-17 18:47:57 +03:00
Code Issues Projects Releases Wiki Activity

Merge pull request #144814 from kevinmehall/pytorch

Browse Source 
PyTorch: Reduce closure size by ~500MB
This commit is contained in:
Rick van Schijndel 2022-04-28 22:57:35 +02:00 committed by GitHub
commit 59ae32d267
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkgs/development/python-modules/pytorch/default.nix
View File

@ -6,7 +6,7 @@
cudaArchList ? null, cudaArchList ? null,
# Native build inputs # Native build inputs
cmake, util-linux, linkFarm, symlinkJoin, which, pybind11, cmake, util-linux, linkFarm, symlinkJoin, which, pybind11, removeReferencesTo,
# Build inputs # Build inputs
numactl, numactl,
@ -220,6 +220,7 @@ in buildPythonPackage rec {
which which
ninja ninja
pybind11 pybind11
removeReferencesTo
] ++ lib.optionals cudaSupport [ cudatoolkit_joined ]; ] ++ lib.optionals cudaSupport [ cudatoolkit_joined ];
buildInputs = [ blas blas.provider ] buildInputs = [ blas blas.provider ]
@ -259,6 +260,8 @@ in buildPythonPackage rec {
]) ])
]; ];
postInstall = '' postInstall = ''
find "$out/${python.sitePackages}/torch/include" "$out/${python.sitePackages}/torch/lib" -type f -exec remove-references-to -t ${stdenv.cc} '{}' +
mkdir $dev mkdir $dev
cp -r $out/${python.sitePackages}/torch/include $dev/include cp -r $out/${python.sitePackages}/torch/include $dev/include
cp -r $out/${python.sitePackages}/torch/share $dev/share cp -r $out/${python.sitePackages}/torch/share $dev/share
@ -273,7 +276,8 @@ in buildPythonPackage rec {
--replace \''${_IMPORT_PREFIX}/lib "$lib/lib" --replace \''${_IMPORT_PREFIX}/lib "$lib/lib"
mkdir $lib mkdir $lib
cp -r $out/${python.sitePackages}/torch/lib $lib/lib mv $out/${python.sitePackages}/torch/lib $lib/lib
ln -s $lib/lib $out/${python.sitePackages}/torch/lib
''; '';
postFixup = lib.optionalString stdenv.isDarwin '' postFixup = lib.optionalString stdenv.isDarwin ''