-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Fix ElemwiseSum for more than 4 inputs #17995
Conversation
Hey @ptrendx , Thanks for submitting the PR
CI supported jobs: [windows-cpu, miscellaneous, unix-gpu, edge, windows-gpu, website, sanity, centos-cpu, clang, centos-gpu, unix-cpu] Note: |
Should we add a test case in the style of #17989 (comment) to ensure no such bug makes it into the codebase again? |
@ptrendx I can confirm that this fixes the issue. |
confirmed it can fix #16708 (comment) as well |
@ptrendx thanks for the fix. Let's add a test for this. |
@mxnet-bot run ci [centos-gpu, unix-gpu, windows-gpu] |
Jenkins CI successfully triggered : [unix-gpu, centos-gpu, windows-gpu] |
Added v1.x label even though this PR is targeted for master in order to track that we want it included in 1.7 release @ciyongch FYI. |
* Fix ElemwiseSum for more than 4 inputs * Added test
* Fix ElemwiseSum for more than 4 inputs * Added test
Description
Fixes #17989
It was caused by a bug in
ElemwiseSum
which for more than 4 inputs andkAddTo
req was counting gradients multiple times.@sxjscience @zhreshold Please test if this fixes the issues you saw.