From 6e0297c883c4d02addd1cfc66ed7812fec334f4f Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Wed, 6 Sep 2023 16:14:45 -0700
Subject: [PATCH] fix: Torch Upgrade to 2.2.0.dev

---
 README.md                           | 2 +-
 cpp/include/torch_tensorrt/macros.h | 2 +-
 dev_dep_versions.yml                | 2 +-
 py/requirements.txt                 | 4 ++--
 pyproject.toml                      | 6 +++---
 version.txt                         | 2 +-
 6 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index a40b7cf755..44cc7c83e0 100644
--- a/README.md
+++ b/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
diff --git a/cpp/include/torch_tensorrt/macros.h b/cpp/include/torch_tensorrt/macros.h
index 0ad6159704..1eb8c6feb5 100644
--- a/cpp/include/torch_tensorrt/macros.h
+++ b/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) \
diff --git a/dev_dep_versions.yml b/dev_dep_versions.yml
index 8a8202ec04..874a27cbcd 100644
--- a/dev_dep_versions.yml
+++ b/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"
diff --git a/py/requirements.txt b/py/requirements.txt
index 0116b23c4f..09b9a39aa3 100644
--- a/py/requirements.txt
+++ b/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
\ No newline at end of file
+pyyaml
diff --git a/pyproject.toml b/pyproject.toml
index 3973f1fdb9..27af0da9cd 100644
--- a/pyproject.toml
+++ b/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"
\ No newline at end of file
+follow_imports = "skip"
diff --git a/version.txt b/version.txt
index 32408126ec..887948350c 100644
--- a/version.txt
+++ b/version.txt
@@ -1 +1 @@
-2.0.0a0
\ No newline at end of file
+2.2.0a0