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

Automated rollback of commit 912db4a625

Browse Source 
PiperOrigin-RevId: 266206858
This commit is contained in:
Benjamin Kramer 2019-08-29 12:52:39 -07:00 committed by TensorFlower Gardener
parent 24f4b15253
commit 988882bb84
1 changed files with 0 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2946,20 +2946,6 @@ port::Status CudnnSupport::DoConvolve(
"This configuration potentially produces incorrect results.");
}
}
// According to the cuDNN documentation algorithm 1 only supports NHWC
// convolutions when using INT8. It doesn't seem to check that before
// accessing memory though, leading to unaligned accesses.
// TODO(b/138726848): File nvidia bug and restrict this to broken versions.
if (algorithm_desc.algo_id() ==
CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM &&
filter_descriptor.layout() == dnn::FilterLayout::kOutputYXInput &&
ToCudnnDataType(element_type) != CUDNN_DATA_INT8 &&
ToCudnnDataType(element_type) != CUDNN_DATA_INT8x4 &&
ToCudnnDataType(element_type) != CUDNN_DATA_UINT8x4) {
return port::Status(
port::error::FAILED_PRECONDITION,
"Data type not supported by algorithm configuration.");
}
return port::Status::OK();
};