WIP

Author: danieldk

overlay.nix

@@ -144,6 +144,11 @@ in
           torchVersion = "2.9";
           xpuPackages = final.xpuPackages_2025_2;
         };
+
+        triton-xpu_2_10 = callPackage ./pkgs/python-modules/triton-xpu {
+          torchVersion = "2.10";
+          xpuPackages = final.xpuPackages_2025_3;
+        };
       }
     )
     (import ./pkgs/python-modules/hooks)

pkgs/python-modules/torch/binary/generic.nix

@@ -48,15 +48,8 @@
   effectiveStdenv ? if cudaSupport then cudaPackages.backendStdenv else stdenv,
 }:
 let
-  effectiveTriton =
-    if cudaSupport then
-      triton-cuda
-    else if xpuSupport then
-      python.pkgs.triton-xpu_2_8
-    else
-      triton;
-
   archs = (import ../archs.nix).${lib.versions.majorMinor version};
+  torchMajorMinor = lib.versions.majorMinor version;
 
   supportedTorchCudaCapabilities =
     let
@@ -68,16 +61,27 @@ let
   supportedCudaCapabilities = lib.intersectLists cudaPackages.flags.cudaCapabilities supportedTorchCudaCapabilities;
   inherit (archs) supportedTorchRocmArchs;
 
+  xpuTritonVersions = {
+    "2.8" = python.pkgs.triton-xpu_2_8;
+    "2.9" = python.pkgs.triton-xpu_2_9;
+    "2.10" = python.pkgs.triton-xpu_2_10;
+  };
+
+  effectiveTriton =
+    if cudaSupport then
+      triton-cuda
+    else if xpuSupport then
+      xpuTritonVersions.${torchMajorMinor}
+    else
+      triton;
+
   aotritonVersions = with rocmPackages; {
     "2.8" = aotriton_0_10;
     "2.9" = aotriton_0_11;
     "2.10" = aotriton_0_11_1;
   };
 
   aotriton =
-    let
-      torchMajorMinor = lib.versions.majorMinor version;
-    in
     aotritonVersions.${torchMajorMinor}
       or (throw "aotriton version is not specified Torch ${torchMajorMinor}");
 

pkgs/python-modules/triton-xpu/default.nix

@@ -56,6 +56,28 @@ let
         hash = "sha256-N8NBAkkpOcbgap4loPJJW6E5bjG+TixCh/HN259RyjI=";
       };
     };
+    "2.10" = {
+      # https://github.com/intel/intel-xpu-backend-for-triton/blob/225cdbde3ea155d5ed4c0aad1f2aa4bd2b3c4a3d/cmake/llvm-hash.txt
+      llvm = {
+        rev = "f6ded0be897e2878612dd903f7e8bb85448269e5";
+        hash = "sha256-T76zHZZ2bp3Ye9GTV+MgbKqMbtmMGElMFsWuCkiWqrM=";
+      };
+      # https://github.com/pytorch/pytorch/tree/v2.10.0-rc7/.ci/docker/ci_commit_pins
+      triton = {
+        rev = "225cdbde3ea155d5ed4c0aad1f2aa4bd2b3c4a3d";
+        hash = "sha256-AuNk4FMBwi7y1zWGhN/P0JsYwPuKV79JBLDDw6IVouA=";
+      };
+      # https://github.com/intel/intel-xpu-backend-for-triton/blob/225cdbde3ea155d5ed4c0aad1f2aa4bd2b3c4a3d/third_party/intel/lib/Target/SPIRV/spirv-llvm-translator.conf
+      spirv_llm = {
+        rev = "daba8b217bc266806ac00095262d1af0ba2ee610";
+        hash = "sha256-X/Pk1GpA1Se6UFp1UIbNAW1JLTj3vgFtg9b7Niv3/ro=";
+      };
+      # https://github.com/KhronosGroup/SPIRV-LLVM-Translator/blob/daba8b217bc266806ac00095262d1af0ba2ee610/spirv-headers-tag.conf
+      spirv_headers = {
+        rev = "9e3836d7d6023843a72ecd3fbf3f09b1b6747a9e";
+        hash = "sha256-N8NBAkkpOcbgap4loPJJW6E5bjG+TixCh/HN259RyjI=";
+      };
+    };
   };
   tritonVersions =
     torchTritonVersions.${torchVersion} or (throw "Unsupported Torch version: ${torchVersion}");
@@ -67,7 +89,7 @@ let
         "SPIRV"
       ];
     }
-    // lib.optionalAttrs (torchVersion == "2.9") {
+    // lib.optionalAttrs (lib.versionAtLeast torchVersion "2.9") {
       llvmProjectsToBuild = [
         "mlir"
         "llvm"
@@ -119,6 +141,10 @@ buildPythonPackage rec {
       sed -i 's/-Werror//g' $NIX_BUILD_TOP/source/CMakeLists.txt
       sed -i 's/ninja==1.11.1.4/ninja>=1.11.1/' $NIX_BUILD_TOP/source/pyproject.toml
     ''}
+    ${lib.optionalString (torchVersion == "2.10") ''
+      sed -i 's/-Werror//g' $NIX_BUILD_TOP/source/CMakeLists.txt
+      sed -i 's/ninja<1.13.0/ninja/' $NIX_BUILD_TOP/source/pyproject.toml
+    ''}
     sed -i '/if (NOT SPIRVToLLVMTranslator_FOUND)/,/endif (NOT SPIRVToLLVMTranslator_FOUND)/c\
       set(SPIRVToLLVMTranslator_SOURCE_DIR "${spirvLlvmTranslatorSrc}")\n\
       set(SPIRVToLLVMTranslator_BINARY_DIR \''${CMAKE_CURRENT_BINARY_DIR}/SPIRVToLLVMTranslator-build)\n\