Skip to content

Conversation

@superxf
Copy link
Contributor

@superxf superxf commented Aug 18, 2023

PR types

others

PR changes

OPs

Description

  • Support masked multiquery attention for transformer decoder stage.
  • Support dtype:[float | float16 | bfloat16]
  • export python api: from paddle.incubate.nn.functional import masked_multiquery_attention

Pcard-74485

@paddle-bot
Copy link

paddle-bot bot commented Aug 18, 2023

你的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 contributor External developers status: proposed labels Aug 18, 2023
@superxf superxf changed the title [Paddle Inference] Add mulit_query_attention kernel [Paddle Inference] Add multi_query_attention kernel Aug 18, 2023
@vivienfanghuagood vivienfanghuagood requested review from carryyu and xiaoxiaohehe001 and removed request for xiaoxiaohehe001 August 21, 2023 08:27
import paddle.incubate.nn.functional as F

# input: [batch_size, num_head, dim_head]
q = paddle.rand(shape=(2, 32, 128), dtype="float32")
Copy link
Contributor

Choose a reason for hiding this comment

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

例子建议给一个group=1和group=2的例子吧

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Done

@luotao1
Copy link
Contributor

luotao1 commented Aug 21, 2023

@superxf 如果是内部同学,可以在 http://rdmp.dltp.baidu-int.com/data-management/github-user 登记下

@CLAassistant
Copy link

CLAassistant commented Aug 22, 2023

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ superxf
❌ lizhenyun


lizhenyun seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

Copy link
Contributor

Choose a reason for hiding this comment

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

可以参考 #56411 ,把 .h 文件消除掉

@paddle-ci-bot
Copy link

paddle-ci-bot bot commented Sep 1, 2023

Sorry to inform you that 22747b5's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually.

@paddle-bot paddle-bot bot closed this Sep 3, 2024
@carryyu
Copy link
Contributor

carryyu commented Sep 3, 2024 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

contributor External developers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants