ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-10-16 09:28:35 +03:00
Code Issues Projects Releases Wiki Activity

Merge pull request #168707 from samuela/samuela/jax

Browse Source 
python3Packages.jax: fix build when testing against jaxlibWithCuda
This commit is contained in:
Samuel Ainsworth 2022-04-17 12:31:12 -07:00 committed by GitHub
commit 06d2cfa7b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,6 +3,7 @@
, blas , blas
, buildPythonPackage , buildPythonPackage
, fetchFromGitHub , fetchFromGitHub
, fetchpatch
, jaxlib , jaxlib
, lapack , lapack
, numpy , numpy
@ -34,6 +35,12 @@ buildPythonPackage rec {
patches = [ patches = [
# See https://github.com/google/jax/issues/7944 # See https://github.com/google/jax/issues/7944
./cache-fix.patch ./cache-fix.patch
# See https://github.com/google/jax/issues/10292
(fetchpatch {
url = "https://github.com/google/jax/commit/cadc8046d56e0c1433cf48a2f106947d5f4ecbfd.patch";
hash = "sha256-jrpIqt4LzWAswt/Cpwtfa5d1Yn31HcXkVH3ETmaigA0=";
})
]; ];
# jaxlib is _not_ included in propagatedBuildInputs because there are # jaxlib is _not_ included in propagatedBuildInputs because there are