From 1061262d964afcaad14d34b7817f6ac678ef0d56 Mon Sep 17 00:00:00 2001
From: Adam Siemieniuk <adam.siemieniuk@intel.com>
Date: Fri, 19 Jun 2026 17:38:22 +0200
Subject: [PATCH] Bump PyTorch version

Updates PyTorch to 2.12.1 release.
Since 2.10 PyTorch packages Triton under its default 'triton' name.
Thus, the explicit 'pytorch_triton' dependency is no longer needed.
---
 pyproject.toml | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 1e8f7b56..18b90ada 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,21 +23,19 @@ ingress_torch_mlir = [
 ]
 # Additional "targets" which pull in optional dependencies -- use `uv sync --extra TARGET`
 ingress_torch_cpu = [
-    "torch==v2.9.1+cpu",
+    "torch==v2.12.1+cpu",
     "lighthouse[ingress_torch_mlir]"
 ]
 ingress_torch_nvidia = [
-    "torch==2.9.1",  # Nvidia-enabled version of torch
+    "torch==2.12.1+cu130",  # Nvidia-enabled version of torch
     "lighthouse[ingress_torch_mlir]"
 ]
 ingress_torch_rocm = [
-    "torch==2.9.1+rocm6.4",  # AMD-enabled version of torch
-    "pytorch_triton_rocm",  # Transitive dependency listed explicitly so that we can state which package repository it is supposed to come from
+    "torch==2.12.1+rocm7.2",  # AMD-enabled version of torch
     "lighthouse[ingress_torch_mlir]"
 ]
 ingress_torch_xpu = [
-    "torch==2.9.1+xpu",  # Intel-enabled version of torch
-    "pytorch_triton_xpu",  # Transitive dependency listed explicitly so that we can state which package repository it is supposed to come from
+    "torch==2.12.1+xpu",  # Intel-enabled version of torch
     "lighthouse[ingress_torch_mlir]"
 ]
 runtime_mpich = [
@@ -50,6 +48,7 @@ runtime_impi = [
 ]
 
 [tool.uv]
+index-strategy = "unsafe-best-match"
 # Declare that the following "targets" are mutually exclusive of one another
 conflicts = [
     [
@@ -64,8 +63,6 @@ conflicts = [
 # Bind packages to particular package repositories
 mlir_python_bindings = { index = "eudsl" }
 torch = { index = "pytorch" }
-pytorch_triton_xpu = { index = "pytorch" }
-pytorch_triton_rocm = { index = "pytorch" }
 torch_mlir = { index = "torch_mlir" }
 
 # The following are the different non-Pypi package repositories we depend on
@@ -78,7 +75,6 @@ format = "flat"
 [[tool.uv.index]]
 name = "pytorch"
 url = "https://download.pytorch.org/whl"
-explicit = true
 
 [[tool.uv.index]]
 name = "torch_mlir"