-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[v1.8.x][BUGFIX] Fix MKLDNN BatchNorm with even number of channels (#19150) #19299 #19425 #19428
Conversation
Even number of channels results in data reordering before batch norm operation. Therefore, if BatchNorm data array is view of another array and the data is stored in MKLDNN format, the data needs to be converted to the default format.
Hey @akarbown , Thanks for submitting the PR
CI supported jobs: [edge, windows-gpu, sanity, unix-gpu, centos-gpu, website, clang, centos-cpu, miscellaneous, unix-cpu, windows-cpu] Note: |
@mxnet-bot run ci [unix-gpu] |
Jenkins CI successfully triggered : [unix-gpu] |
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. Thank you!
@samskalicky Do you want to confirm this is needed for the release? |
Bug fixes are still ok until we finish the license issues. Can someone confirm this fix has already been ported to the 1.x branch already before we merge this? |
I'll create PR with that fix for 1.x today. |
@mxnet-bot run ci [unix-gpu, windows-gpu] |
Jenkins CI successfully triggered : [unix-gpu, windows-gpu] |
Description
It's the change (#19299) ported form mxnet 1.7.
Checklist
Essentials
Changes
Comments