From 3559f78528728f4e54b61a3b12e04f6949545622 Mon Sep 17 00:00:00 2001 From: Davis King Date: Tue, 8 Dec 2015 22:53:13 -0500 Subject: [PATCH] Improved cudnn error messages. --- dlib/dnn/cudnn_dlibapi.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/dlib/dnn/cudnn_dlibapi.cpp b/dlib/dnn/cudnn_dlibapi.cpp index d1b4ed367..eb11c3ed2 100644 --- a/dlib/dnn/cudnn_dlibapi.cpp +++ b/dlib/dnn/cudnn_dlibapi.cpp @@ -22,6 +22,10 @@ static const char* cudnn_get_error_string(cudnnStatus_t s) return "CUDA Resources could not be allocated."; case 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: return "A call to cuDNN failed"; }