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

Automated rollback of commit b2d5c920ab

Browse Source 
PiperOrigin-RevId: 260748949
This commit is contained in:
Tim Shen 2019-07-30 10:46:03 -07:00 committed by TensorFlower Gardener
parent fdc9600788
commit af916f80c0
1 changed files with 0 additions and 7 deletions

7
tensorflow/stream_executor/cuda/cuda_dnn.cc
View File

@ -2533,13 +2533,6 @@ port::StatusOr<dnn::AlgorithmDesc> GetCudnnConvolutionForwardAlgorithm(
return *algo_desc;
}
if (!absl::StrContains(scratch_or.status().ToString(),
"CUDNN_STATUS_ALLOC_FAILED")) {
return port::Status(port::error::INVALID_ARGUMENT,
absl::StrCat("cuDNN returned unexpected error: ",
scratch_or.status().ToString()));
}
algo_desc = algorithm_config.algorithm_no_scratch();
// Failed to allocate workspace for the first algorithm, fall back to the