From 39d82005f7a58fed901526a82bd83ff91a60ea0c Mon Sep 17 00:00:00 2001
From: Rabi Mishra <ramishra@redhat.com>
Date: Thu, 8 Jan 2026 07:58:07 +0530
Subject: [PATCH] fix(rocm): add early return in get_flash_attn_version for
 ROCm (#31286)

Signed-off-by: rabi <ramishra@redhat.com>
---
 vllm/attention/utils/fa_utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/vllm/attention/utils/fa_utils.py b/vllm/attention/utils/fa_utils.py
index 189bf3d4f..00107cd7f 100644
--- a/vllm/attention/utils/fa_utils.py
+++ b/vllm/attention/utils/fa_utils.py
@@ -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,