fix(rocm): add early return in get_flash_attn_version for ROCm (#31286)

Signed-off-by: rabi <ramishra@redhat.com>
This commit is contained in:
Rabi Mishra
2026-01-08 07:58:07 +05:30
committed by GitHub
parent 25eef3dc2e
commit 39d82005f7

View File

@@ -40,6 +40,9 @@ def get_flash_attn_version(requires_alibi: bool = False) -> int | None:
if current_platform.is_xpu():
return 2
if current_platform.is_rocm():
# ROCm doesn't use vllm_flash_attn; return None to skip fa_version arg
return None
try:
from vllm.vllm_flash_attn.flash_attn_interface import (
fa_version_unsupported_reason,