From 7e9108efc6089e6a0c174ccc4e2bd81dca8cc227 Mon Sep 17 00:00:00 2001
From: Kishor Prins <me@68x.ca>
Date: Sun, 29 Jun 2025 17:53:36 -0700
Subject: [PATCH] Revert changes to pyproject toml to use pytorch nightly

---
 pyproject.toml | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 8acdb43..b83b9f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,10 +47,8 @@ gpu = [
     "torch==2.6.0+cu124",
 ]
 rocm = [
-    "torch @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torch-2.6.0%2Brocm6.4.1.git1ded221d-cp310-cp310-linux_x86_64.whl",
-    "pytorch-triton-rocm @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/pytorch_triton_rocm-3.2.0%2Brocm6.4.1.git6da9e660-cp310-cp310-linux_x86_64.whl",
-    "torchvision @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torchvision-0.21.0%2Brocm6.4.1.git4040d51f-cp310-cp310-linux_x86_64.whl",
-    "torchaudio @ https://repo.radeon.com/rocm/manylinux/rocm-rel-6.4.1/torchaudio-2.6.0%2Brocm6.4.1.gitd8831425-cp310-cp310-linux_x86_64.whl",
+    "torch==2.8.0.dev20250627+rocm6.4",
+    "pytorch-triton-rocm>=3.2.0",
 ]
 cpu = [
     "torch==2.6.0",
@@ -73,10 +71,15 @@ conflicts = [
     ],
 ]
 
+
 [tool.uv.sources]
 torch = [
     { index = "pytorch-cpu", extra = "cpu" },
     { index = "pytorch-cuda", extra = "gpu" },
+    { index = "pytorch-rocm", extra = "rocm" },
+]
+pytorch-triton-rocm = [
+    { index = "pytorch-rocm", extra = "rocm" },
 ]
 
 [[tool.uv.index]]
@@ -89,6 +92,11 @@ name = "pytorch-cuda"
 url = "https://download.pytorch.org/whl/cu124"
 explicit = true
 
+[[tool.uv.index]]
+name = "pytorch-rocm"
+url = "https://download.pytorch.org/whl/nightly/rocm6.4"
+explicit = true
+
 [build-system]
 requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"