forked from smilart/nvidia-cdl
-
Notifications
You must be signed in to change notification settings - Fork 1
/
check_cuda.cuh
51 lines (43 loc) · 1.55 KB
/
check_cuda.cuh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
/*
* Copyright (c) 2014 Smilart and others.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Alexander Komarov [email protected] - implementation.
*/
#ifndef CHECK_CUDA_CUH_
#define CHECK_CUDA_CUH_
#include "check.h"
#ifndef CPU
#endif
#ifdef checkCudaCall
#error checkCudaCall macro defined not only in check_cuda.cuh
#endif
#ifdef checkCuCall
#error checkCuCall macro defined not only in check_cuda.cuh
#endif
#define checkCudaCall(function, ...) \
{ \
cudaError_t result = function; \
if (result != cudaSuccess) { \
std::string descriptionOfError = std::string("Description of error - ") + std::string(cudaGetErrorString(result)) + std::string("."); \
throwString(__FILE__, QUOTE_VALUE(__LINE__), #function, descriptionOfError , ##__VA_ARGS__); \
} \
}
#define checkKernelRun(kernelNameAndGridDim, ...) \
{ \
kernelNameAndGridDim, ##__VA_ARGS__; \
cudaError_t result = cudaGetLastError(); \
if (result != cudaSuccess) { \
throwString(__FILE__, QUOTE_VALUE(__LINE__), extractKernelName(#kernelNameAndGridDim) + "<<<>>> start failure", cudaGetErrorString(result)); \
} \
result = cudaThreadSynchronize(); \
if (result != cudaSuccess) { \
throwString(__FILE__, QUOTE_VALUE(__LINE__), extractKernelName(#kernelNameAndGridDim) + "<<<>>> run failure", cudaGetErrorString(result)); \
} \
}
#endif // CHECK_CUDA_CUH_