diff --git a/paddle/fluid/operators/batch_norm_op.cu b/paddle/fluid/operators/batch_norm_op.cu index 32564516556a020e6649d1a91492b4f83f226b26..8f3691dd15359b42dfacca1c26e2b631a48161b0 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,