Revert "fix torch specs in requirements files" (#1150)

ftynse · web-flow · commit 336951908cda · 2026-03-19T17:00:46.000+01:00
Reverts #1134. Let's see if this somehow broke wave runtime tests.
diff --git a/gen-pytorch-rocm-requirements.py b/gen-pytorch-rocm-requirements.py
@@ -29,7 +29,7 @@
 PYTORCH_INDEX = "https://download.pytorch.org/whl/rocm{ver}"
 AMD_FIND_LINKS = "https://repo.radeon.com/rocm/manylinux/rocm-rel-{ver}/"
 
-TORCH_SPEC = "torch>=2.6,<2.10"
+TORCH_SPEC = "torch>=2.6"
 
 # ---------------------------------------------------------------------------
 # ROCm version detection
diff --git a/pytorch-cpu-requirements.txt b/pytorch-cpu-requirements.txt
@@ -1,2 +1,2 @@
 --index-url https://download.pytorch.org/whl/cpu
-torch>=2.6,<2.10
+torch>=2.6,<2.9
diff --git a/pytorch-rocm-requirements.txt b/pytorch-rocm-requirements.txt
@@ -1,2 +1,2 @@
 --index-url https://download.pytorch.org/whl/rocm6.4
-torch>=2.6,<2.10
+torch>=2.6,<2.9

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`	`1`	`--index-url https://download.pytorch.org/whl/cpu`
`2`		`-torch>=2.6,<2.10`
	`2`	`+torch>=2.6,<2.9`