From b6ce4f8b2fa85304cc3b95299d82212e90c663d7 Mon Sep 17 00:00:00 2001
From: Lv Mengsi <ceci3@users.noreply.github.com>
Date: Tue, 26 Nov 2019 16:20:33 +0800
Subject: [PATCH] Fix mistake of batch norm op (#21237)

* fix_bn

* revert unittest,test=develop
---
 paddle/fluid/operators/batch_norm_op.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/batch_norm_op.cu b/paddle/fluid/operators/batch_norm_op.cu
index 95d7f23b2c..4b4fcd487c 100644
--- a/paddle/fluid/operators/batch_norm_op.cu
+++ b/paddle/fluid/operators/batch_norm_op.cu
@@ -418,7 +418,7 @@ class BatchNormGradKernel<platform::CUDADeviceContext, T>
           }
         } else {
           if (d_x) {
-            BNBackwardData<T, block, framework::DataLayout::kNCHW><<<
+            BNBackwardData<T, block, framework::DataLayout::kNHWC><<<
                 grid2, block, 0, dev_ctx.stream()>>>(
                 d_y->data<T>(), scale->data<BatchNormParamType<T>>(),
                 saved_mean_data, x->data<T>(), saved_var_data, C, N, H * W * D,
-- 
GitLab