We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2b56e0f commit 42528b0Copy full SHA for 42528b0
tests/aiu_addons/test_fp8_addon.py
@@ -18,6 +18,7 @@
18
import torch
19
20
# Local
21
+from fms_mo.prep import available_packages
22
import fms_mo.aiu_addons.fp8.fp8_spyre_op # pylint: disable=unused-import
23
24
@@ -33,6 +34,10 @@ def test_fp8_registration() -> None:
33
34
35
36
# 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
+)
41
@pytest.mark.skipif(
42
not torch.cuda.is_available()
43
or (torch.cuda.is_available() and torch.cuda.get_device_capability() < (8, 9)),
0 commit comments