-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[BUGFIX] Add check to make sure num_group is non-zero #20186
Conversation
Hey @Adnios , Thanks for submitting the PR
CI supported jobs: [clang, edge, unix-gpu, website, centos-gpu, centos-cpu, windows-gpu, unix-cpu, windows-cpu, miscellaneous, sanity] Note: |
@mxnet-bot run ci [all] |
@mxnet-bot run ci [windows-cpu] |
1 similar comment
@mxnet-bot run ci [windows-cpu] |
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.
Thanks for the fix
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
@mxnet-bot run ci [windows-gpu] |
@Adnios thank you for the fix |
* add check for group not equal zero * num_group in convolution must be positive
Description
Related issue: #19921
When
group=0
, themxnet.gluon.nn.Conv1D(Conv2D, Conv3D)
will cause crashes(floating point exception)MxNet doesn't seem to have checked whether
group=0
. This pr adds check for group not equal zero insrc/operator/nn/convolution.cc
. Now the error message will be like this.We can use the following code to test.
Checklist
Essentials
Changes
Comments