Update to Torch 2.14.0 and remove Torch 2.12.

.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

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,16 +45,13 @@ 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`
 - `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`
 

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"

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"

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"
     }
   },
   {

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"
     ]

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.
   '';

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 {

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