diff --git a/vllm/_aiter_ops.py b/vllm/_aiter_ops.py index 012a3f36798e..1663b2b6e40b 100644 --- a/vllm/_aiter_ops.py +++ b/vllm/_aiter_ops.py @@ -999,7 +999,9 @@ def is_fp8bmm_enabled(cls) -> bool: @classmethod @if_aiter_supported def is_fp4bmm_enabled(cls) -> bool: - return cls._AITER_ENABLED and cls._FP4BMM_ENABLED + from vllm.platforms.rocm import on_gfx950 + + return cls._AITER_ENABLED and cls._FP4BMM_ENABLED and on_gfx950() @classmethod @if_aiter_supported