From c3dbef66ab6ffa54573431e4da517ed3e45eb8b5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 09:16:50 +0000 Subject: [PATCH 1/6] python3Packages.torch_2_14: update hashes to 2.14.0 --- .../torch/binary/torch-versions-hash.json | 72 ++++++++----------- .../torch/binary/torch-versions.json | 14 ---- 2 files changed, 31 insertions(+), 55 deletions(-) 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 cca1676f..b7451356 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 @@ -146,78 +146,68 @@ }, "2.14": { "x86_64-linux": { - "cu130": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-Rek9qAsDlogrAyWNaX+obKDwZt0x0VdwRNHpTdhymxs=", + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-Fg4bxGrt7TER0oAfiuENyaG5RoQ6fhJrTb9eGcVwbpU=", "version": "2.14.0" }, - "cu132": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-wYSnxRk5txrxNkbv06j3Nki9N2xpIumcTGOE02dgXVg=", + "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=", "version": "2.14.0" }, - "cpu": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-J4TuoRFcRBGEHDvMEbfgUm7R00fcz4GwFZGIXlcmDFM=", + "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=", "version": "2.14.0" }, - "cu134": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu134-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-tF/80uEFZy+BeCg6yUICZlR4iLxGG2G7tQtSIc3dgps=", + "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=", "version": "2.14.0" }, "cu126": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-I8A6bMBJeOyv1onF/k/d0m53elSTRfw0trjxgxJnyPo=", + "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=", "version": "2.14.0" }, "xpu": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bxpu-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-biy7nCq44+TNguilNL+vVh8vUMqxJm62NI9TKwMnYiU=", - "version": "2.14.0" - }, - "rocm714": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Brocm7.14-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-AzDhFg7FWQPXQWTlAALNInmfH6zSGzQ85GOdBeNnr1Q=", + "url": "https://download.pytorch.org/whl/xpu/torch-2.14.0%2Bxpu-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-WvZeKJZY9wmnqvJqwTw/qRQ3NHvzkfTZu9W6jELSJ64=", "version": "2.14.0" }, "rocm72": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Brocm7.2-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-vKWLdiy6vxl7xWeAoXMI41SNINmoRo5J+RSzV/pFtWg=", + "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=", "version": "2.14.0" } }, "aarch64-linux": { - "cu126": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-OHT6clnwB+1o9Kts/uPllSUkXNQvYhnfHMx0LaLWOoQ=", - "version": "2.14.0" - }, - "cu130": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-98TXTngVFaFh4gBBvjuHx/e1Nzl4c3xgMnFseLW83VU=", + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-CS1cEpOIUN+9kKZUs8jaw0wz4wD4jrGe5vTvk5ksY0c=", "version": "2.14.0" }, - "cu134": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu134-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-9FLL2bhpyXxMdrOnmHA0pWoUmNTcMRzA0O/gppiOJnE=", + "cu126": { + "url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-JCJN6evMb4XPpIYUyxgQzv+sVCj32t7ZM7CAi1vQ8wM=", "version": "2.14.0" }, - "cpu": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-nRYFP0amJ/JSrxinD74EGQDDcZhZUvWeS66TrxddtAU=", + "cu130": { + "url": "https://download.pytorch.org/whl/cu130/torch-2.14.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-IOxLuJRKhH3uYObVU2tBVnDdVAM0Lzbb4Ipr5b9YLgg=", "version": "2.14.0" }, "cu132": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-0nJppAuxuA9qxjakegmyeL1sXQvMDxkKWpH21xz8XPM=", + "url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl", + "hash": "sha256-eoHHxugfwqaiZPHYpwv9hDXACQWHPx1kYK73tXFzB/A=", "version": "2.14.0" } }, "aarch64-darwin": { "cpu": { - "url": "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/torch-2.14.0-cp313-cp313-macosx_14_0_arm64.whl", - "hash": "sha256-KRX4fbuKEzAe6HIFDQ7N+S3R6f4kWnZsyQ8Zi23QAas=", + "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0-cp313-cp313-macosx_14_0_arm64.whl", + "hash": "sha256-yvY1nWTAB0vLn4ZBoWkjkRjIFloKD+95EQxyv9ZHS+w=", "version": "2.14.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 60a1560d..6b823d1f 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json @@ -89,55 +89,41 @@ { "torchVersion": "2.14.0", - "torchTesting": "rc6", "cudaVersion": "12.6", "systems": ["x86_64-linux", "aarch64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "cudaVersion": "13.0", "systems": ["x86_64-linux", "aarch64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "cudaVersion": "13.2", "systems": ["x86_64-linux", "aarch64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", - "cudaVersion": "13.4", - "systems": ["x86_64-linux", "aarch64-linux"] - }, - { - "torchVersion": "2.14.0", - "torchTesting": "rc6", "cpu": true, "systems": ["aarch64-linux", "x86_64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "metal": true, "systems": ["aarch64-darwin"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "rocmVersion": "7.2", "systems": ["x86_64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "rocmVersion": "7.14", "systems": ["x86_64-linux"] }, { "torchVersion": "2.14.0", - "torchTesting": "rc6", "xpuVersion": "2026.1.0", "systems": ["x86_64-linux"] } From cab167502e841a4162d9d7f40904f88fbdd298d3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 09:38:55 +0000 Subject: [PATCH 2/6] rocmPackages: fixup ROCm dependencies --- nix-builder/lib/extension/default.nix | 25 +++++++++++++--------- nix-builder/lib/extension/tvm-ffi/arch.nix | 21 ++++++++++++------ 2 files changed, 29 insertions(+), 17 deletions(-) diff --git a/nix-builder/lib/extension/default.nix b/nix-builder/lib/extension/default.nix index d5be56c3..2574799a 100644 --- a/nix-builder/lib/extension/default.nix +++ b/nix-builder/lib/extension/default.nix @@ -55,16 +55,21 @@ in onednn-xpu ] ++ lib.optionals rocmSupport ( - [ - clr - ] - ++ (with rocmPackages; [ - hipcub-devel - hipsparselt - rocprim-devel - rocthrust-devel - rocwmma-devel - ]) + with rocmPackages; + if theRock then + [ + amdrocm-ccl-devel + amdrocm-hipblas-common-devel + ] + else + [ + hipcub-devel + hipsparselt + rocprim-devel + rocthrust-devel + rocshmem-devel + rocwmma-devel + ] ); mkTvmFfiExtension = callPackage ./tvm-ffi/arch.nix { diff --git a/nix-builder/lib/extension/tvm-ffi/arch.nix b/nix-builder/lib/extension/tvm-ffi/arch.nix index c4b39c6a..66ed2a97 100644 --- a/nix-builder/lib/extension/tvm-ffi/arch.nix +++ b/nix-builder/lib/extension/tvm-ffi/arch.nix @@ -219,13 +219,20 @@ stdenv.mkDerivation (prevAttrs: { ) ++ lib.optionals rocmSupport ( with rocmPackages; - [ - hipcub-devel - hipsparselt - rocprim-devel - rocthrust-devel - rocwmma-devel - ] + if theRock then + [ + amdrocm-ccl-devel + amdrocm-hipblas-common-devel + ] + else + [ + hipcub-devel + hipsparselt + rocprim-devel + rocthrust-devel + rocshmem-devel + rocwmma-devel + ] ) ++ lib.optionals xpuSupport ([ oneapi-torch-dev From 802918aecb6e4444841b90594adf48aa69681406 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 09:41:14 +0000 Subject: [PATCH 3/6] nix-builder: remove Torch 2.12 --- docs/source/builder/build-variants.md | 13 --- nix-builder/build-variants.json | 13 --- .../torch/binary/torch-versions-hash.json | 110 ++++-------------- .../torch/binary/torch-versions.json | 41 ------- nix-builder/versions.nix | 67 ----------- 5 files changed, 21 insertions(+), 223 deletions(-) diff --git a/docs/source/builder/build-variants.md b/docs/source/builder/build-variants.md index e99321e4..c41f038c 100644 --- a/docs/source/builder/build-variants.md +++ b/docs/source/builder/build-variants.md @@ -7,27 +7,21 @@ available. This list will be updated as new PyTorch versions are released. ## CPU aarch64-darwin -- `torch212-cpu-aarch64-darwin` - `torch213-cpu-aarch64-darwin` - `torch214-cpu-aarch64-darwin` ## Metal aarch64-darwin -- `torch212-metal-aarch64-darwin` - `torch213-metal-aarch64-darwin` - `torch214-metal-aarch64-darwin` ## CPU aarch64-linux -- `torch212-cxx11-cpu-aarch64-linux` - `torch213-cxx11-cpu-aarch64-linux` - `torch214-cxx11-cpu-aarch64-linux` ## CUDA aarch64-linux -- `torch212-cxx11-cu126-aarch64-linux` -- `torch212-cxx11-cu130-aarch64-linux` -- `torch212-cxx11-cu132-aarch64-linux` - `torch213-cxx11-cu126-aarch64-linux` - `torch213-cxx11-cu130-aarch64-linux` - `torch213-cxx11-cu132-aarch64-linux` @@ -37,15 +31,11 @@ available. This list will be updated as new PyTorch versions are released. ## CPU x86_64-linux -- `torch212-cxx11-cpu-x86_64-linux` - `torch213-cxx11-cpu-x86_64-linux` - `torch214-cxx11-cpu-x86_64-linux` ## CUDA x86_64-linux -- `torch212-cxx11-cu126-x86_64-linux` -- `torch212-cxx11-cu130-x86_64-linux` -- `torch212-cxx11-cu132-x86_64-linux` - `torch213-cxx11-cu126-x86_64-linux` - `torch213-cxx11-cu130-x86_64-linux` - `torch213-cxx11-cu132-x86_64-linux` @@ -55,8 +45,6 @@ available. This list will be updated as new PyTorch versions are released. ## ROCm x86_64-linux -- `torch212-cxx11-rocm71-x86_64-linux` -- `torch212-cxx11-rocm72-x86_64-linux` - `torch213-cxx11-rocm71-x86_64-linux` - `torch213-cxx11-rocm72-x86_64-linux` - `torch214-cxx11-rocm714-x86_64-linux` @@ -64,7 +52,6 @@ available. This list will be updated as new PyTorch versions are released. ## XPU x86_64-linux -- `torch212-cxx11-xpu20253-x86_64-linux` - `torch213-cxx11-xpu20260-x86_64-linux` - `torch214-cxx11-xpu20261-x86_64-linux` diff --git a/nix-builder/build-variants.json b/nix-builder/build-variants.json index 88035549..9e07e6a3 100644 --- a/nix-builder/build-variants.json +++ b/nix-builder/build-variants.json @@ -1,26 +1,20 @@ { "aarch64-darwin": { "cpu": [ - "torch212-cpu-aarch64-darwin", "torch213-cpu-aarch64-darwin", "torch214-cpu-aarch64-darwin" ], "metal": [ - "torch212-metal-aarch64-darwin", "torch213-metal-aarch64-darwin", "torch214-metal-aarch64-darwin" ] }, "aarch64-linux": { "cpu": [ - "torch212-cxx11-cpu-aarch64-linux", "torch213-cxx11-cpu-aarch64-linux", "torch214-cxx11-cpu-aarch64-linux" ], "cuda": [ - "torch212-cxx11-cu126-aarch64-linux", - "torch212-cxx11-cu130-aarch64-linux", - "torch212-cxx11-cu132-aarch64-linux", "torch213-cxx11-cu126-aarch64-linux", "torch213-cxx11-cu130-aarch64-linux", "torch213-cxx11-cu132-aarch64-linux", @@ -31,14 +25,10 @@ }, "x86_64-linux": { "cpu": [ - "torch212-cxx11-cpu-x86_64-linux", "torch213-cxx11-cpu-x86_64-linux", "torch214-cxx11-cpu-x86_64-linux" ], "cuda": [ - "torch212-cxx11-cu126-x86_64-linux", - "torch212-cxx11-cu130-x86_64-linux", - "torch212-cxx11-cu132-x86_64-linux", "torch213-cxx11-cu126-x86_64-linux", "torch213-cxx11-cu130-x86_64-linux", "torch213-cxx11-cu132-x86_64-linux", @@ -47,15 +37,12 @@ "torch214-cxx11-cu132-x86_64-linux" ], "rocm": [ - "torch212-cxx11-rocm71-x86_64-linux", - "torch212-cxx11-rocm72-x86_64-linux", "torch213-cxx11-rocm71-x86_64-linux", "torch213-cxx11-rocm72-x86_64-linux", "torch214-cxx11-rocm714-x86_64-linux", "torch214-cxx11-rocm72-x86_64-linux" ], "xpu": [ - "torch212-cxx11-xpu20253-x86_64-linux", "torch213-cxx11-xpu20260-x86_64-linux", "torch214-cxx11-xpu20261-x86_64-linux" ] 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 b7451356..25f6a327 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 @@ -8,74 +8,6 @@ } } }, - "2.12": { - "x86_64-linux": { - "cu126": { - "url": "https://download.pytorch.org/whl/cu126/torch-2.12.0%2Bcu126-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-hyejkBw0v5glBpqvgvCYTdKP9iLhpyfSU0VVzXePxdw=", - "version": "2.12.0" - }, - "cu130": { - "url": "https://download.pytorch.org/whl/cu130/torch-2.12.0%2Bcu130-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-/l/vt4SjcNG6SVneboe807NUQQQKmb/+MvXNA7vINMA=", - "version": "2.12.0" - }, - "cu132": { - "url": "https://download.pytorch.org/whl/cu132/torch-2.12.0%2Bcu132-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-GBcVE1xToXCmFp/MCCOzBUgn/YLkCGr1eI1/QzKjAV0=", - "version": "2.12.0" - }, - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.12.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-raeAGL36MNHHZlls0y2RDb9bA0JM2FkjG20qAFM96SI=", - "version": "2.12.0" - }, - "rocm71": { - "url": "https://download.pytorch.org/whl/rocm7.1/torch-2.12.0%2Brocm7.1-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-N1sCgonX1LXrhWKDjz4oPAU9/p4DgtdL8Q3ja619sMQ=", - "version": "2.12.0" - }, - "rocm72": { - "url": "https://download.pytorch.org/whl/rocm7.2/torch-2.12.0%2Brocm7.2-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-WMB+TMYMHEfZa7/7kiL0fBCQ/F3qjZQYFG0lCKh7eF4=", - "version": "2.12.0" - }, - "xpu": { - "url": "https://download.pytorch.org/whl/xpu/torch-2.12.0%2Bxpu-cp313-cp313-linux_x86_64.whl", - "hash": "sha256-VvdOfGwJbhp6whXree5ZC3ZL4/u6j0/rwUW8pHGUoIM=", - "version": "2.12.0" - } - }, - "aarch64-linux": { - "cu126": { - "url": "https://download.pytorch.org/whl/cu126/torch-2.12.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-PPZnbUHY1XNQ1F1zizXL7XZx42JCqDaZPNabDBiBkXc=", - "version": "2.12.0" - }, - "cu130": { - "url": "https://download.pytorch.org/whl/cu130/torch-2.12.0%2Bcu130-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-v18GfTpNcTt1zNagFB+BM8dJWgFrkXzm3OwUkuPamLA=", - "version": "2.12.0" - }, - "cu132": { - "url": "https://download.pytorch.org/whl/cu132/torch-2.12.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-uWo2HRjhEXtq67nN8+yOtGkO7RX7lD01wC3K5pL2Fuk=", - "version": "2.12.0" - }, - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.12.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-aLfd1NtGA6A+EG50xwmMjYyJQ9M8HlraAJykzYhXWcM=", - "version": "2.12.0" - } - }, - "aarch64-darwin": { - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.12.0-cp313-cp313-macosx_14_0_arm64.whl", - "hash": "sha256-kN1Yel9hv+EwcUi1geIIT8W8SgbiuQog6aNrgQh/8Ws=", - "version": "2.12.0" - } - } - }, "2.13": { "x86_64-linux": { "cu126": { @@ -146,9 +78,9 @@ }, "2.14": { "x86_64-linux": { - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", - "hash": "sha256-Fg4bxGrt7TER0oAfiuENyaG5RoQ6fhJrTb9eGcVwbpU=", + "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=", "version": "2.14.0" }, "cu130": { @@ -156,38 +88,33 @@ "hash": "sha256-dFAQaV4EWNbyisy2l7U3G2/SRaqGllMBZflyzRJrvY0=", "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=", - "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=", "version": "2.14.0" }, - "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=", - "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=", + "cpu": { + "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl", + "hash": "sha256-Fg4bxGrt7TER0oAfiuENyaG5RoQ6fhJrTb9eGcVwbpU=", "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=", "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=", + "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=", + "version": "2.14.0" } }, "aarch64-linux": { - "cpu": { - "url": "https://download.pytorch.org/whl/cpu/torch-2.14.0%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl", - "hash": "sha256-CS1cEpOIUN+9kKZUs8jaw0wz4wD4jrGe5vTvk5ksY0c=", - "version": "2.14.0" - }, "cu126": { "url": "https://download.pytorch.org/whl/cu126/torch-2.14.0%2Bcu126-cp313-cp313-manylinux_2_28_aarch64.whl", "hash": "sha256-JCJN6evMb4XPpIYUyxgQzv+sVCj32t7ZM7CAi1vQ8wM=", @@ -202,6 +129,11 @@ "url": "https://download.pytorch.org/whl/cu132/torch-2.14.0%2Bcu132-cp313-cp313-manylinux_2_28_aarch64.whl", "hash": "sha256-eoHHxugfwqaiZPHYpwv9hDXACQWHPx1kYK73tXFzB/A=", "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=", + "version": "2.14.0" } }, "aarch64-darwin": { 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 6b823d1f..9961627f 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json @@ -5,47 +5,6 @@ "systems": ["x86_64-linux"] }, - { - "torchVersion": "2.12.0", - "cudaVersion": "12.6", - "systems": ["x86_64-linux", "aarch64-linux"] - }, - { - "torchVersion": "2.12.0", - "cudaVersion": "13.0", - "systems": ["x86_64-linux", "aarch64-linux"] - }, - { - "torchVersion": "2.12.0", - "cudaVersion": "13.2", - "systems": ["x86_64-linux", "aarch64-linux"] - }, - { - "torchVersion": "2.12.0", - "cpu": true, - "systems": ["aarch64-linux", "x86_64-linux"] - }, - { - "torchVersion": "2.12.0", - "metal": true, - "systems": ["aarch64-darwin"] - }, - { - "torchVersion": "2.12.0", - "rocmVersion": "7.1", - "systems": ["x86_64-linux"] - }, - { - "torchVersion": "2.12.0", - "rocmVersion": "7.2", - "systems": ["x86_64-linux"] - }, - { - "torchVersion": "2.12.0", - "xpuVersion": "2025.3.2", - "systems": ["x86_64-linux"] - }, - { "torchVersion": "2.13.0", "cudaVersion": "12.6", diff --git a/nix-builder/versions.nix b/nix-builder/versions.nix index f4dcff92..f62884d5 100644 --- a/nix-builder/versions.nix +++ b/nix-builder/versions.nix @@ -17,73 +17,6 @@ bundleBuild = false; } - { - torchVersion = "2.12"; - cpu = true; - systems = [ - "aarch64-darwin" - "x86_64-linux" - "aarch64-linux" - ]; - bundleBuild = true; - tvmFfiVersion = "0.1"; - } - { - torchVersion = "2.12"; - cudaVersion = "12.6"; - systems = [ - "x86_64-linux" - "aarch64-linux" - ]; - bundleBuild = true; - tvmFfiVersion = "0.1"; - } - { - torchVersion = "2.12"; - cudaVersion = "13.0"; - systems = [ - "x86_64-linux" - "aarch64-linux" - ]; - bundleBuild = true; - tvmFfiVersion = "0.1"; - } - { - torchVersion = "2.12"; - cudaVersion = "13.2"; - systems = [ - "x86_64-linux" - "aarch64-linux" - ]; - bundleBuild = true; - tvmFfiVersion = "0.1"; - } - { - torchVersion = "2.12"; - metal = true; - systems = [ "aarch64-darwin" ]; - bundleBuild = true; - } - { - torchVersion = "2.12"; - rocmVersion = "7.1"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - } - { - torchVersion = "2.12"; - rocmVersion = "7.2"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - } - { - torchVersion = "2.12"; - xpuVersion = "2025.3.2"; - systems = [ "x86_64-linux" ]; - bundleBuild = true; - tvmFfiVersion = "0.1"; - } - { torchVersion = "2.13"; cpu = true; From c739888eadc122eb1f029ff1b2469d81ebaa4465 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 09:58:40 +0000 Subject: [PATCH 4/6] nix-builder: use Torch 2.13 buildsets for TVM-FFI --- nix-builder/versions.nix | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nix-builder/versions.nix b/nix-builder/versions.nix index f62884d5..8e416ed0 100644 --- a/nix-builder/versions.nix +++ b/nix-builder/versions.nix @@ -26,6 +26,7 @@ "aarch64-linux" ]; bundleBuild = true; + tvmFfiVersion = "0.1"; } { torchVersion = "2.13"; @@ -35,6 +36,7 @@ "aarch64-linux" ]; bundleBuild = true; + tvmFfiVersion = "0.1"; } { torchVersion = "2.13"; @@ -44,6 +46,7 @@ "aarch64-linux" ]; bundleBuild = true; + tvmFfiVersion = "0.1"; } { torchVersion = "2.13"; @@ -61,7 +64,6 @@ systems = [ "aarch64-darwin" ]; bundleBuild = true; } - # Broken: https://github.com/ROCm/ROCm/issues/6322 { torchVersion = "2.13"; rocmVersion = "7.1"; @@ -79,6 +81,7 @@ xpuVersion = "2026.0.0"; systems = [ "x86_64-linux" ]; bundleBuild = true; + tvmFfiVersion = "0.1"; } { From 99942bdae97e8365a5cfafcc6ecd728906564cc0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 10:00:34 +0000 Subject: [PATCH 5/6] CI: update all uses of Torch 2.12 to 2.13 --- .github/workflows/test_kernels.yaml | 2 +- examples/kernels/flake.nix | 6 +++--- examples/kernels/relu-specific-torch/flake.nix | 2 +- nix-builder/lib/checks.nix | 2 +- nix-builder/tests/Dockerfile.test-kernel | 2 +- 5 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.github/workflows/test_kernels.yaml b/.github/workflows/test_kernels.yaml index ca5e8f2a..e63b014e 100644 --- a/.github/workflows/test_kernels.yaml +++ b/.github/workflows/test_kernels.yaml @@ -29,7 +29,7 @@ jobs: max-parallel: 4 matrix: python-version: ["3.10", "3.14"] - torch-version: ["2.12.0", "2.13.0"] + torch-version: ["2.13.0", "2.14.0"] env: UV_PYTHON_PREFERENCE: only-managed diff --git a/examples/kernels/flake.nix b/examples/kernels/flake.nix index 8c916408..325c44fe 100644 --- a/examples/kernels/flake.nix +++ b/examples/kernels/flake.nix @@ -17,8 +17,8 @@ cudaVersion = "cu126"; cuda13Version = "cu132"; rocmVersion = "rocm71"; - xpuVersion = "xpu20253"; - torchVersion = "212"; + xpuVersion = "xpu20260"; + torchVersion = "213"; tvmFfiVersion = "01"; # All example kernels to build in CI. @@ -142,7 +142,7 @@ drv = sys: out: out.packages.${sys}.default; torchVersions = _defaultVersions: [ { - torchVersion = "2.12"; + torchVersion = "2.13"; cudaVersion = "12.6"; systems = [ "x86_64-linux" diff --git a/examples/kernels/relu-specific-torch/flake.nix b/examples/kernels/relu-specific-torch/flake.nix index 4ec29c36..426b06d4 100644 --- a/examples/kernels/relu-specific-torch/flake.nix +++ b/examples/kernels/relu-specific-torch/flake.nix @@ -15,7 +15,7 @@ path = ./.; torchVersions = defaultVersions: [ { - torchVersion = "2.12"; + torchVersion = "2.13"; cudaVersion = "12.6"; systems = [ "x86_64-linux" diff --git a/nix-builder/lib/checks.nix b/nix-builder/lib/checks.nix index 88845bfb..d446dc2d 100644 --- a/nix-builder/lib/checks.nix +++ b/nix-builder/lib/checks.nix @@ -64,7 +64,7 @@ let touch $out ''; in -assert lib.assertMsg (builtins.all (buildSet: buildSet.torch.version == "2.12.0") kernelBuildSets) +assert lib.assertMsg (builtins.all (buildSet: buildSet.torch.version == "2.13.0") kernelBuildSets) '' Torch minver/maxver filtering does not work. ''; diff --git a/nix-builder/tests/Dockerfile.test-kernel b/nix-builder/tests/Dockerfile.test-kernel index 371907cd..3462e380 100644 --- a/nix-builder/tests/Dockerfile.test-kernel +++ b/nix-builder/tests/Dockerfile.test-kernel @@ -3,7 +3,7 @@ ARG PYTHON_VERSION=3.10 # Test with the oldest CUDA version supported by the latest PyTorch. ARG CUDA_VERSION=12.6.0 ARG UBI_VERSION=8 -ARG TORCH_VERSION=2.12.0 +ARG TORCH_VERSION=2.13.0 FROM nvidia/cuda:${CUDA_VERSION}-devel-ubi${UBI_VERSION} as base From 7575f6ee8ab2c9b9bd81b84512b5100b7a22718c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 4 Sep 2026 12:30:06 +0000 Subject: [PATCH 6/6] Update test kernel locks --- kernels/tests/kernel_locking/kernels.lock | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kernels/tests/kernel_locking/kernels.lock b/kernels/tests/kernel_locking/kernels.lock index b6a69d1c..e7687e65 100644 --- a/kernels/tests/kernel_locking/kernels.lock +++ b/kernels/tests/kernel_locking/kernels.lock @@ -5,7 +5,7 @@ "version": 1 }, "lock": { - "commit": "0bd00775195dda6f84c16c31f6d94564232d08b3" + "commit": "94986c119039ea0cb65592c210c57255de1089eb" } }, { @@ -14,7 +14,7 @@ "version": 1 }, "lock": { - "commit": "d649efb56fb249ac8f7a57fa1866728ad0c60e52" + "commit": "4f593f99ef250e10e9377b5c10109351b418f318" } }, {