mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-03-14 08:26:37 +01:00
dad7900f20
Diff: https://github.com/google/jax/compare/jax-v0.8.1...jax-v0.8.2 Changelog: https://docs.jax.dev/en/latest/changelog.html
23 lines
770 B
Bash
Executable file
23 lines
770 B
Bash
Executable file
#! /usr/bin/env nix-shell
|
|
#! nix-shell -i sh -p jq
|
|
|
|
prefetch() {
|
|
expr="(import <nixpkgs> { system = \"$2\"; config.cudaSupport = true; }).python$1.pkgs.$3.src.url"
|
|
url=$(NIX_PATH=.. nix-instantiate --eval -E "$expr" | jq -r)
|
|
echo "$url"
|
|
sha256=$(nix-prefetch-url "$url")
|
|
nix --extra-experimental-features nix-command hash convert --to sri --hash-algo sha256 "$sha256"
|
|
echo
|
|
}
|
|
|
|
for py in "311" "312" "313" "314"; do
|
|
prefetch "$py" "x86_64-linux" "jaxlib-bin"
|
|
prefetch "$py" "aarch64-linux" "jaxlib-bin"
|
|
prefetch "$py" "aarch64-darwin" "jaxlib-bin"
|
|
prefetch "$py" "x86_64-linux" "jax-cuda12-plugin"
|
|
prefetch "$py" "aarch64-linux" "jax-cuda12-plugin"
|
|
done
|
|
|
|
for arch in "x86_64-linux" "aarch64-linux"; do
|
|
prefetch "312" "$arch" "jax-cuda12-pjrt"
|
|
done
|