-
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
[Bug fix] Add new unittests for gIOHW format in conv_transpose_mkldnn_op #37344
Conversation
✅ This PR's description meets the template requirements! |
Thanks for your contribution! |
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.
I reran some CI's for you as they seemed to be failing for weird reasons. LGTM.
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
@jczaja please merge |
…ePaddle#37344) * Add new unittests * Replace I with O channel for filter groups * Undo changes affecting other operators * Fix oneDNN namespace typo * Fix code format error
PR types
Bug fixes
PR changes
OPs
Describe
test_conv2d_transpose_mkldnn_op did not cover any case with grouped filter weights. Added two cases covering it. Tests failed because there was bug related to wrong channel division.
Example:
IOHW = [4, 3, 3, 3]
gOIHW = [2, 1, 4, 3, 3] (bug)
gOIHW = [2, 3, 2, 3, 3] (expected)