experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Deprecate TF_USE_CUDNN env var.

Browse Source 
A warning is now printed if it is used. The warning is only printed once.

PiperOrigin-RevId: 309103776
Change-Id: If5b7a94d5cfa532504660bdff782b03879a5419e
This commit is contained in:
Reed Wanderman-Milne 2020-04-29 15:00:10 -07:00 committed by TensorFlower Gardener
parent f98da6aa76
commit 31301ff561
1 changed files with 18 additions and 1 deletions

19
tensorflow/core/util/use_cudnn.cc
View File

@ -22,6 +22,24 @@ 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;
}
#define ADD_BOOL_CUDNN_FLAG(func_name, flag_name, default_value) \
bool func_name() { \
bool value = default_value; \
@ -32,7 +50,6 @@ namespace tensorflow {
return value; \
}
ADD_BOOL_CUDNN_FLAG(CanUseCudnn, TF_USE_CUDNN, true);
ADD_BOOL_CUDNN_FLAG(CudnnUseAutotune, TF_CUDNN_USE_AUTOTUNE, true);
// Whether to auto-tuning Cudnn RNN forward and backward pass to pick
// statistically the best cudnnRNNAlgo_t and cudnnMathType_t.