skip torch versions < 2.5

pytorch · Feb 7, 2025 · 4baed04 · 4baed04
1 parent 77d004e
commit 4baed04
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/float8/test_float8_utils.py b/test/float8/test_float8_utils.py
@@ -4,6 +4,10 @@
 import torch
 
 from torchao.float8.float8_utils import _round_scale_down_to_power_of_2
+from torchao.utils import TORCH_VERSION_AT_LEAST_2_5
+
+if not TORCH_VERSION_AT_LEAST_2_5:
+    pytest.skip("Unsupported PyTorch version", allow_module_level=True)
 
 
 # source for notable single-precision cases: