OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Improved cudnn error messages.

This commit is contained in:
Davis King 2015-12-08 22:53:13 -05:00
parent d179693475
commit 3559f78528
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlib/dnn/cudnn_dlibapi.cpp
View File

@ -22,6 +22,10 @@ static const char* cudnn_get_error_string(cudnnStatus_t s)
return "CUDA Resources could not be allocated."; return "CUDA Resources could not be allocated.";
case CUDNN_STATUS_BAD_PARAM: case CUDNN_STATUS_BAD_PARAM:
return "CUDNN_STATUS_BAD_PARAM"; return "CUDNN_STATUS_BAD_PARAM";
case CUDNN_STATUS_EXECUTION_FAILED:
return "CUDNN_STATUS_EXECUTION_FAILED";
case CUDNN_STATUS_NOT_SUPPORTED:
return "CUDNN_STATUS_NOT_SUPPORTED";
default: default:
return "A call to cuDNN failed"; return "A call to cuDNN failed";
} }