fix bn layout network config caching issue

ROCm / MIOpen

AMD's Machine Intelligence Library

Other

1.09k stars 231 forks source link

Closed bghimireamd closed 3 months ago

bghimireamd commented 3 months ago

pytorch's batch norm was failing because incorrect layout (NCHW was being cached when NHWC). This PR is to fix the issue.

aserio commented 3 months ago

@junliume @DrizztDoUrden, can we get this PR reviewed quickly? It is blocking a feature that needs to be merged ASAP.