From 2f5582e1f925f3ff4fac2356d81f64503f06b651 Mon Sep 17 00:00:00 2001 From: Dmitry Nikolaev Date: Wed, 3 Sep 2025 19:44:36 +0000 Subject: [PATCH] Fix condition to enable NHWC Batchnorm --- aten/src/ATen/native/Normalization.cpp | 2 -- 1 file changed, 2 deletions(-) diff --git a/aten/src/ATen/native/Normalization.cpp b/aten/src/ATen/native/Normalization.cpp index b9cf3c907f09f..4550a2a2e2257 100644 --- a/aten/src/ATen/native/Normalization.cpp +++ b/aten/src/ATen/native/Normalization.cpp @@ -541,8 +541,6 @@ BatchNormBackend _select_batch_norm_backend( && weight.defined() && bias.defined() && ((running_mean.defined() && running_var.defined()) || (!running_mean.defined() && !running_var.defined() && training)) - && input.suggest_memory_format() != MemoryFormat::ChannelsLast - && input.suggest_memory_format() != MemoryFormat::ChannelsLast3d && (input.suggest_memory_format() == MemoryFormat::Contiguous #if (defined(USE_ROCM) && ROCM_VERSION >= 60500) || (input.suggest_memory_format() == MemoryFormat::ChannelsLast && PYTORCH_MIOPEN_SUGGEST_NHWC_BATCHNORM)