Skip to content

nix-builder: update nixpkgs, bumps Python from 3.13 to 3.14 #831

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
danieldk merged 2 commits into from
Sep 17, 2026
Merged

nix-builder: update nixpkgs, bumps Python from 3.13 to 3.14 #831

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b003e31
nix-builder: update nixpkgs, bumps Python from 3.13 to 3.14
danieldk Sep 16, 2026
36142b8
nix-builder: fix cc-wrappers unconditionally injecting `-mtls-dialect…
danieldk Sep 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions flake.lock
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

5 changes: 1 addition & 4 deletions flake.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,7 @@

inputs = {
flake-utils.url = "github:numtide/flake-utils";
# Temporarily pin this commit to fix ucx issues:
# https://github.com/NixOS/nixpkgs/pull/538738
nixpkgs.url = "github:NixOS/nixpkgs/ec1a11210589d294f0ac99d3290a27e6c73dfa1d";
#nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable-small";
nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable-small";

@sayakpaul sayakpaul Sep 17, 2026

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we use pinned commits?

@danieldk danieldk Sep 17, 2026

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

They are pinned in flake.lock.

flake-compat.url = "github:edolstra/flake-compat";
rust-overlay = {
url = "github:oxalica/rust-overlay";
Expand Down
4 changes: 1 addition & 3 deletions kernels/tests/test_entry_points.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,7 @@ def test_egg_info_writer_entry_point_is_importable():
`egg_info` run in an environment where `kernels` is installed, including
builds of unrelated packages. If it points at a module that is not shipped,
every such build fails with `ModuleNotFoundError`."""
entry_points = [
ep for ep in distribution("kernels").entry_points if ep.group == "egg_info.writers"
]
entry_points = [ep for ep in distribution("kernels").entry_points if ep.group == "egg_info.writers"]
assert entry_points, "no egg_info.writers entry point registered"

for ep in entry_points:
Expand Down
6 changes: 3 additions & 3 deletions nix-builder/overlay.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -131,12 +131,12 @@ final: prev:
);

hf-xet = python-super.hf-xet.overridePythonAttrs (prevAttrs: rec {
version = "1.4.3";
version = "1.6.0";
src = final.fetchFromGitHub {
owner = "huggingface";
repo = "xet-core";
tag = "v${version}";
hash = "sha256-zAliMR2d2j6ynHQmAljQ8XgDyjuPxNawI1bZks5aRgs=";
hash = "sha256-SP6Z8iIkrt3FVXxXYdvjeiIAfcrVlfCPQq6C36DfhEM=";
};
cargoDeps = final.rustPlatform.fetchCargoVendor {
inherit (prevAttrs)
Expand All @@ -147,7 +147,7 @@ final: prev:
version
src
;
hash = "sha256-TOgBT0l7TvJamVdIAdAUFRWs8AMRRY+Ydoh6e+3dEp0=";
hash = "sha256-pvtq9mKlmwaqAq281Lin/UgVGcRe2SEvyzCa+xWSwVQ=";
};
});

Expand Down
8 changes: 4 additions & 4 deletions nix-builder/pkgs/python-modules/cuda-bindings/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,15 +19,15 @@ let
cuda_12 = {
version = "12.9.5";
hash = {
x86_64-linux = "sha256-2jdFj4xQdNWQQN7dvQBUYMDPcNHvkN19LIFuRoYDjoc=";
aarch64-linux = "sha256-4G0nKyUUnhqDtEb/Pb789vgvZoZcSejZ23VoO7z/C18=";
x86_64-linux = "sha256-QNx5yMv2Y+JNJgeyqXl4fg1oQLYkpNpIzVMDrLhIKtI=";
aarch64-linux = "sha256-OzqJ/M1swexTWpQLQyQ3YVN65XPZd8PNGOOPhB4fioY=";
};
};
cuda_13 = {
version = "13.2.0";
hash = {
x86_64-linux = "sha256-fcoNoFPTtMxIae/0nGHAPzxduqC81xIxejWNW48/OF0=";
aarch64-linux = "sha256-ZinKLfb3lbeEdSQJvK7b0ip6ZRt0tWoWXrwMncvVBNA=";
x86_64-linux = "sha256-9K+fPhvmA/oS1a1s/KeETJ0jC++peStavffdeZecNiY=";
aarch64-linux = "sha256-pkZLMPRmktbH9l1KDgRQ2B3SneOvwbtRVlOXPQHCzW4=";
};
};
in
Expand Down
6 changes: 3 additions & 3 deletions nix-builder/pkgs/python-modules/libtpu/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,18 +31,18 @@
let
# The wheel ships per-CPython-ABI builds (cp311..cp314); pick the tag
# matching the python this package set is built for. The hash below is
# for cp313 (the nixpkgs default python); if either moves, recompute
# for cp314 (the nixpkgs default python); if either moves, recompute
# via the prefetch script above.
abi = "cp${lib.versions.major python.version}${lib.versions.minor python.version}";
in
buildPythonPackage rec {
pname = "libtpu";
version = "0.0.44.1";
version = "0.0.48";
format = "wheel";

src = requireFile {
name = "libtpu-${version}-${abi}-${abi}-manylinux_2_31_x86_64.whl";
hash = "sha256-qd34JzD0TpphtqLj67/HqYgVg/bXT3gF6E+pYQyaCsA="; # cp313
hash = "sha256-QD0G/ssztiQYMmd/gwI/IXlYnwyZ4lwRocEf/7V7NBY=";
message = ''
libtpu is served from a gated Google Artifact Registry and cannot
be fetched by a pure Nix build. Fetch and register it with:
Expand Down
4 changes: 2 additions & 2 deletions nix-builder/pkgs/python-modules/nvidia-cuda-nvdisasm/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,8 @@ let
format = "wheel";
pyShortVersion = "cp" + builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
hashes = {
cp313-aarch64-linux = "sha256-3UdRiE+QFrm22/AHq961aB0KLtxzHdPS/anW2Hjoj3M=";
cp313-x86_64-linux = "sha256-+hcISwfA3KaKQokvdxtLG0D76bkWYCCWI+Yc6mEcrow=";
cp314-aarch64-linux = "sha256-3UdRiE+QFrm22/AHq961aB0KLtxzHdPS/anW2Hjoj3M=";
cp314-x86_64-linux = "sha256-+hcISwfA3KaKQokvdxtLG0D76bkWYCCWI+Yc6mEcrow=";
};
hash =
hashes."${pyShortVersion}-${stdenv.system}"
Expand Down
8 changes: 4 additions & 4 deletions nix-builder/pkgs/python-modules/nvidia-cutlass-dsl-libs-cu/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,10 @@ let
format = "wheel";
pyShortVersion = "cp" + builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
hashes = {
cp313-x86_64-linux-cu12 = "sha256-JQS1vbLwIQpqG8kjrO/peQBjZmlLc/FFijyFca7zt2U=";
cp313-aarch64-linux-cu12 = "sha256-F9nVvbj1MCZtyEJCNqgFI+4qmAfoUqJMIglkOKEt870=";
cp313-x86_64-linux-cu13 = "sha256-famVhPTw1JjKcKZHAiqylcpUoglJRBGBzQwyiVhNqKM=";
cp313-aarch64-linux-cu13 = "sha256-gTcFEd/Z4cCNojAs8UD0reyJ31BOw0iiOwiqAXRyDsM=";
cp314-x86_64-linux-cu12 = "sha256-Avifeb/8ofZw3D4Z5SOPEggFFjDrTr2DVxzKqjTn6kU=";
cp314-aarch64-linux-cu12 = "sha256-yE7F1PRbY0q+FrLxEcU1L9IexmXruSnkAxiD3PE10xE=";
cp314-x86_64-linux-cu13 = "sha256-iPWAJ6JUrD5rlZLq82Ely27Pd61OQCaEeAQSVUbsSxA=";
cp314-aarch64-linux-cu13 = "sha256-q7bO5j0gI4v4/349rBJTS16LxF1CVic9l5OYrrF6Rr0=";
};
hash =
hashes."${pyShortVersion}-${stdenv.system}-cu${cudaMajor}"
Expand Down
8 changes: 4 additions & 4 deletions nix-builder/pkgs/python-modules/nvidia-cutlass-dsl-libs/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,10 @@ let
format = "wheel";
pyShortVersion = "cp" + builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
hashes = {
cp313-x86_64-linux-cu12 = "sha256-CaaqzoQVlOo5z3X3zuaNwxuM88263/y9SU8I/HPGRm0=";
cp313-aarch64-linux-cu12 = "sha256-fKTlypvQyqLseX67HkBLs+sPdtI1gjYReVyxTY00Hkw=";
cp313-x86_64-linux-cu13 = "sha256-CaaqzoQVlOo5z3X3zuaNwxuM88263/y9SU8I/HPGRm0=";
cp313-aarch64-linux-cu13 = "sha256-fKTlypvQyqLseX67HkBLs+sPdtI1gjYReVyxTY00Hkw=";
cp314-x86_64-linux-cu12 = "sha256-J0Yf9SM4KAtsAkdfh4USc7WhHDh9nao4M3YiWj92fMI=";
cp314-aarch64-linux-cu12 = "sha256-loFa+YsZuARYAv2LClPV4Zfg07zkLHA44oYa98hIP10=";
cp314-x86_64-linux-cu13 = "sha256-J0Yf9SM4KAtsAkdfh4USc7WhHDh9nao4M3YiWj92fMI=";
cp314-aarch64-linux-cu13 = "sha256-loFa+YsZuARYAv2LClPV4Zfg07zkLHA44oYa98hIP10=";
};
hash =
hashes."${pyShortVersion}-${stdenv.system}-cu${cudaPackages.cudaMajorVersion}"
Expand Down
4 changes: 2 additions & 2 deletions nix-builder/pkgs/python-modules/torch/binary/generic.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -151,11 +151,11 @@ let
torch-tpu =
let
abi = "cp${lib.versions.major python.version}${lib.versions.minor python.version}";
version = "0.1.1.dev20260707090224";
version = "0.1.1.dev20260916100444";
in
requireFile {
name = "torch_tpu-${version}-${abi}-${abi}-manylinux_2_31_x86_64.whl";
hash = "sha256-eCjwoX0UKd/L/IccxXn88p0GdyQjFdPl2Er1luZz4H0="; # cp313
hash = "sha256-A8Rw9/QkMCjj9jfb4s+/d04/m/xH5L9rJDa0+giessM=";
message = ''
torch_tpu is served from a gated Google Artifact Registry and
cannot be fetched by a pure Nix build. Fetch and register it with:
Expand Down
126 changes: 63 additions & 63 deletions nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,144 +2,144 @@
"2.11": {
"x86_64-linux": {
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.11.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-RQJdd1LbxrTHhMA6+u6cXxlzDOCEsuQ/yaL+FnfZ/4Y=",
"url": "https://download.pytorch.org/whl/cpu/torch-2.11.0%2Bcpu-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-+EgeqQiOTluBF4p1qr27ZYvehjm8GhX9XY+TCryWZzU=",
"version": "2.11.0"
}
}
},
"2.13": {
"x86_64-linux": {
"cu126": {
"url": "https://download.pytorch.org/whl/cu126/torch-2.13.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-QZjI10eKtHrSVpMJOH2Ish+1U6HPirBiYPvVpqublxI=",
"cu130": {
"url": "https://download.pytorch.org/whl/cu130/torch-2.13.0%2Bcu130-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-4jEwKkVymNAjb3veMQglaPbNBhO2a060aEnorVPC440=",
"version": "2.13.0"
},
"cu130": {
"url": "https://download.pytorch.org/whl/cu130/torch-2.13.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-lb260vB4a9RIky5LcqZASqIikNtuyVTiTt+m0zyDPI8=",
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0%2Bcpu-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-0g+lPudEUC+kxpgYpyCwXKDTer0FXU9uZsrhVRFLxpE=",
"version": "2.13.0"
},
"cu126": {
"url": "https://download.pytorch.org/whl/cu126/torch-2.13.0%2Bcu126-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-r/07SaTkiu+JX/2eGWGBvVBUItfDe0pvmvp2iZ0qzT0=",
"version": "2.13.0"
},
"cu132": {
"url": "https://download.pytorch.org/whl/cu132/torch-2.13.0%2Bcu132-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-GMmhYaD7ZJ167wriy/kQ+H/7pRSa6BJ9D/QSOWmLbIo=",
"url": "https://download.pytorch.org/whl/cu132/torch-2.13.0%2Bcu132-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-JP+fQfRyQJ0pjShJdOI+PlkjML/VNClVfqWb/6SZq6g=",
"version": "2.13.0"
},
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-P7+cnR88EMLVnQSspCbe6czGzrMtJVxh6TrMO091+uY=",
"xpu": {
"url": "https://download.pytorch.org/whl/xpu/torch-2.13.0%2Bxpu-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-ArcpOjS+L13T466NApEZdwetRjTeK9p6FpdfKruQidU=",
"version": "2.13.0"
},
"rocm71": {
"url": "https://download.pytorch.org/whl/rocm7.1/torch-2.13.0%2Brocm7.1-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-NHcAYyaBihRFcXkm2b5o+KD2DCX/zjtGFHGAsfpeRVw=",
"url": "https://download.pytorch.org/whl/rocm7.1/torch-2.13.0%2Brocm7.1-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-FinA4sY15Z1K+iDGb/My2V4inVDK0EplKBJaUD+0HMY=",
"version": "2.13.0"
},
"rocm72": {
"url": "https://download.pytorch.org/whl/rocm7.2/torch-2.13.0%2Brocm7.2-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-kSHk6QZMVP3d7CukzG/B1NNx4lJiflyjngb2+gflHxo=",
"version": "2.13.0"
},
"xpu": {
"url": "https://download.pytorch.org/whl/xpu/torch-2.13.0%2Bxpu-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-DDY+EveM5ng5orILI4eyuq5JGuZHDGfZnJkYA9xZJ70=",
"url": "https://download.pytorch.org/whl/rocm7.2/torch-2.13.0%2Brocm7.2-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-gm3oRE9Me56sC/3oZcTNe4UrohezX7cadmuFrB4zEOI=",
"version": "2.13.0"
}
},
"aarch64-linux": {
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0%2Bcpu-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-ygIfnrL4NFyD+gPjoEWHMIr7jfcb1HJnCz7OAN9YYhw=",
"version": "2.13.0"
},
"cu126": {
"url": "https://download.pytorch.org/whl/cu126/torch-2.13.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-smGCNbrLKe6jc1fOZgqrUpbZcp6s/UYp1QMnOC2hH38=",
"url": "https://download.pytorch.org/whl/cu126/torch-2.13.0%2Bcu126-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-obilv+JA2N/fLU8LcFMFUuGTUFDM1X1FA/kYKeN3M3A=",
"version": "2.13.0"
},
"cu130": {
"url": "https://download.pytorch.org/whl/cu130/torch-2.13.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-zsET+2Dw/p0sBNgGHVYN2uP4v3gphMHnuZMINgjXC5E=",
"url": "https://download.pytorch.org/whl/cu130/torch-2.13.0%2Bcu130-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-CtGIzfp6XMxnNmyOgCadMrRev/x1nCgtEhUWkiQFZio=",
"version": "2.13.0"
},
"cu132": {
"url": "https://download.pytorch.org/whl/cu132/torch-2.13.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-kTV7e3bs14i9diAFqjdOnfzcJcTFz1AZRzQmVOtw6KI=",
"version": "2.13.0"
},
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-C499BCMCeui5DHl3xifzN58yU2OggiTf+tm0staEqD0=",
"url": "https://download.pytorch.org/whl/cu132/torch-2.13.0%2Bcu132-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-mrdBLoddWwRYLr1RSbxGnV0dUYapMEGwLU6783u6U7Q=",
"version": "2.13.0"
}
},
"aarch64-darwin": {
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0-cp313-cp313-macosx_14_0_arm64.whl",
"hash": "sha256-M0SYmc5UlsG4S0hTF52U/RAgKK4UBzFNn7lWu3nnDQk=",
"url": "https://download.pytorch.org/whl/cpu/torch-2.13.0-cp314-cp314-macosx_14_0_arm64.whl",
"hash": "sha256-2EmzkOB9jTM86OyvkbJzxlbFmDeaGcms8TGKiD9rORw=",
"version": "2.13.0"
}
}
},
"2.14": {
"x86_64-linux": {
"cu126": {
"url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-rJ0c4Qk2BvXLHetPYS0vyyc+0Q/0ozcDq7oJzBbF3Mg=",
"url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-sV4eYquELQma6bGTEjdPYleWaBce/ti2dhCLYPp15bA=",
"version": "2.14.0"
},
"cu130": {
"url": "https://download.pytorch.org/whl/cu130/torch-2.14.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-dFAQaV4EWNbyisy2l7U3G2/SRaqGllMBZflyzRJrvY0=",
"url": "https://download.pytorch.org/whl/cu130/torch-2.14.0%2Bcu130-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-ErCfnp+RNHWWOTpt25XpvAqgAyJ+tsfNfyDBROrGJEI=",
"version": "2.14.0"
},
"cu132": {
"url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-wIPAjPcsIH/vvYv8pJNfiDCvOMsca5VIvViZObc/VsQ=",
"url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-qM2Uhb0E0Uz98jW8pxKWQ5bPOPZzYGjZByValShyAr4=",
"version": "2.14.0"
},
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-Fg4bxGrt7TER0oAfiuENyaG5RoQ6fhJrTb9eGcVwbpU=",
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-8VL0HcXcRir+DeeA5FHrtH6otEUfj5GflTeqjiy+HX4=",
"version": "2.14.0"
},
"rocm72": {
"url": "https://download.pytorch.org/whl/rocm7.2/torch-2.14.0%2Brocm7.2-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-+iKhqF9PksT7SdybEXs+syuMaEloUI9YXfBC5wFuuX8=",
"xpu": {
"url": "https://download.pytorch.org/whl/xpu/torch-2.14.0%2Bxpu-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-wxFmDc0QLevXVxG09ffM5epLqLMKe+XHg57jtHGvPC0=",
"version": "2.14.0"
},
"rocm714": {
"url": "https://download.pytorch.org/whl/rocm7.14/torch-2.14.0%2Brocm7.14-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-RziHZNqLXI/U9hzAZrkJBciAsSsVUXQL219x9r42g2I=",
"url": "https://download.pytorch.org/whl/rocm7.14/torch-2.14.0%2Brocm7.14-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-YUNOyreck4O2klfMf9TR6TKjHFKHgc1Ew1QjmPJ9cf0=",
"version": "2.14.0"
},
"xpu": {
"url": "https://download.pytorch.org/whl/xpu/torch-2.14.0%2Bxpu-cp313-cp313-manylinux_2_28_x86_64.whl",
"hash": "sha256-WvZeKJZY9wmnqvJqwTw/qRQ3NHvzkfTZu9W6jELSJ64=",
"rocm72": {
"url": "https://download.pytorch.org/whl/rocm7.2/torch-2.14.0%2Brocm7.2-cp314-cp314-manylinux_2_28_x86_64.whl",
"hash": "sha256-96rbFtnkgPuvhKVctTqk9Vd4Eww93z5bxHTbR4vNV9c=",
"version": "2.14.0"
}
},
"aarch64-linux": {
"cu126": {
"url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-JCJN6evMb4XPpIYUyxgQzv+sVCj32t7ZM7CAi1vQ8wM=",
"url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-8X/SRkWdAVBdNNpglod1ek2xOdfV5WRa6GBHBvm1s2Q=",
"version": "2.14.0"
},
"cu130": {
"url": "https://download.pytorch.org/whl/cu130/torch-2.14.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-IOxLuJRKhH3uYObVU2tBVnDdVAM0Lzbb4Ipr5b9YLgg=",
"url": "https://download.pytorch.org/whl/cu130/torch-2.14.0%2Bcu130-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-mvTo85M8PVFdGUm++tVRJizbPxlcu9tGdDaVhcVew0I=",
"version": "2.14.0"
},
"cu132": {
"url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-eoHHxugfwqaiZPHYpwv9hDXACQWHPx1kYK73tXFzB/A=",
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-J7NGQP6nJohilhtzTHI4c+FbHCZXseuEvPrAU+WEFSc=",
"version": "2.14.0"
},
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl",
"hash": "sha256-CS1cEpOIUN+9kKZUs8jaw0wz4wD4jrGe5vTvk5ksY0c=",
"cu132": {
"url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp314-cp314-manylinux_2_28_aarch64.whl",
"hash": "sha256-lNS9GB45tsaNkWvDZhCPuBNK/LpBDDM868JNoHixEBc=",
"version": "2.14.0"
}
},
"aarch64-darwin": {
"cpu": {
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0-cp313-cp313-macosx_14_0_arm64.whl",
"hash": "sha256-yvY1nWTAB0vLn4ZBoWkjkRjIFloKD+95EQxyv9ZHS+w=",
"url": "https://download.pytorch.org/whl/cpu/torch-2.14.0-cp314-cp314-macosx_14_0_arm64.whl",
"hash": "sha256-0PeHRAAKeGPMPx6vr1EdS5lE8Hnwgnfeb+/ksB9NQbk=",
"version": "2.14.0"
}
}
Expand Down
2 changes: 1 addition & 1 deletion nix-builder/pkgs/python-modules/torch/binary/torch_versions.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from packaging.version import Version

PYTHON_VERSION = "cp313"
PYTHON_VERSION = "cp314"


def cuda_version_to_framework(cuda_version: str) -> str:
Expand Down
Loading
Loading