From a186dc49f4bcd55906d05226dd4f00ca84cfec09 Mon Sep 17 00:00:00 2001 From: Kai Li Date: Mon, 31 Mar 2014 22:07:22 +0800 Subject: [PATCH] Define CUDA_POST_KERNEL_CHECK with CUDA_CHECK --- include/caffe/common.hpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/include/caffe/common.hpp b/include/caffe/common.hpp index e87705b..e436756 100644 --- a/include/caffe/common.hpp +++ b/include/caffe/common.hpp @@ -42,10 +42,7 @@ private:\ i += blockDim.x * gridDim.x) // CUDA: check for error after kernel execution and exit loudly if there is one. -#define CUDA_POST_KERNEL_CHECK \ - if (cudaSuccess != cudaPeekAtLastError()) \ - LOG(FATAL) << "Cuda kernel failed. Error: " \ - << cudaGetErrorString(cudaPeekAtLastError()) +#define CUDA_POST_KERNEL_CHECK CUDA_CHECK(cudaPeekAtLastError()) namespace caffe { -- 2.7.4