diff --git a/tests/quantization/test_cutlass_w4a16.py b/tests/quantization/test_cutlass_w4a16.py index 7557e6fc8f3..b0ccb6c68a9 100644 --- a/tests/quantization/test_cutlass_w4a16.py +++ b/tests/quantization/test_cutlass_w4a16.py @@ -14,7 +14,7 @@ import torch from vllm.platforms import current_platform -if not current_platform.has_device_capability(90): +if not current_platform.has_device_capability(90) or current_platform.is_rocm(): pytest.skip( "Machete W4A16 requires Hopper (sm_90).", allow_module_level=True,