| 16 | namespace cuda { |
| 17 | |
| 18 | const char *errorString(cudnnStatus_t err) { |
| 19 | switch (err) { |
| 20 | case CUDNN_STATUS_SUCCESS: return "CUDNN_STATUS_SUCCESS"; |
| 21 | case CUDNN_STATUS_NOT_INITIALIZED: |
| 22 | return "CUDNN_STATUS_NOT_INITIALIZED"; |
| 23 | case CUDNN_STATUS_ALLOC_FAILED: return "CUDNN_STATUS_ALLOC_FAILED"; |
| 24 | case CUDNN_STATUS_BAD_PARAM: return "CUDNN_STATUS_BAD_PARAM"; |
| 25 | case CUDNN_STATUS_INTERNAL_ERROR: return "CUDNN_STATUS_INTERNAL_ERROR"; |
| 26 | case CUDNN_STATUS_INVALID_VALUE: return "CUDNN_STATUS_INVALID_VALUE"; |
| 27 | case CUDNN_STATUS_ARCH_MISMATCH: return "CUDNN_STATUS_ARCH_MISMATCH"; |
| 28 | case CUDNN_STATUS_MAPPING_ERROR: return "CUDNN_STATUS_MAPPING_ERROR"; |
| 29 | case CUDNN_STATUS_EXECUTION_FAILED: |
| 30 | return "CUDNN_STATUS_EXECUTION_FAILED"; |
| 31 | case CUDNN_STATUS_NOT_SUPPORTED: return "CUDNN_STATUS_NOT_SUPPORTED"; |
| 32 | case CUDNN_STATUS_LICENSE_ERROR: return "CUDNN_STATUS_LICENSE_ERROR"; |
| 33 | #if CUDNN_VERSION >= 6000 |
| 34 | case CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING: |
| 35 | return "CUDNN_STATUS_RUNTIME_PREREQUISITE_MISSING"; |
| 36 | #if CUDNN_VERSION >= 7000 |
| 37 | case CUDNN_STATUS_RUNTIME_IN_PROGRESS: |
| 38 | return "CUDNN_STATUS_RUNTIME_IN_PROGRESS"; |
| 39 | case CUDNN_STATUS_RUNTIME_FP_OVERFLOW: |
| 40 | return "CUDNN_STATUS_RUNTIME_FP_OVERFLOW"; |
| 41 | #if CUDNN_VERSION >= 8000 |
| 42 | case CUDNN_STATUS_VERSION_MISMATCH: |
| 43 | return "CUDNN_STATUS_VERSION_MISMATCH"; |
| 44 | #endif |
| 45 | #endif |
| 46 | #endif |
| 47 | default: return "UNKNOWN"; |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | template<> |
| 52 | cudnnDataType_t getCudnnDataType<float>() { |