Remove environmental variable TF_USE_CUDNN.

reedwm · tensorflower-gardener · commit d3c70a1a2f2d · 2020-08-15T17:35:58.000-07:00
This environmental variable was deprecated and useless. cuDNN was still
required to run TensorFlow even if the variable was set to 0.

I will remove code dead code that uses the CanUseCudnn in a subsequent change.

PiperOrigin-RevId: 326851365
Change-Id: Ie2cde09a8a4913125b6c0bfe8ae52da816f80887
diff --git a/RELEASE.md b/RELEASE.md
@@ -261,6 +261,7 @@ stjohnso98, <NAME>, <HERE>, <USING>, <GITHUB>, <HANDLE>
     * Mutable tables now restore checkpointed values when loaded from SavedModel.
   * GPU
     * TF 2.3 includes PTX kernels only for [compute capability](https://developer.nvidia.com/cuda-gpus) 7.0 to reduce the TF pip binary size.  Earlier releases included PTX for a variety of older compute capabilities.
+    * Remove environmental variable `TF_USE_CUDNN`.
   * Others
     * Retain parent namescope for ops added inside `tf.while_loop`/`tf.cond`/`tf.switch_case`.
     * Update `tf.vectorized_map` to support vectorizing `tf.while_loop` and TensorList operations.
diff --git a/tensorflow/core/util/use_cudnn.cc b/tensorflow/core/util/use_cudnn.cc
@@ -22,23 +22,8 @@ limitations under the License.
 
 namespace tensorflow {
 
-bool CanUseCudnn() {
-  static bool is_enabled = [] {
-    bool is_enabled = true;
-    // TODO(b/155239286): Remove TF_USE_CUDNN after TF 2.3 is released.
-    Status status =
-        ReadBoolFromEnvVar("TF_USE_CUDNN", /*default_val=*/true, &is_enabled);
-    if (!status.ok()) {
-      LOG(ERROR) << status;
-    }
-    if (!is_enabled) {
-      LOG(WARNING) << "The environmental variable TF_USE_CUDNN is deprecated "
-                      "and will be ignored in the future";
-    }
-    return is_enabled;
-  }();
-  return is_enabled;
-}
+// TODO(b/155239286): Remove this function
+bool CanUseCudnn() { return true; }
 
 #define ADD_BOOL_CUDNN_FLAG(func_name, flag_name, default_value)           \
   bool func_name() {                                                       \
