Skip to content

Commit b47133f

Browse files
committed
add CUDNN_STATUS_VERSION_MISMATCH
1 parent 0b76e96 commit b47133f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

include/caffe/util/cudnn.hpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
5050
return "CUDNN_STATUS_RUNTIME_IN_PROGRESS";
5151
case CUDNN_STATUS_RUNTIME_FP_OVERFLOW:
5252
return "CUDNN_STATUS_RUNTIME_FP_OVERFLOW";
53+
#endif
54+
#if CUDNN_VERSION_MIN(8, 0, 1)
55+
case CUDNN_STATUS_VERSION_MISMATCH:
56+
return "CUDNN_STATUS_VERSION_MISMATCH";
5357
#endif
5458
}
5559
return "Unknown cudnn status";

0 commit comments

Comments
 (0)