tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1#!/usr/bin/env nix-shell
2#!nix-shell -i bash -p nix-prefetch-scripts
3
4set -eou pipefail
5
6version=$1
7
8bucket="https://download.pytorch.org/whl"
9
10url_and_key_list=(
11 "x86_64-linux-310 $bucket/triton-${version}-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl triton-${version}-cp310-cp310-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"
12 "x86_64-linux-311 $bucket/triton-${version}-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl triton-${version}-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"
13 "x86_64-linux-312 $bucket/triton-${version}-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl triton-${version}-cp312-cp312-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"
14 "x86_64-linux-313 $bucket/triton-${version}-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl triton-${version}-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"
15 "x86_64-linux-314 $bucket/triton-${version}-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl triton-${version}-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"
16 "aarch64-linux-310 $bucket/triton-${version}-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl triton-${version}-cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"
17 "aarch64-linux-311 $bucket/triton-${version}-cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl triton-${version}-cp311-cp311-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"
18 "aarch64-linux-312 $bucket/triton-${version}-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl triton-${version}-cp312-cp312-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"
19 "aarch64-linux-313 $bucket/triton-${version}-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl triton-${version}-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"
20 "aarch64-linux-314 $bucket/triton-${version}-cp314-cp314-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl triton-${version}-cp314-cp314-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"
21)
22
23hashfile="binary-hashes-$version.nix"
24echo " \"$version\" = {" >>$hashfile
25
26for url_and_key in "${url_and_key_list[@]}"; do
27 key=$(echo "$url_and_key" | cut -d' ' -f1)
28 url=$(echo "$url_and_key" | cut -d' ' -f2)
29 name=$(echo "$url_and_key" | cut -d' ' -f3)
30
31 echo "prefetching ${url}..."
32 hash=$(nix --extra-experimental-features nix-command hash convert --hash-algo sha256 $(nix-prefetch-url "$url" --name "$name"))
33
34 echo " $key = {" >>$hashfile
35 echo " name = \"$name\";" >>$hashfile
36 echo " url = \"$url\";" >>$hashfile
37 echo " hash = \"$hash\";" >>$hashfile
38 echo " };" >>$hashfile
39
40 echo
41done
42
43echo " };" >>$hashfile
44echo "done."