| /* |
| * 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; |
| } |
| |
| // |
| // |
| // |