chore: cutting release branch for 2.1.0

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

Author: narendasan

README.md

@@ -115,8 +115,8 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedd
 
 These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
 
-- Bazel 5.2.0
-- Libtorch 2.2.0.dev (latest nightly) (built with CUDA 12.1)
+- Bazel 6.2.1
+- Libtorch 2.1.0
 - CUDA 12.1
 - cuDNN 8.8.1
 - TensorRT 8.6.1

WORKSPACE

@@ -54,14 +54,14 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu121/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/cu121/libtorch-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu121/libtorch-shared-with-deps-latest.zip"],
 )
 
 # Download these tarballs manually from the NVIDIA website

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 2
+#define TORCH_TENSORRT_MINOR_VERSION 1
 #define TORCH_TENSORRT_PATCH_VERSION 0
 #define TORCH_TENSORRT_VERSION      \
   STR(TORCH_TENSORRT_MAJOR_VERSION) \

dev_dep_versions.yml

@@ -1,4 +1,3 @@
-__version__: "2.2.0.dev0"
 __cuda_version__: "12.1"
 __cudnn_version__: "8.8"
 __tensorrt_version__: "8.6"

py/requirements.txt

@@ -2,8 +2,8 @@ numpy
 packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu121
-torch>=2.2.0.dev,<2.3.0
-torchvision>=0.16.0.dev,<0.17.0
+torch>=2.1.0,<2.2.0
+torchvision>=0.16.0,<0.17.0
 --extra-index-url https://pypi.ngc.nvidia.com
 tensorrt==8.6.1
 pyyaml

pyproject.toml

@@ -9,8 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt>=8.6,<8.7",
-    "torch >=2.2.0.dev,<2.3.0",
-    #"torch==2.1.0.dev20230731",
+    "torch>=2.1.0,<2.2.0",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -42,8 +41,7 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
 requires-python = ">=3.8"
 keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
 dependencies = [
-    "torch >=2.2.0.dev,<2.3.0",
-    #"torch==2.1.0.dev20230731",
+    "torch>=2.1.0,<2.2.0",
     "tensorrt>=8.6,<8.7",
     "packaging>=23",
     "numpy",
@@ -52,7 +50,7 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
-torchvision = ["torchvision >=0.16.dev,<0.17.0"]
+torchvision = ["torchvision>=0.16.0,<0.17.0"]
 
 [project.urls]
 Homepage = "https://pytorch.org/tensorrt"

toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl

@@ -59,14 +59,14 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu121/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/cu121/libtorch-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/test/cu121/libtorch-shared-with-deps-latest.zip"],
 )
 
 ####################################################################################

version.txt

@@ -1 +1 @@
-2.2.0a0
+2.1.0a0