Skip to content

[XPU] support flashmask_attention #71573

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.

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

Merged
merged 4 commits into from
Mar 25, 2025
Merged

Conversation

runzhech
Copy link
Contributor

@runzhech runzhech commented Mar 11, 2025

PR Category

Custom Device

PR Types

New features

Description

support flashmask attention
参考https://github.com/PaddlePaddle/PaddleNLP/blob/develop/llm/docs/flashmask.md

Copy link

paddle-bot bot commented Mar 11, 2025

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

@paddle-bot paddle-bot bot added the XPU label Mar 11, 2025
@runzhech runzhech force-pushed the flashmask branch 5 times, most recently from 04def74 to 8036db4 Compare March 13, 2025 08:03
@runzhech runzhech force-pushed the flashmask branch 4 times, most recently from 6091105 to 927ad19 Compare March 20, 2025 02:12
@runzhech runzhech changed the title [XPU] support flashmask_attention forward [XPU] support flashmask_attention Mar 20, 2025
Copy link
Member

@tizhou86 tizhou86 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Copy link
Contributor

@ZibinGuo ZibinGuo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@houj04 houj04 merged commit 3de252b into PaddlePaddle:develop Mar 25, 2025
30 checks passed
@runzhech runzhech deleted the flashmask branch March 25, 2025 08:34
co63oc pushed a commit to co63oc/Paddle that referenced this pull request Mar 27, 2025
YqGe585 pushed a commit to YqGe585/Paddle that referenced this pull request May 7, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants