cuda_error.hpp

Go to the documentation of this file.

#pragma once
 
#include <cuda_runtime_api.h>
 
#include <uni20/common/diagnostic_error.hpp>
 
#include <optional>
#include <source_location>
#include <string>
#include <string_view>
 
namespace uni20::cuda
{
 
class CudaRuntimeError : public uni20::diagnostic_error {
  public:
    CudaRuntimeError(cudaError_t status, std::string operation, std::optional<int> device = std::nullopt);
 
    [[nodiscard]] cudaError_t status() const noexcept { return status_; }
 
    [[nodiscard]] std::string const& operation() const noexcept { return operation_; }
 
    [[nodiscard]] std::string const& error_name() const noexcept { return error_name_; }
 
    [[nodiscard]] std::string const& reason() const noexcept { return reason_; }
 
    [[nodiscard]] std::optional<int> device() const noexcept { return device_; }
 
  private:
    cudaError_t status_;
    std::string operation_;
    std::string error_name_;
    std::string reason_;
    std::optional<int> device_;
};
 
[[noreturn]] void raise_runtime_error(cudaError_t status, std::string_view operation,
                                      std::optional<int> device = std::nullopt,
                                      std::source_location where = std::source_location::current());
 
void check(cudaError_t status, std::string_view operation, std::optional<int> device = std::nullopt,
           std::source_location where = std::source_location::current());
 
} // namespace uni20::cuda