Skip to content

Add the causal mask fast path in flash attention#99

Merged
Connor1996 merged 1 commit intoskyzh:mainfrom
Connor1996:impl
Feb 24, 2026
Merged

Add the causal mask fast path in flash attention#99
Connor1996 merged 1 commit intoskyzh:mainfrom
Connor1996:impl

Conversation

@Connor1996
Copy link
Collaborator

@Connor1996 Connor1996 commented Feb 24, 2026

Add the causal mask fast path in flash attention

Signed-off-by: Connor1996 <zbk602423539@gmail.com>
@Connor1996 Connor1996 changed the title Ref flash attention: causal mask mode + block-level skip Add the causal mask fast path in flash attention Feb 24, 2026
@Connor1996 Connor1996 merged commit ff9cd45 into skyzh:main Feb 24, 2026
2 checks passed
@Connor1996 Connor1996 deleted the impl branch February 24, 2026 10:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant