mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
e7d5e652df
Co-authored-by: Olli Saarikivi <olsaarik@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Binyang Li <binyli@microsoft.com>
42 lines
1.6 KiB
C++
42 lines
1.6 KiB
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <nanobind/nanobind.h>
|
|
#include <nanobind/stl/string.h>
|
|
|
|
#include <mscclpp/errors.hpp>
|
|
|
|
namespace nb = nanobind;
|
|
using namespace mscclpp;
|
|
|
|
void register_error(nb::module_& m) {
|
|
nb::enum_<ErrorCode>(m, "ErrorCode")
|
|
.value("SystemError", ErrorCode::SystemError)
|
|
.value("InternalError", ErrorCode::InternalError)
|
|
.value("RemoteError", ErrorCode::RemoteError)
|
|
.value("InvalidUsage", ErrorCode::InvalidUsage)
|
|
.value("Timeout", ErrorCode::Timeout)
|
|
.value("Aborted", ErrorCode::Aborted);
|
|
|
|
nb::class_<BaseError>(m, "BaseError")
|
|
.def(nb::init<std::string&, int>(), nb::arg("message"), nb::arg("errorCode"))
|
|
.def("get_error_code", &BaseError::getErrorCode)
|
|
.def("what", &BaseError::what);
|
|
|
|
nb::class_<Error, BaseError>(m, "Error")
|
|
.def(nb::init<const std::string&, ErrorCode>(), nb::arg("message"), nb::arg("errorCode"))
|
|
.def("get_error_code", &Error::getErrorCode);
|
|
|
|
nb::class_<SysError, BaseError>(m, "SysError")
|
|
.def(nb::init<const std::string&, int>(), nb::arg("message"), nb::arg("errorCode"));
|
|
|
|
nb::class_<CudaError, BaseError>(m, "CudaError")
|
|
.def(nb::init<const std::string&, cudaError_t>(), nb::arg("message"), nb::arg("errorCode"));
|
|
|
|
nb::class_<CuError, BaseError>(m, "CuError")
|
|
.def(nb::init<const std::string&, CUresult>(), nb::arg("message"), nb::arg("errorCode"));
|
|
|
|
nb::class_<IbError, BaseError>(m, "IbError")
|
|
.def(nb::init<const std::string&, int>(), nb::arg("message"), nb::arg("errorCode"));
|
|
}
|