Improved CHECK_CUDA() macro

018e85f1 · Davis King · 935a0b4e · 018e85f1
Commit 018e85f1 authored Apr 24, 2016 by Davis King
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

dlib/dnn/cuda_utils.h dlib/dnn/cuda_utils.h +2 -2

No files found.
--- a/dlib/dnn/cuda_utils.h
+++ b/dlib/dnn/cuda_utils.h
@@ -15,7 +15,7 @@

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

 // ----------------------------------------------------------------------------------------