We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 92d6dd8 commit e0afc3aCopy full SHA for e0afc3a
aten/src/ATen/native/Normalization.cpp
@@ -524,7 +524,8 @@ BatchNormBackend _select_batch_norm_backend(
524
// TODO: Remove PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM once ROCm officially supports NHWC in MIOpen
525
// See #64427
526
// non static variable is used to be able to change environment variable in runtime for testing
527
- bool PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM = c10::utils::check_env("PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM").value_or(false);
+ // enabled by default for ROCm >= 7.0.0
528
+ bool PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM = c10::utils::check_env("PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM").value_or(ROCM_VERSION >= 70000);
529
530
if (
531
input.is_cuda()
0 commit comments