Skip to content

Commit 4baed04

Browse files
skip torch versions < 2.5
1 parent 77d004e commit 4baed04

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/float8/test_float8_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44
import torch
55

66
from torchao.float8.float8_utils import _round_scale_down_to_power_of_2
7+
from torchao.utils import TORCH_VERSION_AT_LEAST_2_5
8+
9+
if not TORCH_VERSION_AT_LEAST_2_5:
10+
pytest.skip("Unsupported PyTorch version", allow_module_level=True)
711

812

913
# source for notable single-precision cases:

0 commit comments

Comments
 (0)