-
Notifications
You must be signed in to change notification settings - Fork 5.9k
[Paddle Inference] Add multi_query_attention kernel #56432
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
Conversation
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into pr
|
你的PR提交成功,感谢你对开源项目的贡献! |
| import paddle.incubate.nn.functional as F | ||
|
|
||
| # input: [batch_size, num_head, dim_head] | ||
| q = paddle.rand(shape=(2, 32, 128), dtype="float32") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
例子建议给一个group=1和group=2的例子吧
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
|
@superxf 如果是内部同学,可以在 http://rdmp.dltp.baidu-int.com/data-management/github-user 登记下 |
|
|
|
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. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
可以参考 #56411 ,把 .h 文件消除掉
|
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. |
|
谢谢,已收到!
|
PR types
others
PR changes
OPs
Description
Pcard-74485