[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:
co-authored by
Woosuk Kwon
parent
6a03954b45
commit
791f92135a
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user