-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
Add fused_linear_param_grad_add_kernel #51805
Add fused_linear_param_grad_add_kernel #51805
Conversation
e69115a
to
115cfb1
Compare
… add_fused_linear_param_grad_add
…sneaxiy/Paddle into add_fused_linear_param_grad_add
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.
LGTM
#include "paddle/phi/kernels/fusion/fused_linear_param_grad_add_kernel.h" | ||
|
||
#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11060 | ||
#include "paddle/fluid/operators/fused/fused_gemm_epilogue_op.h" |
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.
这里不能在include fluid头文件了,已经在编译单独的so了
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.
线下讨论,将在下一个PR里修复。感谢指出。
PR types
Performance optimization
PR changes
OPs
Describe
Add
fused_linear_param_grad_add_kernel
for GPT performance optimization.