Unverified Commit f7608d89 authored by Tim Moon's avatar Tim Moon Committed by GitHub
Browse files

Add instructions in cuDNN error message for cuDNN error logging (#204)


Signed-off-by: default avatarTim Moon <tmoon@nvidia.com>
parent 83911ddb
...@@ -42,7 +42,15 @@ inline void check_cublas_(cublasStatus_t status) { ...@@ -42,7 +42,15 @@ inline void check_cublas_(cublasStatus_t status) {
inline void check_cudnn_(cudnnStatus_t status) { inline void check_cudnn_(cudnnStatus_t status) {
if ( status != CUDNN_STATUS_SUCCESS ) { if ( status != CUDNN_STATUS_SUCCESS ) {
NVTE_ERROR("CUDNN Error: " + std::string(cudnnGetErrorString(status))); std::string message;
message.reserve(1024);
message += "CUDNN Error: ";
message += cudnnGetErrorString(status);
message += (". "
"For more information, enable cuDNN error logging "
"by setting CUDNN_LOGERR_DBG=1 and "
"CUDNN_LOGDEST_DBG=stderr in the environment.");
NVTE_ERROR(message);
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment