diff --git a/torchao/__init__.py b/torchao/__init__.py index 6bf616e48e..be9bf5e824 100644 --- a/torchao/__init__.py +++ b/torchao/__init__.py @@ -35,7 +35,7 @@ # dumped)". # TODO(#2901, and before next torchao release): make this generic for # future torchao and torch versions - if __version__.startswith("0.13.0") and torch.__version__ > "2.8": + if __version__.startswith("0.13.0") and torch.__version__ >= "2.9": logger.warning( f"Skipping import of cpp extensions due to incompatible torch version {torch.__version__} for torchao version {__version__}" )