Revert "not use PERSISTENT in batch_norm. test=develop" (#16333)

paddle/fluid/operators/batch_norm_op.cu

@@ -75,15 +75,11 @@ class BatchNormKernel<platform::CUDADeviceContext, T>
                  << "CUDNN_BN_MIN_EPSILON instead.";
     }
     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
+#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<int> dims;
@@ -305,15 +301,11 @@ class BatchNormGradKernel<platform::CUDADeviceContext, T>
                    << "CUDNN_BN_MIN_EPSILON instead.";
       }
       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
+#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<T>::type,