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