fix: Torch Upgrade to 2.2.0.dev

Author: gs-olive

README.md

@@ -116,7 +116,7 @@ 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.1.0.dev20230703 (built with CUDA 12.1)
+- Libtorch 2.2.0.dev (latest nightly) (built with CUDA 12.1)
 - CUDA 12.1
 - cuDNN 8.8.1
 - TensorRT 8.6.1

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

dev_dep_versions.yml

@@ -1,4 +1,4 @@
-__version__: "2.0.0.dev0"
+__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.1.0.dev,<2.2.0
+torch>=2.2.0.dev,<2.3.0
 torchvision>=0.16.0.dev,<0.17.0
 --extra-index-url https://pypi.ngc.nvidia.com
 tensorrt==8.6.1
-pyyaml
+pyyaml

pyproject.toml

@@ -9,7 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt>=8.6,<8.7",
-    "torch >=2.1.0.dev,<2.2.0",
+    "torch >=2.2.0.dev,<2.3.0",
     #"torch==2.1.0.dev20230731",
     "pybind11==2.6.2",
     "numpy",
@@ -42,7 +42,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.1.0.dev,<2.2.0",
+    "torch >=2.2.0.dev,<2.3.0",
     #"torch==2.1.0.dev20230731",
     "tensorrt>=8.6,<8.7",
     "packaging>=23",
@@ -217,4 +217,4 @@ disallow_untyped_calls = false
 [[tool.mypy.overrides]]
 module = "torch_tensorrt.fx.*"
 ignore_errors = true
-follow_imports = "skip"
+follow_imports = "skip"

version.txt

@@ -1 +1 @@
-2.0.0a0
+2.2.0a0