diff --git a/torchao/__init__.py b/torchao/__init__.py index f0f7d05e9a..90694d1f15 100644 --- a/torchao/__init__.py +++ b/torchao/__init__.py @@ -76,10 +76,12 @@ def _parse_version(version_string): # Built against torch 2.9.0 (_parse_version("0.14.1"), _parse_version("2.9.0")), (_parse_version("0.14.1"), _parse_version("2.10.0.dev")), + # Built against torch 2.9.1 + (_parse_version("0.15.0"), _parse_version("2.9.1")), + (_parse_version("0.15.0"), _parse_version("2.10.0.dev")), # Current torchao version - (_parse_version("0.15.0.dev"), _parse_version("2.9.0")), - (_parse_version("0.15.0.dev"), _parse_version("2.9.1")), - (_parse_version("0.15.0.dev"), _parse_version("2.10.0.dev")), + (_parse_version("0.16.0.dev"), _parse_version("2.9.1")), + (_parse_version("0.16.0.dev"), _parse_version("2.10.0.dev")), ] current_torch_version = _parse_version(torch.__version__)