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
, buildPythonPackage
, fetchFromGitHub
, fetchpatch
, jaxlib
, lapack
, numpy
@ -34,6 +35,12 @@ buildPythonPackage rec {
patches = [
# See https://github.com/google/jax/issues/7944
./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