-
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 reduce sum+grad, mean+grad, min and max oneDNN kernels #45536
[PHI] Migrate reduce sum+grad, mean+grad, min and max oneDNN kernels #45536
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
@Aganlengzi A test in the Framework CI is failing and I have no way of debugging it locally. All tests related to converting the kernels to PHI are passing locally. Can you please suggest a solution? |
@chenwhql can you help review? |
We are analyzing this issue and will reply tomorrow |
…nto reduce_onednn_op_phi_migration
… reduce_onednn_op_phi_migration
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
@chenwhql Could you merge this as soon as the Framework CI passes? Coverage CI is complaining about some code not being tested, but I've double checked that there are Python tests that cover it. |
@jczaja Please review |
…addlePaddle#45536) * gaussian random * mkldnn to onednn renaming * fix merge conflicts * Migrate reduce_op oneDNN kernels to phi * Remove unnecessary header * remove fluid code * onednn renaming * Change std::vector<int64_t> to IntArray * Fix code style * Move classes from mkldnn_reuse.h to onednn_reuse.h * Move more functions from mkldnn_helper.h to onednn_helpper.h * Change MKLDNN to OneDNN in VLOG message * Implement reviewer suggestions Co-authored-by: Silv3S <slawomir.siwek@intel.com>
PR types
Others
PR changes
Others
Describe
Migrates the following oneDNN operator kernels to PHI: