Skip to content

[CUDA] Update Flash Attention to support head_sink for smooth softmax in GQA #25358

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
tianleiwu wants to merge 2 commits into from

add test

f8cd59e
Select commit
Loading
Failed to load commit list.
Closed

[CUDA] Update Flash Attention to support head_sink for smooth softmax in GQA #25358

add test
f8cd59e
Select commit
Loading
Failed to load commit list.
Azure Pipelines / Linux Android Emulator QNN CI Pipeline succeeded Jul 10, 2025 in 12m 48s

Build #20250710.30 succeeded

0 errors / 0 warnings

View more details on Azure Pipelines