chore: release branch cut

Signed-off-by: Naren Dasan <[email protected]>

.github/scripts/generate_binary_build_matrix.py

@@ -25,8 +25,8 @@
 }
 CUDA_ARCHES_DICT = {
     "nightly": ["11.8", "12.4", "12.6"],
-    "test": ["11.8", "12.1", "12.4"],
-    "release": ["11.8", "12.1", "12.4"],
+    "test": ["11.8", "12.4", "12.6"],
+    "release": ["11.8", "12.4", "12.6"],
 }
 ROCM_ARCHES_DICT = {
     "nightly": ["6.1", "6.2"],
@@ -64,8 +64,8 @@
 XPU = "xpu"
 
 
-CURRENT_NIGHTLY_VERSION = "2.6.0"
-CURRENT_CANDIDATE_VERSION = "2.5.1"
+CURRENT_NIGHTLY_VERSION = "2.7.0"
+CURRENT_CANDIDATE_VERSION = "2.6.0"
 CURRENT_STABLE_VERSION = "2.5.1"
 CURRENT_VERSION = CURRENT_STABLE_VERSION
 

MODULE.bazel

@@ -55,21 +55,21 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu126/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu126/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu126/libtorch-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu126/libtorch-shared-with-deps-latest.zip"],
 )
 
 http_archive(
     name = "libtorch_win",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu126/libtorch-win-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu126/libtorch-win-shared-with-deps-latest.zip"],
 )
 
 # Download these tarballs manually from the NVIDIA website

docker/dist-build.sh

@@ -5,9 +5,9 @@ set -x
 TOP_DIR=$(cd $(dirname $0); pwd)/..
 
 if [[ -z "${USE_CXX11}" ]]; then
-    BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/nightly/cu124 -w dist"
+    BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/test/cu126 -w dist"
 else
-    BUILD_CMD="python -m pip wheel . --config-setting="--build-option=--use-cxx11-abi" --extra-index-url https://download.pytorch.org/whl/nightly/cu124 -w dist"
+    BUILD_CMD="python -m pip wheel . --config-setting="--build-option=--use-cxx11-abi" --extra-index-url https://download.pytorch.org/whl/test/cu126 -w dist"
 fi
 
 # TensorRT restricts our pip version

py/requirements.txt

@@ -1,7 +1,7 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/nightly/cu124
+--extra-index-url https://download.pytorch.org/whl/test/cu126
 torch>=2.6.0.dev,<2.7.0
 torchvision>=0.20.0.dev,<0.21.0
 --extra-index-url https://pypi.ngc.nvidia.com

pyproject.toml

@@ -9,7 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt-cu12>=10.6.0,<10.8.0",
-    "torch>=2.6.0.dev,<2.7.0",
+    "torch>=2.6.0,<2.7.0",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -54,7 +54,7 @@ keywords = [
     "inference",
 ]
 dependencies = [
-    "torch>=2.6.0.dev,<2.7.0",
+    "torch>=2.6.0,<2.7.0",
     "tensorrt-cu12>=10.6.0,<10.8.0",
     "tensorrt-cu12-bindings>=10.6.0,<10.8.0",
     "tensorrt-cu12-libs>=10.6.0,<10.8.0",
@@ -101,21 +101,16 @@ include-package-data = false
 
 [tool.uv]
 environments = ["sys_platform == 'linux'", "sys_platform == 'windows'"]
-
 prerelease = "if-necessary-or-explicit"
-
+index-strategy = "unsafe-best-match"
 
 [tool.uv.sources]
-torch = [
-    { index = "pytorch-nightly-cu126"},
-]
-torchvision = [
-     { index = "pytorch-nightly-cu126"},
-]
+torch = [{ index = "pytorch-test-cu126" }]
+torchvision = [{ index = "pytorch-test-cu126" }]
 
 [[tool.uv.index]]
-name = "pytorch-nightly-cu126"
-url = "https://download.pytorch.org/whl/nightly/cu126"
+name = "pytorch-test-cu126"
+url = "https://download.pytorch.org/whl/test/cu126"
 explicit = false
 
 # [[tool.uv.index]]