diff --git a/cpp/include/torch_tensorrt/macros.h b/cpp/include/torch_tensorrt/macros.h index 90fab3afcc..bdb0a30115 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 5 +#define TORCH_TENSORRT_MINOR_VERSION 6 #define TORCH_TENSORRT_PATCH_VERSION 0 #define TORCH_TENSORRT_VERSION \ STR(TORCH_TENSORRT_MAJOR_VERSION) \ diff --git a/py/requirements.txt b/py/requirements.txt index 15d359fbc3..361afab365 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -2,7 +2,7 @@ numpy packaging pybind11==2.6.2 --extra-index-url https://download.pytorch.org/whl/nightly/cu124 -torch>=2.5.0.dev,<2.6.0 +torch>=2.6.0.dev,<2.7.0 torchvision>=0.20.0.dev,<0.21.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml diff --git a/py/torch_tensorrt/dynamo/lowering/_decompositions.py b/py/torch_tensorrt/dynamo/lowering/_decompositions.py index 8c391afa5b..534bc3eac5 100644 --- a/py/torch_tensorrt/dynamo/lowering/_decompositions.py +++ b/py/torch_tensorrt/dynamo/lowering/_decompositions.py @@ -3,7 +3,7 @@ from typing import Any, Callable, Dict, List, Optional import torch -from torch._decomp import register_decomposition +from torch._decomp import _decomp_table_to_post_autograd_aten, register_decomposition from torch._ops import OpOverload from torch_tensorrt.dynamo._defaults import default_device from torch_tensorrt.dynamo.conversion.converter_utils import get_positive_dim @@ -394,4 +394,9 @@ def get_decompositions( } return {**CORE_ATEN_DECOMPOSITIONS_FILTERED, **TORCH_TRT_DECOMPOSITIONS} else: - return {**ENABLED_TORCH_DECOMPOSITIONS, **TORCH_TRT_DECOMPOSITIONS} + # changes made here due to torch2.6 changes https://github.com/pytorch/pytorch/pull/135080 + return { + **ENABLED_TORCH_DECOMPOSITIONS, + **_decomp_table_to_post_autograd_aten(), + **TORCH_TRT_DECOMPOSITIONS, + } diff --git a/pyproject.toml b/pyproject.toml index 96bb6838ec..1284e458f4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ requires = [ "typing-extensions>=4.7.0", "future>=0.18.3", "tensorrt-cu12==10.3.0", - "torch>=2.5.0.dev,<2.6.0", + "torch>=2.6.0.dev,<2.7.0", "pybind11==2.6.2", "numpy", ] @@ -54,7 +54,7 @@ keywords = [ "inference", ] dependencies = [ - "torch>=2.5.0.dev,<2.6.0", + "torch>=2.6.0.dev,<2.7.0", "tensorrt-cu12==10.3.0", "tensorrt-cu12-bindings==10.3.0", "tensorrt-cu12-libs==10.3.0", diff --git a/version.txt b/version.txt index b8feefb940..3d87ca93f8 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -2.5.0a0 +2.6.0a0