diff --git a/test/prototype/test_low_bit_optim.py b/test/prototype/test_low_bit_optim.py index d386f3210d..5ce3d08b81 100644 --- a/test/prototype/test_low_bit_optim.py +++ b/test/prototype/test_low_bit_optim.py @@ -43,6 +43,8 @@ except ImportError: lpmm = None +if torch.version.hip is not None: + pytest.skip("Skipping the test in ROCm", allow_module_level=True) _DEVICES = get_available_devices()