Commit 2fd98234 authored by Davis King's avatar Davis King

Improved more macros

parent 018e85f1
......@@ -25,7 +25,7 @@ static const char* cublas_get_error_string(cublasStatus_t s)
// Check the return value of a call to the cuBLAS runtime for an error condition.
#define CHECK_CUBLAS(call) \
{ \
do{ \
const cublasStatus_t error = call; \
if (error != CUBLAS_STATUS_SUCCESS) \
{ \
......@@ -34,7 +34,7 @@ static const char* cublas_get_error_string(cublasStatus_t s)
sout << "code: " << error << ", reason: " << cublas_get_error_string(error);\
throw dlib::cublas_error(sout.str()); \
} \
}
}while(false)
namespace dlib
{
......
......@@ -36,7 +36,7 @@ static const char* cudnn_get_error_string(cudnnStatus_t s)
// Check the return value of a call to the cuDNN runtime for an error condition.
#define CHECK_CUDNN(call) \
{ \
do{ \
const cudnnStatus_t error = call; \
if (error != CUDNN_STATUS_SUCCESS) \
{ \
......@@ -45,7 +45,7 @@ static const char* cudnn_get_error_string(cudnnStatus_t s)
sout << "code: " << error << ", reason: " << cudnn_get_error_string(error);\
throw dlib::cudnn_error(sout.str()); \
} \
}
}while(false)
namespace dlib
......
......@@ -24,7 +24,7 @@ static const char* curand_get_error_string(curandStatus_t s)
// Check the return value of a call to the cuDNN runtime for an error condition.
#define CHECK_CURAND(call) \
{ \
do{ \
const curandStatus_t error = call; \
if (error != CURAND_STATUS_SUCCESS) \
{ \
......@@ -33,7 +33,7 @@ static const char* curand_get_error_string(curandStatus_t s)
sout << "code: " << error << ", reason: " << curand_get_error_string(error);\
throw dlib::curand_error(sout.str()); \
} \
}
}while(false)
namespace dlib
{
......
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