From 7ef805d3b549500791a9455eaa2fbe0a85ed8238 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Mon, 24 Aug 2026 07:32:53 +0000 Subject: [PATCH 1/8] Ingest Torch 2.14.0-rc6 --- .../torch/binary/torch-versions-hash.json | 78 +++++++++++++++++++ .../torch/binary/torch-versions.json | 55 +++++++++++++ 2 files changed, 133 insertions(+) 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 8f03b3e50..cca1676f2 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 @@ -143,5 +143,83 @@ "version": "2.13.0" } } + }, + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "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=", + "version": "2.14.0" + } + } } } \ No newline at end of file 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 88370dd2c..60a1560d9 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json +++ b/nix-builder/pkgs/python-modules/torch/binary/torch-versions.json @@ -85,5 +85,60 @@ "torchVersion": "2.13.0", "xpuVersion": "2025.3.2", "systems": ["x86_64-linux"] + }, + + { + "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 04eff289e4e6c6e285fef9cc22284044a7fe15a3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Mon, 24 Aug 2026 13:54:31 +0000 Subject: [PATCH 2/8] python3Packages.triton_3_8_0: init --- nix-builder/overlay.nix | 1 + nix-builder/pkgs/python-modules/triton/default.nix | 11 ++++++++++- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/nix-builder/overlay.nix b/nix-builder/overlay.nix index cc2f10ff0..5a4b4c701 100644 --- a/nix-builder/overlay.nix +++ b/nix-builder/overlay.nix @@ -78,6 +78,7 @@ final: prev: inherit (triton) triton_3_7_0 triton_3_7_1 + triton_3_8_0 ; inherit (triton-rocm) triton-rocm_3_7_0; inherit (triton-xpu) diff --git a/nix-builder/pkgs/python-modules/triton/default.nix b/nix-builder/pkgs/python-modules/triton/default.nix index afa0d162f..58a37ded2 100644 --- a/nix-builder/pkgs/python-modules/triton/default.nix +++ b/nix-builder/pkgs/python-modules/triton/default.nix @@ -26,7 +26,16 @@ let hash = "sha256-NIlNUa/xq/ewF/vQxen+chEwXDWZFX/KJR4/QryPAM8="; }; }; - + "3.8.0" = { + x86_64-linux = { + url = "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/triton-3.8.0-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; + hash = "sha256-N5GzFsE7IramrxGuYTCvFteG/LORzv0vS38JSqfnQH8="; + }; + aarch64-linux = { + url = "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/triton-3.8.0-cp313-cp313-linux_aarch64.whl"; + hash = "sha256-wSLjP1kxqfERDKia4OpaeXkkGw3RVXDsSfKv9qyLcd0="; + }; + }; }; generic = callPackage ./generic.nix { }; versionAttr = lib.replaceStrings [ "." ] [ "_" ]; From b6febeb90766642f681dc8bf53b89d96afb68aa4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 06:31:19 +0000 Subject: [PATCH 3/8] xpuPackages_2026_1_0: init --- nix-builder/overlay.nix | 10 + .../python-modules/triton-xpu/default.nix | 5 + .../intel-deep-learning-2026.1.0.json | 792 ++++++++++++++++++ nix-builder/pkgs/xpu-packages/onednn-xpu.nix | 4 + 4 files changed, 811 insertions(+) create mode 100644 nix-builder/pkgs/xpu-packages/intel-deep-learning-2026.1.0.json diff --git a/nix-builder/overlay.nix b/nix-builder/overlay.nix index 5a4b4c701..513003e16 100644 --- a/nix-builder/overlay.nix +++ b/nix-builder/overlay.nix @@ -84,6 +84,7 @@ final: prev: inherit (triton-xpu) triton-xpu_3_7_1 triton-xpu_3_7_2 + triton-xpu_3_8_0 ; cuda-bindings = python-self.callPackage ./pkgs/python-modules/cuda-bindings { }; @@ -230,6 +231,14 @@ final: prev: xpuPackages = final.xpuPackages_2026_0_0; }; + torch-bin_2_14 = mkTorch { + version = "2.14"; + triton-cuda = triton_3_8_0; + triton-rocm = triton-rocm_3_7_0; + triton-xpu = triton-xpu_3_8_0; + xpuPackages = final.xpuPackages_2026_1_0; + }; + transformers = python-super.transformers.overridePythonAttrs (prevAttrs: rec { version = "5.3.0"; src = python-super.fetchPypi { @@ -297,6 +306,7 @@ final: prev: xpuVersions = [ "2025.3.2" "2026.0.0" + "2026.1.0" ]; newXpuPackages = final.callPackage ./pkgs/xpu-packages { }; in diff --git a/nix-builder/pkgs/python-modules/triton-xpu/default.nix b/nix-builder/pkgs/python-modules/triton-xpu/default.nix index 91c504615..2ab719f39 100644 --- a/nix-builder/pkgs/python-modules/triton-xpu/default.nix +++ b/nix-builder/pkgs/python-modules/triton-xpu/default.nix @@ -18,4 +18,9 @@ in hash = "sha256-jMwVkt3aI6sunRQ9orS2nrQIIRcsW9S8TiYjI9x/UXI="; }; + triton-xpu_3_8_0 = generic { + version = "3.8.0"; + url = "https://huggingface.co/buckets/danieldk/pytorch-rc/resolve/2.14.0/rc6/triton_xpu-3.8.0-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl"; + hash = "sha256-Yg7PawctmULzjsvM2APnFpZc6dDWo3Jx2GvtK4U0T04="; + }; } diff --git a/nix-builder/pkgs/xpu-packages/intel-deep-learning-2026.1.0.json b/nix-builder/pkgs/xpu-packages/intel-deep-learning-2026.1.0.json new file mode 100644 index 000000000..3e922ab66 --- /dev/null +++ b/nix-builder/pkgs/xpu-packages/intel-deep-learning-2026.1.0.json @@ -0,0 +1,792 @@ +{ + "intel-deep-learning-essentials": { + "deps": [ + "intel-deep-learning-essentials-env", + "intel-deep-learning-essentials-getting-started", + "intel-oneapi-ccl", + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp", + "intel-oneapi-dev-utilities", + "intel-oneapi-libdpstd-devel", + "intel-oneapi-mkl-devel", + "intel-oneapi-tlt", + "intel-pti-dev" + ], + "components": [ + { + "name": "intel-deep-learning-essentials-2026.1", + "sha256": "64b168b2220ef8f9132f13e6100b8939b86d00be8de04b40d2d39f4971228485", + "url": "https://yum.repos.intel.com/oneapi/intel-deep-learning-essentials-2026.1-2026.1.0-177.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-deep-learning-essentials-env": { + "deps": [ + "intel-oneapi-tlt" + ], + "components": [ + { + "name": "intel-deep-learning-essentials-env-2026.1", + "sha256": "2616e3bc23a7ce95e01a8edbf4f9b424eb801fd380f4e26ab5fbbfb555144b8e", + "url": "https://yum.repos.intel.com/oneapi/intel-deep-learning-essentials-env-2026.1-2026.1.0-177.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-deep-learning-essentials-getting-started": { + "deps": [], + "components": [ + { + "name": "intel-deep-learning-essentials-getting-started-2026.1", + "sha256": "ad9e7adfe6eac5f99c9aa8518eddb72e05ffab3ade88ccd9ceb2d97a6f73e5a4", + "url": "https://yum.repos.intel.com/oneapi/intel-deep-learning-essentials-getting-started-2026.1-2026.1.0-177.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-ccl": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mpi" + ], + "components": [ + { + "name": "intel-oneapi-ccl-2022.1", + "sha256": "9f58b162aee6d8dc112050302f5fbf01cc1719f392a0d7e5cc969be1183bc8c8", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-ccl-2022.1-2022.1.1-3.x86_64.rpm", + "version": "2022.1.1" + }, + { + "name": "intel-oneapi-ccl-devel-2022.1", + "sha256": "12a7a32fb8abc15dd6089568fa590c468b96b6d21de4a2538b96586cb704c4fb", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-ccl-devel-2022.1-2022.1.1-3.x86_64.rpm", + "version": "2022.1.1" + } + ], + "version": "2022.1.1" + }, + "intel-oneapi-common-licensing": { + "deps": [], + "components": [ + { + "name": "intel-oneapi-common-licensing-2026.0", + "sha256": "5fa8776f170337086c3f65dd004b2c44dd9ede2fd65dfa7f7ccf8c91ed53e1ff", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-common-licensing-2026.0-2026.0.0-235.noarch.rpm", + "version": "2026.0.0" + } + ], + "version": "2026.0.0" + }, + "intel-oneapi-common-oneapi-vars": { + "deps": [], + "components": [ + { + "name": "intel-oneapi-common-oneapi-vars-2026.0", + "sha256": "cdbfba8aa00a8d002668bc16dc0b57c59180c087b485218ba10258e949a0f0be", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-common-oneapi-vars-2026.0-2026.0.0-235.noarch.rpm", + "version": "2026.0.0" + } + ], + "version": "2026.0.0" + }, + "intel-oneapi-common-vars": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars" + ], + "components": [ + { + "name": "intel-oneapi-common-vars", + "sha256": "49ec4a58ca254dcf99b71b4b7bfe130359b6fd4b478eb74ec06caeb367f05d10", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-common-vars-2026.0.0-235.noarch.rpm", + "version": "2026.0.0" + } + ], + "version": "2026.0.0" + }, + "intel-oneapi-compiler-cpp-eclipse-cfg": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-compiler-cpp-eclipse-cfg-2026.1", + "sha256": "3b5781a56c2ce686c938a0ba7f336970c74d4dc4680e59a9fcf22586a80b87f4", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-cpp-eclipse-cfg-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-dpcpp-cpp": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-dpcpp-cpp", + "intel-oneapi-libdpstd-devel" + ], + "components": [ + { + "name": "intel-oneapi-compiler-dpcpp-cpp-2026.1", + "sha256": "e86f0f6ea6cfad22e851613f80b8a98e2bc8828b4d64493d9f3831fce2e0ad04", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-dpcpp-cpp-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-dpcpp-cpp-common": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-cpp-eclipse-cfg", + "intel-oneapi-compiler-dpcpp-eclipse-cfg", + "intel-oneapi-compiler-shared-common", + "intel-oneapi-icc-eclipse-plugin-cpp" + ], + "components": [ + { + "name": "intel-oneapi-compiler-dpcpp-cpp-common-2026.1", + "sha256": "c43c495e9c13ce48842a92c14e2020f1b3559b018a35055f3f5610492f4faf72", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-dpcpp-cpp-common-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-dpcpp-cpp-runtime": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-shared-runtime", + "intel-oneapi-tbb", + "intel-oneapi-umf" + ], + "components": [ + { + "name": "intel-oneapi-compiler-dpcpp-cpp-runtime-2026.1", + "sha256": "699557d2c33ea0f57001569d4bf45876b335253617f8ec31bcad7e78e9b1f75f", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-dpcpp-cpp-runtime-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-dpcpp-eclipse-cfg": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-compiler-dpcpp-eclipse-cfg-2026.1", + "sha256": "f7b8f7460c1b6d0677a75f6ebd0c59f51245f81a138cf251804c09103645f3f7", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-dpcpp-eclipse-cfg-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-shared": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-shared-common", + "intel-oneapi-compiler-shared-runtime", + "intel-oneapi-dpcpp-debugger" + ], + "components": [ + { + "name": "intel-oneapi-compiler-shared-2026.1", + "sha256": "f36e9a02cb8afc4ecb13720d70cf42659628e9a3589b9bf082ab86f8d01982e2", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-shared-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-shared-common": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-compiler-shared-common-2026.1", + "sha256": "0b94d4421052fb3952715e0c27679db4b061abbc606e621890cf4a6026cd550c", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-shared-common-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-compiler-shared-runtime": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-openmp" + ], + "components": [ + { + "name": "intel-oneapi-compiler-shared-runtime-2026.1", + "sha256": "98931f961d8ebc6d21056547a66ca8f892878e1859e19f9b750b91d0f8bf2034", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-compiler-shared-runtime-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-dev-utilities": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-dev-utilities-eclipse-cfg" + ], + "components": [ + { + "name": "intel-oneapi-dev-utilities-2026.0", + "sha256": "cfefeec9eade215a36b8bee028f894d288c8bb6fa9f3c6aa87a038d21d220a63", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-dev-utilities-2026.0-2026.0.1-16.x86_64.rpm", + "version": "2026.0.1" + } + ], + "version": "2026.0.1" + }, + "intel-oneapi-dev-utilities-eclipse-cfg": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-dev-utilities-eclipse-cfg-2026.0", + "sha256": "fea0e4ba401ba0b4265f4163007f732aed35f375aaa863779ed732f94800282a", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-dev-utilities-eclipse-cfg-2026.0-2026.0.1-16.noarch.rpm", + "version": "2026.0.1" + } + ], + "version": "2026.0.1" + }, + "intel-oneapi-dpcpp-cpp": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-common", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-compiler-shared", + "intel-oneapi-dev-utilities", + "intel-oneapi-tbb" + ], + "components": [ + { + "name": "intel-oneapi-dpcpp-cpp-2026.1", + "sha256": "298f68ba711912815ae21b36b507852820f4421673ec88067cd954a0797fd6e7", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-dpcpp-cpp-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-icc-eclipse-plugin-cpp": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-icc-eclipse-plugin-cpp-2026.1", + "sha256": "32cc6cac04ad6183ff10678999fef952ce4364ac9fb6996be01ddc0b1c6bd3d1", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-icc-eclipse-plugin-cpp-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-libdpstd-devel": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-libdpstd-devel-2022.13", + "sha256": "33a073097137ecb139301b9a1c671c04ccf5bf0ea88c758435de6c066c714278", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-libdpstd-devel-2022.13-2022.13.0-107.x86_64.rpm", + "version": "2022.13.0" + } + ], + "version": "2022.13.0" + }, + "intel-oneapi-mkl-classic-devel": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-classic-include", + "intel-oneapi-mkl-cluster", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-classic-devel-2026.1", + "sha256": "50e4248ba330d3842e13876ddc2c22f95b4c3ac12eafd97946702f5e0e35ab21", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-classic-devel-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-classic-include": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-mkl-classic-include-2026.1", + "sha256": "2fab7105e6e01faa6fbc887a7f710c0475ff59887b1f57202f1565f0a62e6446", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-classic-include-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-cluster": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-cluster-2026.1", + "sha256": "893a4752f1c9d5244877b119c1020450002966c60c41b4648005232f07f84700", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-cluster-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + }, + { + "name": "intel-oneapi-mkl-cluster-devel-2026.1", + "sha256": "0ad071ff7d08aa67626a033b2dde2e24e0f16828b6dc7d445824246ccaa94c82", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-cluster-devel-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-core": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-classic-include", + "intel-oneapi-openmp", + "intel-oneapi-tbb" + ], + "components": [ + { + "name": "intel-oneapi-mkl-core-2026.1", + "sha256": "7e0644f9c67dde03ec00c2a0bf262907b91d83a24235c95583af48642dacae17", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-core-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + }, + { + "name": "intel-oneapi-mkl-core-devel-2026.1", + "sha256": "62201467a9b321373e37ac9008f2a55200693e34c565be42c317cf094c9ae48a", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-core-devel-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-devel": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-classic-devel", + "intel-oneapi-mkl-sycl" + ], + "components": [ + { + "name": "intel-oneapi-mkl-devel-2026.1", + "sha256": "6cda821e96c25c4a7ddd8c84386c1cffebf9d39646fe838ca1f5ecfdaf2481f2", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-devel-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-core", + "intel-oneapi-mkl-sycl-blas", + "intel-oneapi-mkl-sycl-data-fitting", + "intel-oneapi-mkl-sycl-dft", + "intel-oneapi-mkl-sycl-include", + "intel-oneapi-mkl-sycl-lapack", + "intel-oneapi-mkl-sycl-rng", + "intel-oneapi-mkl-sycl-sparse", + "intel-oneapi-mkl-sycl-stats", + "intel-oneapi-mkl-sycl-vm" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-2026.1", + "sha256": "68a186f421d12a5b49a1404755b0d5bf0594e164cc07a9b072902c13268940df", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + }, + { + "name": "intel-oneapi-mkl-sycl-devel-2026.1", + "sha256": "7a29d748df916b24eaa67d0058dfa2da0efbf180f1325677d00809fe4942a681", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-devel-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-blas": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-blas-2026.1", + "sha256": "8c85c7eadfd534fb5419ed0e39c16d783bc28f9c0c7e472501b8ff8c89916cae", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-blas-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-data-fitting": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-data-fitting-2026.1", + "sha256": "b7bf072017bbcb31c4031724fbe9b969eca6aabf464855e6a5737b714a8ceec3", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-data-fitting-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-dft": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-dft-2026.1", + "sha256": "64165fd63fe788146c23d5808b097b75563dc8bf12f076ea833d36ebba63d6e3", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-dft-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-include": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-mkl-classic-include" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-include-2026.1", + "sha256": "41c4739761b63104cc9f0fde185d789cca0622a37dda690a44e645112249d74c", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-include-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-lapack": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core", + "intel-oneapi-mkl-sycl-blas" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-lapack-2026.1", + "sha256": "1960404284226aa8ec3af7f02656d9cbf62f3cad733b0b46d3e353d942c530bf", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-lapack-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-rng": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-rng-2026.1", + "sha256": "4fe720004808bf8545d18c3cfd3c386e7494a5f3bae6d0896d7740fe2d2c41bd", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-rng-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-sparse": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core", + "intel-oneapi-mkl-sycl-blas" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-sparse-2026.1", + "sha256": "bf1daea87758cd3428e099443546da0b2744647e48bc390a6937a130122b414b", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-sparse-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-stats": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-stats-2026.1", + "sha256": "b7ed88fa08ecebc48fb4ad42cd41c2f080964fe94a6fb6aa7304d9e6bc0debd6", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-stats-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mkl-sycl-vm": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-compiler-dpcpp-cpp-runtime", + "intel-oneapi-mkl-core" + ], + "components": [ + { + "name": "intel-oneapi-mkl-sycl-vm-2026.1", + "sha256": "8c3588243a2a63f67fff12225066e3673b7c468137573038408d91ece9243512", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mkl-sycl-vm-2026.1-2026.1.0-236.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-mpi": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-mpi-2021.18", + "sha256": "d0de538635467c4aa3dd5d44a1b1afb07715bf44c8ee776ac74bc745d6b112bf", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mpi-2021.18-2021.18.1-9.x86_64.rpm", + "version": "2021.18.1" + }, + { + "name": "intel-oneapi-mpi-devel-2021.18", + "sha256": "56f56d7870a0b22829f70235b91e6c4eb78921c473c4fa4aab067493c67e4fca", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-mpi-devel-2021.18-2021.18.1-9.x86_64.rpm", + "version": "2021.18.1" + } + ], + "version": "2021.18.1" + }, + "intel-oneapi-openmp": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-openmp-common", + "intel-oneapi-umf" + ], + "components": [ + { + "name": "intel-oneapi-openmp-2026.1", + "sha256": "f14c21d0e76077394d60a17825fb4606ee06565efbee40d2a18240b129c74366", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-openmp-2026.1-2026.1.0-235.x86_64.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-openmp-common": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-openmp-common-2026.1", + "sha256": "376afff03072bb22b16318ed6356210e9f717f09a56904e381017de58ddf635d", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-openmp-common-2026.1-2026.1.0-235.noarch.rpm", + "version": "2026.1.0" + } + ], + "version": "2026.1.0" + }, + "intel-oneapi-tbb": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-tcm" + ], + "components": [ + { + "name": "intel-oneapi-tbb-2023.1", + "sha256": "8b25110f2374ae97cf9cb013ee59a57d2b078e3ce359a1323299558ce161d3d1", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-tbb-2023.1-2023.1.0-151.x86_64.rpm", + "version": "2023.1.0" + }, + { + "name": "intel-oneapi-tbb-devel-2023.1", + "sha256": "a2a72cdb71640025f9b9cca614e88aa21283c08c0fcd5f9097bcc058623a74c6", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-tbb-devel-2023.1-2023.1.0-151.x86_64.rpm", + "version": "2023.1.0" + } + ], + "version": "2023.1.0" + }, + "intel-oneapi-tcm": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-oneapi-tcm-1.5", + "sha256": "52e9cb0a6d1d35ec016a8127066e801e26c32e1936dafa0ea276c5f193a2a672", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-tcm-1.5-1.5.0-489.x86_64.rpm", + "version": "1.5.0" + } + ], + "version": "1.5.0" + }, + "intel-oneapi-tlt": { + "deps": [], + "components": [ + { + "name": "intel-oneapi-tlt-2026.0", + "sha256": "3e57339aa8aed4f1a27364a99402e5c9809cdf9ce0569274541d386f6e19d9fb", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-tlt-2026.0-2026.0.0-220.x86_64.rpm", + "version": "2026.0.0" + } + ], + "version": "2026.0.0" + }, + "intel-oneapi-umf": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars", + "intel-oneapi-tcm" + ], + "components": [ + { + "name": "intel-oneapi-umf-1.1", + "sha256": "f00958235e64afdd6e4b153c4578a954dc58c92a2e6bc695cee35cc1766bd7a4", + "url": "https://yum.repos.intel.com/oneapi/intel-oneapi-umf-1.1-1.1.0-340.x86_64.rpm", + "version": "1.1.0" + } + ], + "version": "1.1.0" + }, + "intel-pti": { + "deps": [ + "intel-oneapi-common-licensing", + "intel-oneapi-common-oneapi-vars", + "intel-oneapi-common-vars" + ], + "components": [ + { + "name": "intel-pti-1.0", + "sha256": "62f2aaea70163ed883d44cb581ccd454247e351fc7ee36c996f22c7746178b99", + "url": "https://yum.repos.intel.com/oneapi/intel-pti-1.0-1.0.1-21.x86_64.rpm", + "version": "1.0.1" + } + ], + "version": "1.0.1" + }, + "intel-pti-dev": { + "deps": [ + "intel-pti" + ], + "components": [ + { + "name": "intel-pti-dev-1.0", + "sha256": "75d8df27a80103579daedb20c27ddcd6c6520455065083b9d52f88103379c291", + "url": "https://yum.repos.intel.com/oneapi/intel-pti-dev-1.0-1.0.1-21.x86_64.rpm", + "version": "1.0.1" + } + ], + "version": "1.0.1" + } +} diff --git a/nix-builder/pkgs/xpu-packages/onednn-xpu.nix b/nix-builder/pkgs/xpu-packages/onednn-xpu.nix index a2a35fa40..43a3f1436 100644 --- a/nix-builder/pkgs/xpu-packages/onednn-xpu.nix +++ b/nix-builder/pkgs/xpu-packages/onednn-xpu.nix @@ -20,6 +20,10 @@ let version = "3.12.2"; hash = "sha256-LrxXRe2La5SZudikqMeGggklLrY2S0okrBX6QXGD+Tc="; }; + "2026.1" = { + version = "3.12.3"; + hash = "sha256-cxlwxlVYoh8ByZJzS+e7rJ8geSoTqC/1AzEYFncq/sA="; + }; }; oneDnnVersion = oneDnnVersions.${lib.versions.majorMinor dpcppVersion} From 273bf3eeeec1e73f2d12eca4dcd660be2f6dd0e6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 07:11:53 +0000 Subject: [PATCH 4/8] rocmPackages.aotriton_0_13: init --- nix-builder/pkgs/aotriton/default.nix | 39 +++++++++++++++++++ .../python-modules/torch/binary/generic.nix | 1 + nix-builder/pkgs/rocm-packages/default.nix | 1 + 3 files changed, 41 insertions(+) diff --git a/nix-builder/pkgs/aotriton/default.nix b/nix-builder/pkgs/aotriton/default.nix index b6e99edf4..21b25cdc5 100644 --- a/nix-builder/pkgs/aotriton/default.nix +++ b/nix-builder/pkgs/aotriton/default.nix @@ -90,4 +90,43 @@ in ]; }; + aotriton_0_13 = generic rec { + version = "0.13b"; + + hashes = { + "7.2" = "sha256-HN7rt+9hq2kfuh2B2pGbnbXYvvKCaciSowvROgSVt6A="; + }; + + images = mkImages version [ + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx90a.tar.gz"; + hash = "sha256-o9GmhozikLqBGGGCBwk+eFJS7/ThimT0lXUstaA//tY="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx942.tar.gz"; + hash = "sha256-zNvH49loOb5Ile4ATyFTHMVdWQyQGJN7njFLujY7OSc="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx950.tar.gz"; + hash = "sha256-UY/QcusFlI/ApsJaIIMlkcZAbfhl47aRoq7/P9TFzh0="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx110x.tar.gz"; + hash = "sha256-7+dz56LIrcmV2Q7NDayi24AShURe4QQy3yspxn1bEdI="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx115x.tar.gz"; + hash = "sha256-G8UOiqi2vaNBDpKIbMyo/UXfPmCmy9qf/FiyxUHv1cI="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx120x.tar.gz"; + hash = "sha256-akZdvAMUi7qKLXjEwqPLgxVeygD09/dJ5nZALXZglow="; + }) + (fetchurl { + url = "https://github.com/ROCm/aotriton/releases/download/0.13b/aotriton-0.13b-images-amd-gfx1250.tar.gz"; + hash = "sha256-Sq9x1uUQVJ1ZN1fl+IWY3x5KKcvNkfcHUO6KdvZcAn8="; + }) + ]; + }; + } diff --git a/nix-builder/pkgs/python-modules/torch/binary/generic.nix b/nix-builder/pkgs/python-modules/torch/binary/generic.nix index 1b8183511..36d461e18 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/generic.nix +++ b/nix-builder/pkgs/python-modules/torch/binary/generic.nix @@ -77,6 +77,7 @@ let aotritonVersions = with rocmPackages; { "2.12" = aotriton_0_11_2; "2.13" = aotriton_0_12; + "2.14" = aotriton_0_13; }; aotriton = diff --git a/nix-builder/pkgs/rocm-packages/default.nix b/nix-builder/pkgs/rocm-packages/default.nix index 9112c882d..25adbcc24 100644 --- a/nix-builder/pkgs/rocm-packages/default.nix +++ b/nix-builder/pkgs/rocm-packages/default.nix @@ -31,6 +31,7 @@ let aotriton_0_11_1 aotriton_0_11_2 aotriton_0_12 + aotriton_0_13 ; }) ]; From b345d8950f5af22b67394fd98fce00ffafd13dca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 07:12:35 +0000 Subject: [PATCH 5/8] python3Packages.torch_2_14: also apply libnuma loading fix --- .../pkgs/python-modules/torch/binary/generic.nix | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/nix-builder/pkgs/python-modules/torch/binary/generic.nix b/nix-builder/pkgs/python-modules/torch/binary/generic.nix index 36d461e18..e1a58493d 100644 --- a/nix-builder/pkgs/python-modules/torch/binary/generic.nix +++ b/nix-builder/pkgs/python-modules/torch/binary/generic.nix @@ -62,6 +62,8 @@ effectiveStdenv ? if cudaSupport then cudaPackages.backendStdenv else stdenv, }: let + inherit (lib) versionAtLeast; + torchMajorMinor = lib.versions.majorMinor version; effectiveTriton = @@ -116,7 +118,7 @@ let rocsparse roctracer ] - ++ lib.optionals (lib.versionAtLeast version "2.12") [ + ++ lib.optionals (versionAtLeast version "2.12") [ rocprofiler-sdk ]; @@ -228,7 +230,7 @@ buildPythonPackage.override { stdenv = effectiveStdenv; } { )) ] ) - ++ lib.optionals (cudaSupport && lib.versionAtLeast version "2.9") [ + ++ lib.optionals (cudaSupport && versionAtLeast version "2.9") [ cudaPackages.libnvshmem ] ++ lib.optionals rocmSupport ([ @@ -274,10 +276,10 @@ buildPythonPackage.override { stdenv = effectiveStdenv; } { ++ lib.optionals tritonSupport [ effectiveTriton ] - ++ lib.optionals (cudaSupport && lib.versionAtLeast version "2.10") [ + ++ lib.optionals (cudaSupport && versionAtLeast version "2.10") [ cuda-bindings ] - ++ lib.optionals (xpuSupport && lib.versionAtLeast version "2.13") [ + ++ lib.optionals (xpuSupport && versionAtLeast version "2.13") [ pyzes ]; @@ -404,7 +406,7 @@ buildPythonPackage.override { stdenv = effectiveStdenv; } { # We need to add this rpath after postFixup (otherwise it might be ordered # before autoPatchelfHook), but before the import check to ensure that the # rpath is used during the import check. - preInstallCheck = lib.optionalString (rocmSupport && torchMajorMinor == "2.13") '' + preInstallCheck = lib.optionalString (rocmSupport && versionAtLeast torchMajorMinor "2.13") '' # libtorch_shmem dynamically loads libnuma. For added fun, if libnuma # cannot be found, it exit()s inside a constructor, calling an atexit() # hook, which then proceeds to deadlock. From b668698b14e85978d187cf4d11db8860711aa88b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 07:13:05 +0000 Subject: [PATCH 6/8] nix-builder: add most Torch 2.14 variants XPU is not building yet. --- nix-builder/versions.nix | 74 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 74 insertions(+) diff --git a/nix-builder/versions.nix b/nix-builder/versions.nix index 6587d31e9..9580b4a9f 100644 --- a/nix-builder/versions.nix +++ b/nix-builder/versions.nix @@ -148,4 +148,78 @@ bundleBuild = true; } + { + torchVersion = "2.14"; + cpu = true; + systems = [ + "aarch64-darwin" + "x86_64-linux" + "aarch64-linux" + ]; + bundleBuild = true; + } + { + torchVersion = "2.14"; + cudaVersion = "12.6"; + systems = [ + "x86_64-linux" + "aarch64-linux" + ]; + bundleBuild = true; + } + { + torchVersion = "2.14"; + cudaVersion = "13.0"; + systems = [ + "x86_64-linux" + "aarch64-linux" + ]; + bundleBuild = true; + } + { + torchVersion = "2.14"; + cudaVersion = "13.2"; + systems = [ + "x86_64-linux" + "aarch64-linux" + ]; + bundleBuild = true; + } + # Since 13.4 is a prerelease, not yet available on: + # https://developer.download.nvidia.com/compute/cuda/redist/ + #{ + # torchVersion = "2.14"; + # cudaVersion = "13.4"; + # systems = [ + # "x86_64-linux" + # "aarch64-linux" + # ]; + # bundleBuild = true; + # tvmFfiVersion = "0.1"; + #} + { + torchVersion = "2.14"; + metal = true; + systems = [ "aarch64-darwin" ]; + bundleBuild = true; + } + { + torchVersion = "2.14"; + rocmVersion = "7.2"; + systems = [ "x86_64-linux" ]; + bundleBuild = true; + } + #{ + # torchVersion = "2.14"; + # rocmVersion = "7.14"; + # systems = [ "x86_64-linux" ]; + # bundleBuild = true; + #} + { + torchVersion = "2.14"; + xpuVersion = "2026.1.0"; + systems = [ "x86_64-linux" ]; + bundleBuild = true; + } + ] From 7af826bafb95018342e31160b466fff76f4d7ca7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 09:56:25 +0000 Subject: [PATCH 7/8] Update variants --- docs/source/builder/build-variants.md | 12 ++++++++++++ nix-builder/build-variants.json | 28 +++++++++++++++++++-------- 2 files changed, 32 insertions(+), 8 deletions(-) diff --git a/docs/source/builder/build-variants.md b/docs/source/builder/build-variants.md index f96a1c462..f414c52cc 100644 --- a/docs/source/builder/build-variants.md +++ b/docs/source/builder/build-variants.md @@ -9,16 +9,19 @@ available. This list will be updated as new PyTorch versions are released. - `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 @@ -28,11 +31,15 @@ available. This list will be updated as new PyTorch versions are released. - `torch213-cxx11-cu126-aarch64-linux` - `torch213-cxx11-cu130-aarch64-linux` - `torch213-cxx11-cu132-aarch64-linux` +- `torch214-cxx11-cu126-aarch64-linux` +- `torch214-cxx11-cu130-aarch64-linux` +- `torch214-cxx11-cu132-aarch64-linux` ## 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 @@ -42,6 +49,9 @@ available. This list will be updated as new PyTorch versions are released. - `torch213-cxx11-cu126-x86_64-linux` - `torch213-cxx11-cu130-x86_64-linux` - `torch213-cxx11-cu132-x86_64-linux` +- `torch214-cxx11-cu126-x86_64-linux` +- `torch214-cxx11-cu130-x86_64-linux` +- `torch214-cxx11-cu132-x86_64-linux` ## ROCm x86_64-linux @@ -49,11 +59,13 @@ available. This list will be updated as new PyTorch versions are released. - `torch212-cxx11-rocm72-x86_64-linux` - `torch213-cxx11-rocm71-x86_64-linux` - `torch213-cxx11-rocm72-x86_64-linux` +- `torch214-cxx11-rocm72-x86_64-linux` ## XPU x86_64-linux - `torch212-cxx11-xpu20253-x86_64-linux` - `torch213-cxx11-xpu20260-x86_64-linux` +- `torch214-cxx11-xpu20261-x86_64-linux` ## Python-only kernels diff --git a/nix-builder/build-variants.json b/nix-builder/build-variants.json index 4de63e6dc..becf280a9 100644 --- a/nix-builder/build-variants.json +++ b/nix-builder/build-variants.json @@ -2,17 +2,20 @@ "aarch64-darwin": { "cpu": [ "torch212-cpu-aarch64-darwin", - "torch213-cpu-aarch64-darwin" + "torch213-cpu-aarch64-darwin", + "torch214-cpu-aarch64-darwin" ], "metal": [ "torch212-metal-aarch64-darwin", - "torch213-metal-aarch64-darwin" + "torch213-metal-aarch64-darwin", + "torch214-metal-aarch64-darwin" ] }, "aarch64-linux": { "cpu": [ "torch212-cxx11-cpu-aarch64-linux", - "torch213-cxx11-cpu-aarch64-linux" + "torch213-cxx11-cpu-aarch64-linux", + "torch214-cxx11-cpu-aarch64-linux" ], "cuda": [ "torch212-cxx11-cu126-aarch64-linux", @@ -20,13 +23,17 @@ "torch212-cxx11-cu132-aarch64-linux", "torch213-cxx11-cu126-aarch64-linux", "torch213-cxx11-cu130-aarch64-linux", - "torch213-cxx11-cu132-aarch64-linux" + "torch213-cxx11-cu132-aarch64-linux", + "torch214-cxx11-cu126-aarch64-linux", + "torch214-cxx11-cu130-aarch64-linux", + "torch214-cxx11-cu132-aarch64-linux" ] }, "x86_64-linux": { "cpu": [ "torch212-cxx11-cpu-x86_64-linux", - "torch213-cxx11-cpu-x86_64-linux" + "torch213-cxx11-cpu-x86_64-linux", + "torch214-cxx11-cpu-x86_64-linux" ], "cuda": [ "torch212-cxx11-cu126-x86_64-linux", @@ -34,17 +41,22 @@ "torch212-cxx11-cu132-x86_64-linux", "torch213-cxx11-cu126-x86_64-linux", "torch213-cxx11-cu130-x86_64-linux", - "torch213-cxx11-cu132-x86_64-linux" + "torch213-cxx11-cu132-x86_64-linux", + "torch214-cxx11-cu126-x86_64-linux", + "torch214-cxx11-cu130-x86_64-linux", + "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" + "torch213-cxx11-rocm72-x86_64-linux", + "torch214-cxx11-rocm72-x86_64-linux" ], "xpu": [ "torch212-cxx11-xpu20253-x86_64-linux", - "torch213-cxx11-xpu20260-x86_64-linux" + "torch213-cxx11-xpu20260-x86_64-linux", + "torch214-cxx11-xpu20261-x86_64-linux" ] } } From b646a923f82f634feeec40ee53084c7926bbbe80 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Tue, 25 Aug 2026 09:59:27 +0000 Subject: [PATCH 8/8] Fix e2e test --- .github/workflows/test_e2e.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test_e2e.yaml b/.github/workflows/test_e2e.yaml index 174cd6966..3bd307591 100644 --- a/.github/workflows/test_e2e.yaml +++ b/.github/workflows/test_e2e.yaml @@ -84,7 +84,8 @@ jobs: id: variant run: | cd /tmp/${{ env.E2E_REPO_NAME }} - VARIANT=$(nix run $GITHUB_WORKSPACE#kernel-builder -- list-variants . | tail -1) + # NOTE: Remove Torch 2.14 grep once it is released. + VARIANT=$(nix run $GITHUB_WORKSPACE#kernel-builder -- list-variants . | grep -v torch214 | tail -1) echo "name=$VARIANT" >> $GITHUB_OUTPUT echo "Building variant: $VARIANT"