release cut 1

Author: zewenli98

docker/dist-build.sh

@@ -5,9 +5,9 @@ set -x
 TOP_DIR=$(cd $(dirname $0); pwd)/..
 
 if [[ -z "${USE_PRE_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/cu128 -w dist"
 else
-    BUILD_CMD="python -m pip wheel . --config-setting="--build-option=--use-pre-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-pre-cxx11-abi" --extra-index-url https://download.pytorch.org/whl/test/cu128 -w dist"
 fi
 
 # TensorRT restricts our pip version

py/requirements.txt

@@ -1,8 +1,8 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/nightly/cu124
-torch>=2.7.0.dev,<2.8.0
-torchvision>=0.22.0.dev,<0.23.0
+--extra-index-url https://download.pytorch.org/whl/test/cu128
+torch>=2.7.0,<2.8.0
+torchvision>=0.22.0,<0.23.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml

pyproject.toml

@@ -9,7 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt-cu12>=10.9.0,<10.10.0",
-    "torch>=2.7.0.dev,<2.8.0",
+    "torch>=2.7.0,<2.8.0",
     "pybind11==2.6.2",
     "numpy",
     "sympy",
@@ -55,7 +55,7 @@ keywords = [
     "inference",
 ]
 dependencies = [
-    "torch>=2.7.0.dev,<2.8.0",
+    "torch>=2.7.0,<2.8.0",
     "tensorrt>=10.9.0,<10.10.0",
     "tensorrt-cu12>=10.9.0,<10.10.0",
     "tensorrt-cu12-bindings>=10.9.0,<10.10.0",
@@ -108,12 +108,12 @@ prerelease = "if-necessary-or-explicit"
 index-strategy = "unsafe-best-match"                                    # Needed for TRT-LLM
 
 [tool.uv.sources]
-torch = [{ index = "pytorch-nightly-cu126" }]
-torchvision = [{ index = "pytorch-nightly-cu126" }]
+torch = [{ index = "pytorch-test-cu128" }]
+torchvision = [{ index = "pytorch-test-cu128" }]
 
 [[tool.uv.index]]
-name = "pytorch-nightly-cu126"
-url = "https://download.pytorch.org/whl/nightly/cu126"
+name = "pytorch-test-cu128"
+url = "https://download.pytorch.org/whl/test/cu128"
 explicit = false
 
 [[tool.uv.index]]
@@ -151,7 +151,6 @@ explicit = false
 # url = "https://download.pytorch.org/whl/cu118"
 # explicit = false
 
-
 [tool.ruff]
 # NOTE: Synchoronize the ignores with .flake8
 lint.ignore = [