-
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
[PHI] Migrate sgd and stack oneDNN kernels #46374
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
@chenwhql Please review |
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.
@piotrekobi To prevent maintenance of duplicate codes, axpy_handler.h/cc
should be deleted in Fluid,may be you need do it in next pr, thanks.
@YuanRisheng I left it there for now because paddle/fluid/operators/math/selected_rows_functor.cc uses it and I had to modify the PHI version slightly so it's not compatible with it. Once selected_rows_functor.cc gets deleted or migrated axpy_handler.h/cc can be deleted from fluid. |
* Convert slice+grad oneDNN fluid kernels to PHI * Change mutable_data to Alloc * Refactor licences
* Convert slice+grad oneDNN fluid kernels to PHI * Change mutable_data to Alloc * Refactor licences
PR types
Others
PR changes
Others
Describe
Migrates the following oneDNN operator fluid kernels to PHI: