push

Author: msaroufim

torchao/utils.py

@@ -22,6 +22,12 @@
     "TORCH_VERSION_AT_LEAST_2_3",
     "TORCH_VERSION_AT_LEAST_2_4",
     "TORCH_VERSION_AT_LEAST_2_5",
+
+    # Needs to be deprecated in the future
+    "TORCH_VERSION_AFTER_2_2",
+    "TORCH_VERSION_AFTER_2_3",
+    "TORCH_VERSION_AFTER_2_4",
+    "TORCH_VERSION_AFTER_2_5",
 ]
 
 
@@ -301,3 +307,13 @@ def torch_version_at_least(min_version):
 TORCH_VERSION_AT_LEAST_2_4 = torch_version_at_least("2.4.0")
 TORCH_VERSION_AT_LEAST_2_3 = torch_version_at_least("2.3.0")
 TORCH_VERSION_AT_LEAST_2_2 = torch_version_at_least("2.2.0")
+
+
+## Deprecated, will be deleted in the future
+def _torch_version_at_least(min_version):
+    return is_fbcode() or version("torch") >= min_version
+
+TORCH_VERSION_AFTER_2_5 = _torch_version_at_least("2.5.0.dev")
+TORCH_VERSION_AFTER_2_4 = _torch_version_at_least("2.4.0.dev")
+TORCH_VERSION_AFTER_2_3 = _torch_version_at_least("2.3.0.dev")
+TORCH_VERSION_AFTER_2_2 = _torch_version_at_least("2.2.0.dev")