“CUDA_CHECK”: 找不到标识符
在cuda c/c++编程时,有时会出现 “error:CUDA_CHECK:找不到标识符”,我们可以自定义一个“cuda_utils.h”的文件,定义CUDA_CHECK。
#include <cuda_runtime_api.h>
#ifndef CUDA_CHECK
#define CUDA_CHECK(callstr)\
{\
cudaError_t error_code = callstr;\
if (error_code != cudaSuccess) {\
std::cerr << "CUDA error " << error_code << " at " << __FILE__ << ":" << __LINE__;\
assert(0);\
}\
}
#endif // CUDA_CHECK