Browse Source

feat(mgb): show more details for cuda driver api call

GitOrigin-RevId: 40e63d9dac
tags/v1.9.0
Megvii Engine Team 3 years ago
parent
commit
f48227c07d
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      dnn/src/cuda/utils.cpp

+ 4
- 1
dnn/src/cuda/utils.cpp View File

@@ -85,7 +85,10 @@ void cuda::__throw_cusolver_error__(cusolverStatus_t err, const char* msg) {
}

void cuda::__throw_cuda_driver_error__(CUresult err, const char* msg) {
auto s = ssprintf("cuda driver error %d occurred; expr: %s", int(err), msg);
const char* err_str = nullptr;
cuGetErrorName(err, &err_str);
err_str = err_str ? err_str : "unknown error";
auto s = ssprintf("cuda driver error %d(%s) occurred; expr: %s", int(err), err_str, msg);
megdnn_throw(s.c_str());
}



Loading…
Cancel
Save