Skip to content
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

[romc6.4_internal_testing] avoid redundant NHWC-NCHW-NHWC conversions for MiopenBatchNormBackward #1948

Draft
wants to merge 1 commit into
base: rocm6.4_internal_testing
Choose a base branch
from

Conversation

dnikolaev-amd
Copy link

@dnikolaev-amd dnikolaev-amd commented Mar 7, 2025

Pytorch called grad.contiguous() before miopen_batch_norm_backward. This led to redundant conversions from NHWC gradient to NCHW and back to NHWC.
Solution based on cudnn_batch_norm_backward description in the same derivatives.yaml file
Fixes SWDEV-520250

Before:
image
After:
image

@rocm-repo-management-api
Copy link

rocm-repo-management-api bot commented Mar 7, 2025

Jenkins build for 605b99b8ef24cf83aca53ffe95fcdea31776f722 commit finished as FAILURE
Links: Blue Ocean view / Build artifacts

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant