cuda_helper.h
731 Bytes
/*
* @Author: yangzilong
* @Date: 2021-12-14 16:41:18
* @Last Modified by: yangzilong
* @Email: yangzilong@objecteye.com
* @Description:
*/
#pragma once
#include <cstdio>
#include <cuda_runtime.h>
#include "define.hpp"
#include "logger.hpp"
#define PRINT_CHECK(call) \
{\
const cudaError_t error_code = call;\
if (cudaSuccess != error_code)\
std::fprintf(stderr, "[%s:%d] %s error code: %d error reason: %s\n", __FILENAME__, __LINE__, __FUNCTION__, error_code, cudaGetErrorString(error_code));\
}
#define CHECK(call) \
{\
const cudaError_t error_code = call;\
if (cudaSuccess != error_code)\
LOG_ERROR("CUDA error, code: {} reason: {}", error_code, cudaGetErrorString(error_code));\
}