[Attention] Set ALWAYS cudagraph support for FA4 (#60908)

Signed-off-by: Giancarlo Delfin <gdelfin@inferact.ai>
Co-authored-by: Woosuk Kwon <woosuk@inferact.ai>
This commit is contained in:
Giancarlo Delfin
2026-10-13 16:22:52 -07:00
committed by GitHub
co-authored by Woosuk Kwon
parent 6a03954b45
commit 791f92135a
+2 -2
View File
@@ -593,7 +593,7 @@ def _maybe_symmetrize_window(
class FlashAttentionMetadataBuilder(AttentionMetadataBuilder[FlashAttentionMetadata]):
# FA3:
# FA3/FA4:
# Supports full cudagraphs for all cases.
#
# FA2:
@@ -613,7 +613,7 @@ class FlashAttentionMetadataBuilder(AttentionMetadataBuilder[FlashAttentionMetad
# https://github.com/vllm-project/vllm/issues/22945
_cudagraph_support = (
AttentionCGSupport.ALWAYS
if get_flash_attn_version() == 3
if get_flash_attn_version() in (3, 4)
else AttentionCGSupport.UNIFORM_BATCH
)
supports_update_block_table: bool = True