Skip to content
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

[cherry pick] add op: fused_feedforward(backward) #36730

Conversation

zkh2016
Copy link
Contributor

@zkh2016 zkh2016 commented Oct 26, 2021

PR types

Others

PR changes

OPs

Describe

This is a fusion operator to compute feed forward layer in transformer model architecture.

reference PR:
add op: fused_feedforward(backward)
Move fused_attention and fused_feedforward functional api path to incubate

@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

zhangkaihuo and others added 2 commits October 26, 2021 08:38
这个PR是fused_feedforward反向的代码

相关kernel实现:fused_dropout_act_bias, fused_residual_dropout_bias, fused_layernorm_residual_dropout_bias

fused_feedforward是一个融合算子,该算子对transformer模型的feed forward层的算子进行融合和封装,使得前端只呈现一个接口,通过融合减少部分访存和kernel launch的时间,以此提升性能。
@zkh2016 zkh2016 force-pushed the release/2.2-cherry-pick-FFN_backward branch from 1201f5e to 4fd7cd2 Compare October 26, 2021 08:40
@lanxianghit lanxianghit merged commit 76c1bae into PaddlePaddle:release/2.2 Oct 26, 2021
@zkh2016 zkh2016 deleted the release/2.2-cherry-pick-FFN_backward branch August 19, 2022 04:03
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.

5 participants