error.hpp

/*
 * Copyright (c) 2020-2024, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <rmm/detail/export.hpp>
 
#include <stdexcept>
#include <string>
 
namespace RMM_NAMESPACE {
 
struct logic_error : public std::logic_error {
  using std::logic_error::logic_error;
};
 
struct cuda_error : public std::runtime_error {
  using std::runtime_error::runtime_error;
};
 
class bad_alloc : public std::bad_alloc {
 public:
  bad_alloc(const char* msg) : _what{std::string{std::bad_alloc::what()} + ": " + msg} {}
 
  bad_alloc(std::string const& msg) : bad_alloc{msg.c_str()} {}
 
  [[nodiscard]] const char* what() const noexcept override { return _what.c_str(); }
 
 private:
  std::string _what;
};
 
class out_of_memory : public bad_alloc {
 public:
  out_of_memory(const char* msg) : bad_alloc{std::string{"out_of_memory: "} + msg} {}
 
  out_of_memory(std::string const& msg) : out_of_memory{msg.c_str()} {}
};
 
class out_of_range : public std::out_of_range {
  using std::out_of_range::out_of_range;
};
 
}  // namespace RMM_NAMESPACE