-
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
Replace EigenBroadcast with ElementwiseBroadcast in ReduceGrad #38959
Replace EigenBroadcast with ElementwiseBroadcast in ReduceGrad #38959
Conversation
Thanks for your contribution! |
1b0ecb6
to
f634c0b
Compare
b75a790
to
e445517
Compare
d5a65d1
to
184261f
Compare
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 for shareDataWith
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 for ci op benchmark
PR types
Performance optimization
PR changes
OPs
Describe
Reduce EigenBroadcastcase with ElementwiseBroadcast in ReduceGrad
为扩大KP算子覆盖率,统一将Reduce_sum/mean 反向的Eigen适配代码替换为ElementwiseBroadcast Kernel
ReduceGrad 性能统计对比:
benchmark 异常说明,与本次PR修改无关,本地测试无性能影响: