CUDAerror.h

Go to the documentation of this file.

//////////////////////////////////////////////////////////////////////////////////////
//// This file is distributed under the University of Illinois/NCSA Open Source License.
//// See LICENSE file in top directory for details.
////
//// Copyright (c) 2019 QMCPACK developers.
////
//// File developed by: Ye Luo, yeluo@anl.gov, Argonne National Laboratory
////
//// File created by: Ye Luo, yeluo@anl.gov, Argonne National Laboratory
////////////////////////////////////////////////////////////////////////////////////////


#ifndef QMCPLUSPLUS_CUDA_ERROR_H
#define QMCPLUSPLUS_CUDA_ERROR_H

#include <iostream>
#include <string>
#include <sstream>
#include <stdexcept>

#define cudaErrorCheck(ans, cause)                \
  {                                               \
    cudaAssert((ans), cause, __FILE__, __LINE__); \
  }

// If the cause is largely redundant with the __FILE__ and __LINE__ information
#define cudaCheck(ans)                         \
  {                                            \
    cudaAssert((ans), "", __FILE__, __LINE__); \
  }

/// prints CUDA error messages. Always use cudaErrorCheck macro.
inline void cudaAssert(cudaError_t code, const std::string& cause, const char* filename, int line, bool abort = true)
{
  if (code != cudaSuccess)
  {
    std::ostringstream err;
    err << "cudaAssert: " << cudaGetErrorName(code) << " " << cudaGetErrorString(code) << ", file " << filename
        << ", line " << line << std::endl
        << cause << std::endl;
    std::cerr << err.str();
    if (abort)
      throw std::runtime_error(cause);
  }
}

#endif