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

Fix typo CUDNN_RETURN_IF_FAIL.

Browse Source
This commit is contained in:
Florian Courtial 2017-02-10 21:17:45 +01:00
parent 714f9b74b5
commit 7fc9bb7e26
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -13,3 +13,4 @@ node_modules
*.pyc *.pyc
__pycache__ __pycache__
*.swp *.swp
.vscode/

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

@ -922,7 +922,7 @@ class CudnnRnnParamsDescriptor : public CudnnDescriptorCommon<void> {
const CudnnRnnDescriptor& rnn_desc); const CudnnRnnDescriptor& rnn_desc);
~CudnnRnnParamsDescriptor() { ~CudnnRnnParamsDescriptor() {
cudnnStatus_t status = wrap::cudnnDestroyFilterDescriptor(parent_, handle_); cudnnStatus_t status = wrap::cudnnDestroyFilterDescriptor(parent_, handle_);
CUDNN_RETURN_IF_FAIL(status, "Failed to destroy RNN filter desciptor"); CUDNN_RETURN_IF_FAIL(status, "Failed to destroy RNN filter descriptor");
} }
cudnnFilterDescriptor_t handle() const { cudnnFilterDescriptor_t handle() const {
if (!ok()) return nullptr; if (!ok()) return nullptr;
@ -1202,7 +1202,7 @@ class CudnnRnnSequenceTensorDescriptor
// Only the first one needs to be destroyed. All others are the same. // Only the first one needs to be destroyed. All others are the same.
cudnnStatus_t status = cudnnStatus_t status =
wrap::cudnnDestroyTensorDescriptor(parent_, handles_[0]); wrap::cudnnDestroyTensorDescriptor(parent_, handles_[0]);
CUDNN_RETURN_IF_FAIL(status, "Failed to destroy sequence tensor desciptor"); CUDNN_RETURN_IF_FAIL(status, "Failed to destroy sequence tensor descriptor");
} }
const cudnnTensorDescriptor_t* handles() const { const cudnnTensorDescriptor_t* handles() const {