Skip to content

Commit 42528b0

Browse files
committed
Add further skips to test
Signed-off-by: Antoni Viros i Martin <[email protected]>
1 parent 2b56e0f commit 42528b0

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/aiu_addons/test_fp8_addon.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import torch
1919

2020
# Local
21+
from fms_mo.prep import available_packages
2122
import fms_mo.aiu_addons.fp8.fp8_spyre_op # pylint: disable=unused-import
2223

2324

@@ -33,6 +34,10 @@ def test_fp8_registration() -> None:
3334

3435

3536
# This test requires an H100 or higher GPU to run
37+
@pytest.mark.skipif(
38+
not available_packages["torchao"] or not available_packages["fms"],
39+
reason="FMS and torchao required to run this test",
40+
)
3641
@pytest.mark.skipif(
3742
not torch.cuda.is_available()
3843
or (torch.cuda.is_available() and torch.cuda.get_device_capability() < (8, 9)),

0 commit comments

Comments
 (0)