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
| /*
| * Copyright 2016 Google Inc.
| *
| * Use of this source code is governed by a BSD-style license that can be
| * found in the LICENSE file.
| *
| */
|
| #include <stdlib.h>
| #include <stdio.h>
|
| //
| //
| //
|
| #include <cuda_runtime_api.h>
|
| //
| //
| //
|
| #include "assert_cuda.h"
|
| //
| //
| //
|
| cudaError_t
| assert_cuda(cudaError_t const code,
| char const * const file,
| int const line,
| bool const abort)
| {
| if (code != cudaSuccess)
| {
| const char* const cuda_err_str = cudaGetErrorString(code);
|
| fprintf(stderr,
| "\"%s\", line %d: assert_cuda ( %d ) = \"%s\"",
| file,line,code,cuda_err_str);
|
| if (abort)
| {
| cudaDeviceReset();
| exit(code);
| }
| }
|
| return code;
| }
|
| //
| //
| //
|
|
