mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-07 01:50:04 +08:00
ad8aef0f98
Mostly in torch/csrc/jit/runtime and in `ATen/cuda/` Pull Request resolved: https://github.com/pytorch/pytorch/pull/110314 Approved by: https://github.com/seemethere
30 lines
614 B
C++
30 lines
614 B
C++
#pragma once
|
|
#include <c10/util/Exception.h>
|
|
#include <stdexcept>
|
|
|
|
namespace torch::jit {
|
|
|
|
struct ExceptionMessage {
|
|
ExceptionMessage(const std::exception& e) : e_(e) {}
|
|
|
|
private:
|
|
const std::exception& e_;
|
|
friend std::ostream& operator<<(
|
|
std::ostream& out,
|
|
const ExceptionMessage& msg);
|
|
};
|
|
|
|
inline std::ostream& operator<<(
|
|
std::ostream& out,
|
|
const ExceptionMessage& msg) {
|
|
auto c10_error = dynamic_cast<const c10::Error*>(&msg.e_);
|
|
if (c10_error) {
|
|
out << c10_error->what_without_backtrace();
|
|
} else {
|
|
out << msg.e_.what();
|
|
}
|
|
return out;
|
|
}
|
|
|
|
} // namespace torch::jit
|