mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
27 lines
586 B
Plaintext
27 lines
586 B
Plaintext
#ifndef CUDANET_HELPER_H
|
|
#define CUDANET_HELPER_H
|
|
|
|
#include <cuda_runtime.h>
|
|
#include <cstdio>
|
|
|
|
#ifndef BLOCK_SIZE
|
|
#define BLOCK_SIZE 128
|
|
#endif // BLOCK_SIZE
|
|
|
|
/**
|
|
* @brief CUDA error checking macro
|
|
*
|
|
*/
|
|
#define CUDA_CHECK(call) \
|
|
do { \
|
|
cudaError_t result = call; \
|
|
if (result != cudaSuccess) { \
|
|
fprintf(stderr, "CUDA error at %s:%d code=%d(%s) \"%s\" \n", \
|
|
__FILE__, __LINE__, static_cast<unsigned int>(result), \
|
|
cudaGetErrorString(result), #call); \
|
|
exit(EXIT_FAILURE); \
|
|
} \
|
|
} while (0)
|
|
|
|
#endif // CUDANET_HELPER_H
|