forked from chenzhi1992/TensorRT-SSD
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cudaUtility.h
86 lines (63 loc) · 1.53 KB
/
cudaUtility.h
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#ifndef __CUDA_UTILITY_H_
#define __CUDA_UTILITY_H_
#include <cuda_runtime.h>
#include <cuda.h>
#include <stdio.h>
#include <string.h>
/**
* Execute a CUDA call and print out any errors
* @return the original cudaError_t result
* @ingroup util
*/
#define CUDA(x) cudaCheckError((x), #x, __FILE__, __LINE__)
/**
* Evaluates to true on success
* @ingroup util
*/
#define CUDA_SUCCESS(x) (CUDA(x) == cudaSuccess)
/**
* Evaluates to true on failure
* @ingroup util
*/
#define CUDA_FAILED(x) (CUDA(x) != cudaSuccess)
/**
* Return from the boolean function if CUDA call fails
* @ingroup util
*/
#define CUDA_VERIFY(x) if(CUDA_FAILED(x)) return false;
/**
* LOG_CUDA string.
* @ingroup util
*/
#define LOG_CUDA "[cuda] "
/*
* define this if you want all cuda calls to be printed
*/
//#define CUDA_TRACE
/**
* cudaCheckError
* @ingroup util
*/
inline cudaError_t cudaCheckError(cudaError_t retval, const char* txt, const char* file, int line )
{
#if !defined(CUDA_TRACE)
if( retval == cudaSuccess)
return cudaSuccess;
#endif
//int activeDevice = -1;
//cudaGetDevice(&activeDevice);
//Log("[cuda] device %i - %s\n", activeDevice, txt);
printf(LOG_CUDA "%s\n", txt);
if( retval != cudaSuccess )
{
printf(LOG_CUDA " %s (error %u) (hex 0x%02X)\n", cudaGetErrorString(retval), retval, retval);
printf(LOG_CUDA " %s:%i\n", file, line);
}
return retval;
}
/**
* iDivUp
* @ingroup util
*/
inline __device__ __host__ int iDivUp( int a, int b ) { return (a % b != 0) ? (a / b + 1) : (a / b); }
#endif