-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.cu
50 lines (42 loc) · 1.51 KB
/
util.cu
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
#ifndef UTIL_H
#define UTIL_H
#include <stdio.h>
#include <stdexcept>
/** Round up to the closest greater multiple. */
int roundup(int value, int multiple) {
if(value % multiple == 0) {
return value;
}
return value + multiple - value % multiple;
}
/** The maxnumthreads limits are only enforced if the total nubmer of threads
* (product of x, y and z) is exceeded. It is therefore well possible to have
* more threads in a given dimension, provided that the other dimensions are
* accordingly smaller. Note that it leads to errors to try and launch a cuda
* kernel with too many threads. */
#ifndef CUDA_MAXNUMTHREADS_X
#define CUDA_MAXNUMTHREADS_X 16
#define CUDA_MAXNUMTHREADS_Y 16
#define CUDA_MAXNUMTHREADS_Z 4
#endif
#define CUDA_CHECK(cmd) do { \
cudaError_t err = cmd; \
if(err != cudaSuccess) { \
fprintf(stderr, "%s, line %d: %s. %s.\n", __FILE__, __LINE__, \
cudaGetErrorName(err), cudaGetErrorString(err)); \
exit(1); \
} \
} while(0)
#define CUDA_CHECK_LAST() CUDA_CHECK(cudaGetLastError())
#define CUDA_THROW(cmd) do { \
cudaError_t err = cmd; \
if(err != cudaSuccess) { \
char msg[128] = ""; \
snprintf(msg, 128, "%s, line %d: %s. %s.", \
__FILE__, __LINE__, \
cudaGetErrorName(err), cudaGetErrorString(err)); \
throw std::runtime_error(msg); \
} \
} while(0)
#define CUDA_THROW_LAST() CUDA_THROW(cudaGetLastError())
#endif