upgrade torch_tensorrt version from 2.8.0.dev to 2.9.0.dev (#3639)

Author: lanluo-nvidia

.github/workflows/docgen.yml

@@ -14,12 +14,12 @@ jobs:
         if: ${{ ! contains(github.actor, 'pytorchbot') }}
         environment: pytorchbot-env
         container:
-            image: docker.io/pytorch/manylinux2_28-builder:cuda12.8
+            image: docker.io/pytorch/manylinux2_28-builder:cuda12.9
             options: --gpus all
         env:
-            CUDA_HOME: /usr/local/cuda-12.8
-            VERSION_SUFFIX: cu128
-            CU_VERSION: cu128
+            CUDA_HOME: /usr/local/cuda-12.9
+            VERSION_SUFFIX: cu129
+            CU_VERSION: cu129
             CHANNEL: nightly
             CI_BUILD: 1
         steps:
@@ -35,14 +35,14 @@ jobs:
             - name: Install base deps
               run: |
                   python3 -m pip install pip --upgrade
-                  python3 -m pip install pyyaml numpy torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu128
+                  python3 -m pip install pyyaml numpy torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu129
                   ./packaging/pre_build_script.sh
             - name: Get HEAD SHA
               id: vars
               run: echo "sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
             - name: Build Python Package
               run: |
-                  python3 -m pip install --pre . --extra-index-url https://download.pytorch.org/whl/nightly/cu128
+                  python3 -m pip install --pre . --extra-index-url https://download.pytorch.org/whl/nightly/cu129
             - name: Generate New Docs
               run: |
                   cd docsrc

MODULE.bazel

@@ -1,6 +1,6 @@
 module(
     name = "torch_tensorrt",
-    version = "2.8.0a0",
+    version = "2.9.0a0",
     repo_name = "org_pytorch_tensorrt",
 )
 

cpp/include/torch_tensorrt/macros.h

@@ -24,7 +24,7 @@
 #define STR(x) XSTR(x)
 
 #define TORCH_TENSORRT_MAJOR_VERSION 2
-#define TORCH_TENSORRT_MINOR_VERSION 6
+#define TORCH_TENSORRT_MINOR_VERSION 9
 #define TORCH_TENSORRT_PATCH_VERSION 0
 #define TORCH_TENSORRT_VERSION      \
   STR(TORCH_TENSORRT_MAJOR_VERSION) \

docker/Dockerfile

@@ -2,9 +2,9 @@
 
 # Base image starts with CUDA
 #TODO: cuda version
-ARG BASE_IMG=nvidia/cuda:12.8.0-devel-ubuntu22.04
+ARG BASE_IMG=nvidia/cuda:12.9.0-devel-ubuntu22.04
 FROM ${BASE_IMG} as base
-ENV BASE_IMG=nvidia/cuda:12.8.0-devel-ubuntu22.04
+ENV BASE_IMG=nvidia/cuda:12.9.0-devel-ubuntu22.04
 
 ARG TENSORRT_VERSION
 ENV TENSORRT_VERSION=${TENSORRT_VERSION}

docker/dist-build.sh

@@ -4,7 +4,7 @@ set -x
 
 TOP_DIR=$(cd $(dirname $0); pwd)/..
 
-BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/nightly/cu128 -w dist"
+BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/nightly/cu129 -w dist"
 
 # TensorRT restricts our pip version
 cd ${TOP_DIR} \

py/requirements.txt

@@ -2,7 +2,7 @@ numpy
 packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu129
-torch>=2.8.0.dev,<2.9.0
+torch>=2.9.0.dev,<2.10.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
 dllist

pyproject.toml

@@ -10,7 +10,7 @@ requires = [
     "future>=0.18.3",
     "tensorrt-cu12>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
     "tensorrt-cu12>=10.3.0,<10.4.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
-    "torch>=2.8.0.dev,<2.9.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
+    "torch>=2.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
     "torch>=2.7.0,<2.8.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
     "pybind11==2.6.2",
     "numpy; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
@@ -59,7 +59,7 @@ keywords = [
     "inference",
 ]
 dependencies = [
-    "torch>=2.8.0.dev,<2.9.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
+    "torch>=2.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
     "torch>=2.7.0,<2.8.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
 
     "tensorrt>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
@@ -135,17 +135,17 @@ index-strategy = "unsafe-best-match"
 
 [tool.uv.sources]
 torch = [
-    { index = "pytorch-nightly-cu128", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" },
+    { index = "pytorch-nightly-cu129", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" },
     { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" },
 ]
 torchvision = [
-    { index = "pytorch-nightly-cu128", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" },
+    { index = "pytorch-nightly-cu129", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" },
     { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" },
 ]
 
 [[tool.uv.index]]
-name = "pytorch-nightly-cu128"
-url = "https://download.pytorch.org/whl/nightly/cu128"
+name = "pytorch-nightly-cu129"
+url = "https://download.pytorch.org/whl/nightly/cu129"
 explicit = false
 
 [[tool.uv.index]]

version.txt

@@ -1 +1 @@
-2.8.0a0
+2.9.0a0