diff --git a/cpp/include/torch_tensorrt/macros.h b/cpp/include/torch_tensorrt/macros.h index 7c37368054..245aa9acb1 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 3 +#define TORCH_TENSORRT_MINOR_VERSION 4 #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 02faa26e05..371c886128 100644 --- a/dev_dep_versions.yml +++ b/dev_dep_versions.yml @@ -1,4 +1,4 @@ -__version__: "2.3.0.dev0" +__version__: "2.4.0.dev0" __cuda_version__: "12.1" __cudnn_version__: "8.9" __tensorrt_version__: "10.0.0.6" diff --git a/pyproject.toml b/pyproject.toml index c307381ef4..d7df99d67b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,7 +8,7 @@ requires = [ "cffi>=1.15.1", "typing-extensions>=4.7.0", "future>=0.18.3", - "tensorrt", + "tensorrt==10.0.0b6", "torch >=2.4.0.dev,<2.5.0", "pybind11==2.6.2", "numpy", diff --git a/version.txt b/version.txt index 00b35529e0..e96f44fb3e 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -2.3.0a0 +2.4.0a0