提交 296b88d6 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!845 add index error type for log interface

Merge pull request !845 from fary86/add_error_type_index_error
...@@ -143,6 +143,7 @@ static std::string ExceptionTypeToString(ExceptionType type) { ...@@ -143,6 +143,7 @@ static std::string ExceptionTypeToString(ExceptionType type) {
_TO_STRING(TimeOutError), _TO_STRING(TimeOutError),
_TO_STRING(ResourceUnavailable), _TO_STRING(ResourceUnavailable),
_TO_STRING(NoPermissionError), _TO_STRING(NoPermissionError),
_TO_STRING(IndexError),
_TO_STRING(ValueError), _TO_STRING(ValueError),
_TO_STRING(TypeError), _TO_STRING(TypeError),
}; };
...@@ -179,7 +180,8 @@ void LogWriter::operator^(const LogStream &stream) const { ...@@ -179,7 +180,8 @@ void LogWriter::operator^(const LogStream &stream) const {
std::ostringstream oss; std::ostringstream oss;
oss << location_.file_ << ":" << location_.line_ << " " << location_.func_ << "] "; oss << location_.file_ << ":" << location_.line_ << " " << location_.func_ << "] ";
if (exception_type_ != NoExceptionType && exception_type_ != TypeError && exception_type_ != ValueError) { if (exception_type_ != NoExceptionType && exception_type_ != IndexError && exception_type_ != TypeError &&
exception_type_ != ValueError) {
oss << ExceptionTypeToString(exception_type_) << " "; oss << ExceptionTypeToString(exception_type_) << " ";
} }
oss << msg.str(); oss << msg.str();
...@@ -187,6 +189,9 @@ void LogWriter::operator^(const LogStream &stream) const { ...@@ -187,6 +189,9 @@ void LogWriter::operator^(const LogStream &stream) const {
trace::TraceGraphInfer(); trace::TraceGraphInfer();
trace::GetInferStackInfo(oss); trace::GetInferStackInfo(oss);
if (exception_type_ == IndexError) {
throw pybind11::index_error(oss.str());
}
if (exception_type_ == ValueError) { if (exception_type_ == ValueError) {
throw pybind11::value_error(oss.str()); throw pybind11::value_error(oss.str());
} }
......
...@@ -54,6 +54,7 @@ enum ExceptionType { ...@@ -54,6 +54,7 @@ enum ExceptionType {
TimeOutError, TimeOutError,
ResourceUnavailable, ResourceUnavailable,
NoPermissionError, NoPermissionError,
IndexError,
ValueError, ValueError,
TypeError, TypeError,
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册