diff options
| -rw-r--r-- | src/caffe/common.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/caffe/common.cpp b/src/caffe/common.cpp index a900d6f0..94fdf924 100644 --- a/src/caffe/common.cpp +++ b/src/caffe/common.cpp @@ -224,6 +224,10 @@ const char* cublasGetErrorString(cublasStatus_t error) { case CUBLAS_STATUS_NOT_SUPPORTED: return "CUBLAS_STATUS_NOT_SUPPORTED"; #endif +#if CUDA_VERSION >= 6050 + case CUBLAS_STATUS_LICENSE_ERROR: + return "CUBLAS_STATUS_LICENSE_ERROR"; +#endif } return "Unknown cublas status"; } |