Skip to content

Commit 900cf5b

Browse files
author
Peter Y. Yeh
committed
Add skip_if_rocm decorator to test_float8_utils
Add ROCm skip decorator to prevent test failures during ongoing ROCm enablement
1 parent 27d0d48 commit 900cf5b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/float8/test_float8_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
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
7+
from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, skip_if_rocm
88

99
if not TORCH_VERSION_AT_LEAST_2_5:
1010
pytest.skip("Unsupported PyTorch version", allow_module_level=True)
@@ -30,6 +30,7 @@
3030
# ("largest subnormal number", [2**-126 * (1 - 2**-23), 1.1754943508222875e-38]),
3131
],
3232
)
33+
@skip_if_rocm("ROCm enablement in progress")
3334
def test_round_scale_down_to_power_of_2_valid_inputs(
3435
test_case: dict,
3536
):

0 commit comments

Comments
 (0)