From 0b6dd2d757b86632e5fd14cf08f49fc9f6d5eb64 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Wed, 28 Oct 2026 19:16:51 +0000 Subject: [PATCH 1/2] python3Packages.torch_215: update to 2.15.0 (final) The final version also removes the CUDA 13.0 variant. --- docs/source/builder/build-variants.md | 2 - nix-builder/build-variants.json | 2 - .../python-modules/torch/binary/generic.nix | 2 +- .../torch/binary/torch-versions-hash.json | 58 ++++++++----------- .../torch/binary/torch-versions.json | 28 +++------ .../python-modules/triton-rocm/default.nix | 4 +- .../python-modules/triton-xpu/default.nix | 4 +- .../pkgs/python-modules/triton/default.nix | 8 +-- nix-builder/versions.nix | 10 +--- 9 files changed, 42 insertions(+), 76 deletions(-) diff --git a/docs/source/builder/build-variants.md b/docs/source/builder/build-variants.md index a6282a71..418ea371 100644 --- a/docs/source/builder/build-variants.md +++ b/docs/source/builder/build-variants.md @@ -49,7 +49,6 @@ available. This list will be updated as new PyTorch versions are released. - `torch214-cu126-aarch64-linux` - `torch214-cu130-aarch64-linux` - `torch214-cu132-aarch64-linux` -- `torch215-cu130-aarch64-linux` - `torch215-cu132-aarch64-linux` - `torch215-cu134-aarch64-linux` @@ -79,7 +78,6 @@ available. This list will be updated as new PyTorch versions are released. - `torch214-cu126-x86_64-linux` - `torch214-cu130-x86_64-linux` - `torch214-cu132-x86_64-linux` -- `torch215-cu130-x86_64-linux` - `torch215-cu132-x86_64-linux` - `torch215-cu134-x86_64-linux` diff --git a/nix-builder/build-variants.json b/nix-builder/build-variants.json index 32f12af1..91a24833 100644 --- a/nix-builder/build-variants.json +++ b/nix-builder/build-variants.json @@ -42,7 +42,6 @@ "torch214-cu126-aarch64-linux", "torch214-cu130-aarch64-linux", "torch214-cu132-aarch64-linux", - "torch215-cu130-aarch64-linux", "torch215-cu132-aarch64-linux", "torch215-cu134-aarch64-linux" ] @@ -72,7 +71,6 @@ "torch214-cu126-x86_64-linux", "torch214-cu130-x86_64-linux", "torch214-cu132-x86_64-linux", - "torch215-cu130-x86_64-linux", "torch215-cu132-x86_64-linux", "torch215-cu134-x86_64-linux" ], diff --git a/nix-builder/pkgs/python-modules/torch/binary/generic.nix b/nix-builder/pkgs/python-modules/torch/binary/generic.nix index 5130734c..1b62380b 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/generic.nix +++ b/nix-builder/pkgs/python-modules/torch/binary/generic.nix @@ -82,7 +82,7 @@ let triton; # For the required version see: - # https://github.com/pytorch/pytorch/blob/v2.15.0-rc3/cmake/External/aotriton.cmake#L13 + # https://github.com/pytorch/pytorch/blob/v2.15.0/cmake/External/aotriton.cmake#L13 aotritonVersions = with rocmPackages; { "2.10" = aotriton_0_11_1; "2.11" = aotriton_0_11_2; diff --git a/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json b/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json index 7965b914..f15572e9 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions-hash.json @@ -341,68 +341,58 @@ }, "2.15": { "x86_64-linux": { - "cpu": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcpu-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-rqmKRJ1V7LCx2/VLzdMiqflTSIl+oIn5p7oSCm/b7Cs=", - "version": "2.15.0" - }, - "cu132": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu132-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-+X0O8rnlY2Gw8YLl7iZtil1Im1Fw/Gi4buiFou6yo3w=", + "cu134": { + "url": "https://download.pytorch.org/whl/cu134/torch-2.15.0%2Bcu134-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-n5CyYClUflFCqUXIOoBENTVZYjPzLOrOQHD+0kGMXpw=", "version": "2.15.0" }, - "cu130": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu130-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-6HYM9X8TwtOSOiNwYcLQcYZlPL3fSKyVKj9t5uaXiOY=", + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.15.0%2Bcpu-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-8InJSRLJVSaaIOX+5rajcSCbYQhJyezibpCKdI2HpyM=", "version": "2.15.0" }, - "cu134": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu134-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-YuMi/K9hlrRD4BM8EYbc4OjUhRWUUsSYu6TAexeRMbg=", + "xpu": { + "url": "https://download.pytorch.org/whl/xpu/torch-2.15.0%2Bxpu-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-IEZrs6bJdC2Y0wbHYMYEg0IR8wjgWh+hFDMYwbzVlts=", "version": "2.15.0" }, - "xpu": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bxpu-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-R7WJWnTEwDSC95WTYxQKQ0WVweGyM0ogRjFgY6k4/FQ=", + "cu132": { + "url": "https://download.pytorch.org/whl/cu132/torch-2.15.0%2Bcu132-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-WT401aVy5DprpRaehRwe7bSqq0P1M4olfwv7MCQqlfw=", "version": "2.15.0" }, "rocm100": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Brocm10.0-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-s3k9tXvpS2tBjF3hXLoWohFKDO67x1HmsjNHASRvdsg=", + "url": "https://download.pytorch.org/whl/rocm10.0/torch-2.15.0%2Brocm10.0-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-ECaQesWB77u4p7Uk1fVMJUr9HaExXE+VxYdFVcXMQjo=", "version": "2.15.0" }, "rocm714": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Brocm7.14-cp314-cp314-manylinux_2_28_x86_64.whl", - "hash": "sha256-E0fCTMtBE/GMVKITnB/H5niTaXYEfRNiHUp03jnxL4A=", + "url": "https://download.pytorch.org/whl/rocm7.14/torch-2.15.0%2Brocm7.14-cp314-cp314-manylinux_2_28_x86_64.whl", + "hash": "sha256-ZMOR47A3GQyoewhwts8Wz6Wo8L/8meDnADEMxlGR4yY=", "version": "2.15.0" } }, "aarch64-linux": { "cpu": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcpu-cp314-cp314-manylinux_2_28_aarch64.whl", - "hash": "sha256-tsXLNsEOoWa1t0hkDaUHp7zuy2UhNJC8X2zzzME3qyk=", - "version": "2.15.0" - }, - "cu130": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu130-cp314-cp314-manylinux_2_28_aarch64.whl", - "hash": "sha256-9qUemUFbrwLZ7EPxP9hvTsM9Q9XvxiON+F2ipSl4Oqg=", + "url": "https://download.pytorch.org/whl/cpu/torch-2.15.0%2Bcpu-cp314-cp314-manylinux_2_28_aarch64.whl", + "hash": "sha256-Ab3H0pHj9rXAPPctihcYpQEtn61fOA5ntWOwp2WSBVo=", "version": "2.15.0" }, "cu134": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu134-cp314-cp314-manylinux_2_28_aarch64.whl", - "hash": "sha256-IIZebBKtc9a8e3xeSXjxWt1UWJJny3elMOiEFeCMuKs=", + "url": "https://download.pytorch.org/whl/cu134/torch-2.15.0%2Bcu134-cp314-cp314-manylinux_2_28_aarch64.whl", + "hash": "sha256-BzlA7qYitNPujWRyW0hm4YflWvQYLTUcXD4K2Rxt6Ic=", "version": "2.15.0" }, "cu132": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0%2Bcu132-cp314-cp314-manylinux_2_28_aarch64.whl", - "hash": "sha256-o3cpAeDHGBBMU9+oNQTbp2TrnZoEfXtS81kDTKmlwxc=", + "url": "https://download.pytorch.org/whl/cu132/torch-2.15.0%2Bcu132-cp314-cp314-manylinux_2_28_aarch64.whl", + "hash": "sha256-CvzFKDnTB+gZblWB2rQHQcrhDw71nS3A4+wM1X+2ais=", "version": "2.15.0" } }, "aarch64-darwin": { "cpu": { - "url": "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc5/torch-2.15.0-cp314-cp314-macosx_14_0_arm64.whl", - "hash": "sha256-ruqM7wdOT2MdDTKF8flooCuE8uu4iwtkwZLOosPSF9I=", + "url": "https://download.pytorch.org/whl/cpu/torch-2.15.0-cp314-cp314-macosx_14_0_arm64.whl", + "hash": "sha256-DSstlnT6C5s/UOgyMaplRaCxnkN6UvCvk5oiT6qkVRU=", "version": "2.15.0" } } diff --git a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json index ac8df581..2fb3e222 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json @@ -205,52 +205,40 @@ "systems": ["x86_64-linux"] }, - { - "torchVersion": "2.15.0", - "cudaVersion": "13.0", - "systems": ["x86_64-linux", "aarch64-linux"], - "torchTesting": "rc5" - }, + { "torchVersion": "2.15.0", "cudaVersion": "13.2", - "systems": ["x86_64-linux", "aarch64-linux"], - "torchTesting": "rc5" + "systems": ["x86_64-linux", "aarch64-linux"] }, { "torchVersion": "2.15.0", "cudaVersion": "13.4", - "systems": ["x86_64-linux", "aarch64-linux"], - "torchTesting": "rc5" + "systems": ["x86_64-linux", "aarch64-linux"] }, { "torchVersion": "2.15.0", "cpu": true, - "systems": ["aarch64-linux", "x86_64-linux"], - "torchTesting": "rc5" + "systems": ["aarch64-linux", "x86_64-linux"] }, { "torchVersion": "2.15.0", "metal": true, - "systems": ["aarch64-darwin"], - "torchTesting": "rc5" + "systems": ["aarch64-darwin"] }, { "torchVersion": "2.15.0", "rocmVersion": "7.14", - "systems": ["x86_64-linux"], - "torchTesting": "rc5" + "systems": ["x86_64-linux"] }, { "torchVersion": "2.15.0", "rocmVersion": "10.0", - "systems": ["x86_64-linux"], - "torchTesting": "rc5" + "systems": ["x86_64-linux"] }, { "torchVersion": "2.15.0", "xpuVersion": "2026.1.2", - "systems": ["x86_64-linux"], - "torchTesting": "rc5" + "systems": ["x86_64-linux"] } ] diff --git a/nix-builder/pkgs/python-modules/triton-rocm/default.nix b/nix-builder/pkgs/python-modules/triton-rocm/default.nix index e7565e5a..99f2e241 100644 --- a/nix-builder/pkgs/python-modules/triton-rocm/default.nix +++ b/nix-builder/pkgs/python-modules/triton-rocm/default.nix @@ -32,8 +32,8 @@ let }; "3.9.0" = { x86_64-linux = { - url = "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc3/triton_rocm-3.9.0-cp312-abi3-linux_x86_64.whl"; - hash = "sha256-PN2kG6Y4PcO3B/jbaBbxfGz5fXYOiulH0fAS9Z98T5A="; + url = "https://download.pytorch.org/whl/triton_rocm-3.9.0-cp312-abi3-linux_x86_64.whl"; + hash = "sha256-lOTCKOqQpNbHntxQY4Mpk9zrNdgiAtHIkejsjiv1Q8M="; }; }; }; diff --git a/nix-builder/pkgs/python-modules/triton-xpu/default.nix b/nix-builder/pkgs/python-modules/triton-xpu/default.nix index f2629274..107e2482 100644 --- a/nix-builder/pkgs/python-modules/triton-xpu/default.nix +++ b/nix-builder/pkgs/python-modules/triton-xpu/default.nix @@ -38,7 +38,7 @@ in triton-xpu_3_9_0 = generic { version = "3.9.0"; - url = "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc3/triton_xpu-3.9.0-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; - hash = "sha256-nz+10HitF8bj3GHu9zabQ1isP4fWtWnSps5et3OkPMg="; + url = "https://download.pytorch.org/whl/triton_xpu-3.9.0-cp314-cp314-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; + hash = "sha256-CoL7VLDomd45COY1Ib1L65+CN6eOVBc54KqVXX5LmqY="; }; } diff --git a/nix-builder/pkgs/python-modules/triton/default.nix b/nix-builder/pkgs/python-modules/triton/default.nix index 46df89aa..8d4abdae 100644 --- a/nix-builder/pkgs/python-modules/triton/default.nix +++ b/nix-builder/pkgs/python-modules/triton/default.nix @@ -48,12 +48,12 @@ let }; "3.9.0" = { x86_64-linux = { - url = "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc3/triton-3.9.0-cp312-abi3-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; - hash = "sha256-MZtpln1qLD8tdGH+hsZMJNSoNfok33gQLoAm8PBRtWk="; + url = "https://download.pytorch.org/whl/triton-3.9.0-cp312-abi3-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; + hash = "sha256-XlKrTi8dZPCsSUyapyNpe3k/gakcTK5MNmCMoLxCZ78="; }; aarch64-linux = { - url = "https://huggingface.co/buckets/kernels-community/pytorch-rc/resolve/2.15.0/rc3/triton-3.9.0-cp312-abi3-linux_aarch64.whl"; - hash = "sha256-Vax66MKJ6ZHXKl+ylIGHpPsy/Bu4JurO8TIROe7J53I="; + url = "https://download.pytorch.org/whl/triton-3.9.0-cp312-abi3-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl"; + hash = "sha256-VUL0MoRy/YJ9ux7EOHM2TCiUaCzD9dHHSmQ3YpAOJD4="; }; }; }; diff --git a/nix-builder/versions.nix b/nix-builder/versions.nix index 26e95563..5abf043a 100644 --- a/nix-builder/versions.nix +++ b/nix-builder/versions.nix @@ -334,15 +334,7 @@ ]; bundleBuild = true; } - { - torchVersion = "2.15"; - cudaVersion = "13.0"; - systems = [ - "x86_64-linux" - "aarch64-linux" - ]; - bundleBuild = true; - } + { torchVersion = "2.15"; cudaVersion = "13.2"; From 21bdf69d97a289b9c581a0c60f63deed131647d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Wed, 28 Oct 2026 19:36:57 +0000 Subject: [PATCH 2/2] python3Packages.torch: slim ROCm output paths a bit --- nix-builder/pkgs/python-modules/torch/binary/generic.nix | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nix-builder/pkgs/python-modules/torch/binary/generic.nix b/nix-builder/pkgs/python-modules/torch/binary/generic.nix index 1b62380b..6fe317fa 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/generic.nix +++ b/nix-builder/pkgs/python-modules/torch/binary/generic.nix @@ -406,7 +406,8 @@ buildPythonPackage.override { stdenv = effectiveStdenv; } { # This keeps the outputs lean and requires downstream to specify # dependencies. rm -rf $out/${python.sitePackages}/torch/lib/{libamd*,libaotriton*,libdrm*,libelf*,libgomp*,libhip*,libhsa*,libMIOpen*,libnuma*,librccl*,libroc*,libtinfo*}.so* - rm -rf $out/${python.sitePackages}/torch/lib/{rocblas,hipblaslt,hipsparselt} + rm -rf $out/${python.sitePackages}/torch/lib/{rocblas,hipblaslt,hipsparselt,aotriton.images} + rm -rf $out/${python.sitePackages}/torch/share/miopen ''; autoPatchelfIgnoreMissingDeps = lib.optionals stdenv.hostPlatform.isLinux [