-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[BUGFIX] Fix MKLDNN BatchNorm with even number of channels (#19150) #19299 #19425
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: [website, windows-gpu, windows-cpu, sanity, clang, unix-cpu, centos-cpu, unix-gpu, centos-gpu, miscellaneous, edge] Note: |
@mxnet-bot run ci [unix-cpu] |
@mxnet-bot run ci [greeting, unix-cpu] |
Jenkins CI successfully triggered : [unix-cpu] |
1 similar comment
Jenkins CI successfully triggered : [unix-cpu] |
…19150) #19299 #19425 (#19428) * Fix MKLDNN BatchNorm with even number of channels (#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
…) #19299 #19425 (#19445) * Fix MKLDNN BatchNorm with even number of channels (#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
@TaoLv, @wkcn, @samskalicky - could you please take a look at this change & review it!? (it's the same change as for the 1.x, 1.7 & 1.8 branches) |
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 @akarbown for porting this fix!
) apache#19299 (apache#19425) * Fix MKLDNN BatchNorm with even number of channels (apache#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
) apache#19299 (apache#19425) * Fix MKLDNN BatchNorm with even number of channels (apache#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
) apache#19299 (apache#19425) * Fix MKLDNN BatchNorm with even number of channels (apache#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
…he#19150) apache#19299 apache#19425 (apache#19445) * Fix MKLDNN BatchNorm with even number of channels (apache#19150) 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. * Add or updated test to verify Batchnorm odd & even number of channels * Fix for Batchnorm odd & even chnls number context
Description
It's the change (#19299) ported form mxnet 1.7.
Checklist
Essentials