Improved more macros

This commit is contained in:
Davis King 2016-04-24 10:07:24 -04:00
parent 018e85f150
commit 2fd982347b
3 changed files with 6 additions and 6 deletions

View File

@ -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
{

View File

@ -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

View File

@ -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
{