cuda_check.h

Go to the documentation of this file.

#pragma once
 
#include <cuda_runtime.h>
#include <stdexcept>
#include <string>
 
#include "log.h"
 
#define FZ_CUDA_CHECK(call)                                                      \
    do {                                                                         \
        cudaError_t _fz_cuda_err_ = (call);                                      \
        if (_fz_cuda_err_ != cudaSuccess) {                                      \
            throw std::runtime_error(                                            \
                std::string("[fzgmod] CUDA error at " __FILE__ ":") +             \
                std::to_string(__LINE__) +                                       \
                " — " #call " → " +                                              \
                cudaGetErrorString(_fz_cuda_err_));                              \
        }                                                                        \
    } while (0)
 
#define FZ_CUDA_CHECK_WARN(call)                                                 \
    do {                                                                         \
        cudaError_t _fz_cuda_err_ = (call);                                      \
        if (_fz_cuda_err_ != cudaSuccess) {                                      \
            FZ_LOG(WARN,                                                         \
                   "CUDA error at %s:%d — " #call " → %s",                      \
                   __FILE__, __LINE__,                                           \
                   cudaGetErrorString(_fz_cuda_err_));                           \
        }                                                                        \
    } while (0)