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
 *
 * https://apache.ac.cn/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