-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
【Hackathon 7th PPSCI No.13】 Backward decomp for put_along_axis_grad op #67891
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
// 如果没有特殊说明,要替换的位置全部指在x这个tensor上的对应位置 | ||
// 对应位置的计算见文件paddle\phi\kernels\funcs\gather_scatter_functor.cc的L293-L319 | ||
// 将要替换的位置全部赋值为0 |
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.
代码注释可以简化一下,并且用英文比较好
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.
这是参考Phi的kernel实现的版本,这个注释是方便老师阅读,后续合入前会删除的,老师看看是否有什么问题。
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.
这是参考Phi的kernel实现的版本,这个注释是方便老师阅读,后续合入前会删除的,老师看看是否有什么问题。
感觉没其他问题
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-CI-Inference 需要麻烦老师帮忙重跑一下,我这没有权限重跑_(:з」∠)_ |
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
PR Category
Operator Mechanism
PR Types
New features
Description
反向拆解put_along_axis_grad op