projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
eeaa38f
)
Define CUDA_POST_KERNEL_CHECK with CUDA_CHECK
author
Kai Li
<kaili_kloud@163.com>
Mon, 31 Mar 2014 14:07:22 +0000
(22:07 +0800)
committer
Kai Li
<kaili_kloud@163.com>
Tue, 1 Apr 2014 02:24:46 +0000
(10:24 +0800)
include/caffe/common.hpp
patch
|
blob
|
history
diff --git
a/include/caffe/common.hpp
b/include/caffe/common.hpp
index
e87705b
..
e436756
100644
(file)
--- 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 {