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

Fix switch statement for CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT_TILING.

Browse Source 
PiperOrigin-RevId: 316522188
Change-Id: I6557506f90d4bf6a91e97b65abd494d58769e3a3
This commit is contained in:
Christian Sigg 2020-06-15 12:35:00 -07:00 committed by TensorFlower Gardener
parent 32c25efaa9
commit 09cf51da59
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -254,10 +254,8 @@ cudnnConvolutionBwdFilterAlgo_t ToConvBackwardFilterAlgo(
// Based on cudnn.h, the following is not implemented.
// case CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD:
case CUDNN_CONVOLUTION_BWD_FILTER_ALGO_WINOGRAD_NONFUSED:
case CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT_TILING:
return algo;
// Produces incorrect results for some shapes. Disabled for now, see
// NVIDIA bug 2072856. TODO(csigg): Only disable for subset of shapes.
// case CUDNN_CONVOLUTION_BWD_FILTER_ALGO_FFT_TILING:
default:
LOG(FATAL)
<< "Unsupported Cudnn convolution backward algorithm for filter: "