From aba2713ffcc623caa30b0fbc40e07592b4625f72 Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Wed, 20 Mar 2019 14:00:34 +0800 Subject: [PATCH] fix comment. test=develop --- paddle/fluid/operators/batch_norm_op.cu | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/paddle/fluid/operators/batch_norm_op.cu b/paddle/fluid/operators/batch_norm_op.cu index 3256451655..8f3691dd15 100644 --- a/paddle/fluid/operators/batch_norm_op.cu +++ b/paddle/fluid/operators/batch_norm_op.cu @@ -79,11 +79,11 @@ class BatchNormKernel // TODO(dengkaipeng): use PERSISTENT mode in training may incur errors // in inference period, cuDNN fixed issues on PERSISTENT mode in version // 7.0.2, 7.0.4 and 7.3.0, we disable this mode currently. -/* #if CUDNN_VERSION_MIN(7, 0, 0) */ -/* mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT; */ -/* #else */ + // #if CUDNN_VERSION_MIN(7, 0, 0) + // mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT; + // #else mode_ = CUDNN_BATCHNORM_SPATIAL; -/* #endif */ + // #endif VLOG(3) << "Setting descriptors."; std::vector dims; @@ -306,14 +306,14 @@ class BatchNormGradKernel } epsilon = std::max(epsilon, CUDNN_BN_MIN_EPSILON); - // TODO(dengkaipeng): use PERSISTENT mode in training may incur errors - // in inference period, cuDNN fixed issues on PERSISTENT mode in version - // 7.0.2, 7.0.4 and 7.3.0, we disable this mode currently. -/* #if CUDNN_VERSION_MIN(7, 0, 0) */ -/* mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT; */ -/* #else */ + // TODO(dengkaipeng): use PERSISTENT mode in training may incur errors + // in inference period, cuDNN fixed issues on PERSISTENT mode in version + // 7.0.2, 7.0.4 and 7.3.0, we disable this mode currently. + // #if CUDNN_VERSION_MIN(7, 0, 0) + // mode_ = CUDNN_BATCHNORM_SPATIAL_PERSISTENT; + // #else mode_ = CUDNN_BATCHNORM_SPATIAL; -/* #endif */ + // #endif CUDNN_ENFORCE(platform::dynload::cudnnSetTensorNdDescriptor( data_desc_, CudnnDataType::type, -- GitLab