python310Packages.jaxlib: 0.4.18 -> 0.4.19
This commit is contained in:
parent
46da009c99
commit
985b2f2bb3
1 changed files with 5 additions and 5 deletions
|
@ -54,7 +54,7 @@ let
|
||||||
inherit (cudaPackages) backendStdenv cudatoolkit cudaFlags cudnn nccl;
|
inherit (cudaPackages) backendStdenv cudatoolkit cudaFlags cudnn nccl;
|
||||||
|
|
||||||
pname = "jaxlib";
|
pname = "jaxlib";
|
||||||
version = "0.4.18";
|
version = "0.4.19";
|
||||||
|
|
||||||
meta = with lib; {
|
meta = with lib; {
|
||||||
description = "JAX is Autograd and XLA, brought together for high-performance machine learning research.";
|
description = "JAX is Autograd and XLA, brought together for high-performance machine learning research.";
|
||||||
|
@ -151,7 +151,7 @@ let
|
||||||
repo = "jax";
|
repo = "jax";
|
||||||
# google/jax contains tags for jax and jaxlib. Only use jaxlib tags!
|
# google/jax contains tags for jax and jaxlib. Only use jaxlib tags!
|
||||||
rev = "refs/tags/${pname}-v${version}";
|
rev = "refs/tags/${pname}-v${version}";
|
||||||
hash = "sha256-rDvWHa8jYCAA9iKbWaFUXdE/9L7AepFiNzmqOcc/090=";
|
hash = "sha256-l5uLPqhg/hqtO9oJSaioow5cH/0jKHDVziGezkfnVcc=";
|
||||||
};
|
};
|
||||||
|
|
||||||
nativeBuildInputs = [
|
nativeBuildInputs = [
|
||||||
|
@ -264,10 +264,10 @@ let
|
||||||
];
|
];
|
||||||
|
|
||||||
sha256 = (if cudaSupport then {
|
sha256 = (if cudaSupport then {
|
||||||
x86_64-linux = "sha256-0CfGWlwKsUFP1DHUN6+6wX3cHr5x3TE6NbqYlV5me1E=";
|
x86_64-linux = "sha256-Z5cSgdRxdKxidaz4b1RlUF4rVcQiUTmQ1OorlBWlpt0=";
|
||||||
} else {
|
} else {
|
||||||
x86_64-linux = "sha256-sljmyIligXC7d9fdlpqR32xyMR0UslWs04gXJBD8FTA=";
|
x86_64-linux = "sha256-sn7p8FFHWIVdBWnsLsVj5jLiSaTlRm7s/qj2RqvQ3jU=";
|
||||||
aarch64-linux = "sha256-eJ4KIkHdcA2EVvyBoNum2cOPcHPFoBOtUTAGufO8FJA=";
|
aarch64-linux = "sha256-oAYF5AeuPHTlwtpDMs2+tAhRAJH0yeSVnB7Ni7wmzS8=";
|
||||||
}).${stdenv.system} or (throw "jaxlib: unsupported system: ${stdenv.system}");
|
}).${stdenv.system} or (throw "jaxlib: unsupported system: ${stdenv.system}");
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue