“CUDA_CHECK”: 找不到辨別符

“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