OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Improved CHECK_CUDA() macro

This commit is contained in:
Davis King 2016-04-24 10:05:37 -04:00
parent 935a0b4e48
commit 018e85f150
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlib/dnn/cuda_utils.h
View File

@ -15,7 +15,7 @@
// Check the return value of a call to the CUDA runtime for an error condition. // Check the return value of a call to the CUDA runtime for an error condition.
#define CHECK_CUDA(call) \ #define CHECK_CUDA(call) \
{ \ do{ \
const cudaError_t error = call; \ const cudaError_t error = call; \
if (error != cudaSuccess) \ if (error != cudaSuccess) \
{ \ { \
@ -24,7 +24,7 @@
sout << "code: " << error << ", reason: " << cudaGetErrorString(error);\ sout << "code: " << error << ", reason: " << cudaGetErrorString(error);\
throw dlib::cuda_error(sout.str()); \ throw dlib::cuda_error(sout.str()); \
} \ } \
} }while(false)
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------