error.hpp

前往此文件的文档

/*
 * 版权所有 (c) 2020-2023, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下位置获取许可证副本
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发是基于“现状”，
 * 不附带任何明示或默示的保证或条件。
 * 请参阅许可证以了解管理权限和
 * 限制的特定语言。
 */
 
#pragma once
 
#include <cuspatial/assert.cuh>
 
#include <cuda_runtime_api.h>
 
#include <stdexcept>
#include <string>
 
namespace cuspatial {


struct logic_error : public std::logic_error {
 logic_error(char const* const message) : std::logic_error(message) {}
 logic_error(std::string const& message) : std::logic_error(message) {}
};

struct cuda_error : public std::runtime_error {
 cuda_error(std::string const& message) : std::runtime_error(message) {}
};

 
} // namespace cuspatial
 
#define CUSPATIAL_STRINGIFY_DETAIL(x) #x
#define CUSPATIAL_STRINGIFY(x) CUSPATIAL_STRINGIFY_DETAIL(x)

#define CUSPATIAL_EXPECTS(cond, reason) \
 (!!(cond)) ? static_cast<void>(0) \
 : throw cuspatial::logic_error("cuSpatial failure at: " __FILE__ \
 ":" CUSPATIAL_STRINGIFY(__LINE__) ": " reason)

#ifndef __CUDA_ARCH__
#define CUSPATIAL_HOST_DEVICE_EXPECTS(cond, reason) CUSPATIAL_EXPECTS(cond, reason)
#else
#define CUSPATIAL_HOST_DEVICE_EXPECTS(cond, reason) cuspatial_assert(cond&& reason)
#endif

#define CUSPATIAL_FAIL(reason) \
 throw cuspatial::logic_error("cuSpatial failure at: " __FILE__ \
 ":" CUSPATIAL_STRINGIFY(__LINE__) ": " reason)
 
namespace cuspatial {
namespace detail {
 
inline void throw_cuda_error(cudaError_t error, const char* file, unsigned int line)
{
 throw cuspatial::cuda_error(std::string{
 "遇到 CUDA 错误于: " + std::string{file} + ":" + std::to_string(line) + ": " +
 std::to_string(error) + " " + cudaGetErrorName(error) + " " + cudaGetErrorString(error)});
}
 
} // namespace detail
} // namespace cuspatial

#define CUSPATIAL_CUDA_TRY(call) \
 do { \
 cudaError_t const status = (call); \
 if (cudaSuccess != status) { \
 cudaGetLastError(); \
 cuspatial::detail::throw_cuda_error(status, __FILE__, __LINE__); \
    }                                                                  \
 } while (0);

#ifndef NDEBUG
#define CUSPATIAL_CHECK_CUDA(stream) \
 do { \
 CUSPATIAL_CUDA_TRY(cudaStreamSynchronize(stream)); \
 CUSPATIAL_CUDA_TRY(cudaPeekAtLastError()); \
 } while (0);
#else
#define CUSPATIAL_CHECK_CUDA(stream) CUSPATIAL_CUDA_TRY(cudaPeekAtLastError());
#endif