Merge pull request NVIDIA#250 from drnikolaev/caffe-0.15-status-code-…

…added Hotfix: New status code for cuDNN v6
FreeApe · Oct 3, 2016 · ce08d07 · ce08d07
2 parents 4628e53 + e1c5420
commit ce08d07
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/include/caffe/util/cudnn.hpp b/include/caffe/util/cudnn.hpp
@@ -41,6 +41,12 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
       return "CUDNN_STATUS_NOT_SUPPORTED";
     case CUDNN_STATUS_LICENSE_ERROR:
       return "CUDNN_STATUS_LICENSE_ERROR";
+#if CUDNN_VERSION_MIN(6, 0, 1)
+    case CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING:
+      return "CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING";
+#endif
+    default:
+      break;
   }
   return "Unknown cudnn status";
 }