Skip to content

Conversation

@sneaxiy
Copy link
Collaborator

@sneaxiy sneaxiy commented Jan 8, 2024

PR types

Others

PR changes

Others

Description

Add unpadded GQA Flash-Attention support.

@paddle-bot
Copy link

paddle-bot bot commented Jan 8, 2024

你的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.

@sneaxiy sneaxiy changed the title Add unpadded gpa fa Support unpadded GQA FlashAttention Jan 12, 2024
@sneaxiy sneaxiy requested a review from ForFishes January 12, 2024 09:06
Copy link
Member

@ForFishes ForFishes left a comment

Choose a reason for hiding this comment

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

LGTM

@sneaxiy sneaxiy merged commit 0a55857 into PaddlePaddle:incubate/new_frl Jan 12, 2024
@sneaxiy sneaxiy deleted the unpadded_gqa_fa branch January 12, 2024 13:17
SylarTiaNII added a commit to SylarTiaNII/Paddle that referenced this pull request Feb 6, 2024
ForFishes pushed a commit that referenced this pull request Feb 7, 2024
* Revert "[BugFix] fix compile error when nccl of_fix_version enabled (#61537)"

This reverts commit e27a536.

* Revert "Support unbalance batchsize in 1F1B shcedule (#61379)"

This reverts commit c35796c.

* Revert "Update ncclCommInit function in fix version (#60947)"

This reverts commit ec5ca16.

* Revert "Support unpadded GQA FlashAttention (#60610)"

This reverts commit 0a55857.

* Revert "Modefied reduce op for store temp_data with MpType (#55709) (#60427)"

This reverts commit 51d97a6.
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.

2 participants